FileserveClient.h

#pragma once
 
#include <FileservePlugin/FileservePluginDLL.h>
 
#include <Core/Interfaces/RemoteToolingInterface.h>
#include <Foundation/Communication/RemoteInterface.h>
#include <Foundation/Configuration/Singleton.h>
#include <Foundation/Types/UniquePtr.h>
#include <Foundation/Types/Uuid.h>
 
namespace plDataDirectory
{
  class FileserveType;
}
 
class PL_FILESERVEPLUGIN_DLL plFileserveClient : public plRemoteToolingInterface
{
  PL_DECLARE_SINGLETON_OF_INTERFACE(plFileserveClient, plRemoteToolingInterface);
 
public:
  plFileserveClient();
  ~plFileserveClient();
 
 
  plRemoteInterface* GetRemoteInterface() override { return m_pNetwork.Borrow(); }
 
  plResult SearchForServerAddress(plTime timeout = plTime::MakeFromSeconds(5));
 
  plResult WaitForServerInfo(plTime timeout = plTime::MakeFromSeconds(60.0 * 5));
 
  plResult SaveCurrentConnectionInfoToDisk() const;
 
  static void DisabledFileserveClient() { s_bEnableFileserve = false; }
 
  const char* GetServerConnectionAddress() { return m_sServerConnectionAddress; }
 
  plResult EnsureConnected(plTime timeout = plTime::MakeFromSeconds(-5));
 
  void UpdateClient();
 
  void AddServerAddressToTry(plStringView sAddress);
 
private:
  friend class plDataDirectory::FileserveType;
 
  static bool s_bEnableFileserve;
 
  struct FileCacheStatus
  {
    plInt64 m_TimeStamp = 0;
    plUInt64 m_FileHash = 0;
    plTime m_LastCheck;
  };
 
  struct DataDir
  {
    // plString m_sRootName;
    // plString m_sPathOnClient;
    plString m_sMountPoint;
    bool m_bMounted = false;
 
    plMap<plString, FileCacheStatus> m_CacheStatus;
  };
 
  void DeleteFile(plUInt16 uiDataDir, plStringView sFile);
  plUInt16 MountDataDirectory(plStringView sDataDir, plStringView sRootName);
  void UnmountDataDirectory(plUInt16 uiDataDir);
  static void ComputeDataDirMountPoint(plStringView sDataDir, plStringBuilder& out_sMountPoint);
  void BuildPathInCache(const char* szFile, const char* szMountPoint, plStringBuilder* out_pAbsPath, plStringBuilder* out_pFullPathMeta) const;
  void GetFullDataDirCachePath(const char* szDataDir, plStringBuilder& out_sFullPath, plStringBuilder& out_sFullPathMeta) const;
  void NetworkMsgHandler(plRemoteMessage& msg);
  void HandleFileTransferMsg(plRemoteMessage& msg);
  void HandleFileTransferFinishedMsg(plRemoteMessage& msg);
  static void WriteMetaFile(plStringBuilder sCachedMetaFile, plInt64 iFileTimeStamp, plUInt64 uiFileHash);
  void WriteDownloadToDisk(plStringBuilder sCachedFile);
  plResult DownloadFile(plUInt16 uiDataDirID, const char* szFile, bool bForceThisDataDir, plStringBuilder* out_pFullPath);
  void DetermineCacheStatus(plUInt16 uiDataDirID, const char* szFile, FileCacheStatus& out_Status) const;
  void UploadFile(plUInt16 uiDataDirID, const char* szFile, const plDynamicArray<plUInt8>& fileContent);
  void InvalidateFileCache(plUInt16 uiDataDirID, plStringView sFile, plUInt64 uiHash);
  static plResult TryReadFileserveConfig(const char* szFile, plStringBuilder& out_Result);
  plResult TryConnectWithFileserver(const char* szAddress, plTime timeout) const;
  void FillFileStatusCache(const char* szFile);
  void ShutdownConnection();
  void ClearState();
 
  mutable plMutex m_Mutex;
  mutable plString m_sServerConnectionAddress;
  plString m_sFileserveCacheFolder;
  plString m_sFileserveCacheMetaFolder;
  bool m_bDownloading = false;
  bool m_bFailedToConnect = false;
  bool m_bWaitingForUploadFinished = false;
  plUuid m_CurFileRequestGuid;
  plStringBuilder m_sCurFileRequest;
  plUniquePtr<plRemoteInterface> m_pNetwork;
  plDynamicArray<plUInt8> m_Download;
  plTime m_CurrentTime;
  plHybridArray<plString, 4> m_TryServerAddresses;
 
  plMap<plString, plUInt16> m_FileDataDir;
  plHybridArray<DataDir, 8> m_MountedDataDirs;
};