Download files from direct URLs via a command line parameter (#1873)

* Download files via a command line parameter
* Use a GET parameter to predetermine Content-Disposition header in S3-compatible temporary URLs

1 files changed, 48 insertions, 2 deletions

diff --git a/src/commandline.cpp b/src/commandline.cpp
index 9082d710..1d44e043 100644
--- a/src/commandline.cpp
+++ b/src/commandline.cpp
@@ -2,6 +2,7 @@
 #include "env.h"
 #include "instancemanager.h"
 #include "loglist.h"
+#include "messagedialog.h"
 #include "multiprocess.h"
 #include "organizercore.h"
 #include "shared/appconfig.h"
@@ -49,8 +50,8 @@ CommandLine::CommandLine() : m_command(nullptr)
 {
   createOptions();
 
-  add<RunCommand, ReloadPluginCommand, RefreshCommand, CrashDumpCommand,
-      LaunchCommand>();
+  add<RunCommand, ReloadPluginCommand, DownloadFileCommand, RefreshCommand,
+      CrashDumpCommand, LaunchCommand>();
 }
 
 std::optional<int> CommandLine::process(const std::wstring& line)
@@ -833,6 +834,51 @@ std::optional<int> ReloadPluginCommand::runPostOrganizer(OrganizerCore& core)
   return {};
 }
 
+Command::Meta DownloadFileCommand::meta() const
+{
+  return {"download", "downloads a file", "URL", ""};
+}
+
+po::options_description DownloadFileCommand::getInternalOptions() const
+{
+  po::options_description d;
+
+  d.add_options()("URL", po::value<std::string>()->required(), "file URL");
+
+  return d;
+}
+
+po::positional_options_description DownloadFileCommand::getPositional() const
+{
+  po::positional_options_description d;
+
+  d.add("URL", 1);
+
+  return d;
+}
+
+bool DownloadFileCommand::canForwardToPrimary() const
+{
+  return true;
+}
+
+std::optional<int> DownloadFileCommand::runPostOrganizer(OrganizerCore& core)
+{
+  const QString url = QString::fromStdString(vm()["URL"].as<std::string>());
+
+  if (!url.startsWith("https://")) {
+    reportError(QObject::tr("Download URL must start with https://"));
+    return 1;
+  }
+
+  log::debug("starting direct download from command line: {}", url.toStdString());
+  MessageDialog::showMessage(QObject::tr("Download started"), qApp->activeWindow(),
+                             false);
+  core.downloadManager()->startDownloadURLs(QStringList() << url);
+
+  return {};
+}
+
 Command::Meta RefreshCommand::meta() const
 {
   return {"refresh", "refreshes MO (same as F5)", "", ""};