mirror of
https://github.com/simplex-chat/simplex-chat.git
synced 2025-06-28 20:29:53 +00:00
54 lines
2 KiB
Haskell
54 lines
2 KiB
Haskell
{-# LANGUAGE DuplicateRecordFields #-}
|
|
{-# LANGUAGE FlexibleContexts #-}
|
|
{-# LANGUAGE GADTs #-}
|
|
{-# LANGUAGE NamedFieldPuns #-}
|
|
{-# LANGUAGE OverloadedStrings #-}
|
|
|
|
module Main where
|
|
|
|
import Control.Concurrent.Async
|
|
import Control.Concurrent.STM
|
|
import Control.Monad.Reader
|
|
import qualified Data.Text as T
|
|
import Simplex.Chat.Bot
|
|
import Simplex.Chat.Controller
|
|
import Simplex.Chat.Core
|
|
import Simplex.Chat.Messages
|
|
import Simplex.Chat.Options
|
|
import Simplex.Chat.Terminal (terminalChatConfig)
|
|
import Simplex.Chat.Types
|
|
import System.Directory (getAppUserDataDirectory)
|
|
import Text.Read
|
|
|
|
main :: IO ()
|
|
main = do
|
|
opts <- welcomeGetOpts
|
|
simplexChatCore terminalChatConfig opts Nothing mySquaringBot
|
|
|
|
welcomeGetOpts :: IO ChatOpts
|
|
welcomeGetOpts = do
|
|
appDir <- getAppUserDataDirectory "simplex"
|
|
opts@ChatOpts {dbFilePrefix} <- getChatOpts appDir "simplex_bot"
|
|
putStrLn $ "SimpleX Chat Bot v" ++ versionNumber
|
|
putStrLn $ "db: " <> dbFilePrefix <> "_chat.db, " <> dbFilePrefix <> "_agent.db"
|
|
pure opts
|
|
|
|
mySquaringBot :: User -> ChatController -> IO ()
|
|
mySquaringBot _user cc = do
|
|
initializeBotAddress cc
|
|
race_ (forever $ void getLine) . forever $ do
|
|
(_, resp) <- atomically . readTBQueue $ outputQ cc
|
|
case resp of
|
|
CRContactConnected _ contact _ -> do
|
|
contactConnected contact
|
|
void . sendMsg contact $ "Hello! I am a simple squaring bot - if you send me a number, I will calculate its square"
|
|
CRNewChatItem _ (AChatItem _ SMDRcv (DirectChat contact) ChatItem {content}) -> do
|
|
let msg = T.unpack $ ciContentToText content
|
|
number_ = readMaybe msg :: Maybe Integer
|
|
void . sendMsg contact $ case number_ of
|
|
Nothing -> "\"" <> msg <> "\" is not a number"
|
|
Just n -> msg <> " * " <> msg <> " = " <> show (n * n)
|
|
_ -> pure ()
|
|
where
|
|
sendMsg Contact {contactId} msg = sendChatCmd cc $ "/_send @" <> show contactId <> " text " <> msg
|
|
contactConnected Contact {localDisplayName} = putStrLn $ T.unpack localDisplayName <> " connected"
|