]> gitweb.ps.run Git - ps-cgit/blob - ui-stats.c
ui-refs.c: Refactor print_tag()
[ps-cgit] / ui-stats.c
1 #include "cgit.h"
2 #include "ui-stats.h"
3 #include "html.h"
4 #include "ui-shared.h"
5
6 #ifdef NO_C99_FORMAT
7 #define SZ_FMT "%u"
8 #else
9 #define SZ_FMT "%zu"
10 #endif
11
12 #define MONTHS 6
13
14 struct authorstat {
15         long total;
16         struct string_list list;
17 };
18
19 #define DAY_SECS (60 * 60 * 24)
20 #define WEEK_SECS (DAY_SECS * 7)
21
22 static void trunc_week(struct tm *tm)
23 {
24         time_t t = timegm(tm);
25         t -= ((tm->tm_wday + 6) % 7) * DAY_SECS;
26         gmtime_r(&t, tm);
27 }
28
29 static void dec_week(struct tm *tm)
30 {
31         time_t t = timegm(tm);
32         t -= WEEK_SECS;
33         gmtime_r(&t, tm);
34 }
35
36 static void inc_week(struct tm *tm)
37 {
38         time_t t = timegm(tm);
39         t += WEEK_SECS;
40         gmtime_r(&t, tm);
41 }
42
43 static char *pretty_week(struct tm *tm)
44 {
45         static char buf[10];
46
47         strftime(buf, sizeof(buf), "W%V %G", tm);
48         return buf;
49 }
50
51 static void trunc_month(struct tm *tm)
52 {
53         tm->tm_mday = 1;
54 }
55
56 static void dec_month(struct tm *tm)
57 {
58         tm->tm_mon--;
59         if (tm->tm_mon < 0) {
60                 tm->tm_year--;
61                 tm->tm_mon = 11;
62         }
63 }
64
65 static void inc_month(struct tm *tm)
66 {
67         tm->tm_mon++;
68         if (tm->tm_mon > 11) {
69                 tm->tm_year++;
70                 tm->tm_mon = 0;
71         }
72 }
73
74 static char *pretty_month(struct tm *tm)
75 {
76         static const char *months[] = {
77                 "Jan", "Feb", "Mar", "Apr", "May", "Jun",
78                 "Jul", "Aug", "Sep", "Oct", "Nov", "Dec"
79         };
80         return fmt("%s %d", months[tm->tm_mon], tm->tm_year + 1900);
81 }
82
83 static void trunc_quarter(struct tm *tm)
84 {
85         trunc_month(tm);
86         while (tm->tm_mon % 3 != 0)
87                 dec_month(tm);
88 }
89
90 static void dec_quarter(struct tm *tm)
91 {
92         dec_month(tm);
93         dec_month(tm);
94         dec_month(tm);
95 }
96
97 static void inc_quarter(struct tm *tm)
98 {
99         inc_month(tm);
100         inc_month(tm);
101         inc_month(tm);
102 }
103
104 static char *pretty_quarter(struct tm *tm)
105 {
106         return fmt("Q%d %d", tm->tm_mon / 3 + 1, tm->tm_year + 1900);
107 }
108
109 static void trunc_year(struct tm *tm)
110 {
111         trunc_month(tm);
112         tm->tm_mon = 0;
113 }
114
115 static void dec_year(struct tm *tm)
116 {
117         tm->tm_year--;
118 }
119
120 static void inc_year(struct tm *tm)
121 {
122         tm->tm_year++;
123 }
124
125 static char *pretty_year(struct tm *tm)
126 {
127         return fmt("%d", tm->tm_year + 1900);
128 }
129
130 struct cgit_period periods[] = {
131         {'w', "week", 12, 4, trunc_week, dec_week, inc_week, pretty_week},
132         {'m', "month", 12, 4, trunc_month, dec_month, inc_month, pretty_month},
133         {'q', "quarter", 12, 4, trunc_quarter, dec_quarter, inc_quarter, pretty_quarter},
134         {'y', "year", 12, 4, trunc_year, dec_year, inc_year, pretty_year},
135 };
136
137 /* Given a period code or name, return a period index (1, 2, 3 or 4)
138  * and update the period pointer to the correcsponding struct.
139  * If no matching code is found, return 0.
140  */
141 int cgit_find_stats_period(const char *expr, struct cgit_period **period)
142 {
143         int i;
144         char code = '\0';
145
146         if (!expr)
147                 return 0;
148
149         if (strlen(expr) == 1)
150                 code = expr[0];
151
152         for (i = 0; i < sizeof(periods) / sizeof(periods[0]); i++)
153                 if (periods[i].code == code || !strcmp(periods[i].name, expr)) {
154                         if (period)
155                                 *period = &periods[i];
156                         return i + 1;
157                 }
158         return 0;
159 }
160
161 const char *cgit_find_stats_periodname(int idx)
162 {
163         if (idx > 0 && idx < 4)
164                 return periods[idx - 1].name;
165         else
166                 return "";
167 }
168
169 static void add_commit(struct string_list *authors, struct commit *commit,
170         struct cgit_period *period)
171 {
172         struct commitinfo *info;
173         struct string_list_item *author, *item;
174         struct authorstat *authorstat;
175         struct string_list *items;
176         char *tmp;
177         struct tm *date;
178         time_t t;
179
180         info = cgit_parse_commit(commit);
181         tmp = xstrdup(info->author);
182         author = string_list_insert(authors, tmp);
183         if (!author->util)
184                 author->util = xcalloc(1, sizeof(struct authorstat));
185         else
186                 free(tmp);
187         authorstat = author->util;
188         items = &authorstat->list;
189         t = info->committer_date;
190         date = gmtime(&t);
191         period->trunc(date);
192         tmp = xstrdup(period->pretty(date));
193         item = string_list_insert(items, tmp);
194         if (item->util)
195                 free(tmp);
196         item->util++;
197         authorstat->total++;
198         cgit_free_commitinfo(info);
199 }
200
201 static int cmp_total_commits(const void *a1, const void *a2)
202 {
203         const struct string_list_item *i1 = a1;
204         const struct string_list_item *i2 = a2;
205         const struct authorstat *auth1 = i1->util;
206         const struct authorstat *auth2 = i2->util;
207
208         return auth2->total - auth1->total;
209 }
210
211 /* Walk the commit DAG and collect number of commits per author per
212  * timeperiod into a nested string_list collection.
213  */
214 static struct string_list collect_stats(struct cgit_context *ctx,
215                                         struct cgit_period *period)
216 {
217         struct string_list authors;
218         struct rev_info rev;
219         struct commit *commit;
220         const char *argv[] = {NULL, ctx->qry.head, NULL, NULL, NULL, NULL};
221         int argc = 3;
222         time_t now;
223         long i;
224         struct tm *tm;
225         char tmp[11];
226
227         time(&now);
228         tm = gmtime(&now);
229         period->trunc(tm);
230         for (i = 1; i < period->count; i++)
231                 period->dec(tm);
232         strftime(tmp, sizeof(tmp), "%Y-%m-%d", tm);
233         argv[2] = xstrdup(fmt("--since=%s", tmp));
234         if (ctx->qry.path) {
235                 argv[3] = "--";
236                 argv[4] = ctx->qry.path;
237                 argc += 2;
238         }
239         init_revisions(&rev, NULL);
240         rev.abbrev = DEFAULT_ABBREV;
241         rev.commit_format = CMIT_FMT_DEFAULT;
242         rev.max_parents = 1;
243         rev.verbose_header = 1;
244         rev.show_root_diff = 0;
245         setup_revisions(argc, argv, &rev, NULL);
246         prepare_revision_walk(&rev);
247         memset(&authors, 0, sizeof(authors));
248         while ((commit = get_revision(&rev)) != NULL) {
249                 add_commit(&authors, commit, period);
250                 free(commit->buffer);
251                 free_commit_list(commit->parents);
252         }
253         return authors;
254 }
255
256 static void print_combined_authorrow(struct string_list *authors, int from,
257                                      int to, const char *name,
258                                      const char *leftclass,
259                                      const char *centerclass,
260                                      const char *rightclass,
261                                      struct cgit_period *period)
262 {
263         struct string_list_item *author;
264         struct authorstat *authorstat;
265         struct string_list *items;
266         struct string_list_item *date;
267         time_t now;
268         long i, j, total, subtotal;
269         struct tm *tm;
270         char *tmp;
271
272         time(&now);
273         tm = gmtime(&now);
274         period->trunc(tm);
275         for (i = 1; i < period->count; i++)
276                 period->dec(tm);
277
278         total = 0;
279         htmlf("<tr><td class='%s'>%s</td>", leftclass,
280                 fmt(name, to - from + 1));
281         for (j = 0; j < period->count; j++) {
282                 tmp = period->pretty(tm);
283                 period->inc(tm);
284                 subtotal = 0;
285                 for (i = from; i <= to; i++) {
286                         author = &authors->items[i];
287                         authorstat = author->util;
288                         items = &authorstat->list;
289                         date = string_list_lookup(items, tmp);
290                         if (date)
291                                 subtotal += (size_t)date->util;
292                 }
293                 htmlf("<td class='%s'>%ld</td>", centerclass, subtotal);
294                 total += subtotal;
295         }
296         htmlf("<td class='%s'>%ld</td></tr>", rightclass, total);
297 }
298
299 static void print_authors(struct string_list *authors, int top,
300                           struct cgit_period *period)
301 {
302         struct string_list_item *author;
303         struct authorstat *authorstat;
304         struct string_list *items;
305         struct string_list_item *date;
306         time_t now;
307         long i, j, total;
308         struct tm *tm;
309         char *tmp;
310
311         time(&now);
312         tm = gmtime(&now);
313         period->trunc(tm);
314         for (i = 1; i < period->count; i++)
315                 period->dec(tm);
316
317         html("<table class='stats'><tr><th>Author</th>");
318         for (j = 0; j < period->count; j++) {
319                 tmp = period->pretty(tm);
320                 htmlf("<th>%s</th>", tmp);
321                 period->inc(tm);
322         }
323         html("<th>Total</th></tr>\n");
324
325         if (top <= 0 || top > authors->nr)
326                 top = authors->nr;
327
328         for (i = 0; i < top; i++) {
329                 author = &authors->items[i];
330                 html("<tr><td class='left'>");
331                 html_txt(author->string);
332                 html("</td>");
333                 authorstat = author->util;
334                 items = &authorstat->list;
335                 total = 0;
336                 for (j = 0; j < period->count; j++)
337                         period->dec(tm);
338                 for (j = 0; j < period->count; j++) {
339                         tmp = period->pretty(tm);
340                         period->inc(tm);
341                         date = string_list_lookup(items, tmp);
342                         if (!date)
343                                 html("<td>0</td>");
344                         else {
345                                 htmlf("<td>"SZ_FMT"</td>", (size_t)date->util);
346                                 total += (size_t)date->util;
347                         }
348                 }
349                 htmlf("<td class='sum'>%ld</td></tr>", total);
350         }
351
352         if (top < authors->nr)
353                 print_combined_authorrow(authors, top, authors->nr - 1,
354                         "Others (%ld)", "left", "", "sum", period);
355
356         print_combined_authorrow(authors, 0, authors->nr - 1, "Total",
357                 "total", "sum", "sum", period);
358         html("</table>");
359 }
360
361 /* Create a sorted string_list with one entry per author. The util-field
362  * for each author is another string_list which is used to calculate the
363  * number of commits per time-interval.
364  */
365 void cgit_show_stats(struct cgit_context *ctx)
366 {
367         struct string_list authors;
368         struct cgit_period *period;
369         int top, i;
370         const char *code = "w";
371
372         if (ctx->qry.period)
373                 code = ctx->qry.period;
374
375         i = cgit_find_stats_period(code, &period);
376         if (!i) {
377                 cgit_print_error(fmt("Unknown statistics type: %c", code[0]));
378                 return;
379         }
380         if (i > ctx->repo->max_stats) {
381                 cgit_print_error(fmt("Statistics type disabled: %s",
382                                      period->name));
383                 return;
384         }
385         authors = collect_stats(ctx, period);
386         qsort(authors.items, authors.nr, sizeof(struct string_list_item),
387                 cmp_total_commits);
388
389         top = ctx->qry.ofs;
390         if (!top)
391                 top = 10;
392
393         html("<div class='cgit-panel'>");
394         html("<b>stat options</b>");
395         html("<form method='get' action=''>");
396         cgit_add_hidden_formfields(1, 0, "stats");
397         html("<table><tr><td colspan='2'/></tr>");
398         if (ctx->repo->max_stats > 1) {
399                 html("<tr><td class='label'>Period:</td>");
400                 html("<td class='ctrl'><select name='period' onchange='this.form.submit();'>");
401                 for (i = 0; i < ctx->repo->max_stats; i++)
402                         html_option(fmt("%c", periods[i].code),
403                                     periods[i].name, fmt("%c", period->code));
404                 html("</select></td></tr>");
405         }
406         html("<tr><td class='label'>Authors:</td>");
407         html("<td class='ctrl'><select name='ofs' onchange='this.form.submit();'>");
408         html_intoption(10, "10", top);
409         html_intoption(25, "25", top);
410         html_intoption(50, "50", top);
411         html_intoption(100, "100", top);
412         html_intoption(-1, "all", top);
413         html("</select></td></tr>");
414         html("<tr><td/><td class='ctrl'>");
415         html("<noscript><input type='submit' value='Reload'/></noscript>");
416         html("</td></tr></table>");
417         html("</form>");
418         html("</div>");
419         htmlf("<h2>Commits per author per %s", period->name);
420         if (ctx->qry.path) {
421                 html(" (path '");
422                 html_txt(ctx->qry.path);
423                 html("')");
424         }
425         html("</h2>");
426         print_authors(&authors, top, period);
427 }
428