290 lines
10 KiB
Perl
290 lines
10 KiB
Perl
################################################################################
|
|
# Copyright 2005-2013 MERETHIS
|
|
# Centreon is developped by : Julien Mathis and Romain Le Merlus under
|
|
# GPL Licence 2.0.
|
|
#
|
|
# This program is free software; you can redistribute it and/or modify it under
|
|
# the terms of the GNU General Public License as published by the Free Software
|
|
# Foundation ; either version 2 of the License.
|
|
#
|
|
# This program is distributed in the hope that it will be useful, but WITHOUT ANY
|
|
# WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A
|
|
# PARTICULAR PURPOSE. See the GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License along with
|
|
# this program; if not, see <http://www.gnu.org/licenses>.
|
|
#
|
|
# Linking this program statically or dynamically with other modules is making a
|
|
# combined work based on this program. Thus, the terms and conditions of the GNU
|
|
# General Public License cover the whole combination.
|
|
#
|
|
# As a special exception, the copyright holders of this program give MERETHIS
|
|
# permission to link this program with independent modules to produce an executable,
|
|
# regardless of the license terms of these independent modules, and to copy and
|
|
# distribute the resulting executable under terms of MERETHIS choice, provided that
|
|
# MERETHIS also meet, for each linked independent module, the terms and conditions
|
|
# of the license of that module. An independent module is a module which is not
|
|
# derived from this program. If you modify this program, you may extend this
|
|
# exception to your version of the program, but you are not obliged to do so. If you
|
|
# do not wish to do so, delete this exception statement from your version.
|
|
#
|
|
# For more information : contact@centreon.com
|
|
# Authors : Quentin Garnier <qgarnier@merethis.com>
|
|
#
|
|
####################################################################################
|
|
|
|
package centreon::plugins::script_sql;
|
|
|
|
use strict;
|
|
use warnings;
|
|
use centreon::plugins::misc;
|
|
|
|
sub new {
|
|
my ($class, %options) = @_;
|
|
my $self = {};
|
|
bless $self, $class;
|
|
# $options{package} = parent package caller
|
|
# $options{options} = options object
|
|
# $options{output} = output object
|
|
$self->{options} = $options{options};
|
|
$self->{output} = $options{output};
|
|
|
|
$self->{options}->add_options(
|
|
arguments => {
|
|
'mode:s' => { name => 'mode_name' },
|
|
'dyn-mode:s' => { name => 'dynmode_name' },
|
|
'list-mode' => { name => 'list_mode' },
|
|
'mode-version:s' => { name => 'mode_version' },
|
|
'sqlmode:s' => { name => 'sqlmode_name', default => 'dbi' },
|
|
'list-sqlmode' => { name => 'list_sqlmode' },
|
|
'multiple' => { name => 'multiple' },
|
|
}
|
|
);
|
|
$self->{version} = '1.0';
|
|
%{$self->{modes}} = ();
|
|
%{$self->{sql_modes}} = ('dbi' => 'centreon::plugins::dbi');
|
|
$self->{default} = undef;
|
|
$self->{sqldefault} = {};
|
|
$self->{sqlmode_current} = undef;
|
|
$self->{sqlmode_stored} = [];
|
|
|
|
$self->{options}->parse_options();
|
|
$self->{option_results} = $self->{options}->get_options();
|
|
foreach (keys %{$self->{option_results}}) {
|
|
$self->{$_} = $self->{option_results}->{$_};
|
|
}
|
|
$self->{options}->clean();
|
|
|
|
$self->{options}->add_help(package => $options{package}, sections => 'PLUGIN DESCRIPTION');
|
|
$self->{options}->add_help(package => __PACKAGE__, sections => 'GLOBAL OPTIONS');
|
|
|
|
return $self;
|
|
}
|
|
|
|
sub init {
|
|
my ($self, %options) = @_;
|
|
# $options{version} = string version
|
|
# $options{help} = string help
|
|
|
|
if (defined($options{help}) && !defined($self->{mode_name}) && !defined($self->{dynmode_name})) {
|
|
$self->{options}->display_help();
|
|
$self->{output}->option_exit();
|
|
}
|
|
if (defined($options{version}) && !defined($self->{mode_name})&& !defined($self->{dynmode_name})) {
|
|
$self->version();
|
|
}
|
|
if (defined($self->{list_mode})) {
|
|
$self->list_mode();
|
|
}
|
|
if (defined($self->{list_sqlmode})) {
|
|
$self->list_sqlmode();
|
|
}
|
|
|
|
# Output HELP
|
|
$self->{options}->add_help(package => 'centreon::plugins::output', sections => 'OUTPUT OPTIONS');
|
|
|
|
if (defined($self->{sqlmode_name}) && $self->{sqlmode_name} ne '') {
|
|
$self->is_sqlmode(sqlmode => $self->{sqlmode_name});
|
|
centreon::plugins::misc::mymodule_load(output => $self->{output}, module => $self->{sql_modes}{$self->{sqlmode_name}},
|
|
error_msg => "Cannot load module --sqlmode.");
|
|
$self->{sqlmode_current} = $self->{sql_modes}{$self->{sqlmode_name}}->new(options => $self->{options}, output => $self->{output}, mode => $self->{sqlmode_name});
|
|
} else {
|
|
$self->{output}->add_option_msg(short_msg => "Need to specify '--sqlmode'.");
|
|
$self->{output}->option_exit();
|
|
}
|
|
|
|
# Load mode
|
|
if (defined($self->{mode_name}) && $self->{mode_name} ne '') {
|
|
$self->is_mode(mode => $self->{mode_name});
|
|
centreon::plugins::misc::mymodule_load(output => $self->{output}, module => $self->{modes}{$self->{mode_name}},
|
|
error_msg => "Cannot load module --mode.");
|
|
$self->{mode} = $self->{modes}{$self->{mode_name}}->new(options => $self->{options}, output => $self->{output}, mode => $self->{mode_name});
|
|
} elsif (defined($self->{dynmode_name}) && $self->{dynmode_name} ne '') {
|
|
centreon::plugins::misc::mymodule_load(output => $self->{output}, module => $self->{dynmode_name},
|
|
error_msg => "Cannot load module --dyn-mode.");
|
|
$self->{mode} = $self->{dynmode_name}->new(options => $self->{options}, output => $self->{output}, mode => $self->{mode_name});
|
|
} else {
|
|
$self->{output}->add_option_msg(short_msg => "Need to specify '--mode' or '--dyn-mode' option.");
|
|
$self->{output}->option_exit();
|
|
}
|
|
|
|
if (defined($options{help})) {
|
|
$self->{options}->add_help(package => $self->{modes}{$self->{mode_name}}, sections => 'MODE');
|
|
$self->{options}->display_help();
|
|
$self->{output}->option_exit();
|
|
}
|
|
if (defined($options{version})) {
|
|
$self->{mode}->version();
|
|
$self->{output}->option_exit(nolabel => 1);
|
|
}
|
|
if (centreon::plugins::misc::minimal_version($self->{mode}->{version}, $self->{mode_version}) == 0) {
|
|
$self->{output}->add_option_msg(short_msg => "Not good version for plugin mode. Excepted at least: " . $self->{mode_version} . ". Get: ". $self->{mode}->{version});
|
|
$self->{output}->option_exit();
|
|
}
|
|
|
|
$self->{options}->parse_options();
|
|
$self->{option_results} = $self->{options}->get_options();
|
|
|
|
push @{$self->{sqlmode_stored}}, $self->{sqlmode_current};
|
|
$self->{sqlmode_current}->set_options(option_results => $self->{option_results});
|
|
$self->{sqlmode_current}->set_defaults(default => $self->{sqldefault});
|
|
|
|
while ($self->{sqlmode_current}->check_options()) {
|
|
$self->{sqlmode_current} = $self->{sql_modes}{$self->{sqlmode_name}}->new(noptions => 1, options => $self->{options}, output => $self->{output}, mode => $self->{sqlmode_name});
|
|
$self->{sqlmode_current}->set_options(option_results => $self->{option_results});
|
|
push @{$self->{sqlmode_stored}}, $self->{sqlmode_current};
|
|
}
|
|
$self->{mode}->check_options(option_results => $self->{option_results}, default => $self->{default});
|
|
}
|
|
|
|
sub run {
|
|
my $self = shift;
|
|
|
|
if ($self->{output}->is_disco_format()) {
|
|
$self->{mode}->disco_format();
|
|
$self->{output}->display_disco_format();
|
|
$self->{output}->exit(exit_litteral => 'ok');
|
|
}
|
|
|
|
if ($self->{output}->is_disco_show()) {
|
|
if (defined($self->{multiple})) {
|
|
$self->{mode}->disco_show(sql => $self->{sqlmode});
|
|
} else {
|
|
$self->{mode}->disco_show(sql => $self->{sqlmode_stored}[0]);
|
|
}
|
|
$self->{output}->display_disco_show();
|
|
$self->{output}->exit(exit_litteral => 'ok');
|
|
} else {
|
|
if (defined($self->{multiple})) {
|
|
$self->{mode}->run(sql => $self->{sqlmode_stored});
|
|
} else {
|
|
$self->{mode}->run(sql => $self->{sqlmode_stored}[0]);
|
|
}
|
|
}
|
|
}
|
|
|
|
sub is_mode {
|
|
my ($self, %options) = @_;
|
|
|
|
# $options->{mode} = mode
|
|
if (!defined($self->{modes}{$options{mode}})) {
|
|
$self->{output}->add_option_msg(short_msg => "mode '" . $options{mode} . "' doesn't exist (use --list-mode option to show available modes).");
|
|
$self->{output}->option_exit();
|
|
}
|
|
}
|
|
|
|
sub is_sqlmode {
|
|
my ($self, %options) = @_;
|
|
|
|
# $options->{sqlmode} = mode
|
|
if (!defined($self->{sql_modes}{$options{sqlmode}})) {
|
|
$self->{output}->add_option_msg(short_msg => "mode '" . $options{sqlmode} . "' doesn't exist (use --list-sqlmode option to show available modes).");
|
|
$self->{output}->option_exit();
|
|
}
|
|
}
|
|
|
|
|
|
sub version {
|
|
my $self = shift;
|
|
$self->{output}->add_option_msg(short_msg => "Plugin Version: " . $self->{version});
|
|
$self->{output}->option_exit(nolabel => 1);
|
|
}
|
|
|
|
sub list_mode {
|
|
my $self = shift;
|
|
$self->{options}->display_help();
|
|
|
|
$self->{output}->add_option_msg(long_msg => "Modes Available:");
|
|
foreach (keys %{$self->{modes}}) {
|
|
$self->{output}->add_option_msg(long_msg => " " . $_);
|
|
}
|
|
$self->{output}->option_exit(nolabel => 1);
|
|
}
|
|
|
|
sub list_sqlmode {
|
|
my $self = shift;
|
|
$self->{options}->display_help();
|
|
|
|
$self->{output}->add_option_msg(long_msg => "SQL Modes Available:");
|
|
foreach (keys %{$self->{sql_modes}}) {
|
|
$self->{output}->add_option_msg(long_msg => " " . $_);
|
|
}
|
|
$self->{output}->option_exit(nolabel => 1);
|
|
}
|
|
|
|
1;
|
|
|
|
__END__
|
|
|
|
=head1 NAME
|
|
|
|
-
|
|
|
|
=head1 SYNOPSIS
|
|
|
|
-
|
|
|
|
=head1 GLOBAL OPTIONS
|
|
|
|
=over 8
|
|
|
|
=item B<--mode>
|
|
|
|
Choose a mode.
|
|
|
|
=item B<--list-mode>
|
|
|
|
List available modes.
|
|
|
|
=item B<--version>
|
|
|
|
Display plugin version.
|
|
|
|
=item B<--mode-version>
|
|
|
|
Check minimal version of mode. If not, unknown error.
|
|
|
|
=item B<--dyn-mode>
|
|
|
|
Specify a mode with the path (separated by '::').
|
|
|
|
=item B<--sqlmode>
|
|
|
|
Choose a sql mode (Default: "dbi").
|
|
|
|
=item B<--list-sqlmode>
|
|
|
|
List available sql modes.
|
|
|
|
=item B<--multiple>
|
|
|
|
Multiple database connections (some mode needs it).
|
|
|
|
=back
|
|
|
|
=head1 DESCRIPTION
|
|
|
|
B<>.
|
|
|
|
=cut
|