Fileserver.h

#pragma once
 
#include <FileservePlugin/Fileserver/ClientContext.h>
#include <Foundation/Communication/RemoteInterface.h>
#include <Foundation/Configuration/Singleton.h>
#include <Foundation/Containers/HashTable.h>
#include <Foundation/Types/UniquePtr.h>
#include <Foundation/Types/Uuid.h>
 
class plRemoteMessage;
 
struct plFileserverEvent
{
  enum class Type
  {
    None,
    ServerStarted,
    ServerStopped,
    ClientConnected,
    ClientReconnected, // connected again after a disconnect
    ClientDisconnected,
    MountDataDir,
    MountDataDirFailed,
    UnmountDataDir,
    FileDownloadRequest,
    FileDownloading,
    FileDownloadFinished,
    FileDeleteRequest,
    FileUploadRequest,
    FileUploading,
    FileUploadFinished,
    AreYouThereRequest,
    LogCustomActivity,
  };
 
  Type m_Type = Type::None;
  plUInt32 m_uiClientID = 0;
  const char* m_szName = nullptr;
  const char* m_szPath = nullptr;
  const char* m_szRedirectedPath = nullptr;
  plUInt32 m_uiSizeTotal = 0;
  plUInt32 m_uiSentTotal = 0;
  plFileserveFileState m_FileState = plFileserveFileState::None;
};
 
class PL_FILESERVEPLUGIN_DLL plFileserver
{
  PL_DECLARE_SINGLETON(plFileserver);
 
public:
  plFileserver();
 
  void StartServer();
 
  void StopServer();
 
  bool UpdateServer();
 
  bool IsServerRunning() const;
 
  void SetPort(plUInt16 uiPort);
 
  plUInt16 GetPort() const { return m_uiPort; }
 
  plEvent<const plFileserverEvent&> m_Events;
 
  void BroadcastReloadResourcesCommand();
 
  static plResult SendConnectionInfo(
    const char* szClientAddress, plUInt16 uiMyPort, const plArrayPtr<plStringBuilder>& myIPs, plTime timeout = plTime::MakeFromSeconds(10));
 
  using ClientMessageHandler = plDelegate<void(plFileserveClientContext&, plRemoteMessage&, plRemoteInterface&, plDelegate<void(const char*)>)>;
 
  void SetCustomMessageHandler(plUInt32 uiSystemID, ClientMessageHandler handler);
 
private:
  void NetworkEventHandler(const plRemoteEvent& e);
  plFileserveClientContext& DetermineClient(plRemoteMessage& msg);
  void NetworkMsgHandler(plRemoteMessage& msg);
  void UnknownNetworkMsgHandler(plRemoteMessage& msg);
  void HandleMountRequest(plFileserveClientContext& client, plRemoteMessage& msg);
  void HandleUnmountRequest(plFileserveClientContext& client, plRemoteMessage& msg);
  void HandleFileRequest(plFileserveClientContext& client, plRemoteMessage& msg);
  void HandleDeleteFileRequest(plFileserveClientContext& client, plRemoteMessage& msg);
  void HandleUploadFileHeader(plFileserveClientContext& client, plRemoteMessage& msg);
  void HandleUploadFileTransfer(plFileserveClientContext& client, plRemoteMessage& msg);
  void HandleUploadFileFinished(plFileserveClientContext& client, plRemoteMessage& msg);
  void LogCustomActivity(const char* szText);
 
  plHashTable<plUInt32, plFileserveClientContext> m_Clients;
  plUniquePtr<plRemoteInterface> m_pNetwork;
  plDynamicArray<plUInt8> m_SendToClient;   // ie. 'downloads' from server to client
  plDynamicArray<plUInt8> m_SentFromClient; // ie. 'uploads' from client to server
  plStringBuilder m_sCurFileUpload;
  plUuid m_FileUploadGuid;
  plUInt32 m_uiFileUploadSize;
  plUInt16 m_uiPort = 1042;
  plMap<plUInt32, ClientMessageHandler> m_CustomMessageHandlers;
};