-rw-r--r-- | ui-snapshot.c | 92 |
1 files changed, 80 insertions, 12 deletions
diff --git a/ui-snapshot.c b/ui-snapshot.c index 7a597ff..9c4d086 100644 --- a/ui-snapshot.c +++ b/ui-snapshot.c | |||
@@ -59,16 +59,32 @@ static int write_tar_bzip2_archive(struct archiver_args *args) | |||
59 | const struct cgit_snapshot_format cgit_snapshot_formats[] = { | 59 | const struct cgit_snapshot_format cgit_snapshot_formats[] = { |
60 | { ".zip", "application/x-zip", write_zip_archive, 0x1 }, | 60 | { ".zip", "application/x-zip", write_zip_archive, 0x1 }, |
61 | { ".tar.gz", "application/x-tar", write_tar_gzip_archive, 0x2 }, | 61 | { ".tar.gz", "application/x-tar", write_tar_gzip_archive, 0x2 }, |
62 | { ".tar.bz2", "application/x-tar", write_tar_bzip2_archive, 0x4 }, | 62 | { ".tar.bz2", "application/x-tar", write_tar_bzip2_archive, 0x4 }, |
63 | { ".tar", "application/x-tar", write_tar_archive, 0x8 }, | 63 | { ".tar", "application/x-tar", write_tar_archive, 0x8 }, |
64 | {} | 64 | {} |
65 | }; | 65 | }; |
66 | 66 | ||
67 | static const struct cgit_snapshot_format *get_format(const char *filename) | ||
68 | { | ||
69 | const struct cgit_snapshot_format *fmt; | ||
70 | int fl, sl; | ||
71 | |||
72 | fl = strlen(filename); | ||
73 | for(fmt = cgit_snapshot_formats; fmt->suffix; fmt++) { | ||
74 | sl = strlen(fmt->suffix); | ||
75 | if (sl >= fl) | ||
76 | continue; | ||
77 | if (!strcmp(fmt->suffix, filename + fl - sl)) | ||
78 | return fmt; | ||
79 | } | ||
80 | return NULL; | ||
81 | } | ||
82 | |||
67 | static int make_snapshot(const struct cgit_snapshot_format *format, | 83 | static int make_snapshot(const struct cgit_snapshot_format *format, |
68 | const char *hex, const char *prefix, | 84 | const char *hex, const char *prefix, |
69 | const char *filename) | 85 | const char *filename) |
70 | { | 86 | { |
71 | struct archiver_args args; | 87 | struct archiver_args args; |
72 | struct commit *commit; | 88 | struct commit *commit; |
73 | unsigned char sha1[20]; | 89 | unsigned char sha1[20]; |
74 | 90 | ||
@@ -93,28 +109,80 @@ static int make_snapshot(const struct cgit_snapshot_format *format, | |||
93 | args.time = commit->date; | 109 | args.time = commit->date; |
94 | ctx.page.mimetype = xstrdup(format->mimetype); | 110 | ctx.page.mimetype = xstrdup(format->mimetype); |
95 | ctx.page.filename = xstrdup(filename); | 111 | ctx.page.filename = xstrdup(filename); |
96 | cgit_print_http_headers(&ctx); | 112 | cgit_print_http_headers(&ctx); |
97 | format->write_func(&args); | 113 | format->write_func(&args); |
98 | return 0; | 114 | return 0; |
99 | } | 115 | } |
100 | 116 | ||
117 | char *dwim_filename = NULL; | ||
118 | const char *dwim_refname = NULL; | ||
119 | |||
120 | static int ref_cb(const char *refname, const unsigned char *sha1, int flags, | ||
121 | void *cb_data) | ||
122 | { | ||
123 | const char *r = refname; | ||
124 | while (r && *r) { | ||
125 | fprintf(stderr, " cmp %s with %s:", dwim_filename, r); | ||
126 | if (!strcmp(dwim_filename, r)) { | ||
127 | fprintf(stderr, "MATCH!\n"); | ||
128 | dwim_refname = refname; | ||
129 | return 1; | ||
130 | } | ||
131 | fprintf(stderr, "no match\n"); | ||
132 | if (isdigit(*r)) | ||
133 | break; | ||
134 | r++; | ||
135 | } | ||
136 | return 0; | ||
137 | } | ||
138 | |||
139 | /* Try to guess the requested revision by combining repo name and tag name | ||
140 | * and comparing this to the requested snapshot name. E.g. the requested | ||
141 | * snapshot is "cgit-0.7.2.tar.gz" while repo name is "cgit" and tag name | ||
142 | * is "v0.7.2". First, the reponame is stripped off, leaving "-0.7.2.tar.gz". | ||
143 | * Next, any '-' and '_' characters are stripped, leaving "0.7.2.tar.gz". | ||
144 | * Finally, the requested format suffix is removed and we end up with "0.7.2". | ||
145 | * Then we test each tag against this dwimmed filename, and for each tag | ||
146 | * we even try to remove any leading characters which are non-digits. I.e. | ||
147 | * we first compare with "v0.7.2", then with "0.7.2" and we've got a match. | ||
148 | */ | ||
149 | static const char *get_ref_from_filename(const char *url, const char *filename, | ||
150 | const struct cgit_snapshot_format *fmt) | ||
151 | { | ||
152 | const char *reponame = cgit_repobasename(url); | ||
153 | fprintf(stderr, "reponame=%s, filename=%s\n", reponame, filename); | ||
154 | if (prefixcmp(filename, reponame)) | ||
155 | return NULL; | ||
156 | filename += strlen(reponame); | ||
157 | while (filename && (*filename == '-' || *filename == '_')) | ||
158 | filename++; | ||
159 | dwim_filename = xstrdup(filename); | ||
160 | dwim_filename[strlen(filename) - strlen(fmt->suffix)] = '\0'; | ||
161 | for_each_tag_ref(ref_cb, NULL); | ||
162 | return dwim_refname; | ||
163 | } | ||
164 | |||
101 | void cgit_print_snapshot(const char *head, const char *hex, const char *prefix, | 165 | void cgit_print_snapshot(const char *head, const char *hex, const char *prefix, |
102 | const char *filename, int snapshots) | 166 | const char *filename, int snapshots, int dwim) |
103 | { | 167 | { |
104 | const struct cgit_snapshot_format* f; | 168 | const struct cgit_snapshot_format* f; |
105 | int sl, fnl; | ||
106 | 169 | ||
107 | fnl = strlen(filename); | 170 | f = get_format(filename); |
171 | if (!f) { | ||
172 | ctx.page.mimetype = "text/html"; | ||
173 | cgit_print_http_headers(&ctx); | ||
174 | cgit_print_docstart(&ctx); | ||
175 | cgit_print_pageheader(&ctx); | ||
176 | cgit_print_error(fmt("Unsupported snapshot format: %s", filename)); | ||
177 | cgit_print_docend(); | ||
178 | return; | ||
179 | } | ||
180 | |||
181 | if (!hex && dwim) | ||
182 | hex = get_ref_from_filename(ctx.repo->url, filename, f); | ||
183 | |||
108 | if (!hex) | 184 | if (!hex) |
109 | hex = head; | 185 | hex = head; |
110 | for (f = cgit_snapshot_formats; f->suffix; f++) { | 186 | |
111 | if (!(snapshots & f->bit)) | ||
112 | continue; | ||
113 | sl = strlen(f->suffix); | ||
114 | if(fnl < sl || strcmp(&filename[fnl-sl], f->suffix)) | ||
115 | continue; | ||
116 | make_snapshot(f, hex, prefix, filename); | 187 | make_snapshot(f, hex, prefix, filename); |
117 | return; | ||
118 | } | ||
119 | cgit_print_error(fmt("Unsupported snapshot format: %s", filename)); | ||
120 | } | 188 | } |