commit-heatmap-gen

main.c (6330B)

---

 #define _DEFAULT_SOURCE
 
 #include <dirent.h>
 #include <errno.h>
 #include <git2.h>
 #include <git2/commit.h>
 #include <git2/repository.h>
 #include <git2/signature.h>
 #include <stdio.h>
 #include <string.h>
 #include <time.h>
 
 #define DAYS_PER_YEAR 365
 #define HOURS_PER_DAY 24
 #define MINUTES_PER_HOUR 60
 #define SECONDS_PER_MINUTE 60
 
 char *bubble_variant(int count) {
   if (count > 13) {
     return "-heavy";
   }
 
   if (count > 8) {
     return "-medium";
   }
 
   if (count > 4) {
     return "-light";
   }
 
   if (count > 0) {
     return "-lightest";
   }
 
   return "";
 }
 
 typedef struct {
   char date[11]; // "YYYY-MM-DD\0"
   int count;
 } commits_by_date;
 
 int main(int argc, char *argv[]) {
   int error;
 
   if (argc == 1) {
     fprintf(stderr, "err: author email required\n");
     exit(1);
   }
   char *author_email = argv[1];
 
   if (argc == 2) {
     fprintf(stderr, "err: abs repo path required");
     exit(1);
   }
   char *repos_dir = argv[2];
 
   commits_by_date commit_tracker[DAYS_PER_YEAR];
   int unique_dates = 0;
 
   const time_t now = time(NULL);
   const time_t year_behind = now - (DAYS_PER_YEAR * HOURS_PER_DAY *
                                     MINUTES_PER_HOUR * SECONDS_PER_MINUTE);
 
   git_libgit2_init();
 
   DIR *dirp = opendir(repos_dir);
   if (dirp == NULL) {
     fprintf(stderr, "err: failed to open directory: %s\n", strerror(errno));
     exit(1);
   }
 
   for (;;) {
     errno = 0;
 
     struct dirent *dp;
     dp = readdir(dirp);
     if (dp == NULL) {
       if (errno == 0) {
         break;
       }
       fprintf(stderr, "err: failed to iterate directory: %s\n",
               strerror(errno));
       exit(1);
     }
 
     if (dp->d_type != DT_DIR) {
       continue;
     }
 
     if (strcmp(dp->d_name, ".") == 0 || strcmp(dp->d_name, "..") == 0) {
       continue;
     }
 
     char path[1024];
     int path_len = snprintf(path, sizeof(path), "%s/%s", repos_dir, dp->d_name);
 
     if (path_len < 0 || (size_t)path_len >= sizeof(path)) {
       fprintf(stderr, "err: path too long\n");
       exit(1);
     }
 
     git_repository *repo = NULL;
     if ((error = git_repository_open_ext(
              &repo, path, GIT_REPOSITORY_OPEN_NO_SEARCH, NULL)) < 0) {
       if (error == GIT_ENOTFOUND) {
         continue;
       }
       const git_error *e = git_error_last();
       fprintf(stderr, "err: %d/%d: %s\n", error, e->klass, e->message);
       exit(error);
     }
 
     git_object *obj = NULL;
     if ((error = git_revparse_single(&obj, repo, "HEAD")) < 0) {
       const git_error *e = git_error_last();
       fprintf(stderr, "err: %d/%d: %s\n", error, e->klass, e->message);
       exit(error);
     }
 
     const git_oid *head = NULL;
     head = git_object_id(obj);
     git_object_free(obj);
 
     git_revwalk *walker = NULL;
     if ((error = git_revwalk_new(&walker, repo)) < 0) {
       const git_error *e = git_error_last();
       fprintf(stderr, "err: %d/%d: %s\n", error, e->klass, e->message);
       exit(error);
     }
 
     if ((error = git_revwalk_push(walker, head)) < 0) {
       const git_error *e = git_error_last();
       fprintf(stderr, "err: %d/%d: %s\n", error, e->klass, e->message);
       exit(error);
     }
 
     git_oid id;
     while (!git_revwalk_next(&id, walker)) {
       git_commit *commit;
       if ((error = git_commit_lookup(&commit, repo, &id)) < 0) {
         const git_error *e = git_error_last();
         fprintf(stderr, "err: %d/%d: %s\n", error, e->klass, e->message);
         exit(error);
       }
 
       const git_signature *author = git_commit_author(commit);
 
       if (strcmp(author_email, author->email) != 0) {
         continue;
       }
 
       if (author->when.time < year_behind) {
         break;
       }
 
       struct tm *lt;
       char buf[11];
 
       lt = localtime(&author->when.time);
       if (strftime(buf, sizeof(buf), "%Y-%m-%d", lt) == 0) {
         fprintf(stderr, "err: failed to get print time\n");
         exit(1);
       }
 
       int found = 0;
       for (int j = 0; j < unique_dates; j++) {
         if (strcmp(commit_tracker[j].date, buf) == 0) {
           commit_tracker[j].count++;
           found = 1;
           break;
         }
       }
 
       if (found == 0) {
         if (unique_dates < DAYS_PER_YEAR) {
           strcpy(commit_tracker[unique_dates].date, buf);
           commit_tracker[unique_dates].count = 1;
           unique_dates++;
         }
       }
     }
   }
 
   if (dirp != NULL) {
     closedir(dirp);
   }
 
   printf("const template = document.createElement(\"template\");"
          "template.innerHTML = `<style>"
          ".heatmap-grid {"
          "display: grid;"
          "grid-template-rows: repeat(7, minmax(0, 1fr));"
          "grid-auto-flow: column;"
          "gap: 0.25rem;"
          "}"
          ".heatmap-bubble {"
          "width: 1rem;"
          "height: 1rem;"
          "border-radius: 0.25rem;"
          "background-color: #eff2f5;"
          "&.-lightest { background-color: #aceebb; }"
          "&.-light { background-color: #4ac26b; }"
          "&.- medium { background-color: #2da44e; }"
          "&.-heavy { background-color: #116329; }"
          "}"
          "</style>");
 
   printf("<div class=\"heatmap-grid\">");
 
   time_t day = year_behind;
   while (day < now) {
     struct tm *lt;
     char buf[11];
 
     lt = localtime(&day);
     if (strftime(buf, sizeof(buf), "%Y-%m-%d", lt) == 0) {
       fprintf(stderr, "err: failed to get print time\n");
       exit(1);
     }
 
     int found = -1;
     for (int j = 0; j < unique_dates; j++) {
       if (strcmp(commit_tracker[j].date, buf) == 0) {
         found = j;
         break;
       }
     }
 
     if (found == -1) {
       printf("<div class=\"heatmap-bubble %s\" title=\"0 sessions on "
              "%s\"></div>",
              bubble_variant(0), buf);
     } else {
       int commits = commit_tracker[found].count;
       printf("<div class=\"heatmap-bubble %s\" title=\"%d sessions on "
              "%s\"></div>",
              bubble_variant(commits), commits, buf);
     }
 
     day += (24 * 60 * 60);
   }
 
   printf("</div>`;");
 
   printf("class CommitHeatmap extends HTMLElement {"
          "constructor() {"
          "super();"
          "const shadowRoot = this.attachShadow({ mode: \"closed\" });"
          "shadowRoot.appendChild(document.importNode(template.content, true));"
          "}}"
          "customElements.define(\"commit-heatmap\", CommitHeatmap);");
 
   git_libgit2_shutdown();
 
   return 0;
 }