XD/XD.c

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#ifdef ERR
#include <stdarg.h>
#endif

#ifdef GIT
#include <git2.h>
#endif

#ifdef ERR
void
l(const char *fmt, ...)
{
  va_list ap;

  va_start(ap, fmt);
  vfprintf(stderr, fmt, ap);
  va_end(ap);

  if (fmt[0] && fmt[strlen(fmt) - 1] == ':') {
    fputc(' ', stderr);
    perror(NULL);
  } else {
    fputc('\n', stderr);
  }
}
#define L(...) l(__VA_ARGS__)
#else
#define L(...)
#endif

#ifdef GIT
/**
 * @brief open git repo if one is available at the current path
 *
 * @return a pointer to the git repo object
 */
git_repository
*init_git()
{
  git_buf buf = GIT_BUF_INIT_CONST(NULL, 0);
  git_repository *repo;

  if (git_libgit2_init() < 0) {
    L("Failed to initalize libgit2, proceeding without git functionality enabled.");
    return NULL;
  }

  if (git_repository_discover(&buf, ".", 0, NULL) < 0) {
    L("Failed to discover git repo: %s", git_error_last()->message);
    return NULL;
  }

  if (git_repository_open(&repo, buf.ptr) < 0) {
    L("Failed to open git repo: %s", git_error_last()->message);
    git_buf_dispose(&buf);
    return NULL;
  }

  /* get rid of object containing git repo path and return the repo */
  git_buf_dispose(&buf);
  return repo;
}

/**
 * @brief check for any existing stashes in the provided git repo
 *
 * @param repo git repo to check for existing stashes
 * @return 1 if stashes found 0 otherwise
 */
int
has_stashes(git_repository *repo)
{
	git_reference *stash = NULL;
  int e;

  e = git_reference_lookup(&stash, repo, "refs/stash");
  if (e == GIT_ENOTFOUND) {
    return 0;
  } else if (e < 0) {
    L("Error looking up stash reference: %s", git_error_last()->message);
    return 0;
  } else {
    e = 1;
  }

  git_reference_free(stash);
  return e;
}

/**
 * @brief check for any untracked changes in the current git repo
 *
 * @param repo git repository object
 * @return 1 if any untracked changes have been found 0 otherwise
 */
int
has_untracked(git_repository *repo)
{
  git_status_options opts = GIT_STATUS_OPTIONS_INIT;
  git_status_list *list = NULL;
  int r = 0;

  opts.show = GIT_STATUS_SHOW_INDEX_AND_WORKDIR;
  opts.flags = GIT_STATUS_OPT_INCLUDE_UNTRACKED |
    GIT_STATUS_OPT_RENAMES_HEAD_TO_INDEX;

  if (git_status_list_new(&list, repo, &opts) < 0) {
    L("Error checking for untracked changes: %s", git_error_last()->message);
    return 0;
  }

  /* if any changes are found return 1 */
  if (git_status_list_entrycount(list) > 0) {
    r = 1;
  }

  git_status_list_free(list);
  return r;
}

/**
 * @brief check for staged changes
 *
 * @param repo git repository object
 * @return 1 if any staged changes found 0 otherwise
 */
int
has_staged(git_repository *repo)
{
  git_status_entry entry;
  git_status_list *list = NULL;
  int i, c, r = 0;

  if (git_status_list_new(&list, repo, NULL) < 0) {
    L("Error checking for staged changes: %s", git_error_last()->message);
    return 0;
  }

  /* if any staged changes are found return 1 */
  if ((c = git_status_list_entrycount(list)) > 0) {
    for (i = 0; i < c; i++) {
      entry = *git_status_byindex(list, i);

      if (entry.status & GIT_STATUS_INDEX_NEW
        || entry.status & GIT_STATUS_INDEX_DELETED
        || entry.status & GIT_STATUS_INDEX_MODIFIED) {
        r = 1;
        break;
      }
    }
  }

  git_status_list_free(list);
  return r;
}
#endif

int
main(int argc, char *argv[])
{
  int i, code = -1;

  /* print version information */
  for (i = 1; i < argc; i++) {
    if (strcmp(argv[i], "-v") == 0) {
      printf("XD v%s\n", VERSION);
      return 0;
    }
  }

  #ifdef GIT
  git_repository *repo;
  int git_ok = 0;

  if ((repo = init_git())) {
    git_ok = 1;
  }

  if (git_ok) {
    /* change the eyes depending on the current git repo's status */
    if (has_stashes(repo)) {
      printf("8"); /* goggle eyes if we have some stashed changes */
    } else if (git_repository_is_empty(repo)) {
      printf("B"); /* sunglasses if we're in a new repo with no HEAD */
    } else {
      printf(";"); /* wink when we're in a git repo */
    }

    /* change the nose depending on the current git repo's status */
    if (has_staged(repo)) {
      printf("*"); /* change to broken nose for staged changes */
    } else if (has_untracked(repo)) {
      printf("^"); /* add a little nose when there are untracked changes in the repo */
    } else if (git_repository_head_detached(repo)) {
      printf("-"); /* add a minus nose when the HEAD is detached */
    }
  } else
  #endif
  if (1) {
    printf(":");
  }

  /* get exit code from user args */
  if (argv[1]) {
    code = atoi(argv[1]);
  }

  /* change mouth based on exit code */
  if (code >= 0) {
    switch (code) {
      case 0: printf(")"); break; /* all good */
      case 130: printf("O"); break; /* Ctrl-c pressed (SIGTERM) */
      case 126: printf("P"); break; /* permission denied */
      case 127: printf("/"); break; /* command not found */
      default: printf("("); break; /* all other codes (usually the program saying it has failed) */
    }
  } else {
    printf("|"); /* no code info */
  }

  #ifdef GIT
  git_repository_free(repo);
  git_libgit2_shutdown();
  #endif
}