aboutsummaryrefslogtreecommitdiff
path: root/src/zen/cmds/projectstore_cmd.h
blob: 1ba98b39ed6a004af084f423c6f6402b9edc2837 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
// Copyright Epic Games, Inc. All Rights Reserved.

#pragma once

#include "../zen.h"

#include "../authutils.h"

namespace zen {

class ProjectStoreCommand : public ZenCmdBase
{
	virtual ZenCmdCategory& CommandCategory() const override { return g_ProjectStoreCategory; }
};

class DropProjectCommand : public ProjectStoreCommand
{
public:
	static constexpr char Name[]		= "project-drop";
	static constexpr char Description[] = "Drop project or project oplog";

	DropProjectCommand();
	~DropProjectCommand();

	virtual void			  Run(const ZenCliOptions& GlobalOptions, int argc, char** argv) override;
	virtual cxxopts::Options& Options() override { return m_Options; }

private:
	cxxopts::Options m_Options{Name, Description};
	std::string		 m_HostName;
	std::string		 m_ProjectName;
	std::string		 m_OplogName;
	bool			 m_DryRun = true;
};

class ProjectInfoCommand : public ProjectStoreCommand
{
public:
	static constexpr char Name[]		= "project-info";
	static constexpr char Description[] = "Info on project or project oplog";

	ProjectInfoCommand();
	~ProjectInfoCommand();
	virtual void			  Run(const ZenCliOptions& GlobalOptions, int argc, char** argv) override;
	virtual cxxopts::Options& Options() override { return m_Options; }

private:
	cxxopts::Options m_Options{Name, Description};
	std::string		 m_HostName;
	std::string		 m_ProjectName;
	std::string		 m_OplogName;
};

class CreateProjectCommand : public ProjectStoreCommand
{
public:
	static constexpr char Name[]		= "project-create";
	static constexpr char Description[] = "Create a project";

	CreateProjectCommand();
	~CreateProjectCommand();

	virtual void			  Run(const ZenCliOptions& GlobalOptions, int argc, char** argv) override;
	virtual cxxopts::Options& Options() override { return m_Options; }

private:
	cxxopts::Options m_Options{Name, Description};
	std::string		 m_HostName;
	std::string		 m_ProjectId;
	std::string		 m_RootDir;
	std::string		 m_EngineRootDir;
	std::string		 m_ProjectRootDir;
	std::string		 m_ProjectFile;
	bool			 m_ForceUpdate = false;
};

class CreateOplogCommand : public ProjectStoreCommand
{
public:
	static constexpr char Name[]		= "oplog-create";
	static constexpr char Description[] = "Create a project oplog";

	CreateOplogCommand();
	~CreateOplogCommand();

	virtual void			  Run(const ZenCliOptions& GlobalOptions, int argc, char** argv) override;
	virtual cxxopts::Options& Options() override { return m_Options; }

private:
	cxxopts::Options m_Options{Name, Description};
	std::string		 m_HostName;
	std::string		 m_ProjectId;
	std::string		 m_OplogId;
	std::string		 m_GcPath;
	bool			 m_ForceUpdate = false;
};

class ExportOplogCommand : public ProjectStoreCommand
{
public:
	static constexpr char Name[]		= "oplog-export";
	static constexpr char Description[] = "Export project store oplog";

	ExportOplogCommand();
	~ExportOplogCommand();

	virtual void			  Run(const ZenCliOptions& GlobalOptions, int argc, char** argv) override;
	virtual cxxopts::Options& Options() override { return m_Options; }

private:
	cxxopts::Options m_Options{Name, Description};
	std::string		 m_HostName;
	std::string		 m_ProjectName;
	std::string		 m_OplogName;
	uint64_t		 m_MaxBlockSize		 = 0;
	uint64_t		 m_MaxChunksPerBlock = 0;
	uint64_t		 m_MaxChunkEmbedSize = 0;
	bool			 m_EmbedLooseFiles	 = false;
	bool			 m_Force			 = false;
	bool			 m_DisableBlocks	 = false;
	bool			 m_Async			 = false;

