{-# LANGUAGE LambdaCase #-}
{-# LANGUAGE OverloadedStrings #-}
{-# LANGUAGE QuasiQuotes #-}
{-# LANGUAGE NoImplicitPrelude #-}

-- : out agent
-- : dep temporary
module Omni.Agent where

import Alpha
import qualified Data.Text as Text
import qualified Omni.Agent.Core as Core
import qualified Omni.Agent.Worker as Worker
import qualified Omni.Cli as Cli
import qualified Omni.Test as Test
import qualified System.Console.Docopt as Docopt

main :: IO ()
main = Cli.main plan

plan :: Cli.Plan ()
plan =
  Cli.Plan
    { Cli.help = help,
      Cli.move = move,
      Cli.test = test,
      Cli.tidy = \_ -> pure ()
    }

help :: Cli.Docopt
help =
  [Cli.docopt|
agent

Usage:
  agent start <name> [--path=<path>]
  agent test
  agent --help

Options:
  --path=<path>   Path to the worker directory [default: .]
  --help          Show this help
|]

move :: Cli.Arguments -> IO ()
move args
  | args `Cli.has` Cli.command "start" = do
      name <-
        Cli.getArg args (Cli.argument "name") |> \case
          Just n -> pure (Text.pack n)
          Nothing -> panic "Name required"
      let path = Cli.getArgWithDefault args "." (Cli.longOption "path")

      let worker =
            Core.Worker
              { Core.workerName = name,
                Core.workerPid = Nothing,
                Core.workerStatus = Core.Idle,
                Core.workerPath = path
              }

      Worker.start worker
  | otherwise = putStrLn (Cli.usage help)

test :: Test.Tree
test = Test.group "Omni.Agent" [unitTests]

unitTests :: Test.Tree
unitTests =
  Test.group
    "Unit tests"
    [ Test.unit "can parse start command" <| do
        let result = Docopt.parseArgs help ["start", "worker-1"]
        case result of
          Left err -> Test.assertFailure <| "Failed to parse 'start': " <> show err
          Right args -> args `Cli.has` Cli.command "start" Test.@?= True
    ]