mirror of
https://github.com/jarun/nnn.git
synced 2024-10-05 12:12:54 +00:00
221 lines
7.5 KiB
Diff
221 lines
7.5 KiB
Diff
# Description: Add git status column to detail mode.
|
|
# Compatibility patch for the namefirst patch.
|
|
#
|
|
# Dependencies: libgit2
|
|
#
|
|
# Authors: @crides, Luuk van Baal
|
|
|
|
diff --git a/src/nnn.c b/src/nnn.c
|
|
index 22402fa..33be591 100644
|
|
--- a/src/nnn.c
|
|
+++ b/src/nnn.c
|
|
@@ -105,6 +105,7 @@
|
|
#include <wchar.h>
|
|
#include <pwd.h>
|
|
#include <grp.h>
|
|
+#include <git2.h>
|
|
|
|
#if !defined(alloca) && defined(__GNUC__)
|
|
/*
|
|
@@ -244,6 +245,16 @@ typedef unsigned int uint_t;
|
|
typedef unsigned char uchar_t;
|
|
typedef unsigned short ushort_t;
|
|
typedef unsigned long long ulong_t;
|
|
+typedef enum {
|
|
+ GIT_COLUMN_STATUS_UNMOD = 0,
|
|
+ GIT_COLUMN_STATUS_NEW,
|
|
+ GIT_COLUMN_STATUS_MODIFIED,
|
|
+ GIT_COLUMN_STATUS_DELETED,
|
|
+ GIT_COLUMN_STATUS_RENAMED,
|
|
+ GIT_COLUMN_STATUS_TYPE_CHANGE,
|
|
+ GIT_COLUMN_STATUS_IGNORED,
|
|
+ GIT_COLUMN_STATUS_CONFLICTED,
|
|
+} git_column_status_t;
|
|
|
|
/* STRUCTURES */
|
|
|
|
@@ -258,6 +269,8 @@ typedef struct entry {
|
|
ulong_t blocks : 40; /* 5 bytes (enough for 512 TiB in 512B blocks allocated) */
|
|
ulong_t nlen : 16; /* 2 bytes (length of file name) */
|
|
ulong_t flags : 8; /* 1 byte (flags specific to the file) */
|
|
+ git_column_status_t status_indexed;
|
|
+ git_column_status_t status_staged;
|
|
};
|
|
#ifndef NOUG
|
|
uid_t uid; /* 4 bytes */
|
|
@@ -366,7 +379,18 @@ static struct {
|
|
ushort_t maxnameln, maxsizeln, maxuidln, maxgidln, maxentln, uidln, gidln, printguid;
|
|
} dtls;
|
|
|
|
+typedef struct {
|
|
+ char *path;
|
|
+ git_status_t status;
|
|
+} simple_git_status_t;
|
|
+
|
|
+typedef struct {
|
|
+ simple_git_status_t *statuses;
|
|
+ size_t len;
|
|
+} simple_git_statuses_t;
|
|
+
|
|
/* GLOBALS */
|
|
+simple_git_statuses_t git_statuses;
|
|
|
|
/* Configuration, contexts */
|
|
static settings cfg = {
|
|
@@ -3472,6 +3496,85 @@ static char *get_kv_val(kv *kvarr, char *buf, int key, uchar_t max, uchar_t id)
|
|
return NULL;
|
|
}
|
|
|
|
+static git_column_status_t git_get_indexed_status(const uint32_t status) {
|
|
+ if (status & GIT_STATUS_INDEX_NEW) return GIT_COLUMN_STATUS_NEW;
|
|
+ if (status & GIT_STATUS_INDEX_MODIFIED) return GIT_COLUMN_STATUS_MODIFIED;
|
|
+ if (status & GIT_STATUS_INDEX_DELETED) return GIT_COLUMN_STATUS_DELETED;
|
|
+ if (status & GIT_STATUS_INDEX_RENAMED) return GIT_COLUMN_STATUS_RENAMED;
|
|
+ if (status & GIT_STATUS_INDEX_TYPECHANGE) return GIT_COLUMN_STATUS_TYPE_CHANGE;
|
|
+ return GIT_COLUMN_STATUS_UNMOD;
|
|
+}
|
|
+
|
|
+static git_column_status_t git_get_staged_status(const uint32_t status) {
|
|
+ if (status & GIT_STATUS_WT_NEW) return GIT_COLUMN_STATUS_NEW;
|
|
+ if (status & GIT_STATUS_WT_MODIFIED) return GIT_COLUMN_STATUS_MODIFIED;
|
|
+ if (status & GIT_STATUS_WT_DELETED) return GIT_COLUMN_STATUS_DELETED;
|
|
+ if (status & GIT_STATUS_WT_RENAMED) return GIT_COLUMN_STATUS_RENAMED;
|
|
+ if (status & GIT_STATUS_WT_TYPECHANGE) return GIT_COLUMN_STATUS_TYPE_CHANGE;
|
|
+ if (status & GIT_STATUS_IGNORED) return GIT_COLUMN_STATUS_IGNORED;
|
|
+ if (status & GIT_STATUS_CONFLICTED) return GIT_COLUMN_STATUS_CONFLICTED;
|
|
+ return GIT_COLUMN_STATUS_UNMOD;
|
|
+}
|
|
+
|
|
+static void print_gitstatus(git_column_status_t status) {
|
|
+ switch (status) {
|
|
+ case GIT_COLUMN_STATUS_UNMOD: addch('-' | 0); break;
|
|
+ case GIT_COLUMN_STATUS_NEW: addch('N' | (g_state.oldcolor ? COLOR_PAIR(1) : COLOR_PAIR(C_EXE))); break;
|
|
+ case GIT_COLUMN_STATUS_MODIFIED: addch('M' | (g_state.oldcolor ? COLOR_PAIR(1) : COLOR_PAIR(4))); break;
|
|
+ case GIT_COLUMN_STATUS_DELETED: addch('D' | (g_state.oldcolor ? COLOR_PAIR(1) : COLOR_PAIR(C_UND))); break;
|
|
+ case GIT_COLUMN_STATUS_RENAMED: addch('R' | (g_state.oldcolor ? COLOR_PAIR(1) : COLOR_PAIR(C_CHR))); break;
|
|
+ case GIT_COLUMN_STATUS_TYPE_CHANGE: addch('T' | (g_state.oldcolor ? COLOR_PAIR(1) : COLOR_PAIR(C_HRD))); break;
|
|
+ case GIT_COLUMN_STATUS_IGNORED: addch('I' | 0); break;
|
|
+ case GIT_COLUMN_STATUS_CONFLICTED: addch('U' | (g_state.oldcolor ? COLOR_PAIR(1) : COLOR_PAIR(C_UND))); break;
|
|
+ }
|
|
+}
|
|
+
|
|
+static bool starts_with(const char *const s, const char *const start) {
|
|
+ const size_t start_len = xstrlen(start);
|
|
+
|
|
+ return (start_len <= xstrlen(s)) && !strncmp(s, start, start_len);
|
|
+}
|
|
+
|
|
+static simple_git_statuses_t statuses_from_path(const char *path) {
|
|
+ git_repository *repo;
|
|
+ git_buf ret = { .ptr = NULL, .asize = 0, .size = 0 };
|
|
+ simple_git_statuses_t statuses = { .statuses = NULL, .len = 0 };
|
|
+ git_repository_discover(&ret, path, false, NULL);
|
|
+ git_repository_open(&repo, ret.ptr);
|
|
+ git_buf_dispose(&ret);
|
|
+
|
|
+ if (repo) {
|
|
+ char buf[PATH_MAX];
|
|
+ const char *workdir = git_repository_workdir(repo);
|
|
+ git_status_list *status_list;
|
|
+
|
|
+ git_status_list_new(&status_list, repo, NULL);
|
|
+ statuses.len = git_status_list_entrycount(status_list);
|
|
+ statuses.statuses = malloc(statuses.len * sizeof(simple_git_status_t));
|
|
+
|
|
+ for (size_t i = 0; i < statuses.len; ++i) {
|
|
+ const git_status_entry *status_ent = git_status_byindex(status_list, i);
|
|
+ const char *entry_path = status_ent->head_to_index ? status_ent->head_to_index->old_file.path
|
|
+ : status_ent->index_to_workdir->old_file.path;
|
|
+
|
|
+ xstrsncpy(buf, workdir, xstrlen(workdir));
|
|
+ statuses.statuses[i].path = abspath(entry_path, buf);
|
|
+ statuses.statuses[i].status = status_ent->status;
|
|
+ }
|
|
+
|
|
+ git_status_list_free(status_list);
|
|
+ git_repository_free(repo);
|
|
+ }
|
|
+ return statuses;
|
|
+}
|
|
+
|
|
+static void statuses_free(simple_git_statuses_t statuses) {
|
|
+ for (size_t i = 0; i < statuses.len; ++i)
|
|
+ free(statuses.statuses[i].path);
|
|
+
|
|
+ free(statuses.statuses);
|
|
+}
|
|
+
|
|
static void resetdircolor(int flags)
|
|
{
|
|
/* Directories are always shown on top, clear the color when moving to first file */
|
|
@@ -3783,6 +3886,12 @@ static void printent(const struct entry *ent, uint_t namecols, bool sel)
|
|
|
|
addch((ent->flags & FILE_SELECTED) ? '+' | A_REVERSE | A_BOLD : ' ');
|
|
|
|
+ if (cfg.showdetail && git_statuses.len) {
|
|
+ print_gitstatus(ent->status_indexed);
|
|
+ print_gitstatus(ent->status_staged);
|
|
+ addch(' ');
|
|
+ }
|
|
+
|
|
if (g_state.oldcolor)
|
|
resetdircolor(ent->flags);
|
|
else {
|
|
@@ -5125,6 +5234,11 @@ static int dentfill(char *path, struct entry **ppdents)
|
|
attron(COLOR_PAIR(cfg.curctx + 1));
|
|
}
|
|
|
|
+ if (git_statuses.len)
|
|
+ statuses_free(git_statuses);
|
|
+ if (cfg.showdetail)
|
|
+ git_statuses = statuses_from_path(path);
|
|
+
|
|
#if _POSIX_C_SOURCE >= 200112L
|
|
posix_fadvise(fd, 0, 0, POSIX_FADV_SEQUENTIAL);
|
|
#endif
|
|
@@ -5322,6 +5436,18 @@ static int dentfill(char *path, struct entry **ppdents)
|
|
#endif
|
|
}
|
|
|
|
+ if (git_statuses.len) {
|
|
+ uint merged_status = 0;
|
|
+
|
|
+ for (size_t i = 0; i < git_statuses.len; i ++)
|
|
+ if ((dentp->flags & DIR_OR_LINK_TO_DIR) ? starts_with(git_statuses.statuses[i].path, abspath(dentp->name, path))
|
|
+ : !xstrcmp(git_statuses.statuses[i].path, abspath(dentp->name, path)))
|
|
+ merged_status |= git_statuses.statuses[i].status;
|
|
+
|
|
+ dentp->status_indexed = git_get_indexed_status(merged_status);
|
|
+ dentp->status_staged = git_get_staged_status(merged_status);
|
|
+ }
|
|
+
|
|
++ndents;
|
|
} while ((dp = readdir(dirp)));
|
|
|
|
@@ -5972,7 +6098,7 @@ static void redraw(char *path)
|
|
#endif
|
|
}
|
|
}
|
|
- dtls.maxentln = dtls.maxnameln + dtls.maxsizeln + (dtls.printguid ? (dtls.maxuidln + dtls.maxgidln) : 0) + (g_state.uidgid ? 26 : 23);
|
|
+ dtls.maxentln = dtls.maxnameln + dtls.maxsizeln + (dtls.printguid ? (dtls.maxuidln + dtls.maxgidln) : 0) + (g_state.uidgid ? 29 : 26);
|
|
|
|
ncols = adjust_cols(ncols);
|
|
|
|
@@ -7714,6 +7840,8 @@ static void cleanup(void)
|
|
fflush(stdout);
|
|
}
|
|
#endif
|
|
+ statuses_free(git_statuses);
|
|
+ git_libgit2_shutdown();
|
|
free(selpath);
|
|
free(plgpath);
|
|
free(cfgpath);
|
|
@@ -7909,6 +8037,7 @@ int main(int argc, char *argv[])
|
|
return EXIT_FAILURE;
|
|
|
|
atexit(cleanup);
|
|
+ git_libgit2_init();
|
|
|
|
/* Check if we are in path list mode */
|
|
if (!isatty(STDIN_FILENO)) {
|