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
|
// Copyright Epic Games, Inc. All Rights Reserved.
#include "vfs_cmd.h"
#include "zenserviceclient.h"
#include <zencore/compactbinarybuilder.h>
#include <zencore/fmtutils.h>
#include <zencore/logging.h>
#include <zencore/string.h>
#include <zencore/uid.h>
#include <zenhttp/formatters.h>
#include <zenhttp/httpclient.h>
#include <zenutil/zenserverprocess.h>
namespace zen {
using namespace std::literals;
VfsCommand::VfsCommand()
{
m_Options.add_option("", "", "verb", "VFS management verb (mount, unmount, info)", cxxopts::value(m_Verb), "<verb>");
m_Options.add_option("", "u", "hosturl", kHostUrlHelp, cxxopts::value(m_HostName)->default_value(""), "<url>");
m_Options.add_option("", "", "vfs-path", "Specify VFS mount point path", cxxopts::value(m_MountPath), "<path>");
m_Options.parse_positional({"verb", "vfs-path"});
}
VfsCommand::~VfsCommand()
{
}
void
VfsCommand::Run(const ZenCliOptions& GlobalOptions, int argc, char** argv)
{
ZEN_UNUSED(GlobalOptions, argc, argv);
if (!ParseOptions(argc, argv))
{
return;
}
// Validate arguments
ZenServiceClient Service({.HostSpec = m_HostName, .CommandName = Name});
HttpClient& Http = Service.Http();
if (m_Verb == "mount"sv)
{
if (m_MountPath.empty())
throw OptionParseException("'--vfs-path' is required", m_Options.help());
CbObjectWriter Cbo;
Cbo << "method"
<< "mount";
Cbo.BeginObject("params");
Cbo << "path" << m_MountPath;
Cbo.EndObject();
if (HttpClient::Response Result = Http.Post("/vfs"sv, Cbo.Save()))
{
}
else
{
Result.ThrowError("VFS mount request failed"sv);
}
}
else if (m_Verb == "unmount"sv)
{
CbObjectWriter Cbo;
Cbo << "method"
<< "unmount";
if (HttpClient::Response Result = Http.Post("/vfs"sv, Cbo.Save()))
{
}
else
{
Result.ThrowError("VFS unmount request failed"sv);
}
}
else if (m_Verb == "info"sv)
{
if (HttpClient::Response Result = Http.Get(fmt::format("/vfs/info")))
{
ExtendableStringBuilder<256> Json;
Result.AsObject().ToJson(Json);
ZEN_CONSOLE("{}", Json);
}
else
{
Result.ThrowError("VFS info fetch failed"sv);
}
}
}
} // namespace zen
|