mirror of
https://github.com/simplex-chat/simplexmq.git
synced 2026-05-11 03:56:59 +00:00
Alexander Bondarenko
5e0123313c
* log control port commands * auth * add auth to xftp, config and commands * log missing auth * put smp save under auth * corrections --------- Co-authored-by: Evgeny Poberezkin <evgeny@poberezkin.com>
62 lines
1.6 KiB
Haskell
62 lines
1.6 KiB
Haskell
{-# LANGUAGE LambdaCase #-}
|
|
{-# LANGUAGE OverloadedStrings #-}
|
|
|
|
module Simplex.Messaging.Server.Control where
|
|
|
|
import qualified Data.Attoparsec.ByteString.Char8 as A
|
|
import Data.ByteString (ByteString)
|
|
import Simplex.Messaging.Encoding.String
|
|
import Simplex.Messaging.Protocol (BasicAuth)
|
|
|
|
data CPClientRole = CPRNone | CPRUser | CPRAdmin
|
|
|
|
data ControlProtocol
|
|
= CPAuth BasicAuth
|
|
| CPSuspend
|
|
| CPResume
|
|
| CPClients
|
|
| CPStats
|
|
| CPStatsRTS
|
|
| CPThreads
|
|
| CPSockets
|
|
| CPSocketThreads
|
|
| CPDelete ByteString
|
|
| CPSave
|
|
| CPHelp
|
|
| CPQuit
|
|
| CPSkip
|
|
|
|
instance StrEncoding ControlProtocol where
|
|
strEncode = \case
|
|
CPAuth bs -> "auth " <> strEncode bs
|
|
CPSuspend -> "suspend"
|
|
CPResume -> "resume"
|
|
CPClients -> "clients"
|
|
CPStats -> "stats"
|
|
CPStatsRTS -> "stats-rts"
|
|
CPThreads -> "threads"
|
|
CPSockets -> "sockets"
|
|
CPSocketThreads -> "socket-threads"
|
|
CPDelete bs -> "delete " <> strEncode bs
|
|
CPSave -> "save"
|
|
CPHelp -> "help"
|
|
CPQuit -> "quit"
|
|
CPSkip -> ""
|
|
strP =
|
|
A.takeTill (== ' ') >>= \case
|
|
"auth" -> CPAuth <$> (A.space *> strP)
|
|
"suspend" -> pure CPSuspend
|
|
"resume" -> pure CPResume
|
|
"clients" -> pure CPClients
|
|
"stats" -> pure CPStats
|
|
"stats-rts" -> pure CPStatsRTS
|
|
"threads" -> pure CPThreads
|
|
"sockets" -> pure CPSockets
|
|
"socket-threads" -> pure CPSocketThreads
|
|
"delete" -> CPDelete <$> (A.space *> strP)
|
|
"save" -> pure CPSave
|
|
"help" -> pure CPHelp
|
|
"quit" -> pure CPQuit
|
|
"" -> pure CPSkip
|
|
_ -> fail "bad ControlProtocol command"
|