source: trunk/disparmmain.cpp @ 4

/*
   Copyright 2012 Willem Vermin, SARA

   Licensed under the Apache License, Version 2.0 (the "License");
   you may not use this file except in compliance with the License.
   You may obtain a copy of the License at

       http://www.apache.org/licenses/LICENSE-2.0

   Unless required by applicable law or agreed to in writing, software
   distributed under the License is distributed on an "AS IS" BASIS,
   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
   See the License for the specific language governing permissions and
   limitations under the License.
 */

#include <iostream>
#include <getopt.h>
#include <stdlib.h>
#include <libgen.h>
#include "tofromstring.h"
#include "disparm.h"
#include "version.h"

void usage()
{
  disparm d=disparm();
  std::cerr << "Usage:" << std::endl;

  std::cerr << programname << " -h,--help : this message" << std::endl;
  std::cerr << programname << " -c,--create [-p,--pool POOL] [-i,--input   INPUTFILE]" << std::endl;
  std::cerr << programname << " -n,--next   [-p,--pool POOL] [-d --lockdir LOCKDIR  ] [-k,--key KEY] [-m,--max MAX]" << std::endl;
  std::cerr << programname << " -r,--ready  [-p,--pool POOL] [-d --lockdir LOCKDIR  ] [-k,--key KEY]" << std::endl;
  std::cerr << programname << " -s,--stats  [-p,--pool POOL] [-d --lockdir LOCKDIR  ]" << std::endl;
  std::cerr << programname << " -u,--unhandled  [-p,--pool POOL] [-d --lockdir LOCKDIR  ]" << std::endl;
  std::cerr << programname << " -v,--version" << std::endl;
  std::cerr << "Defaults:" << std::endl;
  std::cerr << "  POOL     : " << d.defpath_pool << std::endl;
  std::cerr << "  LOCKDIR  : " << d.defpath_lockdir  << std::endl;
  std::cerr << "  MAX      : " << "1" << std::endl;
  std::cerr << "  INPUTFILE: " << d.definputfilename;
  if (d.definputfilename == "-")
    std::cerr << " (standard input)";
  std::cerr << std::endl;
}

int main(int argc, char*argv[])
{
  disparm disparmthing=disparm();
  bool nextflag       = 0;
  bool createflag     = 0;
  bool readyflag      = 0;
  bool REMOVEFLAG     = 0;
  bool statsflag      = 0;
  bool unhandledflag  = 0;
  bool lockdirset     = 0;

  disparmthing.setarg(argc,argv);

  while(1)
  {
    static struct option long_options[] =
    {
      {"help",     0, 0, 'h'},
      {"pool",     1, 0, 'p'},
      {"create",   0, 0, 'c'},
      {"input",    1, 0, 'i'},
      {"next",     0, 0, 'n'},
      {"max",      1, 0, 'm'},
      {"ready",    0, 0, 'r'},
      {"stats",    0, 0, 's'},
      {"lockdir",  1, 0, 'd'},
      {"key",      1, 0, 'k'},
      {"REMOVE",   0, 0, 'R'},
      {"unhandled",0, 0, 'u'},
      {"debug",    0, 0, 'D'},
      {"version",  0, 0, 'v'},
      {0,          0, 0, 0}
    };
    int c = getopt_long(argc,argv,"hp:ci:nm:rsuDvd:k:R",long_options,0);
    if (c == -1)
    {
      if (optind < argc)  // unexpected extra argument
      {
        std::cerr << programname << ": Unexpected argument:" << argv[optind] << std::endl;
        return 1;
      }
      break;
    }
    if ( c == '?' || c == ':' )
    {
      std::cerr << "Invalid parameter, try "<< argv[0]<<" --help " <<std::endl;
      return 1;
    }
    switch(c)
    {
      case 'h':
        usage();
        break;
      case 'p':
        disparmthing.setpath_pool(std::string(optarg));
        if (!lockdirset)
        {
          std::string d = dirname(optarg);
          int rc = disparmthing.setpath_lockdir(d);
          if (rc != 0)
          {
            std::cerr << programname << ": Cannot create directory '"<< d << "'" << std::endl;
            return 1;
          }
        }
        break;
      case 'c':
        createflag = 1;
        break;
      case 'k':
        disparmthing.setkey(from_string <long long>(optarg));
        break;
      case 'i':
         disparmthing.setinputfilename(std::string(optarg));
        break;
      case 'n':
        nextflag = 1;
        break;
      case 'm':
        disparmthing.setmaxiter(atoi(optarg));
        break;
      case 'r':
        readyflag = 1;
        break;
      case 'R':
        REMOVEFLAG = 1;
        break;
      case 's':
        statsflag = 1;
        break;
      case 'd':
        {
          int rc = disparmthing.setpath_lockdir(std::string(optarg));
          if (rc != 0)
          {
            std::cerr << programname << ": Cannot create directory '"<<optarg<<"'"<<std::endl;
            return 1;
          }
        }
        lockdirset = 1;
        break;
      case 'u':
        unhandledflag = 1;
        break;
      case 'D':
        disparmthing.setdebug();
        break;
      case 'v':
        std::cerr << programversion << std::endl;
        break;
    }
  }

  int rc=0;
  if(createflag)
  {
    std::string s;
    rc |= disparmthing.create(s);
    std::cout << s;
    if (rc != 0)
      exit(1);
  }
  if(nextflag)
  {
    std::string s;
    rc |= disparmthing.next(s);
    std::cout << s;
  }
  if(readyflag)
  {
    rc |= disparmthing.ready();
  }
  if(REMOVEFLAG)
  {
    rc |= disparmthing.REMOVE();
  }
  if(statsflag)
  {
    std::cerr << disparmthing << std::endl;
  }
  if(unhandledflag)
  {
    rc |=disparmthing.echounhandled();
  }
  return rc;
}