// Created on: 2001-07-09
// Created by: Julia DOROVSKIKH
// Copyright (c) 2001-2014 OPEN CASCADE SAS
//
// This file is part of Open CASCADE Technology software library.
//
// This library is free software; you can redistribute it and/or modify it under
// the terms of the GNU Lesser General Public License version 2.1 as published
// by the Free Software Foundation, with special exception defined in the file
// OCCT_LGPL_EXCEPTION.txt. Consult the file LICENSE_LGPL_21.txt included in OCCT
// distribution for complete text of the license and disclaimer of any warranty.
//
// Alternatively, this file may be used under the terms of Open CASCADE
// commercial license or contractual agreement.

#include <CDM_Application.hxx>
#include <Message_Messenger.hxx>
#include <Message_ProgressScope.hxx>
#include <CDM_MetaData.hxx>
#include <OSD_FileSystem.hxx>
#include <OSD_Path.hxx>
#include <PCDM_DOMHeaderParser.hxx>
#include <Standard_Type.hxx>
#include <TCollection_AsciiString.hxx>
#include <TCollection_ExtendedString.hxx>
#include <TDF_Data.hxx>
#include <TDocStd_Document.hxx>
#include <TDocStd_Owner.hxx>
#include <UTL.hxx>
#include <XmlLDrivers.hxx>
#include <XmlLDrivers_DocumentRetrievalDriver.hxx>
#include <XmlMDF.hxx>
#include <XmlMDF_ADriver.hxx>
#include <XmlMDF_ADriverTable.hxx>
#include <XmlObjMgt.hxx>
#include <XmlObjMgt_RRelocationTable.hxx>

IMPLEMENT_STANDARD_RTTIEXT(XmlLDrivers_DocumentRetrievalDriver, PCDM_RetrievalDriver)

#ifdef _MSC_VER
  #include <tchar.h>
#endif // _MSC_VER

#include <Standard_Failure.hxx>
#include <Standard_ErrorHandler.hxx>

#define START_REF "START_REF"
#define END_REF "END_REF"

#define MODIFICATION_COUNTER "MODIFICATION_COUNTER: "
#define REFERENCE_COUNTER "REFERENCE_COUNTER: "

// #define TAKE_TIMES
static void take_time(const int, const char*, const occ::handle<Message_Messenger>&)
#ifdef TAKE_TIMES
  ;
#else
{
}
#endif

