Files
emacs/lisp/treemacs/treemacs-single-file-git-status.py
2020-12-05 21:29:49 +01:00

86 lines
2.6 KiB
Python

from subprocess import Popen, PIPE, DEVNULL
import sys
# There are 3+ command line arguments:
# 1) the file to update
# 2) the file's previous state, to check if things changed at all
# 3) the file's parents that need to be updated as well
FILE = sys.argv[1]
OLD_STATE = sys.argv[2]
PARENTS = [p for p in sys.argv[3:]]
FILE_STATE_CMD = "git status --porcelain --ignored "
IS_IGNORED_CMD = "git check-ignore "
IS_TRACKED_CMD = "git ls-files --error-unmatch "
IS_CHANGED_CMD = "git diff-index --quiet HEAD "
def main():
new_state = determine_file_git_state()
# nothing to do
if OLD_STATE == new_state:
sys.exit(2)
proc_list = []
# for every parent file start all necessary git processes immediately
# even if we don't need them later
for p in PARENTS:
add_git_processes(proc_list, p)
result_list = [(FILE, new_state)]
# iterate through the parents and propagate ignored and untracked states downwards
# the following states are possible for *directories*:
# 0 -> clean
# ! -> ignored
# ? -> untracked
# M -> modified
i = 0
l = len(proc_list)
propagate_state = None
while i < l:
path, ignore_proc, tracked_proc, changed_proc = proc_list[i]
if ignore_proc.wait() == 0:
propagate_state = "!"
result_list.append((path, propagate_state))
break
elif tracked_proc.wait() == 1:
propagate_state = "?"
result_list.append((path, propagate_state))
break
elif changed_proc.wait() == 1:
result_list.append((path, "M"))
else:
result_list.append((path, "0"))
i += 1
if propagate_state:
i += 1
while i < l:
result_list.append((proc_list[i][0], propagate_state))
i += 1
elisp_conses = "".join(['("{}" . "{}")'.format(path, state) for path,state in result_list])
elisp_alist = "({})".format(elisp_conses)
print(elisp_alist)
def add_git_processes(status_listings, path):
ignored_proc = Popen(IS_IGNORED_CMD + path, shell=True, stdout=DEVNULL, stderr=DEVNULL)
tracked_proc = Popen(IS_TRACKED_CMD + path, shell=True, stdout=DEVNULL, stderr=DEVNULL)
changed_proc = Popen(IS_CHANGED_CMD + path, shell=True, stdout=DEVNULL, stderr=DEVNULL)
status_listings.append((path, ignored_proc, tracked_proc, changed_proc))
def determine_file_git_state():
proc = Popen(FILE_STATE_CMD + FILE, shell=True, stdout=PIPE, stderr=DEVNULL)
line = proc.stdout.readline()
if line:
state = line.lstrip().split(b" ")[0]
return state.decode('utf-8').strip()[0]
else:
return "0"
main()