Updated footer comment
[metaproxy-moved-to-github.git] / src / metaproxy_prog.cpp
index 0bec077..5ef52a4 100644 (file)
-/* $Id: metaproxy_prog.cpp,v 1.6 2006-06-10 14:29:12 adam Exp $
-   Copyright (c) 2005-2006, Index Data.
+/* This file is part of Metaproxy.
+   Copyright (C) 2005-2008 Index Data
 
-   See the LICENSE file for details
- */
+Metaproxy is free software; you can redistribute it and/or modify it under
+the terms of the GNU General Public License as published by the Free
+Software Foundation; either version 2, or (at your option) any later
+version.
+
+Metaproxy is distributed in the hope that it will be useful, but WITHOUT ANY
+WARRANTY; without even the implied warranty of MERCHANTABILITY or
+FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
+for more details.
+
+You should have received a copy of the GNU General Public License
+along with this program; if not, write to the Free Software
+Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
+*/
 
 #include "config.hpp"
 
-#include <boost/program_options.hpp>
-namespace po = boost::program_options;
+#include <yaz/log.h>
+#include <yaz/options.h>
+#include <yaz/daemon.h>
 
 #include <iostream>
 #include <stdexcept>
+#include <libxml/xinclude.h>
 
 #include "filter.hpp"
 #include "package.hpp"
 #include "router_flexml.hpp"
 #include "factory_static.hpp"
 
+#if HAVE_UNISTD_H
+#include <unistd.h>
+#endif
+#ifdef WIN32
+#include <direct.h>
+#endif
+
 namespace mp = metaproxy_1;
 
+static void handler(void *data)
+{
+    mp::RouterFleXML *routerp = (mp::RouterFleXML*) data;
+
+    mp::Package pack;
+    pack.router(*routerp).move();
+}
+    
 int main(int argc, char **argv)
 {
     try 
     {
-        po::options_description desc("Allowed options");
-        desc.add_options()
-            ("help,h", "produce help message")
-            ("version,V", "show version")
-            ("config", po::value< std::vector<std::string> >(), "xml config")
-            ;
-        
-        po::positional_options_description p;
-        p.add("config", -1);
-        
-        po::variables_map vm;        
-        po::store(po::command_line_parser(argc, argv).
-                  options(desc).positional(p).run(), vm);
-        po::notify(vm);
-        
-        if (vm.count("help")) {
-            std::cout << desc << "\n";
-            return 1;
-        }
-        if (vm.count("version")) {
-            std::cout << "Metaproxy " VERSION "\n";
-            return 0;
-        }
-        xmlDocPtr doc = 0;
-        if (vm.count("config"))
-        {
-            std::vector<std::string> config_fnames = 
-                vm["config"].as< std::vector<std::string> >();
+        const char *fname = 0;
+        int ret;
+        char *arg;
+        unsigned mode = 0;
+        const char *pidfile = 0;
+        const char *uid = 0;
 
-            if (config_fnames.size() != 1)
-            {
-                std::cerr << "Only one configuration must be given\n";
-                std::exit(1);
-            }
-            
-            doc = xmlParseFile(config_fnames[0].c_str());
-            if (!doc)
+        while ((ret = options("c{config}:Dh{help}l:p:u:V{version}w:X", 
+                              argv, argc, &arg)) != -2)
+        {
+            switch (ret)
             {
-                std::cerr << "xmlParseFile failed\n";
+            case 'c':
+                fname = arg;
+                break;
+            case 'D':
+                mode = YAZ_DAEMON_FORK|YAZ_DAEMON_KEEPALIVE;
+                break;
+            case 'h':
+                std::cerr << "metaproxy\n"
+                    " -h|--help     help\n"
+                    " -V|--version  version\n"
+                    " -c|--config f config filename\n"
+                    " -D            daemon and keepalive operation\n"
+                    " -l f          log file f\n"
+                    " -p f          pid file f\n"
+                    " -u id         change uid to id\n"
+                    " -w dir        changes working directory to dir\n"
+                    " -X            debug mode (no fork/daemon mode)\n"
+                          << std::endl;
+                break;
+            case 'l':
+                yaz_log_init_file(arg);
+                break;
+            case 'p':
+                pidfile = arg;
+                break;
+            case 'u':
+                uid = arg;
+                break;
+            case 'V':
+                std::cout << VERSION "\n";
+                std::exit(0);
+                break;
+            case 'w':
+                if (chdir(arg)) 
+                {
+                    std::cerr << "chdir " << arg << " failed" << std::endl;
+                    std::exit(1);
+                }
+            case 'X':
+                mode = YAZ_DAEMON_DEBUG;
+                break;
+            case -1:
+                std::cerr << "bad option: " << arg << std::endl;
                 std::exit(1);
             }
         }
-        else
+        if (!fname)
         {
-            std::cerr << "No configuration given\n";
+            std::cerr << "No configuration given; use -h for help\n";
             std::exit(1);
         }
-        if (doc)
+
+        yaz_log(YLOG_LOG, "Metaproxy " VERSION " started");
+        xmlDocPtr doc = xmlReadFile(fname,
+                                    NULL, 
+                                    XML_PARSE_XINCLUDE + XML_PARSE_NOBLANKS
+                                    + XML_PARSE_NSCLEAN + XML_PARSE_NONET );
+        
+        if (!doc)
         {
-            try {
-                mp::FactoryStatic factory;
-                mp::RouterFleXML router(doc, factory);
-                mp::Package pack;
-                pack.router(router).move();
-            }
-            catch (std::runtime_error &e) {
-                std::cerr << "std::runtime error: " << e.what() << "\n";
-                std::exit(1);
-            }
-            xmlFreeDoc(doc);
+            std::cerr << "XML parsing failed\n";
+            std::exit(1);
         }
-    }
-    catch (po::unknown_option &e) {
-        std::cerr << e.what() << "; use --help for list of options\n";
-        std::exit(1);
+        // and perform Xinclude then
+        if (xmlXIncludeProcess(doc) > 0) {
+            std::cerr << "processing XInclude directive\n";
+        }
+        mp::FactoryStatic factory;
+        mp::RouterFleXML router(doc, factory, false);
+
+        yaz_daemon("metaproxy", mode, handler, &router, pidfile, uid);
     }
     catch (std::logic_error &e) {
         std::cerr << "std::logic error: " << e.what() << "\n";
@@ -108,8 +155,9 @@ int main(int argc, char **argv)
 /*
  * Local variables:
  * c-basic-offset: 4
+ * c-file-style: "Stroustrup"
  * indent-tabs-mode: nil
- * c-file-style: "stroustrup"
  * End:
  * vim: shiftwidth=4 tabstop=8 expandtab
  */
+