projects/smsdaemon/PluginManager.cpp

        
#include "PluginManager.h"

#include "Logger.h"

#include "plugins/EchoPlugin.h"
#include "plugins/SpamPlugin.h"
#include "plugins/ShellExecPlugin.h"
#include "plugins/TogPlugin.h"
#include "plugins/StatusPlugin.h"
#include "plugins/ListPlugin.h"
#include "plugins/HostStatusPlugin.h"

typedef std::map<std::string, Plugin*>::iterator MapIterator;

PluginManager::PluginManager()
{
}
        
PluginManager::~PluginManager()
{
}


void PluginManager::AddPlugin(Plugin* plugin)
{
        if (plugin != 0)
        {
                if (plugin->IsHelper())
                {
                        _helper_plugins.push_back(plugin);
                }
                else
                {
                        std::string command = plugin->GetCommand();

                        if ( _plugins[ command ] == 0)
                                _plugins[ command ] = plugin;
                        else
                                Logger::logMessage( std::string("AddPlugin() -- already have a plugin called ") + command);
                }
        }
        else
        {
                Logger::logMessage("AddPlugin() -- cannot register a null pointer");
        }
}

void PluginManager::LoadPlugins()
{
        static EchoPlugin echo;

        static SpamPlugin spam;

        static ShellExecPlugin wake("wake", "/home/torben/bin/wake", false, false);
        static ShellExecPlugin ping("ping", "nmap -sP 192.168.10.1-255 | grep appears | awk '{print $2}'", false,true);

        static TogPlugin tog;

        static StatusPlugin status;

        static ListPlugin list;

        static HostStatusPlugin hoststatus;

        for(MapIterator it = _plugins.begin(); it != _plugins.end(); ++it)
        {
                Plugin* pl = (*it).second;
                if (pl != 0)
                        Logger::logMessage( std::string("Loaded plugin \"") + pl->GetCommand() + "\" - " + pl->GetDescription() );
        }
}

Plugin* PluginManager::GetPlugin(const std::string& pluginname)
{
        return _plugins[ pluginname ];
}


std::vector<Plugin*> PluginManager::GetPluginList()
{
        typedef std::map<std::string, Plugin*>::iterator MapIterator;
        std::vector<Plugin*> plugin_list;

        for (MapIterator it = _plugins.begin(); it != _plugins.end(); ++it)
        {
                Plugin* pl = (*it).second;
                plugin_list.push_back(pl);
        }


        return plugin_list;
}
1	torben	26
2			#include "PluginManager.h"
3
4	torben	157	#include "Logger.h"
5	torben	26
6	torben	132	#include "plugins/EchoPlugin.h"
7			#include "plugins/SpamPlugin.h"
8			#include "plugins/ShellExecPlugin.h"
9			#include "plugins/TogPlugin.h"
10			#include "plugins/StatusPlugin.h"
11			#include "plugins/ListPlugin.h"
12	torben	137	#include "plugins/HostStatusPlugin.h"
13	torben	26
14			typedef std::map<std::string, Plugin*>::iterator MapIterator;
15
16			PluginManager::PluginManager()
17			{
18			}
19
20			PluginManager::~PluginManager()
21			{
22			}
23
24
25	torben	28	void PluginManager::AddPlugin(Plugin* plugin)
26			{
27	torben	35	if (plugin != 0)
28			{
29	torben	184	if (plugin->IsHelper())
30			{
31			_helper_plugins.push_back(plugin);
32			}
33			else
34			{
35			std::string command = plugin->GetCommand();
36	torben	35
37	torben	184	if ( _plugins[ command ] == 0)
38			_plugins[ command ] = plugin;
39			else
40			Logger::logMessage( std::string("AddPlugin() -- already have a plugin called ") + command);
41			}
42	torben	35	}
43			else
44			{
45	torben	157	Logger::logMessage("AddPlugin() -- cannot register a null pointer");
46	torben	35	}
47	torben	28	}
48
49	torben	26	void PluginManager::LoadPlugins()
50			{
51			static EchoPlugin echo;
52	torben	28
53			static SpamPlugin spam;
54	torben	26
55	torben	107	static ShellExecPlugin wake("wake", "/home/torben/bin/wake", false, false);
56			static ShellExecPlugin ping("ping", "nmap -sP 192.168.10.1-255 \| grep appears \| awk '{print $2}'", false,true);
57	torben	35
58	torben	49	static TogPlugin tog;
59
60	torben	101	static StatusPlugin status;
61
62	torben	114	static ListPlugin list;
63
64	torben	137	static HostStatusPlugin hoststatus;
65
66	torben	26	for(MapIterator it = _plugins.begin(); it != _plugins.end(); ++it)
67			{
68			Plugin* pl = (*it).second;
69	torben	35	if (pl != 0)
70	torben	157	Logger::logMessage( std::string("Loaded plugin \"") + pl->GetCommand() + "\" - " + pl->GetDescription() );
71	torben	26	}
72			}
73
74			Plugin* PluginManager::GetPlugin(const std::string& pluginname)
75			{
76			return _plugins[ pluginname ];
77			}
78
79
80			std::vector<Plugin*> PluginManager::GetPluginList()
81			{
82			typedef std::map<std::string, Plugin*>::iterator MapIterator;
83			std::vector<Plugin*> plugin_list;
84
85			for (MapIterator it = _plugins.begin(); it != _plugins.end(); ++it)
86			{
87			Plugin* pl = (*it).second;
88			plugin_list.push_back(pl);
89			}
90
91
92			return plugin_list;
93			}