// make_file_dataset.cxx
//
// David Adams
// June 2003
//
// Create a ATLAS-POOL event dataset from a file.

#include <string>
#include <iostream>
#include <dlfcn.h>
#include "dataset_util/FileStatus.h"
#include "dataset_util/FileName.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/Url.h"
#include "dataset_file/FileManagementSystem.h"
#include "dataset_credential/GssCredentialManager.h"
#include "dataset_base/SingleFileDataset.h"

using std::string;
using std::cout;
using std::cerr;
using std::endl;
using dset::Url;
using dset::FileManagementSystem;
using dset::GssCredentialManager;
using dset::Dataset;
using dset::SingleFileDataset;

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

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

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

  // First argument.
  string dstype = "";
  string content_label = "";
  string xfile = "dataset.xml";
  bool putfile = false;
  string ifname = "";
  int stat = 0;
  // Print help and exit.
  bool help = false;
  // Flag to insert file into file catalog.
  for ( int iarg=1; iarg<argc; ++iarg ) {
    string arg = argv[iarg];
    // help
    if ( arg == "-h" ) {
      help = true;
    // specify ID
    } else if ( arg == "-t" ) {
      dstype = argv[++iarg];
    // specify content label
    } else if ( arg == "-c" ) {
      if ( iarg < argc ) {
        arg = argv[++iarg];
	content_label = arg;
      }
    // specify file name
    } else if ( arg == "-f" ) {
      if ( iarg < argc ) {
        arg = argv[++iarg];
	xfile = arg;
      }
    // Put file in FMS
    } else if ( arg == "-p" ) {
      putfile = true;
    } else if ( ifname == "" ) {
      ifname = arg;
    } else {
      cerr << "Ignoring argument " << arg << endl;
    }
  }

  if ( help ) {
  } else if ( ifname == "" ) {
    stat = 1;
    cerr << "File name must be provided" << endl;
  }

  if ( help || stat!=0 ) {
    cout << "Usage: " << arg0 << endl;
    cout << "       -t dstype -c clabel [-i ID | -u [DIR]] [-p] URL" << endl;
    cout << "  -t dstype specifies the type of dataset" << endl;
    cout << "  -c content label (ESD, AOD, ...)" << endl;
    cout << "  -f fname  to write the output dataset XML to file fname" << endl;
    cout << "  -p to put file in the FMS SE (DIAL_FMS_SE)" << endl;
    cout << "  URL is the input file URL or an accessible physical file" << endl;
    return stat;
  }

  // Check type.
  if ( dstype == "" ) {
    cerr << "Dataset type must be provided" << endl;
    return 10;
  }

  // Load libraries until we have plugins.
  if ( dstype == "RootHistogramDataset" ) {
    string lib = "libdataset_root.so";
    void* handle = dlopen(lib.c_str(), RTLD_NOW|RTLD_GLOBAL);
    if ( handle == 0 ) {
      cerr << "Unable to load library " << lib << endl;
      cerr << "  " << dlerror() << endl;
    }
  }
  if ( dstype == "AtlasPoolEventDataset" ||
       dstype == "ApecDataset" ) {
    string lib = "libdataset_atlaspool.so";
    void* handle = dlopen(lib.c_str(), RTLD_NOW|RTLD_GLOBAL);
    if ( handle == 0 ) {
      cerr << "Unable to load library " << lib << endl;
      cerr << "  " << dlerror() << endl;
    }
  }

  // Check content label.
  if ( content_label == "" ) {
    cerr << "Content label must be provided" << endl;
    return 11;
  }

  // Initialize the unique ID generator.
  string uid_name = Environment::initial().value("DIAL_UIDS");
  if ( uid_name == "" ) {
    cerr << "Unable to locate $DIAL_UIDS" << endl;
    return 12;
  }
  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 13;
  }

  // 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 14;
  }

  // Parse the input file.
  Url infurl(ifname);
  if ( ! infurl.is_valid() ) {
    if ( ! FileStatus(ifname).is_readable() ) {
      cerr << "Input file is not a valid URL or accessible physical file:" << endl;
      cerr << "  " << ifname << endl;
      return 15;
    }
    infurl = Url("file:" + ifname);
  }
  assert( infurl.is_valid() );

  // Put the file into the FMS.
  Url outurl = infurl;
  if ( putfile ) {
    FileManagementSystem& fms = FileManagementSystem::default_instance();
    outurl = fms.put(infurl);
    if ( ! outurl.is_valid() ) {
      cerr << "Error putting file into FMS:" << endl;
      cerr << "  " << fms.error_message() << endl;
      cerr << fms << endl;
    }
  }
    
  // Create dataset.
  DatasetId parentid;
  const Dataset* pdst = SingleFileDataset::create(dstype, outurl, content_label, parentid);
  if ( pdst == 0 || ! pdst->is_valid() ) {
    cerr << "Unable to create dataset" << endl;
    if ( pdst != 0 ) {
      int err = pdst->error();
      cerr << "Dataset reports error " << err << endl;
    }
    return 15;
  }

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

  // Write XML description.
  if ( xfile.size() ) {
    XmlParser parser;
    int wstat = parser.write(xfile, *pele);
    if ( wstat ) {
      cerr << "Error writing XML: " << wstat << endl;
      cerr << *pele << endl;
    } else {
      cerr << "Output in " << xfile << endl;
    }
    delete pele;
  }

}