1 files changed, 113 insertions, 0 deletions
diff --git a/src/zenserver/storage/zenstorageserver.h b/src/zenserver/storage/zenstorageserver.h
new file mode 100644
index 000000000..e4c31399d
--- /dev/null
+++ b/src/zenserver/storage/zenstorageserver.h
@@ -0,0 +1,113 @@
+// Copyright Epic Games, Inc. All Rights Reserved.
+
+#pragma once
+
+#include "zenserver.h"
+
+#include <zenhttp/auth/authmgr.h>
+#include <zenhttp/auth/authservice.h>
+#include <zenhttp/httptest.h>
+#include <zenstore/cache/structuredcachestore.h>
+#include <zenstore/gc.h>
+#include <zenstore/projectstore.h>
+
+#include "admin/admin.h"
+#include "buildstore/httpbuildstore.h"
+#include "cache/httpstructuredcache.h"
+#include "diag/diagsvcs.h"
+#include "frontend/frontend.h"
+#include "objectstore/objectstore.h"
+#include "projectstore/httpprojectstore.h"
+#include "stats/statsreporter.h"
+#include "upstream/upstream.h"
+#include "vfs/vfsservice.h"
+#include "workspaces/httpworkspaces.h"
+
+namespace zen {
+
+class ZenStorageServer : public ZenServerBase
+{
+	ZenStorageServer& operator=(ZenStorageServer&&) = delete;
+	ZenStorageServer(ZenStorageServer&&)			= delete;
+
+public:
+	ZenStorageServer();
+	~ZenStorageServer();
+
+	void SetDedicatedMode(bool State) { m_IsDedicatedMode = State; }
+	void SetTestMode(bool State) { m_TestMode = State; }
+	void SetDataRoot(std::filesystem::path Root) { m_DataRoot = Root; }
+	void SetContentRoot(std::filesystem::path Root) { m_ContentRoot = Root; }
+
+	int	 Initialize(const ZenStorageServerOptions& ServerOptions, ZenServerState::ZenServerEntry* ServerEntry);
+	void Run();
+	void Cleanup();
+
+private:
+	void InitializeState(const ZenStorageServerOptions& ServerOptions);
+	void InitializeStructuredCache(const ZenStorageServerOptions& ServerOptions);
+	void Flush();
+
+	bool				  m_IsDedicatedMode		   = false;
+	bool				  m_TestMode			   = false;
+	bool				  m_DebugOptionForcedCrash = false;
+	std::string			  m_StartupScrubOptions;
+	CbObject			  m_RootManifest;
+	std::filesystem::path m_DataRoot;
+	std::filesystem::path m_ContentRoot;
+	asio::steady_timer	  m_StateMarkerTimer{m_IoContext};
+
+	void EnqueueStateMarkerTimer();
+	void CheckStateMarker();
+
+	std::unique_ptr<AuthMgr>		 m_AuthMgr;
+	std::unique_ptr<HttpAuthService> m_AuthService;
+	void							 InitializeAuthentication(const ZenStorageServerOptions& ServerOptions);
+
+	void InitializeServices(const ZenStorageServerOptions& ServerOptions);
+	void RegisterServices();
+
+	HttpStatsService				  m_StatsService;
+	std::unique_ptr<JobQueue>		  m_JobQueue;
+	GcManager						  m_GcManager;
+	GcScheduler						  m_GcScheduler{m_GcManager};
+	std::unique_ptr<CidStore>		  m_CidStore;
+	Ref<ZenCacheStore>				  m_CacheStore;
+	std::unique_ptr<OpenProcessCache> m_OpenProcessCache;
+	HttpTestService					  m_TestService;
+	std::unique_ptr<CidStore>		  m_BuildCidStore;
+	std::unique_ptr<BuildStore>		  m_BuildStore;
+
+#if ZEN_WITH_TESTS
+	HttpTestingService m_TestingService;
+#endif
+
+	RefPtr<ProjectStore>						m_ProjectStore;
+	std::unique_ptr<VfsServiceImpl>				m_VfsServiceImpl;
+	std::unique_ptr<HttpProjectService>			m_HttpProjectService;
+	std::unique_ptr<Workspaces>					m_Workspaces;
+	std::unique_ptr<HttpWorkspacesService>		m_HttpWorkspacesService;
+	std::unique_ptr<UpstreamCache>				m_UpstreamCache;
+	std::unique_ptr<HttpUpstreamService>		m_UpstreamService;
+	std::unique_ptr<HttpStructuredCacheService> m_StructuredCacheService;
+	std::unique_ptr<HttpFrontendService>		m_FrontendService;
+	std::unique_ptr<HttpObjectStoreService>		m_ObjStoreService;
+	std::unique_ptr<HttpBuildStoreService>		m_BuildStoreService;
+	std::unique_ptr<VfsService>					m_VfsService;
+	std::unique_ptr<HttpAdminService>			m_AdminService;
+};
+
+class ZenStorageServerMain : public ZenServerMain
+{
+public:
+	ZenStorageServerMain(ZenStorageServerOptions& ServerOptions);
+	virtual void DoRun(ZenServerState::ZenServerEntry* Entry) override;
+
+	ZenStorageServerMain(const ZenStorageServerMain&) = delete;
+	ZenStorageServerMain& operator=(const ZenStorageServerMain&) = delete;
+
+private:
+	ZenStorageServerOptions& m_ServerOptions;
+};
+
+}  // namespace zen