From d6579dcdbd54beaf155e5183023acc86734676ac Mon Sep 17 00:00:00 2001
From: Ben Sima <ben@bsima.me>
Date: Mon, 19 Oct 2020 13:41:34 -0400
Subject: [PATCH] add subscribe function

---
 Urbit/Airlock.hs    | 28 +++++++++++-----------------
 shell.nix           |  2 +-
 urbit-airlock.cabal |  1 +
 3 files changed, 13 insertions(+), 18 deletions(-)

diff --git a/Urbit/Airlock.hs b/Urbit/Airlock.hs
index c5690a3..81bef40 100644
--- a/Urbit/Airlock.hs
+++ b/Urbit/Airlock.hs
@@ -8,6 +8,7 @@ module Urbit.Airlock
     connect,
     poke,
     ack,
+    subscribe,
   )
 where
 
@@ -15,13 +16,16 @@ import Control.Lens ()
 import Data.Aeson ((.=))
 import qualified Data.Aeson as Aeson
 import Data.ByteString (ByteString)
+import qualified Data.ByteString.Char8 as Char8
 import qualified Data.ByteString.Lazy as L
 import Data.Text (Text)
 import qualified Data.Text as Text
-import qualified Data.Text.Encoding as Encoding
+import Network.Http.Client as Client
 import Network.Wreq (FormParam ((:=)))
 import qualified Network.Wreq as Wreq
 import qualified Network.Wreq.Session as Session
+import System.IO.Streams (InputStream, OutputStream)
+import qualified System.IO.Streams as Streams
 
 -- import qualified Network.Wai.EventSource as Event
 
@@ -52,8 +56,6 @@ type Path = Text
 
 type Mark = Text
 
-type Subscription = Text
-
 -- | The `@p` for the ship (no leading ~).
 type ShipName = Text
 
@@ -105,18 +107,10 @@ ack sess ship eventId =
           "event-id" .= eventId
         ]
 
--- TODO
--- ssePipe :: Ship -> IO _
--- ssePipe ship = undefined
-
 -- |
-subscribe :: Ship -> App -> Path -> IO Subscription
-subscribe = undefined
-
--- |
-unsubscribe :: Ship -> Subscription -> IO ()
-unsubscribe = undefined
-
--- |
-delete :: Ship -> IO ()
-delete = undefined
+subscribe :: Ship -> Path -> OutputStream ByteString -> IO ()
+subscribe ship path outfn = Client.get addr handle
+  where
+    handle :: Response -> InputStream ByteString -> IO ()
+    handle _ i = Streams.connect i outfn
+    addr = Char8.pack $ (url ship) ++ "/" ++ Text.unpack path
diff --git a/shell.nix b/shell.nix
index f226fa4..03b4a6b 100644
--- a/shell.nix
+++ b/shell.nix
@@ -5,7 +5,7 @@ nixpkgs.mkShell {
   buildInputs = [
     nixpkgs.ormolu.bin
     (nixpkgs.pkgs.haskell.packages.${compiler}.ghcWithPackages (hp: with hp; [
-      aeson base bytestring lens text uuid wai wai-extra wreq
+      aeson base bytestring http-streams lens text uuid wai wai-extra wreq
     ]))
   ];
 }
diff --git a/urbit-airlock.cabal b/urbit-airlock.cabal
index a759317..6bbadfd 100644
--- a/urbit-airlock.cabal
+++ b/urbit-airlock.cabal
@@ -21,6 +21,7 @@ library
     base >= 4.7 && < 5,
     aeson,
     bytestring,
+    http-streams,
     lens,
     text,
     uuid,