Initial commit

4 files changed, 199 insertions, 0 deletions

diff --git a/fig-monitor-irc/fig-monitor-irc.cabal b/fig-monitor-irc/fig-monitor-irc.cabal
new file mode 100644
index 0000000..618b63b
--- /dev/null
+++ b/fig-monitor-irc/fig-monitor-irc.cabal
@@ -0,0 +1,50 @@
+cabal-version: 3.4
+name: fig-monitor-irc
+version: 0.1.0.0
+
+common defaults
+  ghc-options: -Wall
+  default-language: GHC2021
+  default-extensions: NoImplicitPrelude PackageImports LambdaCase MultiWayIf OverloadedStrings OverloadedLists OverloadedRecordDot DuplicateRecordFields RecordWildCards NoFieldSelectors BlockArguments ViewPatterns TypeFamilies DataKinds GADTs
+
+common deps
+  build-depends:
+      base
+    , aeson
+    , base64
+    , binary
+    , bytestring
+    , containers
+    , data-default-class
+    , directory
+    , filepath
+    , irc-client
+    , megaparsec
+    , microlens
+    , mtl
+    , network
+    , safe-exceptions
+    , text
+    , time
+    , tomland
+    , transformers
+    , unordered-containers
+    , vector
+    , fig-utils
+    , fig-bus
+
+library
+  import: defaults
+  import: deps
+  hs-source-dirs: src
+  exposed-modules:
+    Fig.Monitor.IRC
+    Fig.Monitor.IRC.Utils
+
+executable fig-monitor-irc
+  import: defaults
+  import: deps
+  build-depends: fig-monitor-irc, optparse-applicative
+  hs-source-dirs:
+    main
+  main-is: Main.hs
\ No newline at end of file
diff --git a/fig-monitor-irc/main/Main.hs b/fig-monitor-irc/main/Main.hs
new file mode 100644
index 0000000..c28061b
--- /dev/null
+++ b/fig-monitor-irc/main/Main.hs
@@ -0,0 +1,29 @@
+module Main where
+
+import Fig.Prelude
+
+import Options.Applicative
+
+import Fig.Monitor.IRC
+import Fig.Monitor.IRC.Utils
+
+data Opts = Opts
+  { busHost :: Text
+  , busPort :: Text
+  , config :: FilePath
+  }
+
+parseOpts :: Parser Opts
+parseOpts = Opts
+  <$> strOption (long "bus-host" <> metavar "HOST" <> help "Address of message bus" <> value "localhost")
+  <*> strOption (long "bus-port" <> metavar "PORT" <> help "Message bus port" <> showDefault <> value "32050")
+  <*> strOption (long "config" <> metavar "PATH" <> help "Path to config file" <> showDefault <> value "fig-monitor-irc.toml")
+
+main :: IO ()
+main = do
+  opts <- execParser $ info (parseOpts <**> helper)
+    ( fullDesc
+    <> header "fig-monitor-discord - monitor IRC chat events"
+    )
+  cfg <- loadConfig opts.config
+  ircBot cfg (opts.busHost, opts.busPort)
diff --git a/fig-monitor-irc/src/Fig/Monitor/IRC.hs b/fig-monitor-irc/src/Fig/Monitor/IRC.hs
new file mode 100644
index 0000000..55d17e5
--- /dev/null
+++ b/fig-monitor-irc/src/Fig/Monitor/IRC.hs
@@ -0,0 +1,83 @@
+{-# Language QuasiQuotes #-}
+
+module Fig.Monitor.IRC where
+
+import Fig.Prelude
+
+import qualified Data.Text as Text
+import qualified Data.ByteString.Base64 as BS.Base64
+
+import Lens.Micro ((%~), (.~), (^.))
+
+import qualified Control.Concurrent as Conc
+import qualified Control.Concurrent.Chan as Chan
+
+import qualified Network.IRC.Client as IRC
+
+import Fig.Utils.SExpr
+import Fig.Bus.Client
+import Fig.Monitor.IRC.Utils
+
+data OutgoingMessage = OutgoingMessage
+  { user :: Text
+  , msg :: Text
+  }
+
+srcUser :: IRC.Source a -> Maybe a
+srcUser (IRC.Channel _ user) = Just user
+srcUser (IRC.User user) = Just user
+srcUser _ = Nothing
+
+ircBot :: Config -> (Text, Text) -> IO ()
+ircBot cfg busAddr = do
+  outgoing <- Chan.newChan @OutgoingMessage
+  mircst <- Conc.newEmptyMVar
+  void . Conc.forkIO $ Conc.readMVar mircst >>= \ircst -> forever $ do
+    o <- liftIO $ Chan.readChan outgoing
+    log $ "Sending: " <> o.msg <> " (from " <> o.user <> ")"
+    let msg = IRC.Privmsg cfg.sendchannel . Right . Text.take 400 $ mconcat
+          [ "<", o.user, "> "
+          , Text.replace "\n" " " o.msg
+          ]
+    IRC.runIRCAction (IRC.send msg) ircst
+  busClient busAddr
+    (\cmds -> do
+        cmds.subscribe [sexp|(monitor irc chat outgoing)|]
+        let handler = IRC.EventHandler
+              ( \case
+                  ev
+                    | IRC.Privmsg _ (Right msg) <- ev ^. IRC.message -> Just msg
+                    | otherwise -> Nothing 
+              )
+              ( \src msg -> case srcUser src of
+                  Just user -> do
+                    log $ "Received: " <> msg <> " (from " <> user <> ")"
+                    liftIO $ cmds.publish [sexp|(monitor irc chat incoming)|]
+                      [ SExprString . BS.Base64.encodeBase64 . encodeUtf8 $ user
+                      , SExprList []
+                      , SExprString . BS.Base64.encodeBase64 . encodeUtf8 $ msg
+                      ]
+                  Nothing -> pure ()
+              )
+        ircst <- IRC.newIRCState
+          ( IRC.tlsConnection (IRC.WithDefaultConfig (encodeUtf8 cfg.host) cfg.port)
+          -- ( IRC.plainConnection (encodeUtf8 cfg.host) cfg.port
+          )
+          ( IRC.defaultInstanceConfig cfg.nick
+            & IRC.handlers %~ (handler:)
+            & IRC.channels .~ cfg.channels 
+          )
+          ()
+        Conc.putMVar mircst ircst
+        IRC.runClientWith ircst
+    )
+    (\_cmds d -> do
+        case d of
+          SExprList [ev, SExprString euser, SExprString emsg]
+            | ev == [sexp|(monitor irc chat outgoing)|]
+            , Right user <- decodeUtf8 <$> BS.Base64.decodeBase64 (encodeUtf8 euser)
+            , Right msg <- decodeUtf8 <$> BS.Base64.decodeBase64 (encodeUtf8 emsg) -> do
+                Chan.writeChan outgoing OutgoingMessage { user = user, msg = msg }
+          _ -> log $ "Invalid outgoing message: " <> tshow d
+    )
+    (pure ())
diff --git a/fig-monitor-irc/src/Fig/Monitor/IRC/Utils.hs b/fig-monitor-irc/src/Fig/Monitor/IRC/Utils.hs
new file mode 100644
index 0000000..2cf46b1
--- /dev/null
+++ b/fig-monitor-irc/src/Fig/Monitor/IRC/Utils.hs
@@ -0,0 +1,37 @@
+{-# Language ApplicativeDo #-}
+
+module Fig.Monitor.IRC.Utils
+  ( FigMonitorIRCException(..)
+  , Config(..)
+  , loadConfig
+  ) where
+
+import Fig.Prelude
+
+import qualified Toml
+
+newtype FigMonitorIRCException = FigMonitorIRCException Text
+  deriving (Show, Eq, Ord)
+instance Exception FigMonitorIRCException
+
+data Config = Config
+  { host :: Text
+  , port :: Int
+  , nick :: Text
+  , sendchannel :: Text
+  , channels :: [Text]
+  } deriving (Show, Eq, Ord)
+
+configCodec :: Toml.TomlCodec Config
+configCodec = do
+  host <- Toml.text "host" Toml..= (\a -> a.host)
+  port <- Toml.int "port" Toml..= (\a -> a.port)
+  nick <- Toml.text "nick" Toml..= (\a -> a.nick)
+  sendchannel <- Toml.text "sendchannel" Toml..= (\a -> a.sendchannel)
+  channels <- Toml.arrayOf Toml._Text "channels" Toml..= (\a -> a.channels)
+  pure $ Config{..}
+
+loadConfig :: FilePath -> IO Config
+loadConfig path = Toml.decodeFileEither configCodec path >>= \case
+  Left err -> throwM . FigMonitorIRCException $ tshow err
+  Right config -> pure config