summaryrefslogtreecommitdiff
path: root/fig-monitor-irc
diff options
context:
space:
mode:
Diffstat (limited to 'fig-monitor-irc')
-rw-r--r--fig-monitor-irc/fig-monitor-irc.cabal50
-rw-r--r--fig-monitor-irc/main/Main.hs29
-rw-r--r--fig-monitor-irc/src/Fig/Monitor/IRC.hs83
-rw-r--r--fig-monitor-irc/src/Fig/Monitor/IRC/Utils.hs37
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
generated by cgit v1.2.3 (git 2.25.1) at 2026-05-10 01:07:04 +0000