//  Copyright (c) 2008 Ole Weidner
// 
//  Distributed under the Boost Software License, Version 1.0. (See accompanying 
//  file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)

#include <iostream>
#include <vector>
#include <string>

#include <saga/saga.hpp>
#include "saga-job.hpp"

#include <cstdlib>
#include <cstring>

///////////////////////////////////////////////////////////////////////////////

static int argc_;

void printUsage()
{
  std::cout << "Usage: saga-job <command> <options>"                << std::endl;
  std::cout << ""                                                   << std::endl; 
  std::cout << "  Commands:         Options:\n"                     << std::endl;
  std::cout << "  run               <rm url> <command> <arguments>" << std::endl; 
  std::cout << "  submit            <rm url> <command> <arguments>" << std::endl; 
  std::cout << ""                                                   << std::endl; 
  std::cout << "  state             <rm url> <jobid>              " << std::endl; 
  std::cout << "  suspend           <rm url> <jobid>              " << std::endl; 
  std::cout << "  resume            <rm url> <jobid>              " << std::endl; 
  std::cout << ""                                                   << std::endl; 
  std::cout << ""                                                   << std::endl; 
  std::cout << "  (rm url: scheme://host:port/job/service) "        << std::endl; 
  std::cout << ""                                                   << std::endl;
  std::cout << "This is a SAGA-based tool for job submission & management."  << std::endl;   
  std::cout << "For additional information, see http://saga.cct.lsu.edu."    << std::endl; 
}

void ExitIfArgcLessThan(int n)
{
  if (argc_ < n) {
    printUsage();
    exit(-1);
  }  
}

///////////////////////////////////////////////////////////////////////////////

int main (int argc, char* argv[])
{
#if !defined(SAGA_HAVE_PACKAGE_JOB)
    std::cerr << "saga-advert: saga has been compiled without the job "
                 "package, bailing out" << std::endl;
#else
  argc_ = argc;
  ExitIfArgcLessThan(2);

  SAGA_GUARDED_EXEC
  (
    if(!strcmp(argv[1], "run"))
    {
      ExitIfArgcLessThan(4);
      
      std::vector<std::string> argvec;
      for(int i=4; i<argc_; ++i)
          argvec.push_back(std::string(argv[i]));
          
      return job_run(std::string(argv[2]),std::string(argv[3]),argvec);
    }  
    else if(!strcmp(argv[1], "submit"))
    {
      ExitIfArgcLessThan(4);
      
      std::vector<std::string> argvec;
      for(int i=4; i<argc_; ++i)
          argvec.push_back(std::string(argv[4]));
          
      job_submit(std::string(argv[2]), std::string(argv[3]), argvec);
    }  
    else if(!strcmp(argv[1], "state"))
    {
      ExitIfArgcLessThan(4);
      job_state(std::string(argv[2]), std::string(argv[3]));
    }
    else if(!strcmp(argv[1], "suspend"))
    {
      ExitIfArgcLessThan(4);
      job_suspend(std::string(argv[2]), std::string(argv[3]));
    }
    else if(!strcmp(argv[1], "resume"))
    {
      ExitIfArgcLessThan(4);
      job_resume(std::string(argv[2]), std::string(argv[3]));
    }
    else if(!strcmp(argv[1], "run_tests"))
    {
      ExitIfArgcLessThan(3);
      run_test(std::string(argv[2]));
    }  
    /*else if(!strcmp(argv[1], "run_benchmarks"))
    {
      ExitIfArgcLessThan(2);
      run_benchmarks(std::string(argv[2]));
    }*/  
    
    else
    {
      std::cout << "Unknown command: " << std::string(argv[1]) << std::endl;
      printUsage();
      exit(-1);
    }
  )
#endif
  return 0;
}