1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
|
# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this
# file, You can obtain one at http://mozilla.org/MPL/2.0/.
import re
import sys
TREEHERDER_REVISION_MATCHER = re.compile(
r"remote:.*https://treeherder\.mozilla\.org/jobs\?.*revision=([\w]*)[ \t]*$"
)
HGMO_REVISION_MATCHER = re.compile(
r"remote:.*/try/(?:rev/|pushloghtml\?changeset=)([\w]*)[ \t]*$"
)
class LogProcessor:
def __init__(self):
self.buf = ""
self.stdout = sys.__stdout__
self._treeherder_revision = None
self._hg_revision = None
@property
def revision(self):
return self._treeherder_revision or self._hg_revision
def write(self, buf):
while buf:
try:
newline_index = buf.index("\n")
except ValueError:
# No newline, wait for next call
self.buf += buf
break
# Get data up to next newline and combine with previously buffered data
data = self.buf + buf[: newline_index + 1]
buf = buf[newline_index + 1 :]
# Reset buffer then output line
self.buf = ""
if data.strip() == "":
continue
self.stdout.write(data.strip("\n") + "\n")
# Check if a temporary commit was created and a Treeherder
# link is found with the remote revision
match = TREEHERDER_REVISION_MATCHER.match(data)
if match:
self._treeherder_revision = match.group(1)
# Gather the hg revision in case the Treeherder one
# is missing for some reason
match = HGMO_REVISION_MATCHER.match(data)
if match:
# Last line found is the revision we want for HG patches
self._hg_revision = match.group(1)
def flush(self):
pass
|
