version 0.3.4
[fms.git] / src / fmsapp.cpp
index 04562fc..99712c5 100644 (file)
@@ -9,6 +9,7 @@
 #include "../include/dbmaintenancethread.h"\r
 #include "../include/freenet/freenetmasterthread.h"\r
 #include "../include/threadwrapper/threadedexecutor.h"\r
+#include "../include/db/sqlite3db.h"\r
 \r
 #include <Poco/Util/HelpFormatter.h>\r
 #include <Poco/FileChannel.h>\r
        #include <unistd.h>\r
 #endif\r
 \r
-//debug\r
-#include <Poco/ScopedLock.h>\r
-#include <Poco/Runnable.h>\r
-#include <Poco/Thread.h>\r
-#include <Poco/ThreadPool.h>\r
-#include "../include/threadwrapper/cancelablerunnable.h"\r
-#include "../include/threadwrapper/cancelablethread.h"\r
-\r
-Poco::FastMutex m1;\r
-class TestRunner:public CancelableRunnable\r
-{\r
-public:\r
-       void run()\r
-       {\r
-               do\r
-               {\r
-\r
-               }while(!IsCancelled());\r
-       }\r
-};\r
-\r
-FMSApp::FMSApp():m_displayhelp(false),m_logtype("file")\r
+FMSApp::FMSApp():m_displayhelp(false),m_showoptions(false),m_setoption(false),m_logtype("file")\r
 {\r
 \r
 }\r
@@ -54,8 +34,10 @@ void FMSApp::defineOptions(Poco::Util::OptionSet &options)
        ServerApplication::defineOptions(options);\r
 \r
        // add command line options here\r
-       options.addOption(Poco::Util::Option("help","?","display help for command line arguments",false).repeatable(false).callback(Poco::Util::OptionCallback<FMSApp>(this,&FMSApp::handleHelp)));\r
-       options.addOption(Poco::Util::Option("log","l","select type of log output (file|stdout|stderr)",false,"type",true).repeatable(false).callback(Poco::Util::OptionCallback<FMSApp>(this,&FMSApp::handleLogOption)));\r
+       options.addOption(Poco::Util::Option("help","?","Display help for command line arguments.",false).repeatable(false).callback(Poco::Util::OptionCallback<FMSApp>(this,&FMSApp::handleHelp)));\r
+       options.addOption(Poco::Util::Option("log","l","Select type of log output (file|stdout|stderr).",false,"type",true).repeatable(false).callback(Poco::Util::OptionCallback<FMSApp>(this,&FMSApp::handleLogOption)));\r
+       options.addOption(Poco::Util::Option("showoptions","","Show all options that can be set and their current values.",false).repeatable(false).callback(Poco::Util::OptionCallback<FMSApp>(this,&FMSApp::handleShowOptions)));\r
+       options.addOption(Poco::Util::Option("setoption","","Set an option.  Values are not validated, so be sure to set them correctly.",false,"option=value",true).repeatable(true).callback(Poco::Util::OptionCallback<FMSApp>(this,&FMSApp::handleSetOption)));\r
 }\r
 \r
 void FMSApp::displayHelp()\r
@@ -82,6 +64,28 @@ void FMSApp::handleLogOption(const std::string &name, const std::string &value)
        }\r
 }\r
 \r
+void FMSApp::handleSetOption(const std::string &name, const std::string &value)\r
+{\r
+       std::vector<std::string> valueparts;\r
+       StringFunctions::Split(value,"=",valueparts);\r
+\r
+       if(valueparts.size()==2)\r
+       {\r
+               m_setoptions[valueparts[0]]=valueparts[1];\r
+       }\r
+       else\r
+       {\r
+               std::cout << "Expected option=value but found " << value << std::endl;\r
+       }\r
+\r
+       m_setoption=true;\r
+}\r
+\r
+void FMSApp::handleShowOptions(const std::string &name, const std::string &value)\r
+{\r
+       m_showoptions=true;\r
+}\r
+\r
 void FMSApp::initialize(Poco::Util::Application &self)\r
 {\r
        ServerApplication::initialize(self);\r
@@ -131,7 +135,9 @@ void FMSApp::initializeLogger()
                }\r
        }\r
        \r
