feat: adding libtorrent again

torrent-client/fifo.py

@@ -0,0 +1,35 @@
+import platform
+
+class Fifo:
+    socket_handle = None
+
+    def __init__(self, path: str):
+        if platform.system() == "Windows":
+            import win32file
+
+            self.socket_handle = win32file.CreateFile(path, win32file.GENERIC_READ | win32file.GENERIC_WRITE,
+                                0, None, win32file.OPEN_EXISTING, win32file.FILE_ATTRIBUTE_NORMAL, None)
+        else:
+            import socket
+            self.socket_handle = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
+            self.socket_handle.connect(path)
+
+    def recv(self, bufSize: int):
+        if platform.system() == "Windows":
+            import win32file
+
+            result, data = win32file.ReadFile(self.socket_handle, bufSize)
+            return data
+        else:
+            return self.socket_handle.recv(bufSize)
+
+    def send_message(self, msg: str):
+        buffer = bytearray(1024 * 2)
+        buffer[:len(msg)] = bytes(msg, "utf-8")
+
+        if platform.system() == "Windows":
+            import win32file
+
+            win32file.WriteFile(self.socket_handle, buffer)
+        else:
+            self.socket_handle.send(buffer)

torrent-client/main.py

@@ -0,0 +1,101 @@
+import libtorrent as lt
+import sys
+from fifo import Fifo
+import json
+import threading
+import time
+
+torrent_port = sys.argv[1]
+read_sock_path = sys.argv[2]
+write_sock_path = sys.argv[3]
+
+session = lt.session({'listen_interfaces': '0.0.0.0:{port}'.format(port=torrent_port)})
+read_fifo = Fifo(read_sock_path)
+write_fifo = Fifo(write_sock_path)
+
+torrent_handles = {}
+downloading_game_id = -1
+
+def start_download(game_id: int, magnet: str, save_path: str):
+    global torrent_handles
+    global downloading_game_id
+
+    params = {'url': magnet, 'save_path': save_path}
+    torrent_handle = session.add_torrent(params)
+    torrent_handles[game_id] = torrent_handle
+    torrent_handle.set_flags(lt.torrent_flags.auto_managed)
+    torrent_handle.resume()
+
+    downloading_game_id = game_id
+
+def pause_download(game_id: int):
+    global torrent_handles
+    global downloading_game_id
+
+    torrent_handle = torrent_handles.get(game_id)
+    if torrent_handle:
+        torrent_handle.pause()
+        torrent_handle.unset_flags(lt.torrent_flags.auto_managed)
+        downloading_game_id = -1
+
+def cancel_download(game_id: int):
+    global torrent_handles
+    global downloading_game_id
+
+    torrent_handle = torrent_handles.get(game_id)
+    if torrent_handle:
+        torrent_handle.pause()
+        session.remove_torrent(torrent_handle)
+        torrent_handles[game_id] = None
+        downloading_game_id =-1
+
+def get_download_updates():
+    global torrent_handles
+    global downloading_game_id
+
+    while True:
+        if downloading_game_id == -1:
+            time.sleep(0.5)
+            continue
+
+        torrent_handle = torrent_handles.get(downloading_game_id)
+
+        status = torrent_handle.status()
+        info = torrent_handle.get_torrent_info()
+
+        write_fifo.send_message(json.dumps({
+            'folderName': info.name() if info else "",
+            'fileSize': info.total_size() if info else 0,
+            'gameId': downloading_game_id,
+            'progress': status.progress,
+            'downloadSpeed': status.download_rate,
+            'numPeers': status.num_peers,
+            'numSeeds': status.num_seeds,
+            'status': status.state,
+            'bytesDownloaded': status.progress * info.total_size() if info else status.all_time_download,
+        }))
+
+        if status.progress == 1:
+            cancel_download(downloading_game_id)
+            downloading_game_id = -1
+
+        time.sleep(0.5)
+
+def listen_to_socket():
+    while True:
+        msg = read_fifo.recv(1024 * 2)
+        payload = json.loads(msg.decode("utf-8"))
+
+        if payload['action'] == "start":
+            start_download(payload['game_id'], payload['magnet'], payload['save_path'])
+        elif payload['action'] == "pause":
+            pause_download(payload['game_id'])
+        elif payload['action'] == "cancel":
+            cancel_download(payload['game_id'])
+
+if __name__ == "__main__":
+    p1 = threading.Thread(target=get_download_updates)
+    p2 = threading.Thread(target=listen_to_socket)
+
+    p1.start()
+    p2.start()

torrent-client/setup.py

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