RGBCube/serenity
1
Fork 0
mirror of https://github.com/RGBCube/serenity synced 2025-07-27 12:07:45 +00:00
Code Issues Activity Actions

pro: Add support for sending POST requests and custom headers

Browse source
This commit is contained in:
AnotherTest 2020-12-26 17:13:23 +03:30 • committed by Andreas Kling
parent 4e203f7e2d
commit 36d642ee75
1 changed files with 24 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

25
Userland/pro.cpp
View file

@ -144,15 +144,38 @@ int main(int argc, char** argv)
{ {
const char* url_str = nullptr; const char* url_str = nullptr;
bool save_at_provided_name = false; bool save_at_provided_name = false;
const char* data = nullptr;
String method = "GET";
HashMap<String, String, CaseInsensitiveStringTraits> request_headers;
Core::ArgsParser args_parser; Core::ArgsParser args_parser;
args_parser.set_general_help( args_parser.set_general_help(
"Download a file from an arbitrary URL. This command uses ProtocolServer, " "Download a file from an arbitrary URL. This command uses ProtocolServer, "
"and thus supports at least http, https, and gemini."); "and thus supports at least http, https, and gemini.");
args_parser.add_option(save_at_provided_name, "Write to a file named as the remote file", nullptr, 'O'); args_parser.add_option(save_at_provided_name, "Write to a file named as the remote file", nullptr, 'O');
args_parser.add_option(data, "(HTTP only) Send the provided data via an HTTP POST request", "data", 'd', "data");
args_parser.add_option(Core::ArgsParser::Option {
.requires_argument = true,
.help_string = "Add a header entry to the request",
.long_name = "header",
.short_name = 'H',
.value_name = "header-value",
.accept_value = [&](auto* s) {
StringView header { s };
auto split = header.find_first_of(':');
if (!split.has_value())
return false;
request_headers.set(header.substring_view(0, split.value()), header.substring_view(split.value() + 1));
return true;
} });
args_parser.add_positional_argument(url_str, "URL to download from", "url"); args_parser.add_positional_argument(url_str, "URL to download from", "url");
args_parser.parse(argc, argv); args_parser.parse(argc, argv);
if (data) {
method = "POST";
// FIXME: Content-Type?
}
URL url(url_str); URL url(url_str);
if (!url.is_valid()) { if (!url.is_valid()) {
fprintf(stderr, "'%s' is not a valid URL\n", url_str); fprintf(stderr, "'%s' is not a valid URL\n", url_str);
@ -162,7 +185,7 @@ int main(int argc, char** argv)
Core::EventLoop loop; Core::EventLoop loop;
auto protocol_client = Protocol::Client::construct(); auto protocol_client = Protocol::Client::construct();
auto download = protocol_client->start_download("GET", url.to_string()); auto download = protocol_client->start_download(method, url.to_string(), request_headers, data ? StringView { data }.bytes() : ReadonlyBytes {});
if (!download) { if (!download) {
fprintf(stderr, "Failed to start download for '%s'\n", url_str); fprintf(stderr, "Failed to start download for '%s'\n", url_str);
return 1; return 1;