-       setLogger(Poco::Logger::create("logfile",formatter,initiallevel));\r
+       setLogger(Poco::Logger::create("logfile",formatter,Poco::Message::PRIO_INFORMATION));\r
+       Poco::Logger::get("logfile").information("LogLevel set to "+tempval);\r
+       Poco::Logger::get("logfile").setLevel(initiallevel);\r
 }\r
 \r
 int FMSApp::main(const std::vector<std::string> &args)\r
@@ -142,7 +148,18 @@ int FMSApp::main(const std::vector<std::string> &args)
        // set working directory to program directory\r
        int rval=chdir(config().getString("application.dir").c_str());\r
 \r
-       if(m_displayhelp==false)\r
+       if(m_displayhelp)\r
+       {\r
+       }\r
+       else if(m_showoptions)\r
+       {\r
+               showOptions();\r
+       }\r
+       else if(m_setoption)\r
+       {\r
+               setOptions();\r
+       }\r
+       else\r
        {\r
                logger().information("FMS startup v"FMS_VERSION);\r
 \r
@@ -155,7 +172,9 @@ int FMSApp::main(const std::vector<std::string> &args)
 \r
                waitForTerminationRequest();\r
 \r
+               logger().trace("FMSApp::main cancelling threads");\r
                m_threads.Cancel();\r
+               logger().trace("FMSApp::main joining threads");\r
                m_threads.Join();\r
 \r
                logger().information("FMS shutdown");\r
@@ -164,16 +183,52 @@ int FMSApp::main(const std::vector<std::string> &args)
        return FMSApp::EXIT_OK;\r
 }\r
 \r
+void FMSApp::setOptions()\r
+{\r
+       for(std::map<std::string,std::string>::iterator i=m_setoptions.begin(); i!=m_setoptions.end(); i++)\r
+       {\r
+               std::string tempval="";\r
+               if(Option::Instance()->Get((*i).first,tempval))\r
+               {\r
+                       Option::Instance()->Set((*i).first,(*i).second);\r
+                       std::cout << "Option " << (*i).first << " set to " << (*i).second << std::endl;\r
+               }\r
+               else\r
+               {\r
+                       std::cout << "Unknown option " << (*i).first << std::endl;\r
+               }\r
+       }\r
+}\r
+\r
+void FMSApp::showOptions()\r
+{\r
+       SQLite3DB::Statement st=SQLite3DB::DB::Instance()->Prepare("SELECT Option, OptionValue FROM tblOption;");\r
+       st.Step();\r
+       while(st.RowReturned())\r
+       {\r
+               std::string option="";\r
+               std::string optionvalue="";\r
+               st.ResultText(0,option);\r
+               st.ResultText(1,optionvalue);\r
+\r
+               std::cout << option << " = " << optionvalue << std::endl;\r
+\r
+               st.Step();\r
+       }\r
+}\r
+\r
 void FMSApp::StartThreads()\r
 {\r
        std::string tempval="";\r
 \r
        // always start the DB maintenance thread\r
+       logger().trace("FMSApp::StartThreads starting DBMaintenanceThread");\r
        m_threads.Start(new DBMaintenanceThread());\r
 \r
        Option::Instance()->Get("StartHTTP",tempval);\r
        if(tempval=="true")\r
        {\r
+               logger().trace("FMSApp::StartThreads starting HTTPThread");\r
                m_threads.Start(new HTTPThread());\r
        }\r
 \r
@@ -181,6 +236,7 @@ void FMSApp::StartThreads()
        Option::Instance()->Get("StartNNTP",tempval);\r
        if(tempval=="true")\r
        {\r
+               logger().trace("FMSApp::StartThreads starting NNTPListener");\r
                m_threads.Start(new NNTPListener());\r
        }\r
 \r
@@ -188,6 +244,7 @@ void FMSApp::StartThreads()
        Option::Instance()->Get("StartFreenetUpdater",tempval);\r
        if(tempval=="true")\r
        {\r
+               logger().trace("FMSApp::StartThreads starting FreenetMasterThread");\r
                m_threads.Start(new FreenetMasterThread());\r
        }\r
 \r