/[projects]/smsdaemon/PluginManager.cpp
ViewVC logotype

Diff of /smsdaemon/PluginManager.cpp

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

revision 26 by torben, Mon Jun 9 18:15:53 2008 UTC revision 207 by torben, Sun Dec 21 17:42:30 2008 UTC
# Line 1  Line 1 
           
1  #include "PluginManager.h"  #include "PluginManager.h"
2    
3  #include "common.h"  #include "Logger.h"
4    #include "ConfigFile.h"
5    #include "Common.h"
6    #include "Util.h"
7    
8  #include "plugins/EchoPlugin.h"  #include "plugins/EchoPlugin.h"
9    #include "plugins/SpamPlugin.h"
10    #include "plugins/ShellExecPlugin.h"
11    #include "plugins/TogPlugin.h"
12    #include "plugins/StatusPlugin.h"
13    #include "plugins/ListPlugin.h"
14    #include "plugins/HostStatusPlugin.h"
15    #include "plugins/WeatherPlugin.h"
16    #include "plugins/UrlTriggerPlugin.h"
17    
18  typedef std::map<std::string, Plugin*>::iterator MapIterator;  typedef std::map<std::string, Plugin*>::iterator MapIterator;
19    
20  PluginManager::PluginManager()  PluginManager::PluginManager()
21  {  {
22  }  }
23            
24  PluginManager::~PluginManager()  PluginManager::~PluginManager()
25  {  {
26  }  }
27    
28    
29  void PluginManager::LoadPlugins()  void PluginManager::AddPlugin(Plugin* plugin)
30    {
31            if (plugin != 0)
32            {
33                    if (plugin->IsHelper())
34                    {
35                            _helper_plugins.push_back(plugin);
36                    }
37                    else
38                    {
39                            std::string command = plugin->GetCommand();
40    
41                            if ( _plugins[ command ] == 0)
42                                    _plugins[ command ] = plugin;
43                            else
44                                    Logger::logMessage( std::string("AddPlugin() -- already have a plugin called ") + command);
45                    }
46            }
47            else
48            {
49                    Logger::logMessage("AddPlugin() -- cannot register a null pointer");
50            }
51    }
52    
53    Plugin* PluginManager::CreatePlugin(const std::string& pluginName, const std::map<std::string,std::string>& args)
54  {  {
55          Common* cmn = Common::instance();          if (pluginName == "echo")
56                    return new EchoPlugin();
57    
58            if (pluginName == "spam")
59                    return new SpamPlugin();
60    
61            if (pluginName == "tog")
62                    return new TogPlugin();
63    
64            if (pluginName == "status")
65                    return new StatusPlugin();
66    
67            if (pluginName == "list")
68                    return new ListPlugin();
69    
70            if (pluginName == "hoststatus")
71                    return new HostStatusPlugin();
72    
73            if (pluginName == "weather")
74                    return new WeatherPlugin();
75    
76          static EchoPlugin echo;          if (pluginName == "shellexec")
77          _plugins[ echo.GetCommand() ] = &echo;                  return new ShellExecPlugin(args);
78    
79            if (pluginName == "urltrigger")
80                    return new UrlTriggerPlugin(args);
81                    
82            return 0;
83    }
84    
85    void PluginManager::LoadPlugins()
86    {
87    /*
88    
89            static ShellExecPlugin wake("wake", "/usr/local/bin/wake", false, false);
90            static ShellExecPlugin ping("ping", "nmap -sP 192.168.10.1-255 | grep appears | awk '{print $2}'", false,true);
91    
92    
93    
94    
95          for(MapIterator it = _plugins.begin(); it != _plugins.end(); ++it)          static UrlTriggerPlugin urltrigger("tou", "http://t-hoerup.dk/test.php");
96    */
97    
98            Logger::logMessage("-------- PluginList --------");
99            std::vector<Value> pluginlist = Common::instance()->GetConfigfile()->GetValues("smsdaemon", "plugin");
100    
101            for (unsigned i=0; i<pluginlist.size(); i++)
102            {
103                    std::string current = pluginlist[i];
104                    
105                    std::string name;
106                    unsigned pos = current.find(' ');
107                    std::map<std::string,std::string> args;
108    
109                    std::string argstr;
110    
111                    if (pos == std::string::npos)
112                    {
113                            name = current;
114                    }
115                    else
116                    {
117                            name = Util::str_trim(current.substr(0,pos));
118                            argstr = Util::str_trim(current.substr(pos+1,1024));
119                            args = ConfigHelper::ParseArguments(argstr);
120                    }
121    
122                    Plugin* pl = 0;
123                    try
124                    {      
125                            pl = CreatePlugin(name, args );
126                    }
127                    catch (std::exception& e)
128                    {
129                            Logger::logMessage(std::string("Failed to load plugin ") + name + " with args: " + argstr);
130                            Logger::logMessage(std::string("Reason: ") + e.what());
131                            continue;
132                    }
133    
134                    if (pl)
135                            AddPlugin(pl);
136                    else
137                            Logger::logMessage( std::string("Unknown plugin: ")+name);
138                    
139            }      
140    
141    
142            for (MapIterator it = _plugins.begin(); it != _plugins.end(); ++it)
143          {          {
144                  Plugin* pl = (*it).second;                  Plugin* pl = (*it).second;
145                  cmn->logMessage( std::string("Loaded plugin \"") + pl->GetCommand() + "\" - " + pl->GetDescription() );                  if (pl != 0)
146                            Logger::logMessage( std::string("Loaded plugin \"") + pl->GetCommand() + "\" - " + pl->GetDescription() );
147          }          }
148    }
149    
150    void PluginManager::DestroyPlugins()
151    {
152  }  }
153    
154  Plugin* PluginManager::GetPlugin(const std::string& pluginname)  Plugin* PluginManager::GetPlugin(const std::string& pluginname)

Legend:
Removed from v.26  
changed lines
  Added in v.207

  ViewVC Help
Powered by ViewVC 1.1.20