Left: | ||
Right: |
LEFT | RIGHT |
---|---|
1 # This file is part of Adblock Plus <https://adblockplus.org/>, | 1 # This file is part of Adblock Plus <https://adblockplus.org/>, |
2 # Copyright (C) 2006-2016 Eyeo GmbH | 2 # Copyright (C) 2006-2016 Eyeo GmbH |
3 # | 3 # |
4 # Adblock Plus is free software: you can redistribute it and/or modify | 4 # Adblock Plus is free software: you can redistribute it and/or modify |
5 # it under the terms of the GNU General Public License version 3 as | 5 # it under the terms of the GNU General Public License version 3 as |
6 # published by the Free Software Foundation. | 6 # published by the Free Software Foundation. |
7 # | 7 # |
8 # Adblock Plus is distributed in the hope that it will be useful, | 8 # Adblock Plus is distributed in the hope that it will be useful, |
9 # but WITHOUT ANY WARRANTY; without even the implied warranty of | 9 # but WITHOUT ANY WARRANTY; without even the implied warranty of |
10 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | 10 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
11 # GNU General Public License for more details. | 11 # GNU General Public License for more details. |
12 # | 12 # |
13 # You should have received a copy of the GNU General Public License | 13 # You should have received a copy of the GNU General Public License |
14 # along with Adblock Plus. If not, see <http://www.gnu.org/licenses/>. | 14 # along with Adblock Plus. If not, see <http://www.gnu.org/licenses/>. |
15 | 15 |
16 """Hooks for integrating Mercurial with Trac. | 16 """Hooks for integrating Mercurial with Trac. |
17 | 17 |
18 Update the issues that are referenced in commit messages when the commits | 18 Update the issues that are referenced in commit messages when the commits |
19 are pushed and and "master" bookmark is moved. See README.md for more | 19 are pushed and `master` bookmark is moved. See README.md for more |
20 information on behavior and configuration. | 20 information on behavior and configuration. |
21 """ | 21 """ |
22 | 22 |
23 import collections | 23 import collections |
24 import contextlib | 24 import contextlib |
25 import posixpath | 25 import posixpath |
26 import re | 26 import re |
27 import xmlrpclib | 27 import xmlrpclib |
28 | 28 |
29 from sitescripts.utils import get_config, get_template | 29 from sitescripts.utils import get_config, get_template |
30 | 30 |
31 | 31 |
32 _IssueRef = collections.namedtuple('IssueRef', 'id commits is_fixed') | 32 _IssueRef = collections.namedtuple('IssueRef', 'id commits is_fixed') |
33 | |
34 ISSUE_NUMBER_REGEX = re.compile(r'\b(issue|fixes)\s+(\d+)\b', re.I) | |
35 NOISSUE_REGEX = re.compile(r'^noissue\b', re.I) | |
36 COMMIT_MESSAGE_REGEX = re.compile(r'\[\S+\ ([^\]]+)\]') | |
37 | |
33 | 38 |
34 @contextlib.contextmanager | 39 @contextlib.contextmanager |
kzar
2016/05/02 09:29:41
I'm learning some new tricks here :)
Vasily Kuznetsov
2016/05/02 16:54:04
Yeah, this is a pretty cool way to make context ma
| |
35 def _trac_proxy(ui, config, action_descr): | 40 def _trac_proxy(ui, config, action_descr): |
36 trac_url = config.get('hg', 'trac_xmlrpc_url') | 41 trac_url = config.get('hg', 'trac_xmlrpc_url') |
37 try: | 42 try: |
38 yield xmlrpclib.ServerProxy(trac_url) | 43 yield xmlrpclib.ServerProxy(trac_url) |
39 except Exception as exc: | 44 except Exception as exc: |
40 if getattr(exc, 'faultCode', 0) == 404: # Not found | 45 if getattr(exc, 'faultCode', 0) == 404: |
kzar
2016/05/02 09:29:41
Nit: This comment doesn't really add anything.
Vasily Kuznetsov
2016/05/02 16:54:03
Fair enough, I'll remove it.
| |
41 ui.warn('warning: 404 (not found) while {}\n'.format(action_descr)) | 46 ui.warn('warning: 404 (not found) while {}\n'.format(action_descr)) |
42 else: | 47 else: |
43 ui.warn('error: {} while {}\n'.format(exc, action_descr)) | 48 ui.warn('error: {} while {}\n'.format(exc, action_descr)) |
44 | 49 |
45 | 50 |
46 def _update_issue(ui, config, issue_id, comment, changes): | 51 def _update_issue(ui, config, issue_id, changes, comment=''): |
47 issue_url_template = config.get('hg', 'issue_url_template') | 52 issue_url_template = config.get('hg', 'issue_url_template') |
48 issue_url = issue_url_template.format(id=issue_id) | 53 issue_url = issue_url_template.format(id=issue_id) |
49 | 54 |
50 updates = [] | 55 updates = [] |
51 if comment: | 56 if comment: |
52 updates.append('posted comment') | 57 for message in COMMIT_MESSAGE_REGEX.findall(comment): |
kzar
2016/04/29 14:05:43
Supposing there's a "feature branch" bookmark that
Vasily Kuznetsov
2016/05/02 16:54:04
Unfortunately the commit messages didn't make it i
kzar
2016/05/03 06:47:35
Nice, looks good to me. Will mean I can even click
| |
58 updates.append(' - referenced a commit: ' + message) | |
53 if 'milestone' in changes: | 59 if 'milestone' in changes: |
54 updates.append('set milestone: {}'.format(changes['milestone'])) | 60 updates.append(' - set milestone to ' + changes['milestone']) |
55 if changes['action'] == 'resolve': | 61 if changes['action'] == 'resolve': |
56 updates.append('closed') | 62 updates.append(' - closed') |
57 if not updates: | 63 if not updates: |
58 return | 64 return |
59 | 65 |
60 with _trac_proxy(ui, config, 'updating issue {}'.format(issue_id)) as tp: | 66 with _trac_proxy(ui, config, 'updating issue {}'.format(issue_id)) as tp: |
61 tp.ticket.update(issue_id, comment, changes, True) | 67 tp.ticket.update(issue_id, comment, changes, True) |
62 ui.status('updated {} ({})\n'.format(issue_url, ', '.join(updates))) | 68 ui.status('updated {}:\n{}\n'.format(issue_url, '\n'.join(updates))) |
63 | 69 |
64 | 70 |
65 def _post_comments(ui, repo, config, refs): | 71 def _post_comments(ui, repo, config, refs): |
66 repo_name = posixpath.split(repo.url())[1] | 72 repo_name = posixpath.split(repo.url())[1] |
67 template = get_template('hg/template/issue_commit_comment.tmpl', | 73 template = get_template('hg/template/issue_commit_comment.tmpl', |
68 autoescape=False) | 74 autoescape=False) |
69 for ref in refs: | 75 for ref in refs: |
70 comment_text = template.render({'repository_name': repo_name, | 76 comment_text = template.render({'repository_name': repo_name, |
71 'changes': ref.commits}) | 77 'changes': ref.commits}) |
72 with _trac_proxy(ui, config, 'getting issue {}'.format(ref.id)) as tp: | 78 with _trac_proxy(ui, config, 'getting issue {}'.format(ref.id)) as tp: |
73 attrs = tp.ticket.get(ref.id)[3] | 79 attrs = tp.ticket.get(ref.id)[3] |
74 _update_issue(ui, config, ref.id, comment_text, | 80 changes = {'_ts': attrs['_ts'], 'action': 'leave'} |
75 {'_ts': attrs['_ts'], 'action': 'leave'}) | 81 _update_issue(ui, config, ref.id, changes, comment_text) |
76 | 82 |
77 | 83 |
78 def _compile_module_regexps(ui, config, modules): | 84 def _compile_module_regexps(ui, config, modules): |
79 for module, regexp in config.items('hg_module_milestones'): | 85 for module, regexp in config.items('hg_module_milestones'): |
80 try: | 86 try: |
81 yield module, re.compile('^' + regexp + '$', re.I) | 87 yield module, re.compile('^{}$'.format(regexp), re.I) |
82 except Exception as exc: | 88 except Exception as exc: |
83 ui.warn('warning: invalid module milestone regexp in config:' | 89 ui.warn('warning: skipped invalid regexp for module {} in ' |
kzar
2016/05/02 09:29:40
This message looks pretty clear but maybe we could
Vasily Kuznetsov
2016/05/02 16:54:03
I suppose you meant "Skipped invalid regexp for MO
| |
84 "'{}' ({})\n".format(regexp, exc)) | 90 "[hg_module_milestones] config: '{}' ({})\n" |
91 .format(module, regexp, exc)) | |
85 | 92 |
86 | 93 |
87 def _get_module_milestones(ui, config, modules): | 94 def _get_module_milestones(ui, config, modules): |
88 module_regexps = dict(_compile_module_regexps(ui, config, modules)) | 95 module_regexps = dict(_compile_module_regexps(ui, config, modules)) |
89 modules = module_regexps.keys() | 96 modules = module_regexps.keys() |
90 if not modules: | 97 if not modules: |
91 return [] | 98 return [] |
92 | 99 |
93 milestones_by_module = {} | 100 milestones_by_module = {} |
94 with _trac_proxy(ui, config, 'getting milestones') as tp: | 101 with _trac_proxy(ui, config, 'getting milestones') as tp: |
95 milestone_names = [ | 102 milestone_names = [ |
96 name for name in tp.ticket.milestone.getAll() | 103 name for name in tp.ticket.milestone.getAll() |
97 if any(regexp.search(name) for regexp in module_regexps.values()) | 104 if any(regexp.search(name) for regexp in module_regexps.values()) |
98 ] | 105 ] |
99 # Using a MultiCall is better because we might have many | 106 # Using a MultiCall is better because we might have many milestones. |
100 # milestones. | 107 get_milestones = xmlrpclib.MultiCall(tp) |
101 multicall = xmlrpclib.MultiCall(tp) | |
102 for name in milestone_names: | 108 for name in milestone_names: |
103 multicall.ticket.milestone.get(name) | 109 get_milestones.ticket.milestone.get(name) |
104 milestones = filter(lambda m: not m['completed'], multicall()) | 110 milestones = [ms for ms in get_milestones() if not ms['completed']] |
kzar
2016/04/29 14:05:43
Nit: I've been told to use list comprehension inst
Vasily Kuznetsov
2016/05/02 16:54:04
Yep, it's definitely better here. I also renamed `
| |
105 for module in modules: | 111 for module in modules: |
106 for milestone in milestones: | 112 for milestone in milestones: |
107 if module_regexps[module].search(milestone['name']): | 113 if module_regexps[module].search(milestone['name']): |
108 milestones_by_module[module] = milestone['name'] | 114 milestones_by_module[module] = milestone['name'] |
109 break | 115 break |
110 | 116 |
111 return milestones_by_module.items() | 117 return milestones_by_module.items() |
112 | 118 |
113 | 119 |
114 def _declare_fixed(ui, config, refs): | 120 def _declare_fixed(ui, config, refs): |
(...skipping 14 matching lines...) Expand all Loading... | |
129 if not attrs['milestone']: | 135 if not attrs['milestone']: |
130 need_milestones[attrs['component']].append(changes) | 136 need_milestones[attrs['component']].append(changes) |
131 updates.append((ref.id, changes)) | 137 updates.append((ref.id, changes)) |
132 | 138 |
133 for module, milestone in _get_module_milestones(ui, config, | 139 for module, milestone in _get_module_milestones(ui, config, |
134 need_milestones.keys()): | 140 need_milestones.keys()): |
135 for changes in need_milestones[module]: | 141 for changes in need_milestones[module]: |
136 changes['milestone'] = milestone | 142 changes['milestone'] = milestone |
137 | 143 |
138 for issue_id, changes in updates: | 144 for issue_id, changes in updates: |
139 _update_issue(ui, config, issue_id, '', changes) | 145 _update_issue(ui, config, issue_id, changes) |
kzar
2016/05/02 09:29:40
Since comment seems like the only optional paramet
Vasily Kuznetsov
2016/05/02 16:54:03
Done.
| |
140 | 146 |
141 | 147 |
142 def _collect_references(ui, commits): | 148 def _collect_references(ui, commits): |
143 issue_number_regex = re.compile(r'\b(issue|fixes)\s+(\d+)\b', re.I) | |
kzar
2016/05/02 09:29:41
Maybe these regexps should be compiled as constant
Vasily Kuznetsov
2016/05/02 16:54:04
Actually it's not called in a loop, only once from
| |
144 noissue_regex = re.compile(r'^noissue\b', re.I) | |
145 | |
146 commits_by_issue = collections.defaultdict(list) | 149 commits_by_issue = collections.defaultdict(list) |
147 fixed_issues = set() | 150 fixed_issues = set() |
148 | 151 |
149 for commit in commits: | 152 for commit in commits: |
150 message = commit.description() | 153 message = commit.description() |
151 if ' - ' not in message: | 154 if ' - ' not in message: |
152 ui.warn("warning: invalid commit message format: '{}'\n" | 155 ui.warn("warning: invalid commit message format: '{}'\n" |
153 .format(message)) | 156 .format(message)) |
154 continue | 157 continue |
155 | 158 |
156 refs, rest = message.split(" - ", 1) | 159 refs, rest = message.split(' - ', 1) |
157 issue_refs = issue_number_regex.findall(refs) | 160 issue_refs = ISSUE_NUMBER_REGEX.findall(refs) |
158 if issue_refs: | 161 if issue_refs: |
159 for ref_type, issue_id in issue_refs: | 162 for ref_type, issue_id in issue_refs: |
160 issue_id = int(issue_id) | 163 issue_id = int(issue_id) |
161 commits_by_issue[issue_id].append(commit) | 164 commits_by_issue[issue_id].append(commit) |
162 if ref_type.lower() == 'fixes': | 165 if ref_type.lower() == 'fixes': |
163 fixed_issues.add(issue_id) | 166 fixed_issues.add(issue_id) |
164 elif not noissue_regex.search(refs): | 167 elif not NOISSUE_REGEX.search(refs): |
165 ui.warn("warning: no issue reference in commit message: '{}'\n" | 168 ui.warn("warning: no issue reference in commit message: '{}'\n" |
166 .format(message)) | 169 .format(message)) |
167 | 170 |
168 for issue_id, commits in sorted(commits_by_issue.items()): | 171 for issue_id, commits in sorted(commits_by_issue.items()): |
169 yield _IssueRef(issue_id, commits, is_fixed=issue_id in fixed_issues) | 172 yield _IssueRef(issue_id, commits, is_fixed=issue_id in fixed_issues) |
170 | 173 |
171 | 174 |
172 def changegroup_hook(ui, repo, node, **kwargs): | 175 def changegroup_hook(ui, repo, node, **kwargs): |
173 config = get_config() | 176 config = get_config() |
174 first_rev = repo[node].rev() | 177 first_rev = repo[node].rev() |
175 pushed_commits = repo[first_rev:] | 178 pushed_commits = repo[first_rev:] |
176 refs = _collect_references(ui, pushed_commits) | 179 refs = _collect_references(ui, pushed_commits) |
177 _post_comments(ui, repo, config, refs) | 180 _post_comments(ui, repo, config, refs) |
178 | 181 |
179 | 182 |
180 def pushkey_hook(ui, repo, **kwargs): | 183 def pushkey_hook(ui, repo, **kwargs): |
181 if kwargs['namespace'] != 'bookmarks': | 184 if (kwargs['namespace'] != 'bookmarks' or # Not a bookmark move. |
kzar
2016/04/29 14:05:43
Nit: Replace these three with one if?
Vasily Kuznetsov
2016/05/02 16:54:03
Done.
| |
182 return # Not a bookmark move. | 185 kwargs['key'] != 'master' or # Not `master` bookmark. |
183 if kwargs['key'] != 'master': | 186 not kwargs['old']): # The bookmark is just created. |
184 return # Not master bookmark. | 187 return |
185 if not kwargs['old']: | |
186 return # The bookmark is just created -- don't do anything. | |
187 | 188 |
188 config = get_config() | 189 config = get_config() |
189 old_master_rev = repo[kwargs['old']].rev() | 190 old_master_rev = repo[kwargs['old']].rev() |
190 new_master_rev = repo[kwargs['new']].rev() | 191 new_master_rev = repo[kwargs['new']].rev() |
191 added_revs = repo.changelog.findmissingrevs([old_master_rev], | 192 added_revs = repo.changelog.findmissingrevs([old_master_rev], |
192 [new_master_rev]) | 193 [new_master_rev]) |
193 added_commits = [repo[rev] for rev in added_revs] | 194 added_commits = [repo[rev] for rev in added_revs] |
194 refs = [ref for ref in _collect_references(ui, added_commits) | 195 refs = [ref for ref in _collect_references(ui, added_commits) |
195 if ref.is_fixed] | 196 if ref.is_fixed] |
196 _declare_fixed(ui, config, refs) | 197 _declare_fixed(ui, config, refs) |
197 | 198 |
198 | 199 |
199 # Alias for backward compatibility. | 200 # Alias for backward compatibility. |
200 hook = changegroup_hook | 201 hook = changegroup_hook |
LEFT | RIGHT |