/****************************************************************************** * Icinga 2 * * Copyright (C) 2012 Icinga Development Team (http://www.icinga.org/) * * * * 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 "i2-base.h" using namespace icinga; boost::signal ScriptFunction::OnRegistered; boost::signal ScriptFunction::OnUnregistered; ScriptFunction::ScriptFunction(const Callback& function) : m_Callback(function), m_ArgumentCount(-1) { } void ScriptFunction::Register(const String& name, const ScriptFunction::Ptr& function) { GetFunctions()[name] = function; Application::GetEQ().Post(boost::bind(boost::ref(OnRegistered), name, function)); } void ScriptFunction::Unregister(const String& name) { GetFunctions().erase(name); Application::GetEQ().Post(boost::bind(boost::ref(OnUnregistered), name)); } ScriptFunction::Ptr ScriptFunction::GetByName(const String& name) { map::iterator it; it = GetFunctions().find(name); if (it == GetFunctions().end()) return ScriptFunction::Ptr(); return it->second; } void ScriptFunction::Invoke(const ScriptTask::Ptr& task, const vector& arguments) { m_Callback(task, arguments); } map& ScriptFunction::GetFunctions(void) { static map functions; return functions; } void ScriptFunction::SetArgumentCount(int count) { if (m_ArgumentCount >= 0) m_ArgumentHints.resize(count); m_ArgumentCount = count; } int ScriptFunction::GetArgumentCount(void) const { return m_ArgumentCount; } void ScriptFunction::SetArgumentHint(int index, const ScriptArgumentHint& hint) { assert(index >= 0 && index < m_ArgumentCount); m_ArgumentHints[index] = hint; } ScriptArgumentHint ScriptFunction::GetArgumentHint(int index) const { if (m_ArgumentCount == -1 || index >= m_ArgumentCount) return ScriptArgumentHint(); assert(index >= 0 && index < m_ArgumentHints.size()); return m_ArgumentHints[index]; } void ScriptFunction::SetReturnHint(const ScriptArgumentHint& hint) { m_ReturnHint = hint; } ScriptArgumentHint ScriptFunction::GetReturnHint(void) const { return m_ReturnHint; }