5 files changed, 94 insertions, 60 deletions

diff --git a/cmd.c b/cmd.c
index 5b3c14c..8914fa5 100644
--- a/cmd.c
+++ b/cmd.c
@@ -105,6 +105,5 @@ static void refs_fn(struct cgit_context *ctx)
 static void snapshot_fn(struct cgit_context *ctx)
 {
-        cgit_print_snapshot(ctx->qry.head, ctx->qry.sha1,
-                            cgit_repobasename(ctx->repo->url), ctx->qry.path,
+        cgit_print_snapshot(ctx->qry.head, ctx->qry.sha1, ctx->qry.path,
                             ctx->repo->snapshots, ctx->qry.nohead);
 }
diff --git a/ui-refs.c b/ui-refs.c
index 32e0429..0805fc8 100644
--- a/ui-refs.c
+++ b/ui-refs.c
@@ -79,5 +79,5 @@ static void print_tag_header()
 {
         html("<tr class='nohover'><th class='left'>Tag</th>"
-             "<th class='left'>Reference</th>"
+             "<th class='left'>Download</th>"
              "<th class='left'>Author</th>"
              "<th class='left' colspan='2'>Age</th></tr>\n");
@@ -85,4 +85,29 @@ static void print_tag_header()
 }
 
+static void print_tag_downloads(const struct cgit_repo *repo, const char *ref)
+{
+        const struct cgit_snapshot_format* f;
+            char *filename;
+        const char *basename;
+
+        if (!ref || strlen(ref) < 2)
+                return;
+
+        basename = cgit_repobasename(repo->url);
+        if (prefixcmp(ref, basename) != 0) {
+                if ((ref[0] == 'v' || ref[0] == 'V') && isdigit(ref[1]))
+                        ref++;
+                if (isdigit(ref[0]))
+                        ref = xstrdup(fmt("%s-%s", basename, ref));
+        }
+
+        for (f = cgit_snapshot_formats; f->suffix; f++) {
+                if (!(repo->snapshots & f->bit))
+                        continue;
+                filename = fmt("%s%s", ref, f->suffix);
+                cgit_snapshot_link(filename, NULL, NULL, NULL, NULL, filename);
+                html("&nbsp;&nbsp;");
+        }
+}
 static int print_tag(struct refinfo *ref)
 {
@@ -99,5 +124,8 @@ static int print_tag(struct refinfo *ref)
                 cgit_tag_link(name, NULL, NULL, ctx.qry.head, name);
                 html("</td><td>");
-                cgit_object_link(tag->tagged);
+                if (ctx.repo->snapshots && (tag->tagged->type == OBJ_COMMIT))
+                        print_tag_downloads(ctx.repo, name);
+                else
+                        cgit_object_link(tag->tagged);
                 html("</td><td>");
                 if (info->tagger)
@@ -113,5 +141,8 @@ static int print_tag(struct refinfo *ref)
                 html_txt(name);
                 html("</td><td>");
-                cgit_object_link(ref->object);
+                if (ctx.repo->snapshots && (tag->tagged->type == OBJ_COMMIT))
+                        print_tag_downloads(ctx.repo, name);
+                else
+                        cgit_object_link(ref->object);
                 html("</td></tr>\n");
         }
diff --git a/ui-shared.c b/ui-shared.c
index 224e5f3..9319881 100644
--- a/ui-shared.c
+++ b/ui-shared.c
@@ -366,9 +366,12 @@ void cgit_patch_link(char *name, char *title, char *class, char *head,
 void cgit_object_link(struct object *obj)
 {
-        char *page, *rev, *name;
+        char *page, *shortrev, *fullrev, *name;
 
+        fullrev = sha1_to_hex(obj->sha1);
+        shortrev = xstrdup(fullrev);
+        shortrev[10] = '\0';
         if (obj->type == OBJ_COMMIT) {
-                cgit_commit_link(fmt("commit %s", sha1_to_hex(obj->sha1)), NULL, NULL,
-                                 ctx.qry.head, sha1_to_hex(obj->sha1));
+                cgit_commit_link(fmt("commit %s...", shortrev), NULL, NULL,
+                                 ctx.qry.head, fullrev);
                 return;
         } else if (obj->type == OBJ_TREE)
@@ -378,7 +381,6 @@ void cgit_object_link(struct object *obj)
         else
                 page = "blob";
-        rev = sha1_to_hex(obj->sha1);
-        name = fmt("%s %s", typename(obj->type), rev);
-        reporevlink(page, name, NULL, NULL, ctx.qry.head, rev, NULL);
+        name = fmt("%s %s...", typename(obj->type), shortrev);
+        reporevlink(page, name, NULL, NULL, ctx.qry.head, fullrev, NULL);
 }
 
@@ -708,6 +710,5 @@ void cgit_print_snapshot_links(const char *repo, const char *head,
                 filename = fmt("%s-%s%s", cgit_repobasename(repo), hex,
                                f->suffix);
-                cgit_snapshot_link(filename, NULL, NULL, (char *)head,
-                                   (char *)hex, filename);
+                cgit_snapshot_link(filename, NULL, NULL, NULL, NULL, filename);
                 html("<br/>");
         }
diff --git a/ui-snapshot.c b/ui-snapshot.c
index 9c4d086..6f09151 100644
--- a/ui-snapshot.c
+++ b/ui-snapshot.c
@@ -115,56 +115,51 @@ static int make_snapshot(const struct cgit_snapshot_format *format,
 }
 
-char *dwim_filename = NULL;
-const char *dwim_refname = NULL;
-
-static int ref_cb(const char *refname, const unsigned char *sha1, int flags,
-                  void *cb_data)
-{
-        const char *r = refname;
-        while (r && *r) {
-                fprintf(stderr, "  cmp %s with %s:", dwim_filename, r);
-                if (!strcmp(dwim_filename, r)) {
-                        fprintf(stderr, "MATCH!\n");
-                        dwim_refname = refname;
-                        return 1;
-                }
-                fprintf(stderr, "no match\n");
-                if (isdigit(*r))
-                        break;
-                r++;
-        }
-        return 0;
-}
-
-/* Try to guess the requested revision by combining repo name and tag name
- * and comparing this to the requested snapshot name. E.g. the requested
- * snapshot is "cgit-0.7.2.tar.gz" while repo name is "cgit" and tag name
- * is "v0.7.2". First, the reponame is stripped off, leaving "-0.7.2.tar.gz".
- * Next, any '-' and '_' characters are stripped, leaving "0.7.2.tar.gz".
- * Finally, the requested format suffix is removed and we end up with "0.7.2".
- * Then we test each tag against this dwimmed filename, and for each tag
- * we even try to remove any leading characters which are non-digits. I.e.
- * we first compare with "v0.7.2", then with "0.7.2" and we've got a match.
+/* Try to guess the requested revision from the requested snapshot name.
+ * First the format extension is stripped, e.g. "cgit-0.7.2.tar.gz" become
+ * "cgit-0.7.2". If this is a valid commit object name we've got a winner.
+ * Otherwise, if the snapshot name has a prefix matching the result from
+ * repo_basename(), we strip the basename and any following '-' and '_'
+ * characters ("cgit-0.7.2" -> "0.7.2") and check the resulting name once
+ * more. If this still isn't a valid commit object name, we check if pre-
+ * pending a 'v' to the remaining snapshot name ("0.7.2" -> "v0.7.2") gives
+ * us something valid.
  */
 static const char *get_ref_from_filename(const char *url, const char *filename,
-                                         const struct cgit_snapshot_format *fmt)
+                                         const struct cgit_snapshot_format *format)
 {
-        const char *reponame = cgit_repobasename(url);
-        fprintf(stderr, "reponame=%s, filename=%s\n", reponame, filename);
-        if (prefixcmp(filename, reponame))
-                return NULL;
-        filename += strlen(reponame);
-        while (filename && (*filename == '-' || *filename == '_'))
-                filename++;
-        dwim_filename = xstrdup(filename);
-        dwim_filename[strlen(filename) - strlen(fmt->suffix)] = '\0';
-        for_each_tag_ref(ref_cb, NULL);
-        return dwim_refname;
+        const char *reponame;
+        unsigned char sha1[20];
+        char *snapshot;
+
+        snapshot = xstrdup(filename);
+        snapshot[strlen(snapshot) - strlen(format->suffix)] = '\0';
+        fprintf(stderr, "snapshot=%s\n", snapshot);
+
+        if (get_sha1(snapshot, sha1) == 0)
+                return snapshot;
+
+        reponame = cgit_repobasename(url);
+        fprintf(stderr, "reponame=%s\n", reponame);
+        if (prefixcmp(snapshot, reponame) == 0) {
+                snapshot += strlen(reponame);
+                while (snapshot && (*snapshot == '-' || *snapshot == '_'))
+                        snapshot++;
+        }
+
+        if (get_sha1(snapshot, sha1) == 0)
+                return snapshot;
+
+        snapshot = fmt("v%s", snapshot);
+        if (get_sha1(snapshot, sha1) == 0)
+                return snapshot;
+
+        return NULL;
 }
 
-void cgit_print_snapshot(const char *head, const char *hex, const char *prefix,
+void cgit_print_snapshot(const char *head, const char *hex,
                          const char *filename, int snapshots, int dwim)
 {
         const struct cgit_snapshot_format* f;
+        char *prefix = NULL;
 
         f = get_format(filename);
@@ -179,10 +174,19 @@ void cgit_print_snapshot(const char *head, const char *hex, const char *prefix,
         }
 
-        if (!hex && dwim)
+        if (!hex && dwim) {
                 hex = get_ref_from_filename(ctx.repo->url, filename, f);
+                if (hex != NULL) {
+                        prefix = xstrdup(filename);
+                        prefix[strlen(filename) - strlen(f->suffix)] = '\0';
+                }
+        }
 
         if (!hex)
                 hex = head;
 
+        if (!prefix)
+                prefix = xstrdup(cgit_repobasename(ctx.repo->url));
+
         make_snapshot(f, hex, prefix, filename);
+        free(prefix);
 }
diff --git a/ui-snapshot.h b/ui-snapshot.h
index 3540303..b6ede52 100644
--- a/ui-snapshot.h
+++ b/ui-snapshot.h
@@ -3,6 +3,5 @@
 
 extern void cgit_print_snapshot(const char *head, const char *hex,
-                                const char *prefix, const char *filename,
-                                int snapshot, int dwim);
+                                const char *filename, int snapshot, int dwim);
 
 #endif /* UI_SNAPSHOT_H */