1 files changed, 77 insertions, 0 deletions
diff --git a/Omni/Agent.hs b/Omni/Agent.hs
new file mode 100644
index 0000000..d53bccd
--- /dev/null
+++ b/Omni/Agent.hs
@@ -0,0 +1,77 @@
+{-# 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
+    ]