mirror of
https://github.com/Cacti/cacti.git
synced 2026-11-08 21:16:17 +09:00
Larry Adams
b9cbbbdedf
git-svn-id: svn+ssh://repo.cacti.net/var/data/svnroot/cacti/branches/BRANCH_0_8_6/cacti@3900 860744bd-22fc-0310-8c96-e9fe5004b5ca
216 lines
7.2 KiB
PHP
216 lines
7.2 KiB
PHP
#!/usr/bin/php -q
|
|
<?php
|
|
/*
|
|
+-------------------------------------------------------------------------+
|
|
| Copyright (C) 2004 Ian Berry |
|
|
| |
|
|
| 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: a php-based graphing solution |
|
|
+-------------------------------------------------------------------------+
|
|
| Most of this code has been designed, written and is maintained by |
|
|
| Ian Berry. See about.php for specific developer credit. Any questions |
|
|
| or comments regarding this code should be directed to: |
|
|
| - iberry@raxnet.net |
|
|
+-------------------------------------------------------------------------+
|
|
| - raXnet - http://www.raxnet.net/ |
|
|
+-------------------------------------------------------------------------+
|
|
*/
|
|
|
|
/* do NOT run this script through a web browser */
|
|
if (!isset($_SERVER["argv"][0]) || isset($_SERVER['REQUEST_METHOD']) || isset($_SERVER['REMOTE_ADDR'])) {
|
|
die("<br><strong>This script is only meant to run at the command line.</strong>");
|
|
}
|
|
|
|
$no_http_headers = true;
|
|
|
|
include(dirname(__FILE__)."/include/config.php");
|
|
include_once($config["base_path"]."/lib/api_automation_tools.php");
|
|
include_once($config["base_path"].'/lib/tree.php');
|
|
|
|
if ($_SERVER["argc"] == 1) {
|
|
usage();
|
|
return(1);
|
|
}else{
|
|
$type = ''; # tree or node
|
|
$name = ''; # Name of a tree or node
|
|
$sortMethod = 'a'; # a = alpha, n = numeric, m = manual
|
|
$parentNode = 0; # When creating a node, the parent node of this node (or zero for root-node)
|
|
$treeId = 0; # When creating a node, it has to go in a tree
|
|
$nodeType = ''; # Should be 'header', 'graph' or 'host' when creating a node
|
|
$graphId = 0; # The ID of the graph to add (gets added to parentNode)
|
|
$rra_id = 0; # The rra_id for the graph to display: 1 = daily, 2 = weekly, 3 = monthly, 4 = yearly
|
|
|
|
$sortTypes = array('a' => 2, 'n' => 3, 'm' => 1);
|
|
$nodeTypes = array('header' => 1, 'graph' => 2, 'host' => 3);
|
|
|
|
$hostId = 0;
|
|
$hostGroupStyle = 1; # 1 = Graph Template, 2 = Data Query Index
|
|
|
|
$hosts = getHosts();
|
|
|
|
for ($i = 1; $i < $_SERVER["argc"]; $i++) {
|
|
switch($_SERVER["argv"][$i]) {
|
|
case "--type":
|
|
$i++;
|
|
$type = $_SERVER["argv"][$i];
|
|
break;
|
|
case "--name":
|
|
$i++;
|
|
$name = $_SERVER["argv"][$i];
|
|
break;
|
|
case "--sort-method":
|
|
$i++;
|
|
$sortMethod = $_SERVER["argv"][$i];
|
|
break;
|
|
case "--parent-node":
|
|
$i++;
|
|
$parentNode = $_SERVER["argv"][$i];
|
|
break;
|
|
case "--tree-id":
|
|
$i++;
|
|
$treeId = $_SERVER["argv"][$i];
|
|
break;
|
|
case "--node-type":
|
|
$i++;
|
|
$nodeType = $_SERVER["argv"][$i];
|
|
break;
|
|
case "--graph-id":
|
|
$i++;
|
|
$graphId = $_SERVER["argv"][$i];
|
|
break;
|
|
case "--rra-id":
|
|
$i++;
|
|
$rra_id = $_SERVER["argv"][$i];
|
|
break;
|
|
case "--host-id":
|
|
$i++;
|
|
$hostId = $_SERVER["argv"][$i];
|
|
break;
|
|
case "--list-hosts":
|
|
displayHosts($hosts);
|
|
return 0;
|
|
case "--host-group-style":
|
|
$i++;
|
|
$hostGroupStyle = $_SERVER["argv"][$i];
|
|
break;
|
|
default:
|
|
usage();
|
|
return 0;
|
|
}
|
|
}
|
|
|
|
if ($type == 'tree') {
|
|
# Add a new tree
|
|
if (empty($name)) {
|
|
printf("You must supply a name with --name\n");
|
|
return 1;
|
|
}
|
|
|
|
$treeOpts = array();
|
|
$treeOpts["id"] = 0; # Zero means create a new one rather than save over an existing one
|
|
$treeOpts["name"] = $name;
|
|
$treeOpts["sort_type"] = $sortTypes[$sortMethod];
|
|
|
|
if (!isset($treeOpts["sort_type"]) || empty($treeOpts["sort_type"])) {
|
|
printf("Invalid sort-method: %s\n", $sortMethod);
|
|
return 1;
|
|
}
|
|
|
|
$existsAlready = db_fetch_cell("select id from graph_tree where name = '$name'");
|
|
if ($existsAlready) {
|
|
printf("Not adding tree - it already exists - tree-id: (%d)\n", $existsAlready);
|
|
return 1;
|
|
}
|
|
|
|
$treeId = sql_save($treeOpts, "graph_tree");
|
|
|
|
sort_tree(SORT_TYPE_TREE, $treeId, $treeOpts["sort_type"]);
|
|
|
|
printf("Tree Created - tree-id: (%d)\n", $treeId);
|
|
|
|
return 0;
|
|
}else if ($type == 'node') {
|
|
# Add a new node to a tree
|
|
$itemType = $nodeTypes[$nodeType];
|
|
|
|
if (!isset($itemType) || empty($itemType)) {
|
|
printf("Invalid node-type: %s\n", $nodeType);
|
|
return 1;
|
|
}
|
|
|
|
if ($parentNode > 0) {
|
|
$parentNodeExists = db_fetch_cell("SELECT id FROM graph_tree_items WHERE graph_tree_id = $treeId AND id = $parentNode");
|
|
if (!isset($parentNodeExists)) {
|
|
printf("parent-node %d does not exist\n");
|
|
return 1;
|
|
}
|
|
}
|
|
|
|
if ($nodeType == 'header') {
|
|
# Blank out the graphId, rra_id, hostID and host_grouping_style fields
|
|
$graphId = 0;
|
|
$rra_id = 0;
|
|
$hostId = 0;
|
|
$hostGroupStyle = 1;
|
|
}else if($nodeType == 'graph') {
|
|
# Blank out name, hostID, host_grouping_style
|
|
$name = '';
|
|
$hostId = 0;
|
|
$hostGroupStyle = 1;
|
|
}else if ($nodeType == 'host') {
|
|
# Blank out graphId, rra_id, name fields
|
|
$graphId = 0;
|
|
$rra_id = 0;
|
|
$name = '';
|
|
|
|
if (!isset($hosts[$hostId])) {
|
|
printf("No such host-id (%s) exists. Try --list-hosts\n", $hostId);
|
|
return 1;
|
|
}
|
|
|
|
if ($hostGroupStyle != 1 && $hostGroupStyle != 2) {
|
|
printf("Host Group Style must be 1 or 2 (Graph Template or Data Query Index)\n");
|
|
return 1;
|
|
}
|
|
}
|
|
|
|
# $nodeId could be a Header Node, a Graph Node, or a Host node.
|
|
$nodeId = api_tree_item_save(0, $treeId, $itemType, $parentNode, $name, $graphId, $rra_id, $hostId, $hostGroupStyle, 1, false);
|
|
printf("Added Node node-id: (%d)\n", $nodeId);
|
|
return 0;
|
|
}else{
|
|
printf("Unknown type: $type\n");
|
|
usage();
|
|
return 1;
|
|
}
|
|
}
|
|
|
|
function usage() {
|
|
echo "Usage:\n";
|
|
echo "add_tree.php --type [tree|node] [type-options]\n\n";
|
|
echo "tree options: --name [Tree Name] --sort-method [a|n|m]\n";
|
|
echo "(sort methods: a = Alphabetic, n = numeric, m = manual)\n\n";
|
|
echo "node options: --node-type [header|graph|host] --tree-id [ID] --parent-node [ID] [Node Type Options]\n";
|
|
echo "header node options: --name [Name]\n";
|
|
echo "graph node options: --graph-id [ID] --rra-id [ID]\n";
|
|
echo "host node options: --host-id [ID] --host-group-style [1 | 2]\n";
|
|
echo "(Host group styles: 1 = Graph template, 2 = Data Query Index)\n\n";
|
|
echo "List Options: --list-hosts\n";
|
|
echo " --list-trees *\n";
|
|
echo " --tree-id [ID] --list-nodes *\n";
|
|
echo " --list-rras *\n";
|
|
echo " --host-id [ID] --list-graphs *\n\n";
|
|
echo "* = not yet implemented - sorry.\n";
|
|
}
|
|
|
|
?>
|