Add support for tree listing via h parameter

This makes the tree page benefit from repo.defbranch.

Signed-off-by: Lars Hjemli <hjemli@gmail.com>

1 files changed, 10 insertions, 0 deletions

diff --git a/ui-tree.c b/ui-tree.c
index db63e13..032a966 100644
--- a/ui-tree.c
+++ b/ui-tree.c
@@ -66,14 +66,24 @@ static int print_entry(const unsigned char *sha1, const char *base,
 }
 
 void cgit_print_tree(const char *rev, const char *hex, char *path)
 {
         struct tree *tree;
         unsigned char sha1[20];
+        struct commit *commit;
 
         curr_rev = xstrdup(rev);
+        get_sha1(rev, sha1);
+        commit = lookup_commit_reference(sha1);
+        if (!commit || parse_commit(commit)) {
+                cgit_print_error(fmt("Invalid head: %s", rev));
+                return;
+        }
+        if (!hex)
+                hex = sha1_to_hex(commit->tree->object.sha1);
+
         if (get_sha1_hex(hex, sha1)) {
                 cgit_print_error(fmt("Invalid object id: %s", hex));
                 return;
         }
         tree = parse_tree_indirect(sha1);
         if (!tree) {