From 2e724fcf6651ef157cbc4b005e36bebf66032fc1 Mon Sep 17 00:00:00 2001
From: Luke Murphy <lukewm@riseup.net>
Date: Thu, 23 Jul 2020 09:29:52 +0200
Subject: [PATCH] Wire up code gathering

---
 dropship.py | 22 +++++++++++++++++++---
 1 file changed, 19 insertions(+), 3 deletions(-)

diff --git a/dropship.py b/dropship.py
index d75fb27..e1ad2f3 100644
--- a/dropship.py
+++ b/dropship.py
@@ -101,14 +101,30 @@ class DropShip:
         """Schedule a task on the event loop."""
         loop.call_soon_threadsafe(asyncio.ensure_future, function)
 
+    async def read_lines(self, stream, pattern):
+        """Read stdout from a command and match lines."""
+        # TODO(decentral1se): if pattern doesnt match, trapped forever
+        while True:
+            line = await stream.readline()
+            decoded = line.decode("utf-8").strip()
+            if pattern in decoded:
+                return decoded
+
     async def wormhole_send(self, widget, fpath):
         """Run `wormhole send` on a local file path."""
         process = await asyncio.create_subprocess_exec(
-            "wormhole", "send", fpath, stdout=subprocess.PIPE
+            "wormhole",
+            "send",
+            fpath,
+            stdout=asyncio.subprocess.PIPE,
+            stderr=asyncio.subprocess.PIPE,
         )
+        line = await self.read_lines(process.stderr, "wormhole receive")
+        code = line.split()[-1]
+        log.info(f"Wormhole send produced f{code}")
 
-        async for line in process.stdout:
-            log.info(line)
+        # TODO(decentral1se): waits forever...
+        await process.wait()
 
 
 async def main():