From 4af96da5670566c111fa5c7dac9572eaac021a4b Mon Sep 17 00:00:00 2001
From: Julien Dessaux
Date: Tue, 18 Jul 2023 00:02:33 +0200
Subject: [haskell] Refactored everything with a ReaderT pattern

---
 haskell/src/SpaceTraders/Database/Tokens.hs | 22 ++++++++++++++--------
 1 file changed, 14 insertions(+), 8 deletions(-)

(limited to 'haskell/src/SpaceTraders/Database/Tokens.hs')

diff --git a/haskell/src/SpaceTraders/Database/Tokens.hs b/haskell/src/SpaceTraders/Database/Tokens.hs
index b907609..e99e49e 100644
--- a/haskell/src/SpaceTraders/Database/Tokens.hs
+++ b/haskell/src/SpaceTraders/Database/Tokens.hs
@@ -2,17 +2,23 @@
 {-# LANGUAGE QuasiQuotes #-}
 
 module SpaceTraders.Database.Tokens
-  ( getToken
-  , setToken
+  ( addToken
+  , getToken
   ) where
 
+import Control.Monad.Reader
 import qualified Database.SQLite.Simple as S
 import qualified Data.Text as T
 
-getToken :: S.Connection -> IO (T.Text)
-getToken conn = do
-  [[token]] <- S.query_ conn "SELECT data FROM tokens;"
-  return token
+import SpaceTraders
+
+addToken :: (HasDatabaseConn env, MonadIO m, MonadReader env m) => T.Text -> m ()
+addToken value = do
+  env <- ask
+  liftIO $ S.execute (getConn env) "INSERT INTO tokens(data) VALUES (?);" (S.Only value)
 
-setToken :: S.Connection -> T.Text -> IO ()
-setToken conn value = S.execute conn "INSERT INTO tokens(data) VALUES (?);" (S.Only value)
+getToken :: (HasDatabaseConn env, MonadFail m, MonadIO m, MonadReader env m) => m T.Text
+getToken = do
+  env <- ask
+  [[token]] <- liftIO $ S.query_ (getConn env) "SELECT data FROM tokens;"
+  return token
-- 
cgit v1.2.3