implement untracked

TODO.md

@@ -9,7 +9,7 @@
 'gitstatus.branch': <Prompt: gitstatus.branch>, # DONE
 'gitstatus.ahead': <Prompt: gitstatus.ahead>, # DONE
 'gitstatus.behind': <Prompt: gitstatus.behind>, # DONE
-'gitstatus.untracked': <Prompt: gitstatus.untracked>, # TODO
+'gitstatus.untracked': <Prompt: gitstatus.untracked>, # DONE
 'gitstatus.changed': <Prompt: gitstatus.changed>, # TODO
 'gitstatus.deleted': <Prompt: gitstatus.deleted>, # TODO
 'gitstatus.conflicts': <Prompt: gitstatus.conflicts>, # TODO

xontrib/pygitstatus/entrypoint.py

@@ -2,7 +2,7 @@
 
 from xonsh.built_ins import XonshSession
 
-from .prompt import ahead, behind, branch, clean, repo_path, short_head, tag
+from .prompt import (ahead, behind, branch, clean, repo_path, short_head, tag, untracked)
 
 
 def _load_xontrib_(xsh: XonshSession, **_):
@@ -24,6 +24,7 @@ def _load_xontrib_(xsh: XonshSession, **_):
     prompt_fields['pygitstatus.repo_path'] = repo_path
     prompt_fields['pygitstatus.short_head'] = short_head
     prompt_fields['pygitstatus.tag'] = tag
+    prompt_fields['pygitstatus.untracked'] = untracked
 
 
 def _unload_xontrib_(xsh: XonshSession, **kwargs) -> dict:

xontrib/pygitstatus/prompt.py

@@ -97,3 +97,13 @@ def tag(fld: PromptField, ctx: PromptFields):
             if head_commit == tag_commit:
                 fld.value = _tag.partition('refs/tags/')[-1]
                 break
+
+
+@PromptField.wrap(prefix="…", info="untracked")
+def untracked(fld: PromptField, ctx: PromptFields):
+    with contextlib.suppress(GitError):
+        fld.value = ''
+        repo = Repo('.')
+        untracked_count = len([v for k, v in repo.status().items() if v == GIT_STATUS_WT_NEW])
+        if untracked_count > 0:
+            fld.value = str(untracked_count)