/* $Id: zpresent.cxx,v 1.7 2000/10/24 18:40:36 cnidr Exp $ */
/************************************************************************
Copyright (c) A/WWW Enterprises, 2000
************************************************************************/

/************************************************************************
Original Copyright Notice

Copyright (c) MCNC, Clearinghouse for Networked Information Discovery and
Retrieval, 1994. 

Permission to use, copy, modify, distribute, and sell this software and
its documentation, in whole or in part, for any purpose is hereby granted
without fee, provided that

1. The above copyright notice and this permission notice appear in all
copies of the software and related documentation. Notices of copyright
and/or attribution which appear at the beginning of any file included in
this distribution must remain intact. 

2. Users of this software agree to make their best efforts (a) to return
to MCNC any improvements or extensions that they make, so that these may
be included in future releases; and (b) to inform MCNC/CNIDR of noteworthy
uses of this software. 

3. The names of MCNC and Clearinghouse for Networked Information Discovery
and Retrieval may not be used in any advertising or publicity relating to
the software without the specific, prior written permission of MCNC/CNIDR. 

THE SOFTWARE IS PROVIDED "AS-IS" AND WITHOUT WARRANTY OF ANY KIND,
EXPRESS, IMPLIED OR OTHERWISE, INCLUDING WITHOUT LIMITATION, ANY WARRANTY
OF MERCHANTABILITY OR FITNESS FOR A PARTICULAR PURPOSE. 

IN NO EVENT SHALL MCNC/CNIDR BE LIABLE FOR ANY SPECIAL, INCIDENTAL,
INDIRECT OR CONSEQUENTIAL DAMAGES OF ANY KIND, OR ANY DAMAGES WHATSOEVER
RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER OR NOT ADVISED OF THE
POSSIBILITY OF DAMAGE, AND ON ANY THEORY OF LIABILITY, ARISING OUT OF OR
IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. 
************************************************************************/

/*@@@
File:		zpresent.cxx
Version:	1.00
$Revision: 1.7 $
Description:	Command-line search utility w/ XML results
Author:		A. Warnock (warnock@awcubed.com), adapted from Isearch.cxx
                by Nassib Nassar, nrn@cnidr.org
@@@*/

#include <stdio.h>
#include <string.h>
#include <locale.h>

#include "isearch.hxx"

#include "common.hxx"
#include "dtreg.hxx"
#include "index.hxx"
#include "registry.hxx"
#include "idb.hxx"
#include "vidb.hxx"

void
XmlBail(const STRING& DBName, const STRING& error_message) {
    cout << "\t<isearch:present status=\"Error\" dbname=\"" << DBName
	 << "\">" << endl;
    cout << "\t\t<isearch:error_block>" << endl;
    cout << error_message;
    cout << "\t\t</isearch:error_block>" << endl;
    cout << "\t</isearch:present>" << endl;
    cout << "</zpresent>" << endl;
    exit;
}

