Parser.hs

module Kiroku.Cli.Parser
  ( kirokuCommandParser,
    kirokuParserInfo,
    kirokuSubparser,
  )
where

import Control.Applicative (optional)
import Data.Text qualified as T
import Kiroku.Cli.Command (KirokuCommand (..))
import Kiroku.Cli.Command qualified as Command
import Options.Applicative
  ( CommandFields,
    Mod,
    Parser,
    ParserInfo,
    command,
    eitherReader,
    fullDesc,
    header,
    help,
    helper,
    info,
    long,
    metavar,
    option,
    progDesc,
    strOption,
    subparser,
    value,
    (<**>),
  )

kirokuCommandParser :: Parser KirokuCommand
kirokuCommandParser =
  maybe KirokuNoCommand id
    <$> optional
      ( subparser
          ( command
              "subscriptions"
              ( info
                  (KirokuSubscriptions <$> subscriptionCommandParser)
                  (fullDesc <> progDesc "Inspect live subscriptions in this process's KirokuStore registry.")
              )
          )
      )

subscriptionCommandParser :: Parser Command.SubscriptionCommand
subscriptionCommandParser =
  subparser
    ( command
        "status"
        ( info
            (Command.SubscriptionStatus <$> statusOptionsParser <**> helper)
            (fullDesc <> progDesc "List live subscription phases and global cursor positions.")
        )
    )

statusOptionsParser :: Parser Command.StatusOptions
statusOptionsParser =
  Command.StatusOptions
    <$> option
      (eitherReader parseOutputFormat)
      ( long "format"
          <> metavar "table|json"
          <> value Command.OutputTable
          <> help "Render as a human table or script-friendly JSON."
      )
    <*> optional
      ( Command.RemoteEndpoint . T.pack
          <$> strOption
            ( long "remote-url"
                <> metavar "URL"
                <> help
                  "Query a running worker's kiroku-metrics /subscriptions endpoint \
                  \(e.g. http://worker:9091) instead of this process's local registry."
            )
      )

parseOutputFormat :: String -> Either String Command.OutputFormat
parseOutputFormat "table" = Right Command.OutputTable
parseOutputFormat "json" = Right Command.OutputJson
parseOutputFormat other = Left ("unsupported output format " <> show other <> "; expected table or json")

kirokuParserInfo :: ParserInfo KirokuCommand
kirokuParserInfo =
  info
    (kirokuCommandParser <**> helper)
    ( fullDesc
        <> progDesc "Run Kiroku operator commands."
        <> header "kiroku - operator commands for Kiroku event stores"
    )

kirokuSubparser :: (KirokuCommand -> command) -> Mod CommandFields command
kirokuSubparser wrap =
  command
    "kiroku"
    ( info
        (wrap <$> kirokuCommandParser <**> helper)
        ( fullDesc
            <> progDesc "Run Kiroku operator commands."
        )
    )