From 2d06ed6a9492eeecc49fb0d21c93b6ae5e0f7edd Mon Sep 17 00:00:00 2001
From: rra <rscmbbng@riseup.net>
Date: Sun, 16 Jan 2022 13:28:26 +0100
Subject: [PATCH] use croc as a backend

---
 dropship/croc.py     | 47 ++++++++++++++++++++++++++++++++++++++++++++
 dropship/dropship.py |  6 +++---
 2 files changed, 50 insertions(+), 3 deletions(-)
 create mode 100644 dropship/croc.py

diff --git a/dropship/croc.py b/dropship/croc.py
new file mode 100644
index 0000000..a420a2d
--- /dev/null
+++ b/dropship/croc.py
@@ -0,0 +1,47 @@
+from subprocess import PIPE
+
+from trio import TASK_STATUS_IGNORED, CancelScope, open_process, run_process
+
+from dropship.logger import log
+
+
+async def croc_send(fpath, parent, task_status=TASK_STATUS_IGNORED):
+    """Run `croc send` on a local file path."""
+    with CancelScope() as scope:
+        command = ["croc", "send", fpath]
+        process = await open_process(command, stderr=PIPE)
+        
+        #output = await process.stderr.receive_some()
+        #this is some nasty hacky shit
+        async for i in process.stderr:
+            if b"Code is:" in i:
+                output = i
+                break
+        code = output.decode().split()[-1]
+        task_status.started((code, scope,))
+        log.info(f"croc_send: now waiting for other side ({code})")
+        await process.wait()
+        log.info(f"croc_send: succesfully transfered ({code})")
+
+    if scope.cancel_called:
+        process.terminate()
+        log.info(f"croc_send: succesfully terminated process ({code})")
+
+    parent._remove_pending_transfer(code)
+
+
+async def croc_recv(code, parent, task_status=TASK_STATUS_IGNORED):
+    """Run `croc receive` on a pending transfer code."""
+    with CancelScope() as scope:
+        command = ["croc", "--yes", code]
+        process = await open_process(command, stderr=PIPE)
+        task_status.started((scope,))
+        log.info(f"croc_recv: now starting receiving process ({code})")
+        await process.wait()
+        log.info(f"croc_recv: succesfully received ({code})")
+
+    if scope.cancel_called:
+        process.terminate()
+        log.info(f"croc_recv: succesfully terminated process ({code})")
+
+    parent._remove_pending_transfer(code)
diff --git a/dropship/dropship.py b/dropship/dropship.py
index 91484cd..bf47d9a 100644
--- a/dropship/dropship.py
+++ b/dropship/dropship.py
@@ -11,7 +11,7 @@ from gi.repository import Gdk, GLib, Gtk
 from dropship.constant import UI_DIR
 from dropship.transfer import PendingTransfer
 from dropship.ui_templates import PendingTransferRow
-from dropship.wormhole import wormhole_recv, wormhole_send
+from dropship.croc import croc_recv, croc_send
 
 
 class DropShip:
@@ -139,12 +139,12 @@ class DropShip:
 
     async def send(self, fpath):
         self._send_spinner_on()
-        code, scope = await self.nursery.start(wormhole_send, fpath, self)
+        code, scope = await self.nursery.start(croc_send, fpath, self)
         self._create_pending_transfer(fpath, code, scope)
         self.clipboard.set_text(code, -1)
         self._send_spinner_off(code)
         log.info(f"send: successfully initiated transfer send ({code})")
 
     async def receive(self, code):
-        await self.nursery.start(wormhole_recv, code, self)
+        await self.nursery.start(croc_recv, code, self)
         log.info(f"send: successfully initiated receive ({code})")