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