// Copyright (c) 2021 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 <OSD_FileSystemSelector.hxx>

IMPLEMENT_STANDARD_RTTIEXT(OSD_FileSystemSelector, OSD_FileSystem)

//=======================================================================
// function : AddProtocol
// purpose :
//=======================================================================
void OSD_FileSystemSelector::AddProtocol (const Handle(OSD_FileSystem)& theFileSystem, bool theIsPreferred)
{
  myProtocols.Remove (theFileSystem); // avoid duplicates
  if (theIsPreferred)
  {
    myProtocols.Prepend (theFileSystem);
  }
  else
  {
    myProtocols.Append (theFileSystem);
  }
}

//=======================================================================
// function : RemoveProtocol
// purpose :
//=======================================================================
void OSD_FileSystemSelector::RemoveProtocol (const Handle(OSD_FileSystem)& theFileSystem)
{
  myProtocols.Remove (theFileSystem);
}

//=======================================================================
// function : IsSupportedPath
// purpose :
//=======================================================================
Standard_Boolean OSD_FileSystemSelector::IsSupportedPath (const TCollection_AsciiString& theUrl) const
{
  for (NCollection_List<Handle(OSD_FileSystem)>::Iterator aProtIter(myProtocols); aProtIter.More(); aProtIter.Next())
  {
    if (aProtIter.Value()->IsSupportedPath (theUrl))
    {
      return true;
    }
  }
  return false;
}

//=======================================================================
// function : IsOpenIStream
// purpose :
//=======================================================================
Standard_Boolean OSD_FileSystemSelector::IsOpenIStream (const opencascade::std::shared_ptr<std::istream>& theStream) const
{
  opencascade::std::shared_ptr<OSD_IStreamBuffer> aFileStream = opencascade::std::dynamic_pointer_cast<OSD_IStreamBuffer> (theStream);
  if (aFileStream.get() == NULL)
  {
    return false;
  }
  for (NCollection_List<Handle(OSD_FileSystem)>::Iterator aProtIter(myProtocols); aProtIter.More(); aProtIter.Next())
  {
    const Handle(OSD_FileSystem)& aFileSystem = aProtIter.Value();
    if (aFileSystem->IsSupportedPath (TCollection_AsciiString (aFileStream->Url().c_str())))
    {
      if (aFileSystem->IsOpenIStream (theStream))
      {
        return true;
      }
    }
  }
  return false;
}

//=======================================================================
// function : OpenIStream
// purpose :
//=======================================================================
opencascade::std::shared_ptr<std::istream> OSD_FileSystemSelector::OpenIStream (const TCollection_AsciiString& theUrl,
                                                                                const std::ios_base::openmode theMode,
                                                                                const int64_t theOffset,
                                                                                const opencascade::std::shared_ptr<std::istream>& theOldStream)
{
  for (NCollection_List<Handle(OSD_FileSystem)>::Iterator aProtIter (myProtocols); aProtIter.More(); aProtIter.Next())
  {
    const Handle(OSD_FileSystem)& aFileSystem = aProtIter.Value();
    if (aFileSystem->IsSupportedPath (theUrl))
    {
      opencascade::std::shared_ptr<std::istream> aStream = aFileSystem->OpenIStream (theUrl, theMode, theOffset, theOldStream);
      if (aStream.get() != NULL)
      {
        return aStream;
      }
    }
  }
  return opencascade::std::shared_ptr<std::istream>();
}


//=======================================================================
// function : OpenStreamBuffer
// purpose :
//=======================================================================
opencascade::std::shared_ptr<std::streambuf> OSD_FileSystemSelector::OpenStreamBuffer (const TCollection_AsciiString& theUrl,
                                                                                       const std::ios_base::openmode theMode,
                                                                                       const int64_t theOffset,
                                                                                       int64_t* theOutBufSize)
{
  for (NCollection_List<Handle(OSD_FileSystem)>::Iterator aProtIter (myProtocols); aProtIter.More(); aProtIter.Next())
  {
    const Handle(OSD_FileSystem)& aFileSystem = aProtIter.Value();
    if (aFileSystem->IsSupportedPath (theUrl))
    {
      opencascade::std::shared_ptr<std::streambuf> aBuf = aFileSystem->OpenStreamBuffer (theUrl, theMode, theOffset, theOutBufSize);
      if (aBuf.get() != NULL)
      {
        return aBuf;
      }
    }
  }
  return opencascade::std::shared_ptr<std::streambuf>();
}