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

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

enum face { EYES, NOSE, MOUTH };

void
l(const char *fmt, ...)
{
  #ifdef ERR
	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);
	}
  #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) {
    git_error_clear();
    return 0;
  } else if (e < GIT_OK) {
    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;
}

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 code = -1;
  char face[] = { ':', 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)) {
      face[EYES] = '8'; /* goggle eyes if we have some stashed changes */
    } else if (git_repository_is_empty(repo)) {
      face[EYES] = 'B'; /* sunglasses if we're in a new repo with no HEAD */
    } else {
      face[EYES] = ';'; /* wink when we're in a git repo */
    }

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

  /* 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: face[MOUTH] = ')'; break; /* all good */
      case 130: face[MOUTH] = 'O'; break; /* Ctrl-c pressed (SIGTERM) */
      case 126: face[MOUTH] = 'P'; break; /* permission denied */
      case 127: face[MOUTH] = '/'; break; /* command not found */
      default: face[MOUTH] = '('; break; /* all other codes (usually the program saying it has failed) */
    }
  } else {
    face[MOUTH] = '|'; /* no code info */
  }

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

  printf("%c%c%c", face[EYES], face[NOSE], face[MOUTH]);
}