// fms.cxx

#include <string>
#include <iostream>
#include <sstream>
#include "dataset_util/Environment.h"
#include "dataset_file/FileManagementSystem.h"

using std::string;
using std::cout;
using std::cerr;
using std::endl;
using std::istringstream;
using dset::Url;
using dset::FileManagementSystem;

// Main program to access FMS.

int main(int argc, char* carg[]) {

  // Parse command line.
  bool bad = false;
  string comname = carg[0];
  string com = "";
  bool help = false;
  Url inurl;
  Url durl;
  for ( int iarg=1; iarg<argc; ++iarg ) {
    string word = carg[iarg];
    int remain = argc - iarg - 1;
    // Handle flags and their arguments.
    if ( word[0] == '-' ) {
      if ( bad = word.size() < 2 ) break;
      char flag = word[1];
      switch (flag) {
        // help
        case 'h': help = true; break;
        default: bad = true; break;
      }
      if ( bad || help ) break;
    // Handle command argument.
    // The first is the command, the second the input URL and
    // the third is the destination URL.
    } else {
      if ( com.size() ) {
        if ( inurl.size() ) {
          if ( com!="copy" || durl.size() ) {
            bad = true;
            break;
          } else {
            durl = Url(word);
            if ( ! durl.is_valid() ) {
              cerr << "Destination must be a URL" << endl;
              return 3;
            }
            continue;
          }
        }
        inurl = Url(word);
        if ( ! inurl.is_valid() ) {
          cerr << "Input must be a URL" << endl;
          return 4;
        }
        continue;
      }
      com = word;
      // help
      if ( word == "help" ) {
        help = true;
        break;
      }
    }
  }
  if ( ! help && ! com.size() ) bad = true;

  if ( bad ) {
    cerr << "fms: Invalid syntax." << endl;
    return 1;
  }

  // Show help message.
  if ( help ) {
    cout << "Usage: " << "fms help" << endl;
    cout << "       " << "fms ping" << endl;
    cout << "       " << "fms get URL" << endl;
    cout << "       " << "fms copy URL file:dir" << endl;
    cout << "       " << "fms put URL" << endl;
    return 0;
  }

  // Open FMS.
  Environment env = Environment::current();
  FileManagementSystem fms(env);

  // Ping.
  if ( com == "ping" ) {
    cout << fms << endl;
    return 0;
  }

  // Get.
  if ( com == "get" ) {
    Url purl = fms.get(inurl);
    if ( purl.is_valid() ) {
      cout << purl << endl;
    } else {
      cerr << "fms: Unable to fetch replica for " << inurl << endl;
      return 5;
    }
    return 0;
  }

  // Copy.
  if ( com == "copy" ) {
    Url purl = fms.copy(inurl, durl);
    if ( purl.is_valid() ) {
      cout << purl << endl;
    } else {
      cerr << "fms: Unable to copy replica for " << inurl << endl;
      return 6;
    }
    return 0;
  }

  // Put.
  if ( com == "put" ) {
    Url purl = fms.put(inurl);
    if ( purl.is_valid() ) {
      cout << purl << endl;
    } else {
      cerr << "fms: Unable to put " << inurl << endl;
      return 7;
    }
    return 0;
  }

  cerr << "fms: Unknown command: " << com << endl;
  return 99;
}