Reaktor: add send delay support

2 files changed, 10 insertions, 5 deletions

diff --git a/src/Reaktor.hs b/src/Reaktor.hs
index 042fcfe..0910e0b 100644
--- a/src/Reaktor.hs
+++ b/src/Reaktor.hs
@@ -86,7 +86,7 @@ run Config{..} getPlugins =
           receiver actions putInMsg sockRecv,
           logger cLogHandle takeLog,
           pinger aSend,
-          sender takeOutMsg sockSend,
+          sender cSendDelay takeOutMsg sockSend,
           splitter plugins takeInMsg
         ]
 
@@ -147,9 +147,12 @@ receiver Actions{..} putInMsg sockRecv =
           aLog $ SGR [38,5,235] "< " <> SGR [38,5,244] (Plain bs)
 
 
-sender :: IO Message -> (ByteString -> IO ()) -> IO ()
-sender takeOutMsg sockSend =
-    forever $ takeOutMsg >>= sockSend . T.encodeUtf8 . formatMessage
+sender :: Maybe Int -> IO Message -> (ByteString -> IO ()) -> IO ()
+sender cSendDelay takeOutMsg sockSend =
+    forever send
+  where
+    send = maybe send0 ((send0 >>) . threadDelay) cSendDelay
+    send0 = takeOutMsg >>= sockSend . T.encodeUtf8 . formatMessage
 
 splitter :: [Message -> IO ()] -> IO Message -> IO ()
 splitter plugins takeInMsg =
diff --git a/src/Reaktor/Internal.hs b/src/Reaktor/Internal.hs
index 66e00af..bd3d241 100644
--- a/src/Reaktor/Internal.hs
+++ b/src/Reaktor/Internal.hs
@@ -32,11 +32,12 @@ data Config = Config
     , cNick :: Maybe Text
     , cLogHandle :: Handle
     , cLogTime :: Bool
+    , cSendDelay :: Maybe Int
     }
   deriving Show
 
 instance Default Config where
-  def = Config False "irc.r" "6667" Nothing stderr True
+  def = Config False "irc.r" "6667" Nothing stderr True Nothing
 
 instance FromJSON Config where
   parseJSON = \case
@@ -47,6 +48,7 @@ instance FromJSON Config where
         cNick <- v .:? "nick"
         cLogHandle <- pure (cLogHandle def)
         cLogTime <- v .:? "logTime" .!= cLogTime def
+        cSendDelay <- fmap (round . (*(1e6 :: Float))) <$> v .:? "sendDelaySec"
         pure Config{..}
       invalid -> typeMismatch "Config" invalid
     where