2 files changed, 60 insertions, 0 deletions

diff --git a/cgit.h b/cgit.h
index e96311f..75e919b 100644
--- a/cgit.h
+++ b/cgit.h
@@ -98,6 +98,21 @@ struct taginfo {
         char *msg;
 };
 
+struct refinfo {
+        const char *refname;
+        struct object *object;
+        union {
+                struct taginfo *tag;
+                struct commitinfo *commit;
+        };
+};
+
+struct reflist {
+        struct refinfo **refs;
+        int alloc;
+        int count;
+};
+
 extern const char *cgit_version;
 
 extern struct repolist cgit_repolist;
@@ -162,6 +177,10 @@ extern int chk_non_negative(int result, char *msg);
 extern int hextoint(char c);
 extern char *trim_end(const char *str, char c);
 
+extern void cgit_add_ref(struct reflist *list, struct refinfo *ref);
+extern int cgit_refs_cb(const char *refname, const unsigned char *sha1,
+                        int flags, void *cb_data);
+
 extern void *cgit_free_commitinfo(struct commitinfo *info);
 
 extern int cgit_diff_files(const unsigned char *old_sha1,
diff --git a/shared.c b/shared.c
index 3d4feea..d815cb1 100644
--- a/shared.c
+++ b/shared.c
@@ -291,6 +291,47 @@ char *trim_end(const char *str, char c)
         return s;
 }
 
+void cgit_add_ref(struct reflist *list, struct refinfo *ref)
+{
+        size_t size;
+
+        if (list->count >= list->alloc) {
+                list->alloc += (list->alloc ? list->alloc : 4);
+                size = list->alloc * sizeof(struct refinfo *);
+                list->refs = xrealloc(list->refs, size);
+        }
+        list->refs[list->count++] = ref;
+}
+
+struct refinfo *cgit_mk_refinfo(const char *refname, const unsigned char *sha1)
+{
+        struct refinfo *ref;
+
+        ref = xmalloc(sizeof (struct refinfo));
+        ref->refname = xstrdup(refname);
+        ref->object = parse_object(sha1);
+        switch (ref->object->type) {
+        case OBJ_TAG:
+                ref->tag = cgit_parse_tag((struct tag *)ref->object);
+                break;
+        case OBJ_COMMIT:
+                ref->commit = cgit_parse_commit((struct commit *)ref->object);
+                break;
+        }
+        return ref;
+}
+
+int cgit_refs_cb(const char *refname, const unsigned char *sha1, int flags,
+                  void *cb_data)
+{
+        struct reflist *list = (struct reflist *)cb_data;
+        struct refinfo *info = cgit_mk_refinfo(refname, sha1);
+
+        if (info)
+                cgit_add_ref(list, info);
+        return 0;
+}
+
 void cgit_diff_tree_cb(struct diff_queue_struct *q,
                        struct diff_options *options, void *data)
 {