mirror of
https://github.com/jarun/nnn.git
synced 2024-08-23 07:41:51 +00:00
241 lines
6.9 KiB
Diff
241 lines
6.9 KiB
Diff
# Description: Add git status column to detail mode. Provides additional
|
|
# command line flag -G which will render the git status
|
|
# column also in normal mode. Vim plugin users may consider
|
|
# adding the -G flag to their command override.
|
|
#
|
|
# Authors: Luuk van Baal
|
|
|
|
diff --git a/src/nnn.c b/src/nnn.c
|
|
index 1028906a..c80314de 100644
|
|
--- a/src/nnn.c
|
|
+++ b/src/nnn.c
|
|
@@ -262,6 +262,25 @@
|
|
#define FREE 0
|
|
#define CAPACITY 1
|
|
|
|
+/* Git icons */
|
|
+#ifdef NERD
|
|
+#define GIT_ADD ""
|
|
+#define GIT_DEL ""
|
|
+#define GIT_IGN ""
|
|
+#define GIT_MOD ""
|
|
+#define GIT_NEW ""
|
|
+#define GIT_NON "-"
|
|
+#define GIT_UPD "ﮮ"
|
|
+#else
|
|
+#define GIT_ADD "A"
|
|
+#define GIT_DEL "D"
|
|
+#define GIT_IGN "!"
|
|
+#define GIT_MOD "M"
|
|
+#define GIT_NEW "?"
|
|
+#define GIT_NON "-"
|
|
+#define GIT_UPD "U"
|
|
+#endif
|
|
+
|
|
/* TYPE DEFINITIONS */
|
|
typedef unsigned int uint_t;
|
|
typedef unsigned char uchar_t;
|
|
@@ -286,6 +305,7 @@ typedef struct entry {
|
|
uid_t uid; /* 4 bytes */
|
|
gid_t gid; /* 4 bytes */
|
|
#endif
|
|
+ char git_status[2][5];
|
|
} *pEntry;
|
|
|
|
/* Selection marker */
|
|
@@ -342,6 +362,7 @@ typedef struct {
|
|
uint_t cliopener : 1; /* All-CLI app opener */
|
|
uint_t waitedit : 1; /* For ops that can't be detached, used EDITOR */
|
|
uint_t rollover : 1; /* Roll over at edges */
|
|
+ uint_t normalgit : 1; /* Show git status in normal mode */
|
|
} settings;
|
|
|
|
/* Non-persistent program-internal states (alphabeical order) */
|
|
@@ -391,7 +412,17 @@ typedef struct {
|
|
} session_header_t;
|
|
#endif
|
|
|
|
+typedef struct {
|
|
+ char status[2];
|
|
+ char path[PATH_MAX];
|
|
+} git_status_t;
|
|
+
|
|
/* GLOBALS */
|
|
+struct {
|
|
+ bool show;
|
|
+ size_t len;
|
|
+ git_status_t *statuses;
|
|
+} git_statuses;
|
|
|
|
/* Configuration, contexts */
|
|
static settings cfg = {
|
|
@@ -422,6 +453,7 @@ static settings cfg = {
|
|
0, /* cliopener */
|
|
0, /* waitedit */
|
|
1, /* rollover */
|
|
+ 0, /* normalgit */
|
|
};
|
|
|
|
static context g_ctx[CTX_MAX] __attribute__ ((aligned));
|
|
@@ -3839,6 +3871,56 @@ static int get_kv_key(kv *kvarr, char *val, uchar_t max, uchar_t id)
|
|
return -1;
|
|
}
|
|
|
|
+static size_t get_git_statuses(const char *path)
|
|
+{
|
|
+ static char gitrev[] = "git rev-parse --show-toplevel 2>/dev/null";
|
|
+ char workdir[PATH_MAX], *ret;
|
|
+ FILE *fp = popen(gitrev, "r");
|
|
+
|
|
+ git_statuses.show = FALSE;
|
|
+ ret = fgets(workdir, PATH_MAX, fp);
|
|
+ pclose(fp);
|
|
+ if (!ret)
|
|
+ return 0;
|
|
+
|
|
+ static char gitstat[] = "git -c core.quotePath= status --porcelain --no-renames --ignored=matching -u ";
|
|
+ char pathspec[PATH_MAX], status[PATH_MAX];
|
|
+ size_t i = -1;
|
|
+ workdir[xstrlen(workdir) - 1] = '\0';
|
|
+ snprintf(pathspec, PATH_MAX, "%s\"%s\"%s 2>/dev/null", gitstat, path, cfg.showhidden ? "" : "/*");
|
|
+ fp = popen(pathspec, "r");
|
|
+
|
|
+ while (fgets(status, PATH_MAX, fp)) {
|
|
+ size_t pathindex = (status[3] == '"') ? 4 : 3;
|
|
+ status[xstrlen(status) - pathindex + 2] = '\0';
|
|
+ git_statuses.statuses = xrealloc(git_statuses.statuses, sizeof(git_status_t) * (++i + 1));
|
|
+ git_statuses.statuses[i].status[0] = status[0];
|
|
+ git_statuses.statuses[i].status[1] = status[1];
|
|
+ mkpath(workdir, status + pathindex, git_statuses.statuses[i].path);
|
|
+ }
|
|
+
|
|
+ pclose(fp);
|
|
+ return (i + 1);
|
|
+}
|
|
+
|
|
+static void set_git_status(char status[][5], uint_t nr)
|
|
+{
|
|
+ for (int j = 0; j < 2; j++) {
|
|
+ if (status[j][0] == '-')
|
|
+ switch (git_statuses.statuses[nr].status[j]) {
|
|
+ case ' ': xstrsncpy(status[j], GIT_NON, 4); break;
|
|
+ case 'M': xstrsncpy(status[j], GIT_MOD, 4); break;
|
|
+ case 'A': xstrsncpy(status[j], GIT_ADD, 4); break;
|
|
+ case '?': xstrsncpy(status[j], GIT_NEW, 4); break;
|
|
+ case '!': xstrsncpy(status[j], GIT_IGN, 4); break;
|
|
+ case 'D': xstrsncpy(status[j], GIT_DEL, 4); break;
|
|
+ case 'U': xstrsncpy(status[j], GIT_UPD, 4); break;
|
|
+ }
|
|
+ }
|
|
+ if (git_statuses.statuses[nr].status[1] != '!')
|
|
+ git_statuses.show = TRUE;
|
|
+}
|
|
+
|
|
static void resetdircolor(int flags)
|
|
{
|
|
/* Directories are always shown on top, clear the color when moving to first file */
|
|
@@ -4176,6 +4258,10 @@ static void printent(const struct entry *ent, uint_t namecols, bool sel)
|
|
|
|
uchar_t color_pair = get_color_pair_name_ind(ent, &ind, &attrs);
|
|
|
|
+ if (git_statuses.show && (cfg.showdetail || cfg.normalgit))
|
|
+ printw("%*s%s%s", (cfg.normalgit && !cfg.showdetail) ? 1 : 0, "",
|
|
+ ent->git_status[0], ent->git_status[1]);
|
|
+
|
|
addch((ent->flags & FILE_SELECTED) ? '+' | A_REVERSE | A_BOLD : ' ');
|
|
|
|
if (g_state.oldcolor)
|
|
@@ -5617,6 +5703,11 @@ static int dentfill(char *path, struct entry **ppdents)
|
|
attron(COLOR_PAIR(cfg.curctx + 1));
|
|
}
|
|
|
|
+ char linkpath[PATH_MAX];
|
|
+ if ((git_statuses.len = get_git_statuses(path)))
|
|
+ if (!realpath(path, linkpath))
|
|
+ printwarn(NULL);
|
|
+
|
|
#if _POSIX_C_SOURCE >= 200112L
|
|
posix_fadvise(fd, 0, 0, POSIX_FADV_SEQUENTIAL);
|
|
#endif
|
|
@@ -5815,6 +5906,29 @@ static int dentfill(char *path, struct entry **ppdents)
|
|
#endif
|
|
}
|
|
|
|
+ if (git_statuses.len) {
|
|
+ char dentpath[PATH_MAX];
|
|
+ size_t pathlen = mkpath(linkpath, dentp->name, dentpath);
|
|
+ dentp->git_status[0][0] = dentp->git_status[1][0] = '-';
|
|
+ dentp->git_status[0][1] = dentp->git_status[1][1] = '\0';
|
|
+
|
|
+ if (dentp->flags & DIR_OR_DIRLNK) {
|
|
+ char prefix[PATH_MAX];
|
|
+ memccpy(prefix, dentpath, '\0', PATH_MAX);
|
|
+ prefix[pathlen - 1] = '/';
|
|
+
|
|
+ for (size_t i = 0; i < git_statuses.len; ++i)
|
|
+ if (is_prefix(git_statuses.statuses[i].path, prefix, pathlen))
|
|
+ set_git_status(dentp->git_status, i);
|
|
+ } else {
|
|
+ for (size_t i = 0; i < git_statuses.len; ++i)
|
|
+ if (!xstrcmp(git_statuses.statuses[i].path, dentpath)) {
|
|
+ set_git_status(dentp->git_status, i);
|
|
+ break;
|
|
+ }
|
|
+ }
|
|
+ }
|
|
+
|
|
++ndents;
|
|
} while ((dp = readdir(dirp)));
|
|
|
|
@@ -6360,11 +6474,12 @@ static int adjust_cols(int n)
|
|
#endif
|
|
if (cfg.showdetail) {
|
|
/* Fallback to light mode if less than 35 columns */
|
|
- if (n < 36)
|
|
+ if (n < 38)
|
|
cfg.showdetail ^= 1;
|
|
else /* 2 more accounted for below */
|
|
- n -= 32;
|
|
- }
|
|
+ n -= (git_statuses.show ? 34 : 32);
|
|
+ } else if (cfg.normalgit && git_statuses.show)
|
|
+ n -= 3;
|
|
|
|
/* 2 columns for preceding space and indicator */
|
|
return (n - 2);
|
|
@@ -8126,6 +8241,7 @@ static void usage(void)
|
|
" -F val fifo mode [0:preview 1:explore]\n"
|
|
#endif
|
|
" -g regex filters\n"
|
|
+ " -G always show git status\n"
|
|
" -H show hidden files\n"
|
|
" -i show current file info\n"
|
|
" -J no auto-proceed on select\n"
|
|
@@ -8266,6 +8382,7 @@ static void cleanup(void)
|
|
free(hostname);
|
|
}
|
|
#endif
|
|
+ free(git_statuses.statuses);
|
|
free(selpath);
|
|
free(plgpath);
|
|
free(cfgpath);
|
|
@@ -8310,7 +8427,7 @@ int main(int argc, char *argv[])
|
|
|
|
while ((opt = (env_opts_id > 0
|
|
? env_opts[--env_opts_id]
|
|
- : getopt(argc, argv, "aAb:cCdDeEfF:gHiJKl:nop:P:QrRs:St:T:uUVxh"))) != -1) {
|
|
+ : getopt(argc, argv, "aAb:cCdDeEfF:gGHiJKl:nop:P:QrRs:St:T:uUVxh"))) != -1) {
|
|
switch (opt) {
|
|
#ifndef NOFIFO
|
|
case 'a':
|
|
@@ -8361,6 +8478,9 @@ int main(int argc, char *argv[])
|
|
cfg.regex = 1;
|
|
filterfn = &visible_re;
|
|
break;
|
|
+ case 'G':
|
|
+ cfg.normalgit = 1;
|
|
+ break;
|
|
case 'H':
|
|
cfg.showhidden = 1;
|
|
break;
|