]> git.armaanb.net Git - stagit.git/blob - stagit.c
extra whitespace
[stagit.git] / stagit.c
1 #include <sys/stat.h>
2
3 #include <err.h>
4 #include <errno.h>
5 #include <inttypes.h>
6 #include <libgen.h>
7 #include <limits.h>
8 #include <stdio.h>
9 #include <stdlib.h>
10 #include <string.h>
11 #include <unistd.h>
12
13 #include <git2.h>
14
15 #include "compat.h"
16 #include "config.h"
17
18 struct commitinfo {
19         const git_oid *id;
20
21         char oid[GIT_OID_HEXSZ + 1];
22         char parentoid[GIT_OID_HEXSZ + 1];
23
24         const git_signature *author;
25         const char          *summary;
26         const char          *msg;
27
28         git_diff_stats *stats;
29         git_diff       *diff;
30         git_commit     *commit;
31         git_commit     *parent;
32         git_tree       *commit_tree;
33         git_tree       *parent_tree;
34
35         size_t addcount;
36         size_t delcount;
37         size_t filecount;
38 };
39
40 static git_repository *repo;
41
42 static const char *relpath = "";
43 static const char *repodir;
44
45 static char name[255];
46 static char description[255];
47 static char cloneurl[1024];
48 static int hasreadme, haslicense;
49
50 void
51 commitinfo_free(struct commitinfo *ci)
52 {
53         if (!ci)
54                 return;
55
56         git_diff_stats_free(ci->stats);
57         git_diff_free(ci->diff);
58         git_tree_free(ci->commit_tree);
59         git_tree_free(ci->parent_tree);
60         git_commit_free(ci->commit);
61 }
62
63 struct commitinfo *
64 commitinfo_getbyoid(const git_oid *id)
65 {
66         struct commitinfo *ci;
67         git_diff_options opts;
68         int error;
69
70         if (!(ci = calloc(1, sizeof(struct commitinfo))))
71                 err(1, "calloc");
72
73         ci->id = id;
74         if (git_commit_lookup(&(ci->commit), repo, id))
75                 goto err;
76
77         git_oid_tostr(ci->oid, sizeof(ci->oid), git_commit_id(ci->commit));
78         git_oid_tostr(ci->parentoid, sizeof(ci->parentoid), git_commit_parent_id(ci->commit, 0));
79
80         ci->author = git_commit_author(ci->commit);
81         ci->summary = git_commit_summary(ci->commit);
82         ci->msg = git_commit_message(ci->commit);
83
84         if ((error = git_commit_tree(&(ci->commit_tree), ci->commit)))
85                 goto err;
86         if (!(error = git_commit_parent(&(ci->parent), ci->commit, 0))) {
87                 if ((error = git_commit_tree(&(ci->parent_tree), ci->parent)))
88                         goto err;
89         } else {
90                 ci->parent = NULL;
91                 ci->parent_tree = NULL;
92         }
93
94         git_diff_init_options(&opts, GIT_DIFF_OPTIONS_VERSION);
95         opts.flags |= GIT_DIFF_DISABLE_PATHSPEC_MATCH;
96         if ((error = git_diff_tree_to_tree(&(ci->diff), repo, ci->parent_tree, ci->commit_tree, &opts)))
97                 goto err;
98         if (git_diff_get_stats(&(ci->stats), ci->diff))
99                 goto err;
100
101         ci->addcount = git_diff_stats_insertions(ci->stats);
102         ci->delcount = git_diff_stats_deletions(ci->stats);
103         ci->filecount = git_diff_stats_files_changed(ci->stats);
104
105         return ci;
106
107 err:
108         commitinfo_free(ci);
109         free(ci);
110
111         return NULL;
112 }
113
114 FILE *
115 efopen(const char *name, const char *flags)
116 {
117         FILE *fp;
118
119         if (!(fp = fopen(name, flags)))
120                 err(1, "fopen");
121
122         return fp;
123 }
124
125 /* Escape characters below as HTML 2.0 / XML 1.0. */
126 void
127 xmlencode(FILE *fp, const char *s, size_t len)
128 {
129         size_t i;
130
131         for (i = 0; *s && i < len; s++, i++) {
132                 switch(*s) {
133                 case '<':  fputs("&lt;",   fp); break;
134                 case '>':  fputs("&gt;",   fp); break;
135                 case '\'': fputs("&apos;", fp); break;
136                 case '&':  fputs("&amp;",  fp); break;
137                 case '"':  fputs("&quot;", fp); break;
138                 default:   fputc(*s, fp);
139                 }
140         }
141 }
142
143 /* Some implementations of dirname(3) return a pointer to a static
144  * internal buffer (OpenBSD). Others modify the contents of `path` (POSIX).
145  * This is a wrapper function that is compatible with both versions.
146  * The program will error out if dirname(3) failed, this can only happen
147  * with the OpenBSD version. */
148 char *
149 xdirname(const char *path)
150 {
151         char *p, *b;
152
153         if (!(p = strdup(path)))
154                 err(1, "strdup");
155         if (!(b = dirname(p)))
156                 err(1, "basename");
157         if (!(b = strdup(b)))
158                 err(1, "strdup");
159         free(p);
160
161         return b;
162 }
163
164 /* Some implementations of basename(3) return a pointer to a static
165  * internal buffer (OpenBSD). Others modify the contents of `path` (POSIX).
166  * This is a wrapper function that is compatible with both versions.
167  * The program will error out if basename(3) failed, this can only happen
168  * with the OpenBSD version. */
169 char *
170 xbasename(const char *path)
171 {
172         char *p, *b;
173
174         if (!(p = strdup(path)))
175                 err(1, "strdup");
176         if (!(b = basename(p)))
177                 err(1, "basename");
178         if (!(b = strdup(b)))
179                 err(1, "strdup");
180         free(p);
181
182         return b;
183 }
184
185 int
186 mkdirp(const char *path)
187 {
188         char tmp[PATH_MAX], *p;
189
190         strlcpy(tmp, path, sizeof(tmp));
191         for (p = tmp + (tmp[0] == '/'); *p; p++) {
192                 if (*p != '/')
193                         continue;
194                 *p = '\0';
195                 if (mkdir(tmp, S_IRWXU | S_IRWXG | S_IRWXO) < 0 && errno != EEXIST)
196                         return -1;
197                 *p = '/';
198         }
199         if (mkdir(tmp, S_IRWXU | S_IRWXG | S_IRWXO) < 0 && errno != EEXIST)
200                 return -1;
201         return 0;
202 }
203
204 void
205 printtimeformat(FILE *fp, const git_time *intime, const char *fmt)
206 {
207         struct tm *intm;
208         time_t t;
209         char out[32];
210
211         t = (time_t) intime->time + (intime->offset * 60);
212         intm = gmtime(&t);
213         strftime(out, sizeof(out), fmt, intm);
214         fputs(out, fp);
215 }
216
217 void
218 printtimez(FILE *fp, const git_time *intime)
219 {
220         printtimeformat(fp, intime, "%Y-%m-%dT%H:%M:%SZ");
221 }
222
223 void
224 printtime(FILE *fp, const git_time *intime)
225 {
226         printtimeformat(fp, intime, "%a %b %e %T %Y");
227 }
228
229 void
230 printtimeshort(FILE *fp, const git_time *intime)
231 {
232         printtimeformat(fp, intime, "%Y-%m-%d %H:%M");
233 }
234
235 int
236 writeheader(FILE *fp)
237 {
238         fputs("<!DOCTYPE HTML>"
239                 "<html dir=\"ltr\" lang=\"en\">\n<head>\n"
240                 "<meta http-equiv=\"Content-Type\" content=\"text/html; charset=UTF-8\" />\n"
241                 "<meta http-equiv=\"Content-Language\" content=\"en\" />\n<title>", fp);
242         xmlencode(fp, name, strlen(name));
243         if (description[0])
244                 fputs(" - ", fp);
245         xmlencode(fp, description, strlen(description));
246         fprintf(fp, "</title>\n<link rel=\"icon\" type=\"image/png\" href=\"%sfavicon.png\" />\n", relpath);
247         fprintf(fp, "<link rel=\"alternate\" type=\"application/atom+xml\" title=\"%s Atom Feed\" href=\"%satom.xml\" />\n",
248                 name, relpath);
249         fprintf(fp, "<link rel=\"stylesheet\" type=\"text/css\" href=\"%sstyle.css\" />\n", relpath);
250         fputs("</head>\n<body>\n<table><tr><td>", fp);
251         fprintf(fp, "<a href=\"../%s\"><img src=\"%slogo.png\" alt=\"\" width=\"32\" height=\"32\" /></a>",
252                 relpath, relpath);
253         fputs("</td><td><h1>", fp);
254         xmlencode(fp, name, strlen(name));
255         fputs("</h1><span class=\"desc\">", fp);
256         xmlencode(fp, description, strlen(description));
257         fputs("</span></td></tr>", fp);
258         if (cloneurl[0]) {
259                 fputs("<tr class=\"url\"><td></td><td>git clone <a href=\"", fp);
260                 xmlencode(fp, cloneurl, strlen(cloneurl));
261                 fputs("\">", fp);
262                 xmlencode(fp, cloneurl, strlen(cloneurl));
263                 fputs("</a></td></tr>", fp);
264         }
265         fputs("<tr><td></td><td>\n", fp);
266         fprintf(fp, "<a href=\"%slog.html\">Log</a> | ", relpath);
267         fprintf(fp, "<a href=\"%sfiles.html\">Files</a> | ", relpath);
268         fprintf(fp, "<a href=\"%srefs.html\">Refs/branches</a>", relpath);
269         if (hasreadme)
270                 fprintf(fp, " | <a href=\"%sfile/README.html\">README</a>", relpath);
271         if (haslicense)
272                 fprintf(fp, " | <a href=\"%sfile/LICENSE.html\">LICENSE</a>", relpath);
273         fputs("</td></tr></table>\n<hr/>\n<div id=\"content\">\n", fp);
274
275         return 0;
276 }
277
278 int
279 writefooter(FILE *fp)
280 {
281         return !fputs("</div>\n</body>\n</html>\n", fp);
282 }
283
284 void
285 writeblobhtml(FILE *fp, const git_blob *blob)
286 {
287         off_t i;
288         size_t n = 1;
289         char *nfmt = "<a href=\"#l%d\" id=\"l%d\">%d</a>\n";
290         const char *s = git_blob_rawcontent(blob);
291         git_off_t len = git_blob_rawsize(blob);
292
293         fputs("<table id=\"blob\"><tr><td class=\"num\"><pre>\n", fp);
294
295         if (len) {
296                 fprintf(fp, nfmt, n, n, n);
297                 for (i = 0; i < len - 1; i++) {
298                         if (s[i] == '\n') {
299                                 n++;
300                                 fprintf(fp, nfmt, n, n, n);
301                         }
302                 }
303         }
304
305         fputs("</pre></td><td><pre>\n", fp);
306         xmlencode(fp, s, (size_t)len);
307         fputs("</pre></td></tr></table>\n", fp);
308 }
309
310 void
311 printcommit(FILE *fp, struct commitinfo *ci)
312 {
313         fprintf(fp, "<b>commit</b> <a href=\"%scommit/%s.html\">%s</a>\n",
314                 relpath, ci->oid, ci->oid);
315
316         if (ci->parentoid[0])
317                 fprintf(fp, "<b>parent</b> <a href=\"%scommit/%s.html\">%s</a>\n",
318                         relpath, ci->parentoid, ci->parentoid);
319
320         if (ci->author) {
321                 fputs("<b>Author:</b> ", fp);
322                 xmlencode(fp, ci->author->name, strlen(ci->author->name));
323                 fputs(" &lt;<a href=\"mailto:", fp);
324                 xmlencode(fp, ci->author->email, strlen(ci->author->email));
325                 fputs("\">", fp);
326                 xmlencode(fp, ci->author->email, strlen(ci->author->email));
327                 fputs("</a>&gt;\n<b>Date:</b>   ", fp);
328                 printtime(fp, &(ci->author->when));
329                 fputc('\n', fp);
330         }
331         fputc('\n', fp);
332
333         if (ci->msg)
334                 xmlencode(fp, ci->msg, strlen(ci->msg));
335
336         fputc('\n', fp);
337 }
338
339 void
340 printshowfile(struct commitinfo *ci)
341 {
342         const git_diff_delta *delta;
343         const git_diff_hunk *hunk;
344         const git_diff_line *line;
345         git_patch *patch;
346         git_buf statsbuf;
347         size_t ndeltas, nhunks, nhunklines;
348         FILE *fp;
349         size_t i, j, k;
350         char path[PATH_MAX];
351
352         snprintf(path, sizeof(path), "commit/%s.html", ci->oid);
353         /* check if file exists if so skip it */
354         if (!access(path, F_OK))
355                 return;
356
357         fp = efopen(path, "w");
358         writeheader(fp);
359         fputs("<pre>\n", fp);
360         printcommit(fp, ci);
361
362         memset(&statsbuf, 0, sizeof(statsbuf));
363
364         /* diff stat */
365         if (ci->stats) {
366                 if (!git_diff_stats_to_buf(&statsbuf, ci->stats,
367                     GIT_DIFF_STATS_FULL | GIT_DIFF_STATS_SHORT, 80)) {
368                         if (statsbuf.ptr && statsbuf.ptr[0]) {
369                                 fputs("<b>Diffstat:</b>\n", fp);
370                                 fputs(statsbuf.ptr, fp);
371                         }
372                 }
373         }
374
375         fputs("<hr/>", fp);
376
377         ndeltas = git_diff_num_deltas(ci->diff);
378         for (i = 0; i < ndeltas; i++) {
379                 if (git_patch_from_diff(&patch, ci->diff, i)) {
380                         git_patch_free(patch);
381                         break;
382                 }
383
384                 delta = git_patch_get_delta(patch);
385                 fprintf(fp, "<b>diff --git a/<a href=\"%sfile/%s.html\">%s</a> b/<a href=\"%sfile/%s.html\">%s</a></b>\n",
386                         relpath, delta->old_file.path, delta->old_file.path,
387                         relpath, delta->new_file.path, delta->new_file.path);
388
389                 /* check binary data */
390                 if (delta->flags & GIT_DIFF_FLAG_BINARY) {
391                         fputs("Binary files differ\n", fp);
392                         git_patch_free(patch);
393                         continue;
394                 }
395
396                 nhunks = git_patch_num_hunks(patch);
397                 for (j = 0; j < nhunks; j++) {
398                         if (git_patch_get_hunk(&hunk, &nhunklines, patch, j))
399                                 break;
400
401                         fprintf(fp, "<a href=\"#h%zu\" id=\"h%zu\" class=\"h\">", j, j);
402                         xmlencode(fp, hunk->header, hunk->header_len);
403                         fputs("</a>", fp);
404
405                         for (k = 0; ; k++) {
406                                 if (git_patch_get_line_in_hunk(&line, patch, j, k))
407                                         break;
408                                 if (line->old_lineno == -1)
409                                         fprintf(fp, "<a href=\"#h%zu-%zu\" id=\"h%zu-%zu\" class=\"i\">+",
410                                                 j, k, j, k);
411                                 else if (line->new_lineno == -1)
412                                         fprintf(fp, "<a href=\"#h%zu-%zu\" id=\"h%zu-%zu\" class=\"d\">-",
413                                                 j, k, j, k);
414                                 else
415                                         fputc(' ', fp);
416                                 xmlencode(fp, line->content, line->content_len);
417                                 if (line->old_lineno == -1 || line->new_lineno == -1)
418                                         fputs("</a>", fp);
419                         }
420                 }
421                 git_patch_free(patch);
422         }
423         git_buf_free(&statsbuf);
424
425         fputs("</pre>\n", fp);
426         writefooter(fp);
427         fclose(fp);
428         return;
429 }
430
431 int
432 writelog(FILE *fp, const char *branch)
433 {
434         struct commitinfo *ci;
435         const git_oid *oid;
436         git_revwalk *w = NULL;
437         git_object *obj = NULL;
438         git_oid id;
439         size_t len;
440
441         mkdir("commit", 0755);
442
443         if (git_revparse_single(&obj, repo, branch))
444                 return -1;
445         oid = git_object_id(obj);
446
447         git_revwalk_new(&w, repo);
448         git_revwalk_push(w, oid);
449         git_revwalk_sorting(w, GIT_SORT_TIME);
450         git_revwalk_simplify_first_parent(w);
451
452         fputs("<table id=\"log\"><thead>\n<tr><td>Age</td><td>Commit message</td>"
453                   "<td>Author</td><td>Files</td><td class=\"num\">+</td>"
454                   "<td class=\"num\">-</td></tr>\n</thead><tbody>\n", fp);
455
456         while (!git_revwalk_next(&id, w)) {
457                 relpath = "";
458
459                 if (!(ci = commitinfo_getbyoid(&id)))
460                         break;
461
462                 fputs("<tr><td>", fp);
463                 if (ci->author)
464                         printtimeshort(fp, &(ci->author->when));
465                 fputs("</td><td>", fp);
466                 if (ci->summary) {
467                         fprintf(fp, "<a href=\"%scommit/%s.html\">", relpath, ci->oid);
468                         if ((len = strlen(ci->summary)) > summarylen) {
469                                 xmlencode(fp, ci->summary, summarylen - 1);
470                                 fputs("…", fp);
471                         } else {
472                                 xmlencode(fp, ci->summary, len);
473                         }
474                         fputs("</a>", fp);
475                 }
476                 fputs("</td><td>", fp);
477                 if (ci->author)
478                         xmlencode(fp, ci->author->name, strlen(ci->author->name));
479                 fputs("</td><td class=\"num\">", fp);
480                 fprintf(fp, "%zu", ci->filecount);
481                 fputs("</td><td class=\"num\">", fp);
482                 fprintf(fp, "+%zu", ci->addcount);
483                 fputs("</td><td class=\"num\">", fp);
484                 fprintf(fp, "-%zu", ci->delcount);
485                 fputs("</td></tr>\n", fp);
486
487                 relpath = "../";
488                 printshowfile(ci);
489
490                 commitinfo_free(ci);
491         }
492         fputs("</tbody></table>", fp);
493
494         git_revwalk_free(w);
495         git_object_free(obj);
496
497         relpath = "";
498
499         return 0;
500 }
501
502 void
503 printcommitatom(FILE *fp, struct commitinfo *ci)
504 {
505         fputs("<entry>\n", fp);
506
507         fprintf(fp, "<id>%s</id>\n", ci->oid);
508         if (ci->author) {
509                 fputs("<updated>", fp);
510                 printtimez(fp, &(ci->author->when));
511                 fputs("</updated>\n", fp);
512         }
513         if (ci->summary) {
514                 fputs("<title type=\"text\">", fp);
515                 xmlencode(fp, ci->summary, strlen(ci->summary));
516                 fputs("</title>\n", fp);
517         }
518
519         fputs("<content type=\"text\">", fp);
520         fprintf(fp, "commit %s\n", ci->oid);
521         if (ci->parentoid[0])
522                 fprintf(fp, "parent %s\n", ci->parentoid);
523         if (ci->author) {
524                 fputs("Author: ", fp);
525                 xmlencode(fp, ci->author->name, strlen(ci->author->name));
526                 fputs(" &lt;", fp);
527                 xmlencode(fp, ci->author->email, strlen(ci->author->email));
528                 fputs("&gt;\nDate:   ", fp);
529                 printtime(fp, &(ci->author->when));
530         }
531         fputc('\n', fp);
532
533         if (ci->msg)
534                 xmlencode(fp, ci->msg, strlen(ci->msg));
535         fputs("\n</content>\n", fp);
536         if (ci->author) {
537                 fputs("<author><name>", fp);
538                 xmlencode(fp, ci->author->name, strlen(ci->author->name));
539                 fputs("</name>\n<email>", fp);
540                 xmlencode(fp, ci->author->email, strlen(ci->author->email));
541                 fputs("</email>\n</author>\n", fp);
542         }
543         fputs("</entry>\n", fp);
544 }
545
546 int
547 writeatom(FILE *fp)
548 {
549         struct commitinfo *ci;
550         git_revwalk *w = NULL;
551         git_oid id;
552         size_t i, m = 100; /* max */
553
554         fputs("<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n"
555               "<feed xmlns=\"http://www.w3.org/2005/Atom\">\n<title>", fp);
556         xmlencode(fp, name, strlen(name));
557         fputs(", branch master</title>\n<subtitle>", fp);
558
559         xmlencode(fp, description, strlen(description));
560         fputs("</subtitle>\n", fp);
561
562         git_revwalk_new(&w, repo);
563         git_revwalk_push_head(w);
564         git_revwalk_sorting(w, GIT_SORT_TIME);
565         git_revwalk_simplify_first_parent(w);
566
567         for (i = 0; i < m && !git_revwalk_next(&id, w); i++) {
568                 if (!(ci = commitinfo_getbyoid(&id)))
569                         break;
570                 printcommitatom(fp, ci);
571                 commitinfo_free(ci);
572         }
573         git_revwalk_free(w);
574
575         fputs("</feed>", fp);
576
577         return 0;
578 }
579
580 int
581 writeblob(git_object *obj, const char *filename, git_off_t filesize)
582 {
583         char fpath[PATH_MAX];
584         char tmp[PATH_MAX] = "";
585         char *d, *p;
586         FILE *fp;
587
588         snprintf(fpath, sizeof(fpath), "file/%s.html", filename);
589         d = xdirname(fpath);
590         if (mkdirp(d)) {
591                 free(d);
592                 return 1;
593         }
594         free(d);
595
596         p = fpath;
597         while (*p) {
598                 if (*p == '/')
599                         strlcat(tmp, "../", sizeof(tmp));
600                 p++;
601         }
602         relpath = tmp;
603
604         fp = efopen(fpath, "w");
605         writeheader(fp);
606         fputs("<p> ", fp);
607         xmlencode(fp, filename, strlen(filename));
608         fprintf(fp, " (%jub)", (uintmax_t)filesize);
609         fputs("</p><hr/>", fp);
610
611         if (git_blob_is_binary((git_blob *)obj)) {
612                 fputs("<p>Binary file</p>\n", fp);
613         } else {
614                 writeblobhtml(fp, (git_blob *)obj);
615                 if (ferror(fp))
616                         err(1, "fwrite");
617         }
618         writefooter(fp);
619         fclose(fp);
620
621         relpath = "";
622
623         return 0;
624 }
625
626 const char *
627 filemode(git_filemode_t m)
628 {
629         static char mode[11];
630
631         memset(mode, '-', sizeof(mode) - 1);
632         mode[10] = '\0';
633
634         if (S_ISREG(m))
635                 mode[0] = '-';
636         else if (S_ISBLK(m))
637                 mode[0] = 'b';
638         else if (S_ISCHR(m))
639                 mode[0] = 'c';
640         else if (S_ISDIR(m))
641                 mode[0] = 'd';
642         else if (S_ISFIFO(m))
643                 mode[0] = 'p';
644         else if (S_ISLNK(m))
645                 mode[0] = 'l';
646         else if (S_ISSOCK(m))
647                 mode[0] = 's';
648         else
649                 mode[0] = '?';
650
651         if (m & S_IRUSR) mode[1] = 'r';
652         if (m & S_IWUSR) mode[2] = 'w';
653         if (m & S_IXUSR) mode[3] = 'x';
654         if (m & S_IRGRP) mode[4] = 'r';
655         if (m & S_IWGRP) mode[5] = 'w';
656         if (m & S_IXGRP) mode[6] = 'x';
657         if (m & S_IROTH) mode[7] = 'r';
658         if (m & S_IWOTH) mode[8] = 'w';
659         if (m & S_IXOTH) mode[9] = 'x';
660
661         if (m & S_ISUID) mode[3] = (mode[3] == 'x') ? 's' : 'S';
662         if (m & S_ISGID) mode[6] = (mode[6] == 'x') ? 's' : 'S';
663         if (m & S_ISVTX) mode[9] = (mode[9] == 'x') ? 't' : 'T';
664
665         return mode;
666 }
667
668 int
669 writefilestree(FILE *fp, git_tree *tree, const char *branch, const char *path)
670 {
671         const git_tree_entry *entry = NULL;
672         const char *filename;
673         char filepath[PATH_MAX];
674         git_object *obj = NULL;
675         git_off_t filesize;
676         size_t count, i;
677         int ret;
678
679         count = git_tree_entrycount(tree);
680         for (i = 0; i < count; i++) {
681                 if (!(entry = git_tree_entry_byindex(tree, i)) ||
682                     git_tree_entry_to_object(&obj, repo, entry))
683                         return -1;
684                 filename = git_tree_entry_name(entry);
685                 switch (git_object_type(obj)) {
686                 case GIT_OBJ_BLOB:
687                         break;
688                 case GIT_OBJ_TREE:
689                         /* NOTE: recurses */
690                         ret = writefilestree(fp, (git_tree *)obj, branch,
691                                              filename);
692                         git_object_free(obj);
693                         if (ret)
694                                 return ret;
695                         continue;
696                 default:
697                         git_object_free(obj);
698                         continue;
699                 }
700                 if (path[0]) {
701                         snprintf(filepath, sizeof(filepath), "%s/%s",
702                                  path, filename);
703                         filename = filepath;
704                 }
705                 filesize = git_blob_rawsize((git_blob *)obj);
706
707                 fputs("<tr><td>", fp);
708                 fprintf(fp, "%s", filemode(git_tree_entry_filemode(entry)));
709                 fprintf(fp, "</td><td><a href=\"%sfile/", relpath);
710                 xmlencode(fp, filename, strlen(filename));
711                 fputs(".html\">", fp);
712                 xmlencode(fp, filename, strlen(filename));
713                 fputs("</a></td><td class=\"num\">", fp);
714                 fprintf(fp, "%ju", (uintmax_t)filesize);
715                 fputs("</td></tr>\n", fp);
716
717                 writeblob(obj, filename, filesize);
718         }
719
720         return 0;
721 }
722
723 int
724 writefiles(FILE *fp, const char *branch)
725 {
726         const git_oid *id;
727         git_tree *tree = NULL;
728         git_object *obj = NULL;
729         git_commit *commit = NULL;
730         int ret = -1;
731
732         fputs("<table id=\"files\"><thead>\n<tr>"
733               "<td>Mode</td><td>Name</td><td class=\"num\">Size</td>"
734               "</tr>\n</thead><tbody>\n", fp);
735
736         if (git_revparse_single(&obj, repo, branch))
737                 goto err;
738         id = git_object_id(obj);
739         if (git_commit_lookup(&commit, repo, id) ||
740             git_commit_tree(&tree, commit))
741                 goto err;
742         ret = writefilestree(fp, tree, branch, "");
743
744 err:
745         fputs("</tbody></table>", fp);
746
747         git_object_free(obj);
748         git_commit_free(commit);
749         git_tree_free(tree);
750
751         return ret;
752 }
753
754 int
755 refs_cmp(const void *v1, const void *v2)
756 {
757         git_reference *r1 = (*(git_reference **)v1);
758         git_reference *r2 = (*(git_reference **)v2);
759         int t1, t2;
760
761         t1 = git_reference_is_branch(r1);
762         t2 = git_reference_is_branch(r2);
763
764         if (t1 != t2)
765                 return t1 - t2;
766
767         return strcmp(git_reference_shorthand(r1),
768                       git_reference_shorthand(r2));
769 }
770
771 int
772 writerefs(FILE *fp)
773 {
774         struct commitinfo *ci;
775         const git_oid *id = NULL;
776         git_object *obj = NULL;
777         git_reference *dref = NULL, *r, *ref = NULL;
778         git_reference_iterator *it = NULL;
779         git_reference **refs = NULL;
780         size_t count, i, j, len, refcount = 0;
781         const char *cols[] = { "Branch", "Tag" }; /* first column title */
782         const char *titles[] = { "Branches", "Tags" };
783         const char *ids[] = { "branches", "tags" };
784         const char *name;
785
786         if (git_reference_iterator_new(&it, repo))
787                 return -1;
788
789         for (refcount = 0; !git_reference_next(&ref, it); refcount++) {
790                 if (!(refs = reallocarray(refs, refcount + 1, sizeof(git_reference *))))
791                         err(1, "realloc");
792                 refs[refcount] = ref;
793         }
794         git_reference_iterator_free(it);
795
796         /* sort by type then shorthand name */
797         qsort(refs, refcount, sizeof(git_reference *), refs_cmp);
798
799         for (j = 0; j < 2; j++) {
800                 for (i = 0, count = 0; i < refcount; i++) {
801                         if (!(git_reference_is_branch(refs[i]) && j == 0) &&
802                             !(git_reference_is_tag(refs[i]) && j == 1))
803                                 continue;
804
805                         switch (git_reference_type(refs[i])) {
806                         case GIT_REF_SYMBOLIC:
807                                 if (git_reference_resolve(&dref, refs[i]))
808                                         goto err;
809                                 r = dref;
810                                 break;
811                         case GIT_REF_OID:
812                                 r = refs[i];
813                                 break;
814                         default:
815                                 continue;
816                         }
817                         if (!(id = git_reference_target(r)))
818                                 goto err;
819                         if (git_reference_peel(&obj, r, GIT_OBJ_ANY))
820                                 goto err;
821                         if (!(id = git_object_id(obj)))
822                                 goto err;
823                         if (!(ci = commitinfo_getbyoid(id)))
824                                 break;
825
826                         /* print header if it has an entry (first). */
827                         if (++count == 1) {
828                                 fprintf(fp, "<h2>%s</h2><table id=\"%s\"><thead>\n<tr><td>%s</td>"
829                                       "<td>Age</td><td>Commit message</td>"
830                                       "<td>Author</td><td>Files</td><td class=\"num\">+</td>"
831                                       "<td class=\"num\">-</td></tr>\n</thead><tbody>\n",
832                                       titles[j], ids[j], cols[j]);
833                         }
834
835                         relpath = "";
836                         name = git_reference_shorthand(r);
837
838                         fputs("<tr><td>", fp);
839                         xmlencode(fp, name, strlen(name));
840                         fputs("</td><td>", fp);
841                         if (ci->author)
842                                 printtimeshort(fp, &(ci->author->when));
843                         fputs("</td><td>", fp);
844                         if (ci->summary) {
845                                 if (j)
846                                         fprintf(fp, "<a href=\"%scommit/%s.html\">",
847                                                 relpath, ci->oid);
848                                 if ((len = strlen(ci->summary)) > summarylen) {
849                                         xmlencode(fp, ci->summary, summarylen - 1);
850                                         fputs("…", fp);
851                                 } else {
852                                         xmlencode(fp, ci->summary, len);
853                                 }
854                                 if (j)
855                                         fputs("</a>", fp);
856                         }
857                         fputs("</td><td>", fp);
858                         if (ci->author)
859                                 xmlencode(fp, ci->author->name, strlen(ci->author->name));
860                         fputs("</td><td class=\"num\">", fp);
861                         fprintf(fp, "%zu", ci->filecount);
862                         fputs("</td><td class=\"num\">", fp);
863                         fprintf(fp, "+%zu", ci->addcount);
864                         fputs("</td><td class=\"num\">", fp);
865                         fprintf(fp, "-%zu", ci->delcount);
866                         fputs("</td></tr>\n", fp);
867
868                         relpath = "../";
869
870                         commitinfo_free(ci);
871                         git_object_free(obj);
872                         obj = NULL;
873                         git_reference_free(dref);
874                         dref = NULL;
875                 }
876                 /* table footer */
877                 if (count)
878                         fputs("</tbody></table>", fp);
879         }
880
881 err:
882         git_object_free(obj);
883         git_reference_free(dref);
884
885         for (i = 0; i < refcount; i++)
886                 git_reference_free(refs[i]);
887         free(refs);
888
889         return 0;
890 }
891
892 int
893 main(int argc, char *argv[])
894 {
895         git_object *obj = NULL;
896         const git_error *e = NULL;
897         FILE *fp, *fpread;
898         char path[PATH_MAX], *p;
899         int status;
900
901         if (argc != 2) {
902                 fprintf(stderr, "%s <repodir>\n", argv[0]);
903                 return 1;
904         }
905         repodir = argv[1];
906
907         git_libgit2_init();
908
909         if ((status = git_repository_open_ext(&repo, repodir,
910                 GIT_REPOSITORY_OPEN_NO_SEARCH, NULL)) < 0) {
911                 e = giterr_last();
912                 fprintf(stderr, "error %d/%d: %s\n", status, e->klass, e->message);
913                 return status;
914         }
915
916         /* use directory name as name */
917         p = xbasename(repodir);
918         snprintf(name, sizeof(name), "%s", p);
919         free(p);
920
921         /* read description or .git/description */
922         snprintf(path, sizeof(path), "%s%s%s",
923                 repodir, repodir[strlen(repodir)] == '/' ? "" : "/", "description");
924         if (!(fpread = fopen(path, "r"))) {
925                 snprintf(path, sizeof(path), "%s%s%s",
926                         repodir, repodir[strlen(repodir)] == '/' ? "" : "/", ".git/description");
927                 fpread = fopen(path, "r");
928         }
929         if (fpread) {
930                 if (!fgets(description, sizeof(description), fpread))
931                         description[0] = '\0';
932                 fclose(fpread);
933         }
934
935         /* read url or .git/url */
936         snprintf(path, sizeof(path), "%s%s%s",
937                 repodir, repodir[strlen(repodir)] == '/' ? "" : "/", "url");
938         if (!(fpread = fopen(path, "r"))) {
939                 snprintf(path, sizeof(path), "%s%s%s",
940                         repodir, repodir[strlen(repodir)] == '/' ? "" : "/", ".git/url");
941                 fpread = fopen(path, "r");
942         }
943         if (fpread) {
944                 if (!fgets(cloneurl, sizeof(cloneurl), fpread))
945                         cloneurl[0] = '\0';
946                 cloneurl[strcspn(cloneurl, "\n")] = '\0';
947                 fclose(fpread);
948         }
949
950         /* check LICENSE */
951         haslicense = !git_revparse_single(&obj, repo, "HEAD:LICENSE");
952         git_object_free(obj);
953         /* check README */
954         hasreadme = !git_revparse_single(&obj, repo, "HEAD:README");
955         git_object_free(obj);
956
957         /* log for HEAD */
958         fp = efopen("log.html", "w");
959         relpath = "";
960         writeheader(fp);
961         writelog(fp, "HEAD");
962         writefooter(fp);
963         fclose(fp);
964
965         /* files for HEAD */
966         fp = efopen("files.html", "w");
967         writeheader(fp);
968         writefiles(fp, "HEAD");
969         writefooter(fp);
970         fclose(fp);
971
972         /* summary page with branches and tags */
973         fp = efopen("refs.html", "w");
974         writeheader(fp);
975         writerefs(fp);
976         writefooter(fp);
977         fclose(fp);
978
979         /* Atom feed */
980         fp = efopen("atom.xml", "w");
981         writeatom(fp);
982         fclose(fp);
983
984         /* cleanup */
985         git_repository_free(repo);
986         git_libgit2_shutdown();
987
988         return 0;
989 }