icingaweb2/library/Icinga/Protocol/Statusdat/TreeToStatusdatQueryParser.php

101 lines
3.4 KiB
PHP

<?php
// {{{ICINGA_LICENSE_HEADER}}}
/**
* This file is part of Icinga Web 2.
*
* Icinga Web 2 - Head for multiple monitoring backends.
* Copyright (C) 2013 Icinga Development Team
*
* 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.
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, write to the Free Software
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
*
* @copyright 2013 Icinga Development Team <info@icinga.org>
* @license http://www.gnu.org/licenses/gpl-2.0.txt GPL, version 2
* @author Icinga Development Team <info@icinga.org>
*
*/
// {{{ICINGA_LICENSE_HEADER}}}
namespace Icinga\Protocol\Statusdat;
use Icinga\Filter\Filterable;
use Icinga\Filter\Query\Node;
use Icinga\Filter\Query\Tree;
use Icinga\Protocol\Statusdat\Query\Expression;
use Icinga\Protocol\Statusdat\Query\Group;
use Icinga\Protocol\Statusdat\Query\IQueryPart;
/**
* Parser to create statusdat filter expressions from query trees
*
*/
class TreeToStatusdatQueryParser
{
/**
* Create a Statusdat expression from a Tree node
*
* @param Node $node The node to convert to an expression
* @param Filterable $source The filterable to use for field mapping
*
* @return IQueryPart Either a statusdat expression or an expression group
*/
private function nodeToQuery(Node $node, Filterable $source)
{
if ($node->type === Node::TYPE_OPERATOR) {
$op = $node->operator;
$node->left = $source->getMappedField($node->left);
$op = 'IN';
$values = $node->right;
if ($node->operator === NODE::OPERATOR_EQUALS_NOT) {
$op = 'NOT_' . $op;
}
foreach ($values as &$value) {
$value = str_replace('*', '%', $value);
}
$values = array($values);
return new Expression($node->left . ' ' . $op . ' ? ', $values);
} else {
$group = new Group();
$group->setType(($node->type === Node::TYPE_OR) ? Group::TYPE_OR : Group::TYPE_AND);
$group->addItem($this->nodeToQuery($node->left, $source));
$group->addItem($this->nodeToQuery($node->right, $source));
return $group;
}
}
/**
* Create a statusdat specific filter expression for the given query tree and filterable
*
* @param Tree $tree The tree to convert to a query
* @param Filterable $source The filterable to use for tree conversion
*
* @return IQueryPart A statusdat query object
*/
public function treeToQuery(Tree $tree, Filterable $source)
{
$tree = $tree->getCopyForFilterable($source);
if ($tree->root !== null) {
$tree->root = $tree->normalizeTree($tree->root);
return $this->nodeToQuery($tree->root, $source);
}
return null;
}
}