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

---
 CMakeLists.txt                |  1 +
 include/file_uploader.hpp     | 14 ++++++++++++++
 include/rclone_uploader.hpp   | 20 ++++++++++++++++++++
 src/main.cpp                  |  6 ++++++
 src/utils/rclone_uploader.cpp | 33 +++++++++++++++++++++++++++++++++
 5 files changed, 74 insertions(+)
 create mode 100644 include/file_uploader.hpp
 create mode 100644 include/rclone_uploader.hpp
 create mode 100644 src/utils/rclone_uploader.cpp

diff --git a/CMakeLists.txt b/CMakeLists.txt
index 57da80f..107ea02 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -17,6 +17,7 @@ message(STATUS "Binary directory: ${CMAKE_BINARY_DIR}")
 add_executable(${PROJECT_NAME}
     src/main.cpp
     src/markdown_translator.cpp
+    src/utils/rclone_uploader.cpp
 )
 
 target_include_directories(${PROJECT_NAME} PRIVATE ${CMAKE_SOURCE_DIR}/include)
diff --git a/include/file_uploader.hpp b/include/file_uploader.hpp
new file mode 100644
index 0000000..6e46a8a
--- /dev/null
+++ b/include/file_uploader.hpp
@@ -0,0 +1,14 @@
+#pragma once
+#ifndef FILE_UPLOADER_H
+#define FILE_UPLOADER_H
+
+#include <string>
+
+class FileUploader {
+public:
+    virtual ~FileUploader() = default;
+    virtual bool uploadFile(const std::string& filepath, const std::string& remotePath) = 0;
+    virtual bool testConnection() = 0;
+};
+
+#endif // FILE_UPLOADER_H
diff --git a/include/rclone_uploader.hpp b/include/rclone_uploader.hpp
new file mode 100644
index 0000000..03e54fe
--- /dev/null
+++ b/include/rclone_uploader.hpp
@@ -0,0 +1,20 @@
+#ifndef RCLONE_UPLOADER_H
+#define RCLONE_UPLOADER_H
+
+#include "file_uploader.hpp"
+#include <string>
+
+class RcloneUploader : public FileUploader {
+
+public:
+    ~RcloneUploader();
+    RcloneUploader(const std::string& cdnEndpoint, const std::string& sourceName);
+    bool uploadFile(const std::string& filepath, const std::string& remotePath) override;
+    bool testConnection() override;
+
+private:
+    std::string cdnEndpoint;
+    std::string sourceName;
+};
+
+#endif // RCLONE_UPLOADER_H
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