|
121
|
1 # hook.py - hook support for mercurial
|
|
|
2 #
|
|
|
3 # Copyright 2007 Matt Mackall <mpm@selenic.com>
|
|
|
4 #
|
|
|
5 # This software may be used and distributed according to the terms of the
|
|
|
6 # GNU General Public License version 2, incorporated herein by reference.
|
|
|
7
|
|
|
8 from i18n import _
|
|
|
9 import os, sys
|
|
|
10 import extensions, util
|
|
|
11
|
|
|
12 def _pythonhook(ui, repo, name, hname, funcname, args, throw):
|
|
|
13 '''call python hook. hook is callable object, looked up as
|
|
|
14 name in python module. if callable returns "true", hook
|
|
|
15 fails, else passes. if hook raises exception, treated as
|
|
|
16 hook failure. exception propagates if throw is "true".
|
|
|
17
|
|
|
18 reason for "true" meaning "hook failed" is so that
|
|
|
19 unmodified commands (e.g. mercurial.commands.update) can
|
|
|
20 be run as hooks without wrappers to convert return values.'''
|
|
|
21
|
|
|
22 ui.note(_("calling hook %s: %s\n") % (hname, funcname))
|
|
|
23 obj = funcname
|
|
|
24 if not hasattr(obj, '__call__'):
|
|
|
25 d = funcname.rfind('.')
|
|
|
26 if d == -1:
|
|
|
27 raise util.Abort(_('%s hook is invalid ("%s" not in '
|
|
|
28 'a module)') % (hname, funcname))
|
|
|
29 modname = funcname[:d]
|
|
|
30 try:
|
|
|
31 obj = __import__(modname)
|
|
|
32 except ImportError:
|
|
|
33 try:
|
|
|
34 # extensions are loaded with hgext_ prefix
|
|
|
35 obj = __import__("hgext_%s" % modname)
|
|
|
36 except ImportError:
|
|
|
37 raise util.Abort(_('%s hook is invalid '
|
|
|
38 '(import of "%s" failed)') %
|
|
|
39 (hname, modname))
|
|
|
40 try:
|
|
|
41 for p in funcname.split('.')[1:]:
|
|
|
42 obj = getattr(obj, p)
|
|
|
43 except AttributeError:
|
|
|
44 raise util.Abort(_('%s hook is invalid '
|
|
|
45 '("%s" is not defined)') %
|
|
|
46 (hname, funcname))
|
|
|
47 if not hasattr(obj, '__call__'):
|
|
|
48 raise util.Abort(_('%s hook is invalid '
|
|
|
49 '("%s" is not callable)') %
|
|
|
50 (hname, funcname))
|
|
|
51 try:
|
|
|
52 r = obj(ui=ui, repo=repo, hooktype=name, **args)
|
|
|
53 except KeyboardInterrupt:
|
|
|
54 raise
|
|
|
55 except Exception, exc:
|
|
|
56 if isinstance(exc, util.Abort):
|
|
|
57 ui.warn(_('error: %s hook failed: %s\n') %
|
|
|
58 (hname, exc.args[0]))
|
|
|
59 else:
|
|
|
60 ui.warn(_('error: %s hook raised an exception: '
|
|
|
61 '%s\n') % (hname, exc))
|
|
|
62 if throw:
|
|
|
63 raise
|
|
|
64 ui.traceback()
|
|
|
65 return True
|
|
|
66 if r:
|
|
|
67 if throw:
|
|
|
68 raise util.Abort(_('%s hook failed') % hname)
|
|
|
69 ui.warn(_('warning: %s hook failed\n') % hname)
|
|
|
70 return r
|
|
|
71
|
|
|
72 def _exthook(ui, repo, name, cmd, args, throw):
|
|
|
73 ui.note(_("running hook %s: %s\n") % (name, cmd))
|
|
|
74
|
|
|
75 env = {}
|
|
|
76 for k, v in args.iteritems():
|
|
|
77 if hasattr(v, '__call__'):
|
|
|
78 v = v()
|
|
|
79 env['HG_' + k.upper()] = v
|
|
|
80
|
|
|
81 if repo:
|
|
|
82 cwd = repo.root
|
|
|
83 else:
|
|
|
84 cwd = os.getcwd()
|
|
|
85 r = util.system(cmd, environ=env, cwd=cwd)
|
|
|
86 if r:
|
|
|
87 desc, r = util.explain_exit(r)
|
|
|
88 if throw:
|
|
|
89 raise util.Abort(_('%s hook %s') % (name, desc))
|
|
|
90 ui.warn(_('warning: %s hook %s\n') % (name, desc))
|
|
|
91 return r
|
|
|
92
|
|
|
93 _redirect = False
|
|
|
94 def redirect(state):
|
|
|
95 global _redirect
|
|
|
96 _redirect = state
|
|
|
97
|
|
|
98 def hook(ui, repo, name, throw=False, **args):
|
|
|
99 r = False
|
|
|
100
|
|
|
101 if _redirect:
|
|
|
102 # temporarily redirect stdout to stderr
|
|
|
103 oldstdout = os.dup(sys.__stdout__.fileno())
|
|
|
104 os.dup2(sys.__stderr__.fileno(), sys.__stdout__.fileno())
|
|
|
105
|
|
|
106 try:
|
|
|
107 for hname, cmd in ui.configitems('hooks'):
|
|
|
108 if hname.split('.')[0] != name or not cmd:
|
|
|
109 continue
|
|
|
110 if hasattr(cmd, '__call__'):
|
|
|
111 r = _pythonhook(ui, repo, name, hname, cmd, args, throw) or r
|
|
|
112 elif cmd.startswith('python:'):
|
|
|
113 if cmd.count(':') == 2:
|
|
|
114 path, cmd = cmd[7:].split(':')
|
|
|
115 mod = extensions.loadpath(path, 'hgkook.%s' % hname)
|
|
|
116 hookfn = getattr(mod, cmd)
|
|
|
117 else:
|
|
|
118 hookfn = cmd[7:].strip()
|
|
|
119 r = _pythonhook(ui, repo, name, hname, hookfn, args, throw) or r
|
|
|
120 else:
|
|
|
121 r = _exthook(ui, repo, hname, cmd, args, throw) or r
|
|
|
122 finally:
|
|
|
123 if _redirect:
|
|
|
124 os.dup2(oldstdout, sys.__stdout__.fileno())
|
|
|
125 os.close(oldstdout)
|
|
|
126
|
|
|
127 return r
|
