1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
|
#!/usr/bin/env php
<?php
/*
+-------------------------------------------------------------------------+
| Copyright (C) 2004-2024 The Cacti Group |
| |
| 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, or (at your option) any later version. |
| |
| 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. |
+-------------------------------------------------------------------------+
| Cacti: The Complete RRDtool-based Graphing Solution |
+-------------------------------------------------------------------------+
| This code is designed, written, and maintained by the Cacti Group. See |
| about.php and/or the AUTHORS file for specific developer information. |
+-------------------------------------------------------------------------+
| http://www.cacti.net/ |
+-------------------------------------------------------------------------+
*/
require(__DIR__ . '/../site/include/cli_check.php');
require_once($config['base_path'] . '/lib/api_automation_tools.php');
/* switch to main database for cli's */
if ($config['poller_id'] > 1) {
db_switch_remote_to_main();
}
/* process calling arguments */
$parms = $_SERVER['argv'];
array_shift($parms);
if (cacti_sizeof($parms)) {
$displayHosts = false;
$displayGraphTemplates = false;
$quietMode = false;
unset($host_id);
unset($graph_template_id);
foreach($parms as $parameter) {
if (strpos($parameter, '=')) {
list($arg, $value) = explode('=', $parameter, 2);
} else {
$arg = $parameter;
$value = '';
}
switch ($arg) {
case '-d':
$debug = true;
break;
case '--host-id':
$host_id = trim($value);
if (!is_numeric($host_id)) {
print "ERROR: You must supply a valid host-id to run this script!\n";
exit(1);
}
break;
case '--graph-template-id':
$graph_template_id = $value;
if (!is_numeric($graph_template_id)) {
print "ERROR: You must supply a numeric graph-template-id for all hosts!\n";
exit(1);
}
break;
case '--version':
case '-V':
case '-v':
display_version();
exit(0);
case '--help':
case '-H':
case '-h':
display_help();
exit(0);
case '--list-hosts':
$displayHosts = true;
break;
case '--list-graph-templates':
$displayGraphTemplates = true;
break;
case '--quiet':
$quietMode = true;
break;
default:
print "ERROR: Invalid Argument: ($arg)\n\n";
display_help();
exit(1);
}
}
/* list options, recognizing $quiteMode */
if ($displayHosts) {
$hosts = getHosts();
displayHosts($hosts, $quietMode);
exit(0);
}
if ($displayGraphTemplates) {
$graphTemplates = getGraphTemplates();
displayGraphTemplates($graphTemplates, $quietMode);
exit(0);
}
/*
* verify required parameters
* for update / insert options
*/
if (!isset($host_id)) {
print "ERROR: You must supply a valid host-id for all hosts!\n";
exit(1);
}
if (!isset($graph_template_id)) {
print "ERROR: You must supply a valid data-query-id for all hosts!\n";
exit(1);
}
/*
* verify valid host id and get a name for it
*/
$host_name = db_fetch_cell("SELECT hostname FROM host WHERE id = " . $host_id);
if (!isset($host_name)) {
print "ERROR: Unknown Host Id ($host_id)\n";
exit(1);
}
/*
* verify valid graph template and get a name for it
*/
$graph_template_name = db_fetch_cell("SELECT name FROM graph_templates WHERE id = " . $graph_template_id);
if (!isset($graph_template_name)) {
print "ERROR: Unknown Graph Template Id ($graph_template_id)\n";
exit(1);
}
/* check, if graph template was already associated */
$exists_already = db_fetch_cell("SELECT host_id FROM host_graph WHERE graph_template_id=$graph_template_id AND host_id=$host_id");
if ((isset($exists_already)) &&
($exists_already > 0)) {
print "ERROR: Graph Template is already associated for host: ($host_id: $host_name) - graph-template: ($graph_template_id: $graph_template_name)\n";
exit(1);
} else {
db_execute("replace into host_graph (host_id,graph_template_id) values (" . $host_id . "," . $graph_template_id . ")");
automation_hook_graph_template($host_id, $graph_template_id);
api_plugin_hook_function('add_graph_template_to_host', array("host_id" => $host_id, "graph_template_id" => $graph_template_id));
}
if (is_error_message()) {
print "ERROR: Failed to add this graph template for host: ($host_id: $host_name) - graph-template: ($graph_template_id: $graph_template_name)\n";
exit(1);
} else {
print "Success: Graph Template associated for host: ($host_id: $host_name) - graph-template: ($graph_template_id: $graph_template_name)\n";
exit(0);
}
} else {
display_help();
exit(0);
}
/* display_version - displays version information */
function display_version() {
$version = get_cacti_cli_version();
print "Cacti Add Graph Template Utility, Version $version, " . COPYRIGHT_YEARS . "\n";
}
function display_help() {
display_version();
print "\nusage: add_graph_template.php --host-id=[ID] --graph-template-id=[ID]\n";
print " [--quiet]\n\n";
print "Required:\n";
print " --host-id the numerical ID of the host\n";
print " --graph-template-id the numerical ID of the graph template to be added\n\n";
print "List Options:\n";
print " --list-hosts\n";
print " --list-graph-templates\n";
print " --quiet - batch mode value return\n\n";
}
|
