feat: adding libtorrent again

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()