From fa33ef92d0a9490a6ec99309bc539f2d045f0fe7 Mon Sep 17 00:00:00 2001
From: Joscha <brummer@migejolise.de>
Date: Sat, 17 Feb 2018 18:09:02 +0000
Subject: [PATCH] Add bot commands

---
 src/EuphApi/Bot.hs                | 78 +++++++++++++++++++++++++++++--
 test/bot_simple_custom_logging.hs | 25 +++++-----
 2 files changed, 87 insertions(+), 16 deletions(-)

diff --git a/src/EuphApi/Bot.hs b/src/EuphApi/Bot.hs
index 042daec..42c9693 100644
--- a/src/EuphApi/Bot.hs
+++ b/src/EuphApi/Bot.hs
@@ -4,6 +4,15 @@
 -- | This module lets you create bots, although it only contains the bare minimum necessary.
 -- It defines the 'Bot' monad which takes care of a few things common to most bots.
 --
+-- The module is meant to be imported qualified, under a different name than all the
+-- other EuphApi modules.
+-- For example:
+--
+-- > import qualified EuphApi.Bot        as B
+-- > import qualified EuphApi.Connection as E
+-- > import qualified EuphApi.Types      as E
+-- > import qualified EuphApi.Utils      as E
+--
 -- = The 'Bot' monad
 --
 -- This monad takes care of
@@ -62,7 +71,6 @@ module EuphApi.Bot (
     Bot
   , BotConfig(..)
   , runBot
-  -- * Bot commands
   -- * Utilities
   , fork
   , defaultReconnectPolicy
@@ -72,6 +80,15 @@ module EuphApi.Bot (
   , getConnection
   , getConnectionInfo
   , getConnectTime
+  -- * Bot commands
+  , stop
+  , send
+  , reply
+  , nick
+  , getMessage
+  , messageLog
+  , messageLogAfter
+  , who
   ) where
 
 import           Control.Concurrent
@@ -85,6 +102,7 @@ import           Data.Time
 import qualified System.Log.Logger          as L
 
 import qualified EuphApi.Connection         as E
+import qualified EuphApi.Types              as E
 
 -- logging functions
 moduleName :: String
@@ -277,8 +295,8 @@ handleNickStuff (E.SnapshotEvent _ _ _ maybeNick) = do
   con       <- getConnection
   case maybeNick of
     Nothing -> fork $ liftIO $ E.nick con myNick
-    Just nick ->
-      if nick == myNick
+    Just curNick ->
+      if curNick == myNick
         then return ()
         else fork $ liftIO $ E.nick con myNick
 handleNickStuff _ = return ()
@@ -298,3 +316,57 @@ handlePasswordStuff _ = return ()
 {-
  - Commands
  -}
+
+-- | Stop the bot.
+stop :: Bot b c ()
+stop = do
+  stopping <- asks bStopping
+  con <- asks bConnection
+  liftIO $ do
+    atomically $ writeTVar stopping False
+    E.disconnect con
+
+-- | Send a new message.
+send :: T.Text -> Bot b c E.Message
+send content = do
+  con <- asks bConnection
+  liftIO $ E.send con Nothing content
+
+-- | Reply to a message.
+reply :: E.Snowflake -> T.Text -> Bot b c E.Message
+reply parentID content = do
+  con <- asks bConnection
+  liftIO $ E.send con (Just parentID) content
+
+-- | Change the bot's nick.
+nick :: T.Text -> Bot b c (T.Text, T.Text)
+nick newNick = do
+  myNick <- asks bNick
+  con <- asks bConnection
+  liftIO $ do
+    atomically $ writeTVar myNick newNick
+    E.nick con newNick
+
+-- | Request an untruncated message.
+getMessage :: E.Snowflake -> Bot b c E.Message
+getMessage messageID = do
+  con <- asks bConnection
+  liftIO $ E.getMessage con messageID
+
+-- | Request the n most recent messages (similar to the 'E.SnapshotEvent').
+messageLog :: Integer -> Bot b c ([E.Message], Maybe E.Snowflake)
+messageLog n = do
+  con <- asks bConnection
+  liftIO $ E.messageLog con n Nothing
+
+-- | Request the n messages preceding a certain message.
+messageLogAfter :: E.Snowflake -> Integer -> Bot b c ([E.Message], Maybe E.Snowflake)
+messageLogAfter messageID n = do
+  con <- asks bConnection
+  liftIO $ E.messageLog con n (Just messageID)
+
+-- | Request a listing of all currently connected sessions.
+who :: Bot b c [E.SessionView]
+who = do
+  con <- asks bConnection
+  liftIO $ E.who con
diff --git a/test/bot_simple_custom_logging.hs b/test/bot_simple_custom_logging.hs
index ae83173..4ef882b 100644
--- a/test/bot_simple_custom_logging.hs
+++ b/test/bot_simple_custom_logging.hs
@@ -5,19 +5,18 @@ import qualified System.Log.Handler        as LH
 import qualified System.Log.Handler.Simple as LH
 import qualified System.Log.Logger         as L
 
-import qualified EuphApi.Bot               as E
-import qualified EuphApi.Connection        as E
+import qualified EuphApi.Bot               as B
 
-myBotConfig :: E.BotConfig () ()
-myBotConfig = E.BotConfig
-  { E.botAddress           = "euphoria.io"
-  , E.botRoom              = "test"
-  , E.botPassword          = Nothing
-  , E.botNick              = "EuphApiTestBot"
-  , E.botHandler           = const $ return ()
-  , E.botInfo              = ()
-  , E.botNewConnectionInfo = return ()
-  , E.botReconnectPolicy   = E.defaultReconnectPolicy
+myBotConfig :: B.BotConfig () ()
+myBotConfig = B.BotConfig
+  { B.botAddress           = "euphoria.io"
+  , B.botRoom              = "test"
+  , B.botPassword          = Nothing
+  , B.botNick              = "EuphApiTestBot"
+  , B.botHandler           = const $ return ()
+  , B.botInfo              = ()
+  , B.botNewConnectionInfo = return ()
+  , B.botReconnectPolicy   = B.defaultReconnectPolicy
   }
 
 main = do
@@ -26,4 +25,4 @@ main = do
       myFormattedHandler = LH.setFormatter myHandler myFormatter
   L.updateGlobalLogger L.rootLoggerName (L.setHandlers [myFormattedHandler])
   L.updateGlobalLogger L.rootLoggerName (L.setLevel L.INFO)
-  E.runBot myBotConfig
+  B.runBot myBotConfig