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("Unknown statistics type: %c", code[0]);
 378                 return;
 379         }
 380         if (i > ctx->repo->max_stats) {
 381                 cgit_print_error("Statistics type disabled: %s", period->name);
 382                 return;
 383         }
 384         authors = collect_stats(ctx, period);
 385         qsort(authors.items, authors.nr, sizeof(struct string_list_item),
 386                 cmp_total_commits);
 387
 388         top = ctx->qry.ofs;
 389         if (!top)
 390                 top = 10;
 391
 392         html("<div class='cgit-panel'>");
 393         html("<b>stat options</b>");
 394         html("<form method='get' action=''>");
 395         cgit_add_hidden_formfields(1, 0, "stats");
 396         html("<table><tr><td colspan='2'/></tr>");
 397         if (ctx->repo->max_stats > 1) {
 398                 html("<tr><td class='label'>Period:</td>");
 399                 html("<td class='ctrl'><select name='period' onchange='this.form.submit();'>");
 400                 for (i = 0; i < ctx->repo->max_stats; i++)
 401                         html_option(fmt("%c", periods[i].code),
 402                                     periods[i].name, fmt("%c", period->code));
 403                 html("</select></td></tr>");
 404         }
 405         html("<tr><td class='label'>Authors:</td>");
 406         html("<td class='ctrl'><select name='ofs' onchange='this.form.submit();'>");
 407         html_intoption(10, "10", top);
 408         html_intoption(25, "25", top);
 409         html_intoption(50, "50", top);
 410         html_intoption(100, "100", top);
 411         html_intoption(-1, "all", top);
 412         html("</select></td></tr>");
 413         html("<tr><td/><td class='ctrl'>");
 414         html("<noscript><input type='submit' value='Reload'/></noscript>");
 415         html("</td></tr></table>");
 416         html("</form>");
 417         html("</div>");
 418         htmlf("<h2>Commits per author per %s", period->name);
 419         if (ctx->qry.path) {
 420                 html(" (path '");
 421                 html_txt(ctx->qry.path);
 422                 html("')");
 423         }
 424         html("</h2>");
 425         print_authors(&authors, top, period);
 426 }
 427