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