hgkw/pretxnkw.py
branchextension
changeset 42 ba000e29ecf3
parent 41 e1c539e1282b
child 44 dc6e7d0e607f
--- a/hgkw/pretxnkw.py	Wed Dec 20 14:54:38 2006 +0100
+++ b/hgkw/pretxnkw.py	Wed Dec 20 17:41:19 2006 +0100
@@ -2,6 +2,8 @@
 from mercurial import cmdutil, commands, util
 import os.path, re, sys
 
+hgkeywords = 'Id|Header|Author|Date|Revision|RCSFile|Source'
+
 def pretxnkw(ui, repo, hooktype, **args):
     '''Collects candidates for keyword expansion on commit
     and expands keywords in working dir.
@@ -37,20 +39,31 @@
     if not files: # nothing to do
         return False
 
-    cid = args['node'][:12]
     user, date = repo.changelog.read(repo.changelog.tip())[1:3]
-    user = util.shortuser(user)
-    date = util.datestr(date=date, format=util.defaultdateformats[0])
+    strdate = util.datestr(date=date)
+    shortuser = util.shortuser(user)
+    shortdate = util.datestr(date=date, format=util.defaultdateformats[0])
                                                # %Y-%m-%d %H:%M:%S
 
-    re_kw = re.compile(r'\$Hg(: [^$]+?,v [a-z0-9]{12} [^$]+? )?\$')
+    re_kw = re.compile(r'\$(%s)(: [^$]+? )?\$' % hgkeywords)
 
     for f in files:
         data = repo.wfile(f).read()
         if not util.binary(data):
-            kw = '$Hg: %s,v %s %s %s $' % (
-                    os.path.basename(f), cid, date, user)
-            data, kwct = re_kw.subn(kw, data)
+
+            def kwexpand(matchobj):
+                RCSFile = os.path.basename(f)+',v'
+                Source = os.path.join(repo.root, f)+',v'
+                Revision = args['node'][:12]
+                Date = strdate
+                Author = user
+                revdateauth = '%s %s %s' % (Revision, shortdate, shortuser)
+                Header = '%s %s' % (Source, revdateauth)
+                Id = '%s %s' % (RCSFile, revdateauth)
+                return '$%s: %s $' % (
+                        matchobj.group(1), eval(matchobj.group(1)))
+
+            data, kwct = re_kw.subn(kwexpand, data)
             if kwct:
                 ui.note(_('expanding keywords in %s\n' % f))
                 # backup file?