int 
main(int argc, char** argv) {
  if (argc < 2) {
    fprintf(stderr,"zpresent v%s\n", IsearchVersion);
    fprintf(stderr,"Copyright (c) 2000 MCNC/CNIDR and A/WWW Enterprises\n");
    fprintf(stderr,"-d (X)   # Search database with root name (X).\n");
    fprintf(stderr,"-id (X)  # Request document(s) with docid (X).\n");
    fprintf(stderr,"-l (X)   # Request document(s) listed in file (X).\n");
    fprintf(stderr,"-p (X)   # Present element set (X) with results.\n");
    fprintf(stderr,"-f (X)   # Present results in format (X).\n");
    fprintf(stderr,"Document Types Supported:");
    DTREG dtreg(0);
    STRLIST DocTypeList;
    dtreg.GetDocTypeList(&DocTypeList);
    STRING s;
    INT x;
    INT y = DocTypeList.GetTotalEntries();
    for (x=1; x<=y; x++) {
      DocTypeList.GetEntry(x, &s);
      fprintf(stderr,"\t ");
      s.Print(stderr);
    }
    fprintf(stderr,"\n");
    fprintf(stderr,"zpresent is currently experimental and should be used\n");
    fprintf(stderr,"cautiously.  Suggestions and improvements are welcomed\n");
    fprintf(stderr,"\n");
    fflush(stdout); fflush(stderr); exit (0);
  }


  printf("<?xml version=\"1.0\" encoding=\"ISO-8859-1\"?>\n");
  printf("<!-- Generated by zpresent v%s, part of the Isite package from -->\n",IsearchVersion);
  printf("<!-- A/WWW Enterprises, http://www.awcubed.com/Isite/ -->\n");
  printf("<!DOCTYPE zpresent SYSTEM \"zpresent.dtd\">\n");
  printf("<zpresent xmlns:isearch=\"http://www.awcubed.com/dtd\">\n");

  STRLIST DocTypeOptions;
  STRING Flag;
  STRING DBName="";
  STRING RecordKey="";
  STRING ESet, RecordSyntax;
  INT x = 0;
  INT LastUsed = 0;
  STRING DBPathName, DBFileName;
  STRING PathName, FileName;
  RESULT RsRecord;
  STRING Record;
  GDT_BOOLEAN Error=GDT_FALSE;
  STRING error_message="";
  STRLIST RecordKeyList;
  INT n_keys;
  STRING ListFile;

  ESet = "B";
  while (x < argc) {
    if (argv[x][0] == '-') {
      Flag = argv[x];
      if (Flag.Equals("-d")) {
	if (++x >= argc) {
	  error_message.Cat("\t\t\t<isearch:error_text>ERROR: No database name specified after -d.</isearch:error_text>\n");
	  XmlBail(DBName,error_message);
	}
	DBName = argv[x];
	LastUsed = x;
      }
      if (Flag.Equals("-id")) {
	if (++x >= argc) {
	  error_message.Cat("\t\t\t<isearch:error_text>ERROR: No document ID specified after -id.</isearch:error_text>\n");
	  XmlBail(DBName,error_message);
	}
	RecordKey = argv[x];
	LastUsed = x;
      }
      if (Flag.Equals("-p")) {
	if (++x >= argc) {
	  error_message.Cat("\t\t\t<isearch:error_text>ERROR: No element set specified after -p.</isearch:error_text>\n");
	  XmlBail(DBName,error_message);
	}
	ESet = argv[x];
	LastUsed = x;
      }
      if (Flag.Equals("-f")) {
	if (++x >= argc) {
	  error_message.Cat("\t\t\t<isearch:error_text>ERROR: No format specified after -f.</isearch:error_text>\n");
	  XmlBail(DBName,error_message);
	}
	RecordSyntax = argv[x];
	LastUsed = x;
      }
      if (Flag.Equals("-l")) {
	if (++x >= argc) {
	  error_message.Cat("\t\t\t<isearch:error_text>ERROR: No list file specified after -l.</isearch:error_text>\n");
	  XmlBail(DBName,error_message);
	}
	ListFile = argv[x];
	LastUsed = x;
      }
    }
    x++;
  }
	
  if (DBName.GetLength() == 0) {
    error_message.Cat("\t\t\t<isearch:error_text>No database name given.</isearch:error_text>\n");
    XmlBail(DBName,error_message);
  }

  if ((RecordKey.GetLength() > 0) && (ListFile.GetLength() > 0)) {
    error_message.Cat("\t\t\t<isearch:error_text>Cannot specify both -id and -l.</isearch:error_text>\n");
    XmlBail(DBName,error_message);
  }

  if (ListFile.GetLength() > 0) {
    RecordKey.ReadFile(ListFile);
    RecordKey.Replace("\n","");
  }

  if (RecordKey.GetLength() == 0) {
    error_message.Cat("\t\t\t<isearch:error_text>No record keys given.</isearch:error_text>\n");
    XmlBail(DBName,error_message);
  }

  RecordKeyList.Split(',',RecordKey);
  n_keys = RecordKeyList.GetTotalEntries();
  if (n_keys <= 0) {
    error_message.Cat("\t\t\t<isearch:error_text>No record ids given.</isearch:error_text>\n");
    XmlBail(DBName,error_message);
  }
  
  /*	
  if (!setlocale(LC_CTYPE,"")) {
    fprintf(stderr,"Warning: Failed to set the locale!\n");
  }
  */

  x = LastUsed + 1;
  if (x > argc) {
    error_message.Cat("\t\t\t<isearch:error_text>Unrecognized arguments</isearch:error_text>\n");
    XmlBail(DBName,error_message);
  }
  
  DBPathName = DBName;
  DBFileName = DBName;
  RemovePath(&DBFileName);
  RemoveFileName(&DBPathName);

  // See if we have a legitimate file
  if (!DBExists(DBName)) {
    // The file does not exist
    error_message.Cat("\t\t\t<isearch:error_text>The specified database was not found: ");
    error_message.Cat(DBName);
    error_message.Cat("</isearch:error_text>\n");
    XmlBail(DBName,error_message);
  }

  VIDB *pdb;
  pdb = new VIDB(DBPathName, DBFileName, DocTypeOptions);
  /*
  if (DebugFlag) {
    pdb->DebugModeOn();
  }
  */
  if (!pdb->IsDbCompatible()) {
    Error=GDT_TRUE;
    error_message.Cat("\t\t\t<isearch:error_text>The specified database is not compatible with this version of zpresent.</isearch:error_text>\n");
    delete pdb;
    XmlBail(DBName,error_message);
  }
  
  // Set the record syntax to SUTRS if it is not specified, and
  // convert OIDs to text, if necessary
  if (RecordSyntax.GetLength() == 0)
    RecordSyntax = XmlRecordSyntax;
  else if (RecordSyntax.CaseEquals("TEXT"))
    RecordSyntax = SutrsRecordSyntax;
  else if (RecordSyntax.CaseEquals(SutrsRecordSyntaxOID))
    RecordSyntax = SutrsRecordSyntax;
  else if (RecordSyntax.CaseEquals(MimeRecordSyntaxOID))
    RecordSyntax = SutrsRecordSyntax;
  else if (RecordSyntax.CaseEquals(UsmarcRecordSyntaxOID))
    RecordSyntax = UsmarcRecordSyntax;
  else if (RecordSyntax.CaseEquals(HtmlRecordSyntaxOID))
    RecordSyntax = HtmlRecordSyntax;
  else if (RecordSyntax.CaseEquals(SgmlRecordSyntaxOID))
    RecordSyntax = SgmlRecordSyntax;
  else if (RecordSyntax.CaseEquals(XmlRecordSyntaxOID))
    RecordSyntax = XmlRecordSyntax;
  else if (RecordSyntax.CaseEquals(GRS1RecordSyntaxOID))
    RecordSyntax = GRS1RecordSyntax;
  else if (RecordSyntax.CaseEquals(OldHtmlRecordSyntaxOID))
    RecordSyntax = HtmlRecordSyntax;
  else if (RecordSyntax.CaseEquals(CNIDRHtmlRecordSyntaxOID))
    RecordSyntax = HtmlRecordSyntax;
  else if (RecordSyntax.CaseEquals(CNIDRSgmlRecordSyntaxOID))
    RecordSyntax = SgmlRecordSyntax;

  printf("\t<isearch:present status=\"OK\" dbname=\"");
  DBName.Print();
  printf("\">\n");

  printf("\t\t<isearch:results count=\"%d\" eset=\"",n_keys);
  ESet.Print();
  printf("\" prs=\"");
  RecordSyntax.Print();
  printf("\">\n");

  RecordKey = "";
  for (INT i=1;i<=n_keys;i++) {

    RecordKeyList.GetEntry(i,&RecordKey);
    pdb->KeyLookup(RecordKey, &RsRecord);
    pdb->Present(RsRecord, ESet, XmlRecordSyntax, &Record);
    if (Record.GetLength() > 0) {
      Record.Print();

    } else {
      error_message.Cat("\t\t\t\t<isearch:error_text>Record ");
      error_message.Cat("not found.</isearch:error_text>\n");
      cout << "\t\t\t<isearch:result docid=\"" << RecordKey <<
	"\" status=\"Error\">\n";
      cout << error_message;
      cout << "\t\t\t</isearch:result>" << endl;
    }
  }

  printf("\t\t</isearch:results>\n");
  printf("\t</isearch:present>\n");
  printf("</zpresent>\n");
  delete pdb;
  fflush(stdout); fflush(stderr); exit (0);
}