src/repo.c

   1 /* Copyright (c) 2006-2014 Jonas Fonseca <jonas.fonseca@gmail.com>
   2  *
   3  * This program is free software; you can redistribute it and/or
   4  * modify it under the terms of the GNU General Public License as
   5  * published by the Free Software Foundation; either version 2 of
   6  * the License, or (at your option) any later version.
   7  *
   8  * This program is distributed in the hope that it will be useful,
   9  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  10  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  11  * GNU General Public License for more details.
  12  */
  13
  14 #include "tig/tig.h"
  15 #include "tig/repo.h"
  16 #include "tig/io.h"
  17 #include "tig/refdb.h"
  18 #include "tig/git.h"
  19
  20 #define REPO_INFO_GIT_DIR       "--git-dir"
  21 #define REPO_INFO_WORK_TREE     "--is-inside-work-tree"
  22 #define REPO_INFO_SHOW_CDUP     "--show-cdup"
  23 #define REPO_INFO_SHOW_PREFIX   "--show-prefix"
  24 #define REPO_INFO_SYMBOLIC_HEAD "--symbolic-full-name"
  25 #define REPO_INFO_RESOLVED_HEAD "HEAD"
  26
  27 struct repo_info_state {
  28         const char **argv;
  29 };
  30
  31 static int
  32 read_repo_info(char *name, size_t namelen, char *value, size_t valuelen, void *data)
  33 {
  34         struct repo_info_state *state = data;
  35         const char *arg = *state->argv ? *state->argv++ : "";
  36
  37         if (!strcmp(arg, REPO_INFO_GIT_DIR)) {
  38                 string_ncopy(repo.git_dir, name, namelen);
  39
  40         } else if (!strcmp(arg, REPO_INFO_WORK_TREE)) {
  41                 /* This can be 3 different values depending on the
  42                  * version of git being used. If git-rev-parse does not
  43                  * understand --is-inside-work-tree it will simply echo
  44                  * the option else either "true" or "false" is printed.
  45                  * Default to true for the unknown case. */
  46                 repo.is_inside_work_tree = strcmp(name, "false") ? TRUE : FALSE;
  47
  48         } else if (!strcmp(arg, REPO_INFO_SHOW_CDUP)) {
  49                 string_ncopy(repo.cdup, name, namelen);
  50
  51         } else if (!strcmp(arg, REPO_INFO_SHOW_PREFIX)) {
  52                 /* Some versions of Git does not emit anything for --show-prefix
  53                  * when the user is in the repository root directory. Try to detect
  54                  * this special case by looking at the emitted value. If it looks
  55                  * like a commit ID and there's no cdup path assume that no value
  56                  * was emitted. */
  57                 if (!*repo.cdup && namelen == 40 && iscommit(name))
  58                         return read_repo_info(name, namelen, value, valuelen, data);
  59
  60                 string_ncopy(repo.prefix, name, namelen);
  61
  62         } else if (!strcmp(arg, REPO_INFO_RESOLVED_HEAD)) {
  63                 string_ncopy(repo.head_id, name, namelen);
  64
  65         } else if (!strcmp(arg, REPO_INFO_SYMBOLIC_HEAD)) {
  66                 if (!prefixcmp(name, "refs/heads/")) {
  67                         add_ref(repo.head_id, name, repo.remote, repo.head);
  68                         name += STRING_SIZE("refs/heads/");
  69                 }
  70                 string_ncopy(repo.head, name, strlen(name) + 1);
  71                 state->argv++;
  72         }
  73
  74         return OK;
  75 }
  76
  77 static int
  78 reload_repo_info(const char **rev_parse_argv)
  79 {
  80         struct repo_info_state state = { rev_parse_argv + 2 };
  81
  82         return io_run_load(rev_parse_argv, "=", read_repo_info, &state);
  83 }
  84
  85 int
  86 load_repo_info(void)
  87 {
  88         const char *rev_parse_argv[] = {
  89                 "git", "rev-parse", REPO_INFO_GIT_DIR, REPO_INFO_WORK_TREE,
  90                         REPO_INFO_SHOW_CDUP, REPO_INFO_SHOW_PREFIX, \
  91                         REPO_INFO_RESOLVED_HEAD, REPO_INFO_SYMBOLIC_HEAD, "HEAD",
  92                         NULL
  93         };
  94
  95         return reload_repo_info(rev_parse_argv);
  96 }
  97
  98 int
  99 load_repo_head(void)
 100 {
 101         const char *rev_parse_argv[] = {
 102                 "git", "rev-parse", REPO_INFO_RESOLVED_HEAD,
 103                         REPO_INFO_SYMBOLIC_HEAD, "HEAD", NULL
 104         };
 105
 106         return reload_repo_info(rev_parse_argv);
 107 }
 108
 109 struct repo_info repo;
 110
 111 /*
 112  * Git index utils.
 113  */
 114
 115 bool
 116 update_index(void)
 117 {
 118         const char *update_index_argv[] = {
 119                 "git", "update-index", "-q", "--unmerged", "--refresh", NULL
 120         };
 121
 122         return io_run_bg(update_index_argv);
 123 }
 124
 125 bool
 126 index_diff(struct index_diff *diff, bool untracked, bool count_all)
 127 {
 128         const char *untracked_arg = !untracked ? "--untracked-files=no" :
 129                                      count_all ? "--untracked-files=all" :
 130                                                  "--untracked-files=normal";
 131         const char *status_argv[] = {
 132                 "git", "status", "--porcelain", "-z", untracked_arg, NULL
 133         };
 134         struct io io;
 135         struct buffer buf;
 136         bool ok = TRUE;
 137
 138         memset(diff, 0, sizeof(*diff));
 139
 140         if (!io_run(&io, IO_RD, repo.cdup, NULL, status_argv))
 141                 return FALSE;
 142
 143         while (io_get(&io, &buf, 0, TRUE) && (ok = buf.size > 3)) {
 144                 if (buf.data[0] == '?')
 145                         diff->untracked++;
 146                 /* Ignore staged but unmerged entries. */
 147                 else if (buf.data[0] != ' ' && buf.data[0] != 'U')
 148                         diff->staged++;
 149                 if (buf.data[1] != ' ')
 150                         diff->unstaged++;
 151                 if (!count_all && diff->staged && diff->unstaged &&
 152                     (!untracked || diff->untracked))
 153                         break;
 154         }
 155
 156         if (io_error(&io))
 157                 ok = FALSE;
 158
 159         io_done(&io);
 160         return ok;
 161 }
 162
 163 /* vim: set ts=8 sw=8 noexpandtab: */