gitweb.ps.run Git - ps-cgit/blob - 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};
 199         time_t now;
 200         long i;
 201         struct tm *tm;
 202         char tmp[11];
 203
 204         time(&now);
 205         tm = gmtime(&now);
 206         period->trunc(tm);
 207         for (i = 1; i < period->count; i++)
 208                 period->dec(tm);
 209         strftime(tmp, sizeof(tmp), "%Y-%m-%d", tm);
 210         argv[2] = xstrdup(fmt("--since=%s", tmp));
 211         init_revisions(&rev, NULL);
 212         rev.abbrev = DEFAULT_ABBREV;
 213         rev.commit_format = CMIT_FMT_DEFAULT;
 214         rev.no_merges = 1;
 215         rev.verbose_header = 1;
 216         rev.show_root_diff = 0;
 217         setup_revisions(3, argv, &rev, NULL);
 218         prepare_revision_walk(&rev);
 219         memset(&authors, 0, sizeof(authors));
 220         while ((commit = get_revision(&rev)) != NULL) {
 221                 add_commit(&authors, commit, period);
 222                 free(commit->buffer);
 223                 free_commit_list(commit->parents);
 224         }
 225         return authors;
 226 }
 227
 228 void print_combined_authorrow(struct string_list *authors, int from, int to,
 229         const char *name, const char *leftclass, const char *centerclass,
 230         const char *rightclass, struct Period *period)
 231 {
 232         struct string_list_item *author;
 233         struct authorstat *authorstat;
 234         struct string_list *items;
 235         struct string_list_item *date;
 236         time_t now;
 237         long i, j, total, subtotal;
 238         struct tm *tm;
 239         char *tmp;
 240
 241         time(&now);
 242         tm = gmtime(&now);
 243         period->trunc(tm);
 244         for (i = 1; i < period->count; i++)
 245                 period->dec(tm);
 246
 247         total = 0;
 248         htmlf("<tr><td class='%s'>%s</td>", leftclass,
 249                 fmt(name, to - from + 1));
 250         for (j = 0; j < period->count; j++) {
 251                 tmp = period->pretty(tm);
 252                 period->inc(tm);
 253                 subtotal = 0;
 254                 for (i = from; i <= to; i++) {
 255                         author = &authors->items[i];
 256                         authorstat = author->util;
 257                         items = &authorstat->list;
 258                         date = string_list_lookup(tmp, items);
 259                         if (date)
 260                                 subtotal += (size_t)date->util;
 261                 }
 262                 htmlf("<td class='%s'>%d</td>", centerclass, subtotal);
 263                 total += subtotal;
 264         }
 265         htmlf("<td class='%s'>%d</td></tr>", rightclass, total);
 266 }
 267
 268 void print_authors(struct string_list *authors, int top, struct Period *period)
 269 {
 270         struct string_list_item *author;
 271         struct authorstat *authorstat;
 272         struct string_list *items;
 273         struct string_list_item *date;
 274         time_t now;
 275         long i, j, total;
 276         struct tm *tm;
 277         char *tmp;
 278
 279         time(&now);
 280         tm = gmtime(&now);
 281         period->trunc(tm);
 282         for (i = 1; i < period->count; i++)
 283                 period->dec(tm);
 284
 285         html("<table class='stats'><tr><th>Author</th>");
 286         for (j = 0; j < period->count; j++) {
 287                 tmp = period->pretty(tm);
 288                 htmlf("<th>%s</th>", tmp);
 289                 period->inc(tm);
 290         }
 291         html("<th>Total</th></tr>\n");
 292
 293         if (top <= 0 || top > authors->nr)
 294                 top = authors->nr;
 295
 296         for (i = 0; i < top; i++) {
 297                 author = &authors->items[i];
 298                 html("<tr><td class='left'>");
 299                 html_txt(author->string);
 300                 html("</td>");
 301                 authorstat = author->util;
 302                 items = &authorstat->list;
 303                 total = 0;
 304                 for (j = 0; j < period->count; j++)
 305                         period->dec(tm);
 306                 for (j = 0; j < period->count; j++) {
 307                         tmp = period->pretty(tm);
 308                         period->inc(tm);
 309                         date = string_list_lookup(tmp, items);
 310                         if (!date)
 311                                 html("<td>0</td>");
 312                         else {
 313                                 htmlf("<td>%d</td>", date->util);
 314                                 total += (size_t)date->util;
 315                         }
 316                 }
 317                 htmlf("<td class='sum'>%d</td></tr>", total);
 318         }
 319
 320         if (top < authors->nr)
 321                 print_combined_authorrow(authors, top, authors->nr - 1,
 322                         "Others (%d)", "left", "", "sum", period);
 323
 324         print_combined_authorrow(authors, 0, authors->nr - 1, "Total",
 325                 "total", "sum", "sum", period);
 326         html("</table>");
 327 }
 328
 329 /* Create a sorted string_list with one entry per author. The util-field
 330  * for each author is another string_list which is used to calculate the
 331  * number of commits per time-interval.
 332  */
 333 void cgit_show_stats(struct cgit_context *ctx)
 334 {
 335         struct string_list authors;
 336         struct Period *period;
 337         int top, i;
 338
 339         period = &periods[0];
 340         if (ctx->qry.period) {
 341                 for (i = 0; i < sizeof(periods) / sizeof(periods[0]); i++)
 342                         if (periods[i].code == ctx->qry.period[0]) {
 343                                 period = &periods[i];
 344                                 break;
 345                         }
 346         }
 347         authors = collect_stats(ctx, period);
 348         qsort(authors.items, authors.nr, sizeof(struct string_list_item),
 349                 cmp_total_commits);
 350
 351         top = ctx->qry.ofs;
 352         if (!top)
 353                 top = 10;
 354         htmlf("<h2>Commits per author per %s</h2>", period->name);
 355
 356         html("<form method='get' action='.' style='float: right; text-align: right;'>");
 357         if (strcmp(ctx->qry.head, ctx->repo->defbranch))
 358                 htmlf("<input type='hidden' name='h' value='%s'/>", ctx->qry.head);
 359         html("Period: ");
 360         html("<select name='period' onchange='this.form.submit();'>");
 361         for (i = 0; i < sizeof(periods) / sizeof(periods[0]); i++)
 362                 htmlf("<option value='%c'%s>%s</option>",
 363                         periods[i].code,
 364                         period == &periods[i] ? " selected" : "",
 365                         periods[i].name);
 366         html("</select><br/><br/>");
 367         html("Authors: ");
 368         html("");
 369         html("<select name='ofs' onchange='this.form.submit();'>");
 370         htmlf("<option value='10'%s>10</option>", top == 10 ? " selected" : "");
 371         htmlf("<option value='25'%s>25</option>", top == 25 ? " selected" : "");
 372         htmlf("<option value='50'%s>50</option>", top == 50 ? " selected" : "");
 373         htmlf("<option value='100'%s>100</option>", top == 100 ? " selected" : "");
 374         htmlf("<option value='-1'%s>All</option>", top == -1 ? " selected" : "");
 375         html("</select>");
 376         html("<noscript>&nbsp;&nbsp;<input type='submit' value='Reload'/></noscript>");
 377         html("</form>");
 378         print_authors(&authors, top, period);
 379 }
 380