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 |
|
|
std::string command = plugin->GetCommand(); |
30 |
|
|
|
31 |
|
|
if ( _plugins[ command ] == 0) |
32 |
|
|
_plugins[ command ] = plugin; |
33 |
|
|
else |
34 |
torben |
157 |
Logger::logMessage( std::string("AddPlugin() -- already have a plugin called ") + command); |
35 |
torben |
35 |
} |
36 |
|
|
else |
37 |
|
|
{ |
38 |
torben |
157 |
Logger::logMessage("AddPlugin() -- cannot register a null pointer"); |
39 |
torben |
35 |
} |
40 |
torben |
28 |
} |
41 |
|
|
|
42 |
torben |
26 |
void PluginManager::LoadPlugins() |
43 |
|
|
{ |
44 |
|
|
static EchoPlugin echo; |
45 |
torben |
28 |
|
46 |
|
|
static SpamPlugin spam; |
47 |
torben |
26 |
|
48 |
torben |
107 |
static ShellExecPlugin wake("wake", "/home/torben/bin/wake", false, false); |
49 |
|
|
static ShellExecPlugin ping("ping", "nmap -sP 192.168.10.1-255 | grep appears | awk '{print $2}'", false,true); |
50 |
torben |
35 |
|
51 |
torben |
49 |
static TogPlugin tog; |
52 |
|
|
|
53 |
torben |
101 |
static StatusPlugin status; |
54 |
|
|
|
55 |
torben |
114 |
static ListPlugin list; |
56 |
|
|
|
57 |
torben |
137 |
static HostStatusPlugin hoststatus; |
58 |
|
|
|
59 |
torben |
26 |
for(MapIterator it = _plugins.begin(); it != _plugins.end(); ++it) |
60 |
|
|
{ |
61 |
|
|
Plugin* pl = (*it).second; |
62 |
torben |
35 |
if (pl != 0) |
63 |
torben |
157 |
Logger::logMessage( std::string("Loaded plugin \"") + pl->GetCommand() + "\" - " + pl->GetDescription() ); |
64 |
torben |
26 |
} |
65 |
|
|
} |
66 |
|
|
|
67 |
|
|
Plugin* PluginManager::GetPlugin(const std::string& pluginname) |
68 |
|
|
{ |
69 |
|
|
return _plugins[ pluginname ]; |
70 |
|
|
} |
71 |
|
|
|
72 |
|
|
|
73 |
|
|
std::vector<Plugin*> PluginManager::GetPluginList() |
74 |
|
|
{ |
75 |
|
|
typedef std::map<std::string, Plugin*>::iterator MapIterator; |
76 |
|
|
std::vector<Plugin*> plugin_list; |
77 |
|
|
|
78 |
|
|
for (MapIterator it = _plugins.begin(); it != _plugins.end(); ++it) |
79 |
|
|
{ |
80 |
|
|
Plugin* pl = (*it).second; |
81 |
|
|
plugin_list.push_back(pl); |
82 |
|
|
} |
83 |
|
|
|
84 |
|
|
|
85 |
|
|
return plugin_list; |
86 |
|
|
} |