# HG changeset patch
# User Yuya Nishihara <yuya@tcha.org>
# Date 2018-07-29 13:04:01
# Node ID 87428152e820da5bea1eed45e2240dbbb274850d
# Parent  a5da906306c9deb9d32dbc4b40beb7a3f6070185

templatekw: add experimental {status} keyword

This is another example of fctx-based keywords. I think this is somewhat
useful in log templates.

diff --git a/mercurial/templatekw.py b/mercurial/templatekw.py
--- a/mercurial/templatekw.py
+++ b/mercurial/templatekw.py
@@ -301,6 +301,15 @@ def _getfilestatus(context, mapping, lis
         revcache['filestatusall'] = listall
     return revcache['filestatus']
 
+def _getfilestatusmap(context, mapping, listall=False):
+    revcache = context.resource(mapping, 'revcache')
+    if 'filestatusmap' not in revcache or revcache['filestatusall'] < listall:
+        stat = _getfilestatus(context, mapping, listall=listall)
+        revcache['filestatusmap'] = statmap = {}
+        for char, files in zip(pycompat.iterbytestr('MAR!?IC'), stat):
+            statmap.update((f, char) for f in files)
+    return revcache['filestatusmap']  # {path: statchar}
+
 def _showfilesbystat(context, mapping, name, index):
     stat = _getfilestatus(context, mapping)
     files = stat[index]
@@ -595,6 +604,19 @@ def showsize(context, mapping):
     fctx = context.resource(mapping, 'fctx')
     return fctx.size()
 
+# requires 'fctx' to denote {status} depends on (ctx, path) pair
+@templatekeyword('status', requires={'ctx', 'fctx', 'revcache'})
+def showstatus(context, mapping):
+    """String. Status code of the current file. (EXPERIMENTAL)"""
+    path = templateutil.runsymbol(context, mapping, 'path')
+    path = templateutil.stringify(context, mapping, path)
+    if not path:
+        return
+    statmap = _getfilestatusmap(context, mapping)
+    if path not in statmap:
+        statmap = _getfilestatusmap(context, mapping, listall=True)
+    return statmap.get(path)
+
 @templatekeyword("successorssets", requires={'repo', 'ctx'})
 def showsuccessorssets(context, mapping):
     """Returns a string of sets of successors for a changectx. Format used
diff --git a/tests/test-template-keywords.t b/tests/test-template-keywords.t
--- a/tests/test-template-keywords.t
+++ b/tests/test-template-keywords.t
@@ -789,12 +789,19 @@ Test file copies dict:
 
 Test file attributes:
 
-  $ hg log -l1 -T '{files % "{pad(size, 3, left=True)} {path}\n"}'
-      a
-    0 b
-    7 fifth
-      fourth
-   13 third
+  $ hg log -l1 -T '{files % "{status} {pad(size, 3, left=True)} {path}\n"}'
+  R     a
+  A   0 b
+  A   7 fifth
+  R     fourth
+  M  13 third
+
+Test file status including clean ones:
+
+  $ hg log -r9 -T '{files("**") % "{status} {path}\n"}'
+  A a
+  C fourth
+  C third
 
 Test index keyword: