Feature/seed completed downloads

.env.example

@@ -1,4 +1,3 @@
 MAIN_VITE_API_URL=API_URL
 MAIN_VITE_AUTH_URL=AUTH_URL
-MAIN_VITE_STEAMGRIDDB_API_KEY=YOUR_API_KEY
 RENDERER_VITE_INTERCOM_APP_ID=YOUR_APP_ID

package.json

@@ -38,6 +38,7 @@
     "@hookform/resolvers": "^3.9.0",
     "@intercom/messenger-js-sdk": "^0.0.14",
     "@primer/octicons-react": "^19.9.0",
+    "@radix-ui/react-dropdown-menu": "^2.1.2",
     "@reduxjs/toolkit": "^2.2.3",
     "@vanilla-extract/css": "^1.14.2",
     "@vanilla-extract/dynamic": "^2.1.1",

postinstall.cjs

@@ -2,6 +2,7 @@ const { default: axios } = require("axios");
 const util = require("node:util");
 const fs = require("node:fs");
 const path = require("node:path");
+const { spawnSync } = require("node:child_process");
 
 const exec = util.promisify(require("node:child_process").exec);
 
@@ -46,4 +47,76 @@ const downloadLudusavi = async () => {
   });
 };
 
+const downloadAria2WindowsAndLinux = async () => {
+  if (fs.existsSync("aria2")) {
+    console.log("Aria2 already exists, skipping download...");
+    return;
+  }
+
+  const file =
+    process.platform === "win32"
+      ? "aria2-1.37.0-win-64bit-build1.zip"
+      : "aria2-1.37.0-1-x86_64.pkg.tar.zst";
+
+  const downloadUrl =
+    process.platform === "win32"
+      ? `https://github.com/aria2/aria2/releases/download/release-1.37.0/${file}`
+      : "https://archlinux.org/packages/extra/x86_64/aria2/download/";
+
+  console.log(`Downloading ${file}...`);
+
+  const response = await axios.get(downloadUrl, { responseType: "stream" });
+
+  const stream = response.data.pipe(fs.createWriteStream(file));
+
+  stream.on("finish", async () => {
+    console.log(`Downloaded ${file}, extracting...`);
+
+    if (process.platform === "win32") {
+      await exec(`npx extract-zip ${file}`);
+      console.log("Extracted. Renaming folder...");
+
+      fs.renameSync(file.replace(".zip", ""), "aria2");
+    } else {
+      await exec(`tar --zstd -xvf ${file} usr/bin/aria2c`);
+      console.log("Extracted. Copying binary file...");
+      fs.mkdirSync("aria2");
+      fs.copyFileSync("usr/bin/aria2c", "aria2/aria2c");
+      fs.rmSync("usr", { recursive: true });
+    }
+
+    console.log(`Extracted ${file}, removing compressed downloaded file...`);
+    fs.rmSync(file);
+  });
+};
+
+const copyAria2Macos = async () => {
+  console.log("Checking if aria2 is installed...");
+
+  const isAria2Installed = spawnSync("which", ["aria2c"]).status;
+
+  if (isAria2Installed != 0) {
+    console.log("Please install aria2");
+    console.log("brew install aria2");
+    return;
+  }
+
+  console.log("Copying aria2 binary...");
+  fs.mkdirSync("aria2");
+  await exec(`cp $(which aria2c) aria2/aria2c`);
+};
+
+if (process.platform === "win32") {
+  fs.copyFileSync(
+    "node_modules/ps-list/vendor/fastlist-0.3.0-x64.exe",
+    "fastlist.exe"
+  );
+}
+
+if (process.platform == "darwin") {
+  copyAria2Macos();
+} else {
+  downloadAria2WindowsAndLinux();
+}
+
 downloadLudusavi();

python_rpc/http_downloader.py

@@ -0,0 +1,47 @@
+import aria2p
+
+class HttpDownloader:
+    def __init__(self):
+        self.download = None
+        self.aria2 = aria2p.API(
+            aria2p.Client(
+                host="http://localhost",
+                port=6800,
+                secret=""
+            )
+        )
+
+    def start_download(self, url: str, save_path: str, header: str):
+        if self.download:
+            self.aria2.resume([self.download])
+        else:
+            downloads = self.aria2.add(url, options={"header": header, "dir": save_path})
+            self.download = downloads[0]
+
+    def pause_download(self):
+        if self.download:
+            self.aria2.pause([self.download])
+
+    def cancel_download(self):
+        if self.download:
+            self.aria2.remove([self.download])
+            self.download = None
+
+    def get_download_status(self):
+        if self.download == None:
+            return None
+
+        download = self.aria2.get_download(self.download.gid)
+
+        response = {
+            'folderName': str(download.dir) + "/" + download.name,
+            'fileSize': download.total_length,
+            'progress': download.completed_length / download.total_length if download.total_length else 0,
+            'downloadSpeed': download.download_speed,
+            'numPeers': 0,
+            'numSeeds': 0,
+            'status': download.status,
+            'bytesDownloaded': download.completed_length,
+        }
+
+        return response

python_rpc/main.py

@@ -0,0 +1,136 @@
+from flask import Flask, request, jsonify
+import sys, json, urllib.parse, psutil
+from torrent_downloader import TorrentDownloader
+from http_downloader import HttpDownloader
+from profile_image_processor import ProfileImageProcessor
+import libtorrent as lt
+
+app = Flask(__name__)
+
+# Retrieve command line arguments
+torrent_port = sys.argv[1]
+http_port = sys.argv[2]
+rpc_password = sys.argv[3]
+
+downloads = {}
+# This can be streamed down from Node
+downloading_game_id = -1
+
+torrent_session = lt.session({'listen_interfaces': '0.0.0.0:{port}'.format(port=torrent_port)})
+
+def validate_rpc_password():
+    """Middleware to validate RPC password."""
+    header_password = request.headers.get('x-hydra-rpc-password')
+    if header_password != rpc_password:
+        return jsonify({"error": "Unauthorized"}), 401
+
+@app.route("/status", methods=["GET"])
+def status():
+    auth_error = validate_rpc_password()
+    if auth_error:
+        return auth_error
+
+    downloader = downloads.get(downloading_game_id)
+    if downloader:
+        status = downloads.get(downloading_game_id).get_download_status()
+        return jsonify(status), 200
+    else:
+        return jsonify(None)
+
+@app.route("/seed-status", methods=["GET"])
+def seed_status():
+    auth_error = validate_rpc_password()
+    if auth_error:
+        return auth_error
+
+    status = torrent_downloader.get_seed_status()
+    return jsonify(status), 200
+
+@app.route("/healthcheck", methods=["GET"])
+def healthcheck():
+    return "", 200
+
+@app.route("/process-list", methods=["GET"])
+def process_list():
+    auth_error = validate_rpc_password()
+    if auth_error:
+        return auth_error
+
+    process_list = [proc.info for proc in psutil.process_iter(['exe', 'pid', 'username'])]
+    return jsonify(process_list), 200
+
+@app.route("/profile-image", methods=["POST"])
+def profile_image():
+    auth_error = validate_rpc_password()
+    if auth_error:
+        return auth_error
+
+    data = request.get_json()
+    image_path = data.get('image_path')
+
+    try:
+        processed_image_path, mime_type = ProfileImageProcessor.process_image(image_path)
+        return jsonify({'imagePath': processed_image_path, 'mimeType': mime_type}), 200
+    except Exception as e:
+        return jsonify({"error": str(e)}), 400
+
+@app.route("/action", methods=["POST"])
+def action():
+    global torrent_session
+    global downloading_game_id
+
+    auth_error = validate_rpc_password()
+    if auth_error:
+        return auth_error
+
+    data = request.get_json()
+    action = data.get('action')
+    game_id = data.get('game_id')
+
+    print(data)
+
+    if action == 'start':
+        url = data.get('url')
+
+        existing_downloader = downloads.get(game_id)
+
+        if existing_downloader:
+            # This will resume the download
+            existing_downloader.start_download(url, data['save_path'], data.get('header'))
+        else:
+            if url.startswith('magnet'):
+                torrent_downloader = TorrentDownloader(torrent_session)
+                downloads[game_id] = torrent_downloader
+                torrent_downloader.start_download(url, data['save_path'], "")
+            else:
+                http_downloader = HttpDownloader()
+                downloads[game_id] = http_downloader
+                http_downloader.start_download(url, data['save_path'], data.get('header'))
+
+        downloading_game_id = game_id
+
+    elif action == 'pause':
+        downloader = downloads.get(game_id)
+        if downloader:
+            downloader.pause_download()
+            downloading_game_id = -1
+    elif action == 'cancel':
+        downloader = downloads.get(game_id)
+        if downloader:
+            downloader.cancel_download()
+
+    # elif action == 'kill-torrent':
+    #     torrent_downloader.abort_session()
+    #     torrent_downloader = None
+    # elif action == 'pause-seeding':
+    #     torrent_downloader.pause_seeding(game_id)
+    # elif action == 'resume-seeding':
+    #     torrent_downloader.resume_seeding(game_id, data['url'], data['save_path'])
+    else:
+        return jsonify({"error": "Invalid action"}), 400
+
+    return "", 200
+
+if __name__ == "__main__":
+    app.run(host="0.0.0.0", port=int(http_port))
+

torrent-client/setup.py → python_rpc/setup.py

@@ -3,18 +3,18 @@
 # Dependencies are automatically detected, but it might need fine tuning.
 build_exe_options = {
     "packages": ["libtorrent"],
-    "build_exe": "hydra-download-manager",
+    "build_exe": "hydra-python-rpc",
     "include_msvcr": True
 }
 
 setup(
-    name="hydra-download-manager",
+    name="hydra-python-rpc",
     version="0.1",
     description="Hydra",
     options={"build_exe": build_exe_options},
     executables=[Executable(
-      "torrent-client/main.py",
-      target_name="hydra-download-manager",
+      "python_rpc/main.py",
+      target_name="hydra-python-rpc",
       icon="build/icon.ico"
     )]
 )