aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--CMakeLists.txt1
-rw-r--r--include/file_uploader.hpp14
-rw-r--r--include/rclone_uploader.hpp20
-rw-r--r--src/main.cpp6
-rw-r--r--src/utils/rclone_uploader.cpp33
5 files changed, 74 insertions, 0 deletions
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;
+}
send patches to the email below
yukais@pinapelz.com
include the subject [PATCH repo_name]
pinapelz.com
homepage