Test repo subtree

2 files changed, 48 insertions, 0 deletions

diff --git a/src/Command/Subtree.hs b/src/Command/Subtree.hs
new file mode 100644
index 0000000..8d42d73
--- /dev/null
+++ b/src/Command/Subtree.hs
@@ -0,0 +1,46 @@
+module Command.Subtree (
+    SubtreeCommand,
+) where
+
+import Data.Text (Text)
+import Data.Text qualified as T
+
+import Command
+import Output
+import Repo
+
+
+data SubtreeCommand = SubtreeCommand SubtreeOptions [ Text ]
+
+data SubtreeOptions = SubtreeOptions
+
+instance Command SubtreeCommand where
+    commandName _ = "subtree"
+    commandDescription _ = "Resolve subdirectory of given repo tree"
+
+    type CommandArguments SubtreeCommand = [ Text ]
+
+    commandUsage _ = T.pack $ unlines $
+        [ "Usage: minici subtree <tree> <path>"
+        ]
+
+    type CommandOptions SubtreeCommand = SubtreeOptions
+    defaultCommandOptions _ = SubtreeOptions
+
+    commandInit _ opts = SubtreeCommand opts
+    commandExec = cmdSubtree
+
+
+cmdSubtree :: SubtreeCommand -> CommandExec ()
+cmdSubtree (SubtreeCommand SubtreeOptions args) = do
+    [ treeParam, path ] <- return args
+    out <- getOutput
+    repo <- getDefaultRepo
+
+    let ( tree, subdir ) =
+            case T.splitOn "(" treeParam of
+                (t : param : _) -> ( t, T.unpack $ T.takeWhile (/= ')') param )
+                _ -> ( treeParam, "" )
+
+    subtree <- getSubtree Nothing (T.unpack path) =<< readTree repo subdir tree
+    outputMessage out $ textTreeId $ treeId subtree
diff --git a/src/Main.hs b/src/Main.hs
index e273715..ec03acb 100644
--- a/src/Main.hs
+++ b/src/Main.hs
@@ -25,6 +25,7 @@ import Command.Extract
 import Command.JobId
 import Command.Log
 import Command.Run
+import Command.Subtree
 import Config
 import Output
 import Repo
@@ -92,6 +93,7 @@ commands =
     , SC $ Proxy @ExtractCommand
     , SC $ Proxy @JobIdCommand
     , SC $ Proxy @LogCommand
+    , SC $ Proxy @SubtreeCommand
     ]
 
 lookupCommand :: String -> Maybe SomeCommandType