git: Add ls-tree

1 files changed, 68 insertions, 0 deletions

diff --git a/server/common/src/git.rs b/server/common/src/git.rs
index c6a44e3..053ff3f 100644
--- a/server/common/src/git.rs
+++ b/server/common/src/git.rs
@@ -64,6 +64,20 @@ pub struct User {
     pub username: String,
 }
 
+#[derive(Debug, PartialEq)]
+pub enum ObjectType {
+    BLOB,
+    COMMIT,
+    TREE,
+}
+
+#[derive(Debug, PartialEq)]
+pub struct TreeEntry {
+    pub object_type: ObjectType,
+    pub object_name: String,
+    pub path: String,
+}
+
 fn io_err(action: &str, e: std::io::Error) -> Error {
     Error::new(format!("{action}: {e}"))
 }
@@ -80,6 +94,34 @@ fn parse_user(output: String) -> User {
     }
 }
 
+fn parse_tree_entries(output: String) -> Vec<TreeEntry> {
+    let mut ret = Vec::new();
+    for line in output.split_terminator('\0') {
+        let mut parts = line.split_terminator('\t');
+        let mut parts2 = parts.next().unwrap_or("").split_terminator(' ');
+        parts2.next(); // object mode
+        let object_type = match parts2.next() {
+            Some(value) => match value {
+                "blob" => ObjectType::BLOB,
+                "commit" => ObjectType::COMMIT,
+                "tree" => ObjectType::TREE,
+                _ => continue,
+            },
+            None => continue,
+        };
+        if let Some(object_name) = parts2.next() {
+            if let Some(path) = parts.next() {
+                ret.push(TreeEntry {
+                    object_type,
+                    object_name: object_name.to_string(),
+                    path: path.to_string(),
+                });
+            }
+        }
+    }
+    ret
+}
+
 impl RepoData {
     fn new() -> Self {
         Self {
@@ -342,6 +384,21 @@ impl RepoData {
         self.run(&mut cmd).await
     }
 
+    async fn ls_tree(
+        &self,
+        repo: &Repository,
+        commit: &str,
+        recursive: bool,
+    ) -> Result<Vec<TreeEntry>, Error> {
+        let mut cmd = self.git_cmd(repo);
+        cmd.arg("ls-tree").arg("-z");
+        if recursive {
+            cmd.arg("-r");
+        }
+        cmd.arg(commit);
+        self.output(&mut cmd).map_ok(parse_tree_entries).await
+    }
+
     async fn get_log_format(
         &self,
         repo: &Repository,
@@ -594,4 +651,15 @@ impl Repository {
 
         data.delete_branch(self, branch.as_str()).await
     }
+
+    pub async fn ls_tree(
+        &self,
+        commit: impl Into<String>,
+        recursive: bool,
+    ) -> Result<Vec<TreeEntry>, Error> {
+        let commit = commit.into();
+        let data = self.lock.read().await;
+
+        data.ls_tree(self, commit.as_str(), recursive).await
+    }
 }