1 files changed, 8 insertions, 0 deletions

diff --git a/ui-stats.c b/ui-stats.c
index 9fc06d3..bdaf9cc 100644
--- a/ui-stats.c
+++ b/ui-stats.c
@@ -93,128 +93,136 @@ static void dec_quarter(struct tm *tm)
 static void inc_quarter(struct tm *tm)
 {
         inc_month(tm);
         inc_month(tm);
         inc_month(tm);
 }
 
 static char *pretty_quarter(struct tm *tm)
 {
         return fmt("Q%d %d", tm->tm_mon / 3 + 1, tm->tm_year + 1900);
 }
 
 static void trunc_year(struct tm *tm)
 {
         trunc_month(tm);
         tm->tm_mon = 0;
 }
 
 static void dec_year(struct tm *tm)
 {
         tm->tm_year--;
 }
 
 static void inc_year(struct tm *tm)
 {
         tm->tm_year++;
 }
 
 static char *pretty_year(struct tm *tm)
 {
         return fmt("%d", tm->tm_year + 1900);
 }
 
 struct cgit_period periods[] = {
         {'w', "week", 12, 4, trunc_week, dec_week, inc_week, pretty_week},
         {'m', "month", 12, 4, trunc_month, dec_month, inc_month, pretty_month},
         {'q', "quarter", 12, 4, trunc_quarter, dec_quarter, inc_quarter, pretty_quarter},
         {'y', "year", 12, 4, trunc_year, dec_year, inc_year, pretty_year},
 };
 
 /* Given a period code or name, return a period index (1, 2, 3 or 4)
  * and update the period pointer to the correcsponding struct.
  * If no matching code is found, return 0.
  */
 int cgit_find_stats_period(const char *expr, struct cgit_period **period)
 {
         int i;
         char code = '\0';
 
         if (!expr)
                 return 0;
 
         if (strlen(expr) == 1)
                 code = expr[0];
 
         for (i = 0; i < sizeof(periods) / sizeof(periods[0]); i++)
                 if (periods[i].code == code || !strcmp(periods[i].name, expr)) {
                         if (period)
                                 *period = &periods[i];
                         return i+1;
                 }
         return 0;
 }
 
+const char *cgit_find_stats_periodname(int idx)
+{
+        if (idx > 0 && idx < 4)
+                return periods[idx - 1].name;
+        else
+                return "";
+}
+
 static void add_commit(struct string_list *authors, struct commit *commit,
         struct cgit_period *period)
 {
         struct commitinfo *info;
         struct string_list_item *author, *item;
         struct authorstat *authorstat;
         struct string_list *items;
         char *tmp;
         struct tm *date;
         time_t t;
 
         info = cgit_parse_commit(commit);
         tmp = xstrdup(info->author);
         author = string_list_insert(tmp, authors);
         if (!author->util)
                 author->util = xcalloc(1, sizeof(struct authorstat));
         else
                 free(tmp);
         authorstat = author->util;
         items = &authorstat->list;
         t = info->committer_date;
         date = gmtime(&t);
         period->trunc(date);
         tmp = xstrdup(period->pretty(date));
         item = string_list_insert(tmp, items);
         if (item->util)
                 free(tmp);
         item->util++;
         authorstat->total++;
         cgit_free_commitinfo(info);
 }
 
 static int cmp_total_commits(const void *a1, const void *a2)
 {
         const struct string_list_item *i1 = a1;
         const struct string_list_item *i2 = a2;
         const struct authorstat *auth1 = i1->util;
         const struct authorstat *auth2 = i2->util;
 
         return auth2->total - auth1->total;
 }
 
 /* Walk the commit DAG and collect number of commits per author per
  * timeperiod into a nested string_list collection.
  */
 struct string_list collect_stats(struct cgit_context *ctx,
         struct cgit_period *period)
 {
         struct string_list authors;
         struct rev_info rev;
         struct commit *commit;
         const char *argv[] = {NULL, ctx->qry.head, NULL, NULL, NULL, NULL};
         int argc = 3;
         time_t now;
         long i;
         struct tm *tm;
         char tmp[11];
 
         time(&now);
         tm = gmtime(&now);
         period->trunc(tm);
         for (i = 1; i < period->count; i++)
                 period->dec(tm);
         strftime(tmp, sizeof(tmp), "%Y-%m-%d", tm);