mirror of https://github.com/Icinga/icinga2.git
73 lines
3.2 KiB
C++
73 lines
3.2 KiB
C++
/******************************************************************************
|
|
* Icinga 2 *
|
|
* Copyright (C) 2012-2017 Icinga Development Team (https://www.icinga.com/) *
|
|
* *
|
|
* 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 St, Fifth Floor, Boston, MA 02110-1301, USA. *
|
|
******************************************************************************/
|
|
|
|
#include "icinga/objectutils.hpp"
|
|
#include "icinga/host.hpp"
|
|
#include "icinga/user.hpp"
|
|
#include "icinga/checkcommand.hpp"
|
|
#include "icinga/eventcommand.hpp"
|
|
#include "icinga/notificationcommand.hpp"
|
|
#include "icinga/hostgroup.hpp"
|
|
#include "icinga/servicegroup.hpp"
|
|
#include "icinga/usergroup.hpp"
|
|
|
|
using namespace icinga;
|
|
|
|
REGISTER_SCRIPTFUNCTION_NS(System, get_host, &Host::GetByName, "name");
|
|
REGISTER_SCRIPTFUNCTION_NS(System, get_service, &ObjectUtils::GetService, "host:name");
|
|
REGISTER_SCRIPTFUNCTION_NS(System, get_services, &ObjectUtils::GetServices, "host");
|
|
REGISTER_SCRIPTFUNCTION_NS(System, get_user, &User::GetByName, "name");
|
|
REGISTER_SCRIPTFUNCTION_NS(System, get_check_command, &CheckCommand::GetByName, "name");
|
|
REGISTER_SCRIPTFUNCTION_NS(System, get_event_command, &EventCommand::GetByName, "name");
|
|
REGISTER_SCRIPTFUNCTION_NS(System, get_notification_command, &NotificationCommand::GetByName, "name");
|
|
REGISTER_SCRIPTFUNCTION_NS(System, get_host_group, &HostGroup::GetByName, "name");
|
|
REGISTER_SCRIPTFUNCTION_NS(System, get_service_group, &ServiceGroup::GetByName, "name");
|
|
REGISTER_SCRIPTFUNCTION_NS(System, get_user_group, &UserGroup::GetByName, "name");
|
|
REGISTER_SCRIPTFUNCTION_NS(System, get_time_period, &TimePeriod::GetByName, "name");
|
|
|
|
Service::Ptr ObjectUtils::GetService(const Value& host, const String& name)
|
|
{
|
|
Host::Ptr hostObj;
|
|
|
|
if (host.IsObjectType<Host>())
|
|
hostObj = host;
|
|
else
|
|
hostObj = Host::GetByName(host);
|
|
|
|
if (!hostObj)
|
|
return Service::Ptr();
|
|
|
|
return hostObj->GetServiceByShortName(name);
|
|
}
|
|
|
|
Array::Ptr ObjectUtils::GetServices(const Value& host)
|
|
{
|
|
Host::Ptr hostObj;
|
|
|
|
if (host.IsObjectType<Host>())
|
|
hostObj = host;
|
|
else
|
|
hostObj = Host::GetByName(host);
|
|
|
|
if (!hostObj)
|
|
return Array::Ptr();
|
|
|
|
return Array::FromVector(hostObj->GetServices());
|
|
}
|