// make_cbnt_dataset.cxx
//
// David Adams
// June 2003
//
// Create a CBNT dataset from a CBNT hbook file.

#include <string>
#include <iostream>
#include "dataset_util/FileStatus.h"
#include "dataset_util/Environment.h"
#include "dataset_util/XmlElement.h"
#include "dataset_util/DtdRegistry.h"
#include "dataset_xml/XmlParser.h"
#include "dataset_id/UniqueIdGenerator.h"
#include "dataset_file/FileCatalog.h"
#include "dataset_credential/GssCredentialManager.h"
#include "dataset_CBNT/CbntDataset.h"

using std::string;
using std::cout;
using std::cerr;
using std::endl;
using dset::FileId;
using dset::FileCatalog;
using dset::LogicalFile;
using dset::GssCredentialManager;
using dset::CbntDataset;

//**********************************************************************

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

  // Fetch arguments.
  string arg0 = argv[0];

  // First argument.
  string hbname = "";
  string fcname = "";
  string id_name = "";
  bool use_uid = true;
  string uid_name = "";
  FileCatalog* pfc = 0;
  int stat = 0;
  // Print help and exit.
  bool help = false;
  // ID for dataset.
  DatasetId dstid;
  assert( ! dstid.is_valid() );
  // Name for dataset XML file.
  string xfile = "dataset.xml";
  for ( int iarg=1; iarg<argc; ++iarg ) {
    string arg = argv[iarg];
    // help
    if ( arg == "-h" ) {
      help = true;
    // specify ID
    } else if ( arg == "-i" ) {
      if ( iarg<argc && argv[iarg+1][0]!='-' ) {
        id_name = argv[++iarg];
	use_uid = false;
      } else {
	cerr << "Value missing for option -i" << endl;
	stat = 4;
      }
    // specify unique ID generator
    } else if ( arg == "-u" ) {
      use_uid = true;
      id_name = "";
      if ( iarg<argc && argv[iarg+1][0]!='-' ) {
        uid_name = argv[++iarg];
      }
    // specify file name
    } else if ( arg == "-f" ) {
      if ( iarg < argc ) {
        arg = argv[++iarg];
	xfile = arg;
      }
    } else if ( arg == "-l" ) {
      fcname = "LocalFileCatalog";
    } else if ( arg == "-a" ) {
      fcname = "AfsFileCatalog";
    } else if ( arg == "-m" ) {
      fcname = "MagdaFileCatalog";
    } else if ( arg == "-n" ) {
      fcname = "NfsFileCatalog";
    } else if ( hbname == "" ) {
      hbname = arg;
    } else if ( fcname == "" ) {
      fcname = arg;
    } else {
      cerr << "Ignoring argument " << arg << endl;
    }
  }

  if ( pfc == 0 ) {
    pfc = FileCatalog::locate(fcname);
  }

  if ( help ) {
  } else if ( hbname == "" ) {
    stat = 1;
    cerr << "File name must be provided" << endl;
  } else if ( pfc == 0 ) {
    if ( fcname == "" ) {
      stat = 2;
      cerr << "File catalog must be specified" << endl;
    } else {
      stat = 3;
      cerr << "Unable to locate file catalog " << fcname << endl;
    }
  }

  if ( help || stat!=0 ) {
    cout << "Usage: " << arg0
         << " [-l|-a|-n|-m] [-u [UID_DIR]] [-i iii.jjj] filename [fcname]"
         << endl;
    cout << "       -i iii-jjj sets ID to iii-jjj" << endl;
    cout << "                  (normally better to use option -u)" << endl;
    cout << "       -u UID_DIR to use global dataset ID from UID_DIR" << endl;
    cout << "                  default for UID_DIR is $DIAL_UIDS" << endl;
    cout << "       -f fname writes XML to file fname" << endl;
    cout << "       -l for local file catalog" << endl;
    cout << "       -a for AFS file catalog" << endl;
    cout << "       -n for NFS file catalog for cwd" << endl;
    cout << "       -m for Magda file catalog" << endl;
    cout << "       or fcname = LocalFileCatalog, AfsFileCatalog, ..." << endl;
    return stat;
  }

  // Check that unique ID generator and explicit ID are not both
  // specified.
  bool use_id = id_name.size();
  if ( use_id && use_uid ) {
    cerr << "Use only one of -i and -u" << endl;
    return 5;
  }

  // Initialize the unique ID gnerator.
  if ( use_uid ) {
    if ( ! uid_name.size() ) {
      uid_name = Environment::initial().value("DIAL_UIDS");
      if ( uid_name == "" ) {
        cerr << "Unable to locate $DIAL_UIDS" << endl;
        stat = 4;
      }
    }
    int gstat = UniqueIdGenerator::set_generator(uid_name);
    if ( gstat != 0 ) {
      cerr << "Unable to set unique ID generator" << endl;
      cerr << "  " << uid_name << endl;
      cerr << "  Error code is " << gstat << endl;
      return 17;
    }
  }

  // Set the credential for this thread.
  // This is needed to find owner for task.
  int cstat = GssCredentialManager::set_default();
  if ( cstat != 0 ) {
    cerr << "Unable to find credential" << endl;
    return 5;
  }

  // Create logical file.
  LogicalFile lfile(*pfc, FileId(hbname));
  if ( lfile.get_readonly().size() == 0 ) {
    cerr << "Unable to read file" << endl << lfile << endl;
    return 4;
  }

  // Create dataset.
  CbntDataset dst(lfile, false);
  if ( ! dst.is_valid() ) {
    cerr << "Unable to create dataset" << endl;
    int err = dst.error();
    cerr << "CbntDataset reports error " << err << endl;
    string msg = "Unknown error";
    if ( err == 51 ) {
      msg = "Event count is zero";
    }
    if ( err == 52 ) {
      msg = "Too many events (> 10000)";
    }
    if ( err == 54 ) {
      msg = "Duplicate event ID";
    }
    cerr << "  " << msg << endl;
    return 5;
  }

  // Set ID.
  if ( use_id ) {
    DatasetId newid(id_name);
    int setstat = dst.set_id(newid);
    if ( setstat != 0 ) {
      cerr << "Error " << setstat << " setting dataset ID " << dstid << endl;
      return 6;
    }
  }
  dst.lock();

  // Create XML.
  const XmlElement* pele = dst.xml();
  if ( pele == 0 ) {
    cerr << "Unable to create XML" << endl;
    return 7;
  }

  // Write XML description.
  if ( xfile.size() ) {
    XmlParser parser;
    parser.write(xfile, *pele);
    delete pele;
    cerr << "Output in " << xfile << endl;
  }

}