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

Diff of /smsdaemon/PluginManager.cpp

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

revision 56 by torben, Wed Jun 11 15:55:18 2008 UTC revision 520 by torben, Sat Dec 26 23:01:01 2009 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    
9    #include "ProxyPlugin.h"
10    #include "plugins/EchoPlugin.h"
11    #include "plugins/SpamPlugin.h"
12    #include "plugins/ShellExecPlugin.h"
13    #include "plugins/TogPlugin.h"
14    #include "plugins/StatusPlugin.h"
15    #include "plugins/ListPlugin.h"
16    #include "plugins/HostStatusPlugin.h"
17    #include "plugins/WeatherPlugin.h"
18    #include "plugins/UrlTriggerPlugin.h"
19    #include "plugins/DelaySpamPlugin.h"
20    
21    
22  #include "EchoPlugin.h"  #include "AccessManager.h"
 #include "SpamPlugin.h"  
 #include "ShellExecPlugin.h"  
 #include "TogPlugin.h"  
23    
24  typedef std::map<std::string, Plugin*>::iterator MapIterator;  typedef std::map<std::string, Plugin*>::iterator MapIterator;
25    
26  PluginManager::PluginManager()  PluginManager::PluginManager()
27  {  {
28  }  }
29            
30  PluginManager::~PluginManager()  PluginManager::~PluginManager()
31  {  {
32  }  }
# Line 21  PluginManager::~PluginManager() Line 34  PluginManager::~PluginManager()
34    
35  void PluginManager::AddPlugin(Plugin* plugin)  void PluginManager::AddPlugin(Plugin* plugin)
36  {  {
         Common* cmn = Common::instance();  
   
37          if (plugin != 0)          if (plugin != 0)
38          {          {
39                  std::string command = plugin->GetCommand();                  if (plugin->IsHelper())
40                    {
41                  if ( _plugins[ command ] == 0)                          _helper_plugins.push_back(plugin);
42                          _plugins[ command ] = plugin;                  }
43                  else                  else
44                          cmn->logMessage( std::string("AddPlugin() -- already have a plugin called ") + command);                  {
45                            std::string command = plugin->GetCommand();
46    
47                            if ( _plugins[ command ] == 0)
48                                    _plugins[ command ] = plugin;
49                            else
50                                    Logger::logMessage( std::string("AddPlugin() -- already have a plugin called ") + command);
51                    }
52          }          }
53          else          else
54          {          {
55                  cmn->logMessage("AddPlugin() -- cannot register a null pointer");                  Logger::logMessage("AddPlugin() -- cannot register a null pointer");
56          }          }
57  }  }
58    
59  void PluginManager::LoadPlugins()  Plugin* PluginManager::CreatePlugin(const std::string& pluginName, const std::map<std::string,std::string>& args)
60  {  {
61          Common* cmn = Common::instance();          if (pluginName == "echo")
62                    return new EchoPlugin();
63    
64            if (pluginName == "spam")
65                    return new SpamPlugin();
66    
67            if (pluginName == "tog")
68                    return new TogPlugin();
69    
70            if (pluginName == "status")
71                    return new StatusPlugin();
72    
73          static EchoPlugin echo;          if (pluginName == "list")
74                    return new ListPlugin();
75    
76          static SpamPlugin spam;          if (pluginName == "hoststatus")
77                    return new HostStatusPlugin();
78    
79          static ShellExecPlugin wake("wake", "/home/torben/bin/wake", false);          if (pluginName == "weather")
80                    return new WeatherPlugin();
81    
82          static TogPlugin tog;          if (pluginName == "shellexec")
83                    return new ShellExecPlugin(args);
84    
85          for(MapIterator it = _plugins.begin(); it != _plugins.end(); ++it)          if (pluginName == "urltrigger")
86                    return new UrlTriggerPlugin(args);
87    
88            if (pluginName == "delayspam")
89                    return new DelaySpamPlugin();
90            
91            return 0;
92    }
93    
94    
95    void PluginManager::DestroyPlugins()
96    {
97            for (std::map<std::string,Plugin*>::iterator it=_plugins.begin(); it!=_plugins.end(); ++it)
98            {
99                    delete it->second;
100            }
101            _plugins.clear();
102    }
103    
104    void PluginManager::LoadPlugins()
105    {
106            Logger::logMessage("-------- PluginList --------");
107            std::vector<Value> pluginlist = Common::instance()->GetConfigfile()->GetValues("smsdaemon", "plugin");
108    
109            for (unsigned i=0; i<pluginlist.size(); i++)
110            {
111                    std::string current = pluginlist[i];
112                    
113                    std::string name;
114                    unsigned pos = current.find(' ');
115                    std::map<std::string,std::string> args;
116    
117                    std::string argstr;
118    
119                    if (pos == std::string::npos)
120                    {
121                            name = current;
122                    }
123                    else
124                    {
125                            name = Util::str_trim(current.substr(0,pos));
126                            argstr = Util::str_trim(current.substr(pos+1,1024));
127                            args = ConfigHelper::ParseArguments(argstr);
128                    }
129    
130                    Plugin* pl = 0;
131                    try
132                    {      
133                            pl = CreatePlugin(name, args );
134                    }
135                    catch (std::exception& e)
136                    {
137                            Logger::logMessage(std::string("Failed to load plugin ") + name + " with args: " + argstr);
138                            Logger::logMessage(std::string("Reason: ") + e.what());
139                            continue;
140                    }
141    
142                    if (pl)
143                    {
144                            AddPlugin(pl);
145                            ParseCommonOptions(name,args);
146                    }
147                    else
148                            Logger::logMessage( std::string("Unknown plugin: ")+name);
149                    
150            }      
151    
152    
153            for (MapIterator it = _plugins.begin(); it != _plugins.end(); ++it)
154          {          {
155                  Plugin* pl = (*it).second;                  Plugin* pl = (*it).second;
156                  if (pl != 0)                  if (pl != 0)
157                          cmn->logMessage( std::string("Loaded plugin \"") + pl->GetCommand() + "\" - " + pl->GetDescription() );                          Logger::logMessage( std::string("Loaded plugin \"") + pl->GetCommand() + "\" - " + pl->GetDescription() );
158            }
159    }
160    
161    void PluginManager::ParseCommonOptions(const std::string& pluginName, std::map<std::string,std::string>& args)
162    {
163            if (args["privileged"] == "1")
164            {
165                    AccessManager::AddPrivPlugin(pluginName);
166          }          }
167  }  }
168    
169  Plugin* PluginManager::GetPlugin(const std::string& pluginname)  Plugin* PluginManager::GetPlugin(const std::string& pluginname)
170  {  {
171          return _plugins[ pluginname ];          static ProxyPlugin proxy;
172    
173            std::map<std::string,Plugin*>::iterator it = _plugins.find(pluginname );
174    
175            if (it != _plugins.end() )
176            {
177                    proxy.SetPlugin(it->second);
178                    return &proxy;
179            }
180            else
181            {
182                    return 0;
183            }
184  }  }
185    
186    
Legend:
Removed from v.56  
changed lines
  Added in v.520
  ViewVC Help
Powered by ViewVC 1.1.20