	std::string m_JupiterNamespace;
	std::string m_JupiterBucket;
	std::string m_JupiterOpenIdProvider;
	std::string m_JupiterAccessToken;
	std::string m_JupiterAccessTokenEnv;
	std::string m_JupiterAccessTokenPath;
	std::string m_OidcTokenAuthExecutablePath;
	bool		m_JupiterAssumeHttp2	   = false;
	bool		m_JupiterDisableTempBlocks = false;

	std::string m_CloudUrl;
	std::string m_CloudKey;
	std::string m_BaseCloudKey;

	std::string m_BuildsUrl;
	std::string m_BuildsId;
	std::string m_BuildsMetadataPath;
	std::string m_BuildsMetadata;

	bool m_BoostWorkerCount			= false;
	bool m_BoostWorkerMemory		= false;
	bool m_BoostWorkers				= false;
	bool m_IgnoreMissingAttachments = false;

	std::string m_ZenUrl;
	std::string m_ZenProjectName;
	std::string m_ZenOplogName;
	bool		m_ZenClean = false;

	std::string m_FileDirectoryPath;
	std::string m_FileName;
	std::string m_BaseFileName;
	bool		m_FileForceEnableTempBlocks = false;
	bool		m_PlainProgress				= false;
};

class ImportOplogCommand : public ProjectStoreCommand
{
public:
	static constexpr char Name[]		= "oplog-import";
	static constexpr char Description[] = "Import project store oplog";

	ImportOplogCommand();
	~ImportOplogCommand();

	virtual void			  Run(const ZenCliOptions& GlobalOptions, int argc, char** argv) override;
	virtual cxxopts::Options& Options() override { return m_Options; }

private:
	cxxopts::Options m_Options{Name, Description};
	std::string		 m_HostName;
	std::string		 m_ProjectName;
	std::string		 m_OplogName;
	std::string		 m_GcPath;
	bool			 m_Force					= false;
	bool			 m_Async					= false;
	bool			 m_IgnoreMissingAttachments = false;
	bool			 m_Clean					= false;
	bool			 m_PlainProgress			= false;

	std::string m_JupiterNamespace;
	std::string m_JupiterBucket;

	std::string m_JupiterOpenIdProvider;
	std::string m_JupiterAccessToken;
	std::string m_JupiterAccessTokenEnv;
	std::string m_JupiterAccessTokenPath;
	std::string m_OidcTokenAuthExecutablePath;
	bool		m_JupiterAssumeHttp2 = false;

	std::string m_CloudUrl;
	std::string m_ZenCacheHost;
	bool		m_UploadToZenCache = false;
	std::string m_CloudKey;

	std::string m_BuildsHost;
	std::string m_BuildsOverrideHost;
	std::string m_BuildsId;

	std::string m_ZenUrl;
	std::string m_ZenProjectName;
	std::string m_ZenOplogName;

	std::string m_FileDirectoryPath;
	std::string m_FileName;

	bool m_BoostWorkerCount	 = false;
	bool m_BoostWorkerMemory = false;
	bool m_BoostWorkers		 = false;

	std::string m_AllowPartialBlockRequests = "true";
};

class SnapshotOplogCommand : public ProjectStoreCommand
{
public:
	static constexpr char Name[]		= "oplog-snapshot";
	static constexpr char Description[] = "Snapshot project store oplog";

	SnapshotOplogCommand();
	~SnapshotOplogCommand();
	virtual void			  Run(const ZenCliOptions& GlobalOptions, int argc, char** argv) override;
	virtual cxxopts::Options& Options() override { return m_Options; }

private:
	cxxopts::Options m_Options{Name, Description};
	std::string		 m_HostName;
	std::string		 m_ProjectName;
	std::string		 m_OplogName;
};

class ProjectStatsCommand : public ProjectStoreCommand
{
public:
	static constexpr char Name[]		= "project-stats";
	static constexpr char Description[] = "Stats on project store";