static int RemoveExtraSeparator(TCollection_AsciiString& aString)
{

  int i, j, len;

  len = aString.Length();
#ifdef _WIN32
  // Case of network path, such as \\MACHINE\dir
  for (i = j = 2; j <= len; i++, j++)
  {
#else
  for (i = j = 1; j <= len; i++, j++)
  {
#endif
    char c = aString.Value(j);
    aString.SetValue(i, c);
    if (c == '/')
    {
      while (j < len && aString.Value(j + 1) == '/')
      {
        j++;
      }
    }
  }
  len = i - 1;
  if (aString.Value(len) == '/')
  {
    len--;
  }
  aString.Trunc(len);
  return len;
}

static TCollection_AsciiString GetDirFromFile(const TCollection_ExtendedString& aFileName)
{
  TCollection_AsciiString theCFile = UTL::CString(aFileName);
  TCollection_AsciiString theDirectory;
  int                     i = theCFile.SearchFromEnd("/");
#ifdef _WIN32
  //    if(i==-1) i=theCFile.SearchFromEnd("\\");
  if (theCFile.SearchFromEnd("\\") > i)
    i = theCFile.SearchFromEnd("\\");
#endif
  if (i != -1)
  {
    theDirectory = theCFile.SubString(1, i);
  }
  return theDirectory;
}

static TCollection_AsciiString AbsolutePath(const TCollection_AsciiString& aDirPath,
                                            const TCollection_AsciiString& aRelFilePath)
{
  TCollection_AsciiString EmptyString = "";
#ifdef _WIN32
  if (aRelFilePath.Search(":") == 2
      || (aRelFilePath.Search("\\") == 1 && aRelFilePath.Value(2) == '\\'))
  {
#else
  if (aRelFilePath.Search("/") == 1)
  {
#endif
    return aRelFilePath;
  }
  TCollection_AsciiString DirPath = aDirPath, RelFilePath = aRelFilePath;
  int                     i, len;

#ifdef _WIN32
  if (DirPath.Search(":") != 2 && (DirPath.Search("\\") != 1 || DirPath.Value(2) != '\\'))
  {
#else
  if (DirPath.Search("/") != 1)
  {
#endif
    return EmptyString;
  }

#ifdef _WIN32
  DirPath.ChangeAll('\\', '/');
  RelFilePath.ChangeAll('\\', '/');
#endif

  RemoveExtraSeparator(DirPath);
  len = RemoveExtraSeparator(RelFilePath);

  while (RelFilePath.Search("../") == 1)
  {
    if (len == 3)
    {
      return EmptyString;
    }
    RelFilePath = RelFilePath.SubString(4, len);
    len -= 3;
    if (DirPath.IsEmpty())
    {
      return EmptyString;
    }
    i = DirPath.SearchFromEnd("/");
    if (i < 0)
    {
      return EmptyString;
    }
    DirPath.Trunc(i - 1);
  }
  TCollection_AsciiString retx;
  retx = DirPath;
  retx += "/";
  retx += RelFilePath;
  return retx;
}

//=================================================================================================

XmlLDrivers_DocumentRetrievalDriver::XmlLDrivers_DocumentRetrievalDriver()
{
  myReaderStatus = PCDM_RS_OK;
}

//=================================================================================================

void XmlLDrivers_DocumentRetrievalDriver::Read(const TCollection_ExtendedString&     theFileName,
                                               const occ::handle<CDM_Document>&      theNewDocument,
                                               const occ::handle<CDM_Application>&   theApplication,
                                               const occ::handle<PCDM_ReaderFilter>& theFilter,
                                               const Message_ProgressRange&          theRange)
{
  myReaderStatus = PCDM_RS_DriverFailure;
  myFileName     = theFileName;

  const occ::handle<OSD_FileSystem>& aFileSystem = OSD_FileSystem::DefaultFileSystem();
  std::shared_ptr<std::istream> aFileStream = aFileSystem->OpenIStream(myFileName, std::ios::in);

  if (aFileStream.get() != nullptr && aFileStream->good())
  {
    Read(*aFileStream, nullptr, theNewDocument, theApplication, theFilter, theRange);
  }
  else
  {
    myReaderStatus = PCDM_RS_OpenError;

    TCollection_ExtendedString aMsg = TCollection_ExtendedString("Error: the file ") + theFileName
                                      + " cannot be opened for reading";

    theApplication->MessageDriver()->Send(aMsg.ToExtString(), Message_Fail);
    throw Standard_Failure("File cannot be opened for reading");
  }
}

//=================================================================================================

void XmlLDrivers_DocumentRetrievalDriver::Read(Standard_IStream& theIStream,
                                               const occ::handle<Storage_Data>& /*theStorageData*/,
                                               const occ::handle<CDM_Document>&    theNewDocument,
                                               const occ::handle<CDM_Application>& theApplication,
                                               const occ::handle<PCDM_ReaderFilter>& /*theFilter*/,
                                               const Message_ProgressRange& theRange)
{
  occ::handle<Message_Messenger> aMessageDriver = theApplication->MessageDriver();
  ::take_time(~0, " +++++ Start RETRIEVE procedures ++++++", aMessageDriver);

  // 1. Read DOM_Document from file
  LDOMParser aParser;

  // if myFileName is not empty, "document" tag is required to be read
  // from the received document
  bool aWithoutRoot = myFileName.IsEmpty();

  if (aParser.parse(theIStream, false, aWithoutRoot))
  {
    TCollection_AsciiString aData;
    std::cout << aParser.GetError(aData) << ": " << aData << '\n';
    myReaderStatus = PCDM_RS_FormatFailure;
    return;
  }
  const XmlObjMgt_Element anElement = aParser.getDocument().getDocumentElement();
  ::take_time(0, " +++++ Fin parsing XML :       ", aMessageDriver);

  ReadFromDomDocument(anElement, theNewDocument, theApplication, theRange);
}

//=======================================================================
// function : ReadFromDomDocument
// purpose  : management of the macro-structure of XML document data
// remark   : If the application needs to use myRelocTable to retrieve additional
//           data from LDOM, this method should be reimplemented
//=======================================================================

void XmlLDrivers_DocumentRetrievalDriver::ReadFromDomDocument(
  const XmlObjMgt_Element&            theElement,
  const occ::handle<CDM_Document>&    theNewDocument,
  const occ::handle<CDM_Application>& theApplication,
  const Message_ProgressRange&        theRange)
{
  const occ::handle<Message_Messenger> aMsgDriver = theApplication->MessageDriver();
  // 1. Read info // to be done
  TCollection_AsciiString anAbsoluteDirectory = GetDirFromFile(myFileName);
  int aCurDocVersion = TDocStd_FormatVersion_VERSION_2; // minimum supported version
  TCollection_ExtendedString anInfo;
  const XmlObjMgt_Element    anInfoElem = theElement.GetChildByTagName("info");
  if (anInfoElem != nullptr)
  {
    XmlObjMgt_DOMString aDocVerStr = anInfoElem.getAttribute("DocVersion");
    if (aDocVerStr != nullptr)
    {
      int anIntegerVersion = 0;
      if (aDocVerStr.GetInteger(anIntegerVersion))
      {
        aCurDocVersion = anIntegerVersion;
      }
      else
      {
        TCollection_ExtendedString aMsg =
          TCollection_ExtendedString("Cannot retrieve the current Document version"
                                     " attribute as \"")
          + aDocVerStr + "\"";
        if (!aMsgDriver.IsNull())
        {
          aMsgDriver->Send(aMsg.ToExtString(), Message_Fail);
        }
      }
    }

    // oan: OCC22305 - check a document version and if it's greater than
    // current version of storage driver set an error status and return
    if (aCurDocVersion > TDocStd_Document::CurrentStorageFormatVersion())
    {
      TCollection_ExtendedString aMsg = TCollection_ExtendedString("error: wrong file version: ")
                                        + aDocVerStr + " while current is "
                                        + TDocStd_Document::CurrentStorageFormatVersion();
      myReaderStatus = PCDM_RS_NoVersion;
      if (!aMsgDriver.IsNull())
      {
        aMsgDriver->Send(aMsg.ToExtString(), Message_Fail);
      }
      return;
    }

    bool isRef = false;
    for (LDOM_Node aNode = anInfoElem.getFirstChild(); aNode != nullptr;
         aNode           = aNode.getNextSibling())
    {
      if (aNode.getNodeType() == LDOM_Node::ELEMENT_NODE)
      {
        if (XmlObjMgt::GetExtendedString((LDOM_Element&)aNode, anInfo))
        {

          // Read ref counter
          if (anInfo.Search(REFERENCE_COUNTER) != -1)
          {
            try
            {
              OCC_CATCH_SIGNALS
              TCollection_AsciiString anInf(anInfo, '?');
              int                     aRefCounter = anInf.Token(" ", 2).IntegerValue();
              theNewDocument->SetReferenceCounter(aRefCounter);
            }
            catch (Standard_Failure const&)
            {
              //    std::cout << "warning: could not read the reference counter in " << aFileName <<
              //    std::endl;
              TCollection_ExtendedString aMsg("Warning: ");
              aMsg = aMsg.Cat("could not read the reference counter").Cat("\0");
              if (!aMsgDriver.IsNull())
              {
                aMsgDriver->Send(aMsg.ToExtString(), Message_Warning);
              }
            }
          }
          else if (anInfo.Search(MODIFICATION_COUNTER) != -1)
          {
            try
            {
              OCC_CATCH_SIGNALS

              TCollection_AsciiString anInf(anInfo, '?');
              int                     aModCounter = anInf.Token(" ", 2).IntegerValue();
              theNewDocument->SetModifications(aModCounter);
            }
            catch (Standard_Failure const&)
            {
              TCollection_ExtendedString aMsg("Warning: could not read the modification counter\0");
              if (!aMsgDriver.IsNull())
              {
                aMsgDriver->Send(aMsg.ToExtString(), Message_Warning);
              }
            }
          }

          if (anInfo == END_REF)
          {
            isRef = false;
          }
          if (isRef)
          { // Process References

            int pos = anInfo.Search(" ");
            if (pos != -1)
            {
              // Parce RefId, DocumentVersion and FileName
              int                        aRefId;
              TCollection_ExtendedString aFileName;
              int                        aDocumentVersion;

              TCollection_ExtendedString aRest = anInfo.Split(pos);
              aRefId                           = UTL::IntegerValue(anInfo);

              int pos2 = aRest.Search(" ");

              aFileName        = aRest.Split(pos2);
              aDocumentVersion = UTL::IntegerValue(aRest);

              TCollection_AsciiString aPath = UTL::CString(aFileName);
              TCollection_AsciiString anAbsolutePath;
              if (!anAbsoluteDirectory.IsEmpty())
              {
                anAbsolutePath = AbsolutePath(anAbsoluteDirectory, aPath);
                if (!anAbsolutePath.IsEmpty())
                {
                  aPath = anAbsolutePath;
                }
              }
              if (!aMsgDriver.IsNull())
              {
                //      std::cout << "reference found; ReferenceIdentifier: " <<
                //      theReferenceIdentifier << "; File:" << thePath << ", version:" <<
                //      theDocumentVersion;
                TCollection_ExtendedString aMsg("Warning: ");
                aMsg = aMsg.Cat("reference found; ReferenceIdentifier:  ")
                         .Cat(aRefId)
                         .Cat("; File:")
                         .Cat(aPath)
                         .Cat(", version:")
                         .Cat(aDocumentVersion)
                         .Cat("\0");
                aMsgDriver->Send(aMsg.ToExtString(), Message_Warning);
              }
              // Add new ref!
              /////////////
              TCollection_ExtendedString theFolder, theName;
              // TCollection_ExtendedString theFile=myReferences(myIterator).FileName();
              TCollection_ExtendedString f(aPath);
#ifndef _WIN32

              int                        i = f.SearchFromEnd("/");
              TCollection_ExtendedString n = f.Split(i);
              f.Trunc(f.Length() - 1);
              theFolder = f;
              theName   = n;
#else
              OSD_Path                   p = UTL::Path(f);
              char16_t                   chr;
              TCollection_ExtendedString dir, dirRet, name;

              dir = UTL::Disk(p);
              dir += UTL::Trek(p);

              for (int i = 1; i <= dir.Length(); ++i)
              {

                chr = dir.Value(i);

                switch (chr)
                {

                  case '|':
                    dirRet += "/";
                    break;

                  case '^':

                    dirRet += "..";
                    break;

                  default:
                    dirRet += chr;
                }
              }
              theFolder = dirRet;
              theName   = UTL::Name(p);
              theName += UTL::Extension(p);
#endif // _WIN32

              occ::handle<CDM_MetaData> aMetaData =
                CDM_MetaData::LookUp(theApplication->MetaDataLookUpTable(),
                                     theFolder,
                                     theName,
                                     aPath,
                                     aPath,
                                     UTL::IsReadOnly(aFileName));
              ////////////
              theNewDocument->CreateReference(aMetaData,
                                              aRefId,
                                              theApplication,
                                              aDocumentVersion,
                                              false);
            }
          }
          if (anInfo == START_REF)
          {
            isRef = true;
          }
        }
      }
    }
  }

  // 2. Read comments
  TCollection_ExtendedString aComment;
  const XmlObjMgt_Element    aCommentsElem = theElement.GetChildByTagName("comments");
  if (aCommentsElem != nullptr)
  {
    for (LDOM_Node aNode = aCommentsElem.getFirstChild(); aNode != nullptr;
         aNode           = aNode.getNextSibling())
    {
      if (aNode.getNodeType() == LDOM_Node::ELEMENT_NODE)
      {
        if (XmlObjMgt::GetExtendedString((LDOM_Element&)aNode, aComment))
        {
          theNewDocument->AddComment(aComment);
        }
      }
    }
  }
  Message_ProgressScope aPS(theRange, "Reading document", 2);
  // 2. Read Shapes section
  if (myDrivers.IsNull())
  {
    myDrivers = AttributeDrivers(aMsgDriver);
  }
  const occ::handle<XmlMDF_ADriver> aNSDriver =
    ReadShapeSection(theElement, aMsgDriver, aPS.Next());
  if (!aNSDriver.IsNull())
  {
    ::take_time(0, " +++++ Fin reading Shapes :    ", aMsgDriver);
  }

  if (!aPS.More())
  {
    myReaderStatus = PCDM_RS_UserBreak;
    return;
  }

  // 2.1. Keep document format version in RT
  occ::handle<Storage_HeaderData> aHeaderData = new Storage_HeaderData();
  aHeaderData->SetStorageVersion(aCurDocVersion);
  myRelocTable.Clear();
  myRelocTable.SetHeaderData(aHeaderData);

  // 5. Read document contents
  try
  {
    OCC_CATCH_SIGNALS
#ifdef OCCT_DEBUG
    TCollection_ExtendedString aMessage("PasteDocument");
    aMsgDriver->Send(aMessage.ToExtString(), Message_Trace);
#endif
    if (!MakeDocument(theElement, theNewDocument, aPS.Next()))
    {
      myReaderStatus = PCDM_RS_MakeFailure;
    }
    else
    {
      myReaderStatus = PCDM_RS_OK;
    }
  }
  catch (Standard_Failure const& anException)
  {
    TCollection_ExtendedString anErrorString(anException.what());
    aMsgDriver->Send(anErrorString.ToExtString(), Message_Fail);
  }
  if (!aPS.More())
  {
    myReaderStatus = PCDM_RS_UserBreak;
    return;
  }

  //    Wipe off the shapes written to the <shapes> section
  ShapeSetCleaning(aNSDriver);

  //    Clean the relocation table.
  //    If the application needs to use myRelocTable to retrieve additional
  //    data from LDOM, this method should be reimplemented avoiding this step
  myRelocTable.Clear();
  ::take_time(0, " +++++ Fin reading data OCAF : ", aMsgDriver);
}

//=================================================================================================

bool XmlLDrivers_DocumentRetrievalDriver::MakeDocument(const XmlObjMgt_Element&         theElement,
                                                       const occ::handle<CDM_Document>& theTDoc,
                                                       const Message_ProgressRange&     theRange)
{
  bool                          aResult = false;
  occ::handle<TDocStd_Document> TDOC    = occ::down_cast<TDocStd_Document>(theTDoc);
  if (!TDOC.IsNull())
  {
    occ::handle<TDF_Data> aTDF = new TDF_Data();
    aResult = XmlMDF::FromTo(theElement, aTDF, myRelocTable, myDrivers, theRange);
    if (aResult)
    {
      TDOC->SetData(aTDF);
      TDocStd_Owner::SetDocument(aTDF, TDOC);
    }
  }
  return aResult;
}

//=================================================================================================

occ::handle<XmlMDF_ADriverTable> XmlLDrivers_DocumentRetrievalDriver::AttributeDrivers(
  const occ::handle<Message_Messenger>& theMessageDriver)
{
  return XmlLDrivers::AttributeDrivers(theMessageDriver);
}

//=======================================================================
// function : take_time
// class    : static
// purpose  : output astronomical time elapsed
//=======================================================================
#ifdef TAKE_TIMES
  #include <time.h>
  #include <sys/timeb.h>
  #include <sys/types.h>
  #include <stdio.h>
  #ifndef _WIN32
extern "C" int ftime(struct timeb* tp);
  #endif
extern struct timeb tmbuf0;

static void take_time(const int                             isReset,
                      const char*                           aHeader,
                      const occ::handle<Message_Messenger>& aMessageDriver)
{
  struct timeb tmbuf;
  ftime(&tmbuf);
  TCollection_ExtendedString aMessage((const char*)aHeader);
  if (isReset)
    tmbuf0 = tmbuf;
  else
  {
    char take_tm_buf[64];
    Sprintf(take_tm_buf,
            "%9.2f s ++++",
            double(tmbuf.time - tmbuf0.time) + double(tmbuf.millitm - tmbuf0.millitm) / 1000.);
    aMessage += take_tm_buf;
  }
  aMessageDriver->Write(aMessage.ToExtString());
}
#endif

//=======================================================================
// function : ReadShapeSection
// purpose  : definition of ReadShapeSection
//=======================================================================
occ::handle<XmlMDF_ADriver> XmlLDrivers_DocumentRetrievalDriver::ReadShapeSection(
  const XmlObjMgt_Element& /*theElement*/,
  const occ::handle<Message_Messenger>& /*aMsgDriver*/,
  const Message_ProgressRange& /*theRange*/)
{
  occ::handle<XmlMDF_ADriver> aDriver;
  // empty; to be redefined
  return aDriver;
}

//=======================================================================
// function : ShapeSetCleaning
// purpose  : definition of ShapeSetCleaning
//=======================================================================
void XmlLDrivers_DocumentRetrievalDriver::ShapeSetCleaning(
  const occ::handle<XmlMDF_ADriver>& /*theDriver*/)
{
}
