From 67f69605b74ad7ade2f09a36e8793b9440de8f8e Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Lars=20Br=C3=BCnjes?= <brunjlar@gmail.com>
Date: Wed, 9 Jun 2021 15:51:57 +0200
Subject: [PATCH] adding liquidity

---
 code/week10/app/uniswap-client.hs | 23 ++++++++++++++++++++---
 1 file changed, 20 insertions(+), 3 deletions(-)

diff --git a/code/week10/app/uniswap-client.hs b/code/week10/app/uniswap-client.hs
index 0bdfb3e..3c0b39e 100644
--- a/code/week10/app/uniswap-client.hs
+++ b/code/week10/app/uniswap-client.hs
@@ -11,14 +11,13 @@ import           Control.Exception
 import           Control.Monad                           (forM_, when)
 import           Control.Monad.IO.Class                  (MonadIO (..))
 import           Data.Aeson                              (Result (..), ToJSON, decode, fromJSON)
-import qualified Data.ByteString.Char8                   as B8
 import qualified Data.ByteString.Lazy                    as LB
 import           Data.Monoid                             (Last (..))
 import           Data.Proxy                              (Proxy (..))
 import           Data.String                             (IsString (..))
 import           Data.Text                               (Text, pack)
 import           Data.UUID                               hiding (fromString)
-import           Ledger.Value                            (AssetClass (..), CurrencySymbol, Value, flattenValue, TokenName (unTokenName))
+import           Ledger.Value                            (AssetClass (..), CurrencySymbol, Value, flattenValue, TokenName)
 import           Network.HTTP.Req
 import qualified Plutus.Contracts.Uniswap                as US
 import           Plutus.PAB.Events.ContractInstanceState (PartiallyDecodedResponse (..))
@@ -50,6 +49,7 @@ main = do
             Funds                    -> getFunds cid
             Pools                    -> getPools cid
             Create amtA tnA amtB tnB -> createPool cid $ toCreateParams cs amtA tnA amtB tnB
+            Add amtA tnA amtB tnB    -> addLiquidity cid $ toAddParams cs amtA tnA amtB tnB
             Swap amtA tnA tnB        -> swap cid $ toSwapParams cs amtA tnA tnB
         go cid cs
 
@@ -57,12 +57,13 @@ data Command =
       Funds
     | Pools
     | Create Integer Char Integer Char
+    | Add Integer Char Integer Char
     | Swap Integer Char Char
     deriving (Show, Read, Eq, Ord)
 
 readCommandIO :: IO Command
 readCommandIO = do
-    putStrLn "Enter a command: Funds, Pools, Create amtA tnA amtB tnB, Swap amtA tnA tnB"
+    putStrLn "Enter a command: Funds, Pools, Create amtA tnA amtB tnB, Add amtA tnA amtB tnB, Swap amtA tnA tnB"
     s <- getLine
     maybe readCommandIO return $ readMaybe s
 
@@ -72,6 +73,9 @@ toCoin cs tn = US.Coin $ AssetClass (cs, fromString [tn])
 toCreateParams :: CurrencySymbol -> Integer -> Char -> Integer -> Char -> US.CreateParams
 toCreateParams cs amtA tnA amtB tnB = US.CreateParams (toCoin cs tnA) (toCoin cs tnB) (US.Amount amtA) (US.Amount amtB)
 
+toAddParams :: CurrencySymbol -> Integer -> Char -> Integer -> Char -> US.AddParams
+toAddParams cs amtA tnA amtB tnB = US.AddParams (toCoin cs tnA) (toCoin cs tnB) (US.Amount amtA) (US.Amount amtB)
+
 toSwapParams :: CurrencySymbol -> Integer -> Char -> Char -> US.SwapParams
 toSwapParams cs amtA tnA tnB = US.SwapParams (toCoin cs tnA) (toCoin cs tnB) (US.Amount amtA) (US.Amount 0)
 
@@ -131,6 +135,19 @@ createPool cid cp = do
             Left err'        -> putStrLn $ "error: " ++ show err'
             _                -> go
 
+addLiquidity :: UUID -> US.AddParams -> IO ()
+addLiquidity cid ap = do
+    callEndpoint cid "add" ap
+    threadDelay 2_000_000
+    go
+  where
+    go = do
+        e <- getStatus cid
+        case e of
+            Right US.Added -> putStrLn "added"
+            Left err'      -> putStrLn $ "error: " ++ show err'
+            _              -> go
+
 swap :: UUID -> US.SwapParams -> IO ()
 swap cid sp = do
     callEndpoint cid "swap" sp