// make_atlaspool_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/FileCatalog.h"
#include "dataset_credential/GssCredentialManager.h"
#include "dataset_atlaspool/AtlasPoolEventDataset.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::AtlasPoolEventDataset;

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

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

  cerr << "make_atlaspool_dataset is obselete" << endl;
  cerr << "  Please use make_file_dataset -t AtlasPoolEventDataset"
       << endl;

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

  // First argument.
  string ifname = "";
  string fctype = "";
  string id_name = "";
  string content_label = "";
  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";
  // Flag to insert file into file catalog.
  bool putfile = true;
  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 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 catalog
    } else if ( arg == "-p" ) {
      putfile = true;
    // Do not put file in catalog
    } else if ( arg == "-g" ) {
      putfile = false;
    } else if ( arg == "-l" ) {
      fctype = "LocalFileCatalog";
    } else if ( arg == "-a" ) {
      fctype = "AfsFileCatalog";
    } else if ( arg == "-m" ) {
      fctype = "MagdaFileCatalog";
    } else if ( arg == "-n" ) {
      fctype = "NfsFileCatalog";
    } else if ( ifname == "" ) {
      ifname = arg;
    } else if ( fctype == "" ) {
      fctype = arg;
    } else {
      cerr << "Ignoring argument " << arg << endl;
    }
  }

  if ( pfc == 0 ) {
    string fcname = "";
    if ( fctype == "NfsFileCatalog" ) {
      fcname = FileName(ifname).fullpath().unlinked().name();
    }
    if ( fctype == "MagdaFileCatalog" ) {
      string lib = "libmysqlpp_dial.so";
      void* handle = dlopen(lib.c_str(), RTLD_NOW|RTLD_GLOBAL);
      if ( handle == 0 ) {
        cerr << "Unable to load library " << lib << endl;
      }
      lib = "libdataset_magda.so";
      handle = dlopen(lib.c_str(), RTLD_NOW|RTLD_GLOBAL);
      if ( handle == 0 ) {
        cerr << "Unable to load library " << lib << endl;
      }
      fcname = "Atlas";
    }
    pfc = FileCatalog::locate(fctype, fcname);
  }

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

  if ( help || stat!=0 ) {
    cout << "Usage: " << arg0 << endl;
    cout << "         [-l|-a|-n|-m] [-u [UID_DIR]] [-i iii.jjj] [-p|-g]"
         << endl;
    cout << "         filename [fctype]" << 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 << "       -c content label (ESD, AOD, ...)" << endl;
    cout << "       -f fname writes XML to file fname" << endl;
    cout << "       -p to put file in specified catalog (default)" << endl;
    cout << "       -g to get (not put) file from specified catalog" << 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 fctype = LocalFileCatalog, AfsFileCatalog, ..." << endl;
    return stat;
  }

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

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

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

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

  // Create logical file.
  FileId fid;
  if ( putfile ) {
    string infile = ifname;
    string lfn = "";
    FileCatalog::Time lifetime = 0;
    fid = pfc->put(ifname, lfn , lifetime);
  } else {
    fid = FileId(ifname);
  }
  if ( ! fid.is_valid() ) {
    string action = "getting";
    if ( putfile ) action = "getting";
    int fcerr = pfc->error();
    cerr << "Error " << fcerr << " " << action << " file" << endl;
    cerr << "  " << ifname << endl;
    cerr << " with catalog" << endl;
    cerr << "  " << *pfc << endl;
    return 13;
  }
  LogicalFile lfile(*pfc, fid);
  if ( lfile.get_readonly().size() == 0 ) {
    cerr << "Unable to read file" << endl << lfile << endl;
    return 14;
  }

  // Create dataset.
  AtlasPoolEventDataset dst(lfile, content_label, false);
  if ( ! dst.is_valid() ) {
    cerr << "Unable to create dataset" << endl;
    int err = dst.error();
    cerr << "AtlasPoolEventDataset reports error " << err << endl;
    string msg = "Unknown error";
    if ( err == 15 ) {
      msg = "Inconsistent content";
    }
    if ( err == 14 ) {
      msg = "Duplicate event ID";
    }
    cerr << "  " << msg << endl;
    return 15;
  }

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

  // Create XML.
  const XmlElement* pele = dst.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;
  }

}