	ProjectStatsCommand();
	~ProjectStatsCommand();
	virtual void			  Run(const ZenCliOptions& GlobalOptions, int argc, char** argv) override;
	virtual cxxopts::Options& Options() override { return m_Options; }

private:
	cxxopts::Options m_Options{Name, Description};
	std::string		 m_HostName;
};

class ProjectOpDetailsCommand : public ProjectStoreCommand
{
public:
	static constexpr char Name[]		= "project-op-details";
	static constexpr char Description[] = "Detail info on ops inside a project store oplog";

	ProjectOpDetailsCommand();
	~ProjectOpDetailsCommand();
	virtual void			  Run(const ZenCliOptions& GlobalOptions, int argc, char** argv) override;
	virtual cxxopts::Options& Options() override { return m_Options; }

private:
	cxxopts::Options m_Options{Name, Description};
	std::string		 m_HostName;
	bool			 m_Details			 = false;
	bool			 m_OpDetails		 = false;
	bool			 m_AttachmentDetails = false;
	bool			 m_CSV				 = false;
	std::string		 m_ProjectName;
	std::string		 m_OplogName;
	std::string		 m_OpId;
};

class OplogMirrorCommand : public ProjectStoreCommand
{
public:
	static constexpr char Name[]		= "oplog-mirror";
	static constexpr char Description[] = "Mirror project store oplog to file system";

	OplogMirrorCommand();
	~OplogMirrorCommand();
	virtual void			  Run(const ZenCliOptions& GlobalOptions, int argc, char** argv) override;
	virtual cxxopts::Options& Options() override { return m_Options; }

private:
	cxxopts::Options m_Options{Name, Description};
	std::string		 m_HostName;
	std::string		 m_ProjectName;
	std::string		 m_OplogName;
	std::string		 m_MirrorRootPath;
	std::string		 m_KeyFilter;
	std::string		 m_FilenameFilter;
	std::string		 m_ChunkIdFilter;
	bool			 m_Decompress		   = false;
	bool			 m_TrimToReferencedSet = true;
};

class OplogValidateCommand : public ProjectStoreCommand
{
public:
	static constexpr char Name[]		= "oplog-validate";
	static constexpr char Description[] = "Validate oplog for missing references";

	OplogValidateCommand();
	~OplogValidateCommand();
	virtual void			  Run(const ZenCliOptions& GlobalOptions, int argc, char** argv) override;
	virtual cxxopts::Options& Options() override { return m_Options; }

private:
	cxxopts::Options m_Options{Name, Description};
	std::string		 m_HostName;
	std::string		 m_ProjectName;
	std::string		 m_OplogName;
};

class OplogDownloadCommand : public ProjectStoreCommand
{
public:
	static constexpr char Name[]		= "oplog-download";
	static constexpr char Description[] = "Download an cloud storage oplog";

	OplogDownloadCommand();
	~OplogDownloadCommand();
	virtual void			  Run(const ZenCliOptions& GlobalOptions, int argc, char** argv) override;
	virtual cxxopts::Options& Options() override { return m_Options; }

private:
	cxxopts::Options m_Options{Name, Description};

	std::filesystem::path m_SystemRootDir;

	bool m_Quiet		 = false;
	bool m_Verbose		 = false;
	bool m_Yes			 = false;
	bool m_PlainProgress = false;
	bool m_LogProgress	 = false;

	AuthCommandLineOptions m_AuthOptions;

	// cloud builds
	std::string m_OverrideHost;
	std::string m_Host;
	std::string m_Url;
	bool		m_AssumeHttp2	= false;
	bool		m_AllowRedirect = false;
	std::string m_ZenCacheHost;
	std::string m_Namespace;
	std::string m_Bucket;
	std::string m_BuildId;
	bool		m_ForceDownload		= false;
	bool		m_BoostWorkerCount	= false;
	bool		m_BoostWorkerMemory = false;
	bool		m_BoostWorkers		= false;
	bool		m_UploadToZenCache	= true;

	std::filesystem::path m_OplogOutputPath;
	bool				  m_DecompressAttachments = true;

	std::vector<std::string> m_Attachments;
	std::filesystem::path	 m_AttachmentsPath;
};

}  // namespace zen