From 71f1c1bb296a506374944b9997c41b9c1fce22d9 Mon Sep 17 00:00:00 2001
From: Pinapelz <yukais@pinapelz.com>
Date: Thu, 30 Oct 2025 23:27:18 -0700
Subject: add rclone file uploader

---
 src/main.cpp                  |  6 ++++++
 src/utils/rclone_uploader.cpp | 33 +++++++++++++++++++++++++++++++++
 2 files changed, 39 insertions(+)
 create mode 100644 src/utils/rclone_uploader.cpp

(limited to 'src')

diff --git a/src/main.cpp b/src/main.cpp
index e926df7..576eb27 100644
--- a/src/main.cpp
+++ b/src/main.cpp
@@ -4,6 +4,8 @@
 #include <sstream>
 #include <unordered_map>
 #include "markdown_translator.hpp"
+#include "file_uploader.hpp"
+#include "rclone_uploader.hpp"
 
 bool parseArguments(int argc, char* argv[], std::unordered_map<std::string, std::string>& params, std::string& inputFile) {
     if (argc < 2) {
@@ -44,6 +46,10 @@ int main(int argc, char* argv[]) {
     std::unordered_map<std::string, std::string> params;
     std::string inputFile;
 
+    // FileUploader* fileupload;
+    // fileupload = new RcloneUploader("", "r2");
+    // std::cout << fileupload->testConnection() << std::endl;
+
     if (!parseArguments(argc, argv, params, inputFile)) {
         return 1;
     }
diff --git a/src/utils/rclone_uploader.cpp b/src/utils/rclone_uploader.cpp
new file mode 100644
index 0000000..5a02697
--- /dev/null
+++ b/src/utils/rclone_uploader.cpp
@@ -0,0 +1,33 @@
+#include <stdexcept>
+#include "rclone_uploader.hpp"
+
+RcloneUploader::~RcloneUploader() = default;
+
+RcloneUploader::RcloneUploader(const std::string& cdnEndpoint,
+                               const std::string& sourceName)
+    : cdnEndpoint(cdnEndpoint), sourceName(sourceName)
+{
+    std::string rcloneInstalledCommand;
+    #ifdef _WIN32
+        rcloneInstalledCommand = "where rclone >nul 2>&1";
+    #else
+        rcloneInstalledCommand = "which rclone >/dev/null 2>&1";
+    #endif
+
+    int result = std::system(rcloneInstalledCommand.c_str());
+    if (result != 0) {
+        throw std::runtime_error("rclone is not installed or not in PATH");
+    }
+}
+
+bool RcloneUploader::uploadFile(const std::string& filepath, const std::string& remotePath=""){
+    std::string command = "rclone copy \"" + filepath + "\" " + sourceName + ":" + remotePath + " 2>/dev/null 1>/dev/null";
+    int result = std::system(command.c_str());
+    return result == 0;
+}
+
+bool RcloneUploader::testConnection(){
+    std::string command = "rclone ls " + sourceName + ": 2>/dev/null 1>/dev/null";
+    int result = std::system(command.c_str());
+    return result == 0;
+}
-- 
cgit v1.2.3