forked from mozilla/gecko-dev
-
Notifications
You must be signed in to change notification settings - Fork 1
/
variables.py
113 lines (85 loc) · 3.51 KB
/
variables.py
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
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
# 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/.
from __future__ import print_function, unicode_literals
import os
import subprocess
import sys
from datetime import datetime
SOURCESTAMP_FILENAME = "sourcestamp.txt"
def buildid_header(output):
buildid = os.environ.get("MOZ_BUILD_DATE")
if buildid and len(buildid) != 14:
print("Ignoring invalid MOZ_BUILD_DATE: %s" % buildid, file=sys.stderr)
buildid = None
if not buildid:
buildid = datetime.now().strftime("%Y%m%d%H%M%S")
output.write("#define MOZ_BUILDID %s\n" % buildid)
def get_program_output(*command):
try:
with open(os.devnull) as stderr:
return subprocess.check_output(
command, stderr=stderr, universal_newlines=True
)
except Exception:
return ""
def get_hg_info(workdir):
repo = get_program_output("hg", "-R", workdir, "path", "default")
if repo:
repo = repo.strip()
if repo.startswith("ssh://"):
repo = "https://" + repo[6:]
repo = repo.rstrip("/")
changeset = get_hg_changeset(workdir)
return repo, changeset
def get_hg_changeset(path):
return get_program_output("hg", "-R", path, "parent", "--template={node}")
def get_info_from_sourcestamp(sourcestamp_path):
"""Read the repository and changelog information from the sourcestamp
file. This assumes that the file exists and returns the results as a list
(either strings or None in case of error).
"""
# Load the content of the file.
lines = None
with open(sourcestamp_path) as f:
lines = f.read().splitlines()
# Parse the repo and the changeset. The sourcestamp file is supposed to
# contain two lines: the first is the build id and the second is the source
# URL.
if len(lines) != 2 or not lines[1].startswith("http"):
# Just return if the file doesn't contain what we expect.
return None, None
# Return the repo and the changeset.
return lines[1].split("/rev/")
def source_repo_header(output):
# We allow the source repo and changeset to be specified via the
# environment (see configure)
import buildconfig
repo = buildconfig.substs.get("MOZ_SOURCE_REPO")
changeset = buildconfig.substs.get("MOZ_SOURCE_CHANGESET")
source = ""
if not repo:
sourcestamp_path = os.path.join(buildconfig.topsrcdir, SOURCESTAMP_FILENAME)
if os.path.exists(os.path.join(buildconfig.topsrcdir, ".hg")):
repo, changeset = get_hg_info(buildconfig.topsrcdir)
elif os.path.exists(sourcestamp_path):
repo, changeset = get_info_from_sourcestamp(sourcestamp_path)
elif not changeset:
changeset = get_hg_changeset(buildconfig.topsrcdir)
if not changeset:
raise Exception(
"could not resolve changeset; " "try setting MOZ_SOURCE_CHANGESET"
)
if changeset:
output.write("#define MOZ_SOURCE_STAMP %s\n" % changeset)
if repo and buildconfig.substs.get("MOZ_INCLUDE_SOURCE_INFO"):
source = "%s/rev/%s" % (repo, changeset)
output.write("#define MOZ_SOURCE_REPO %s\n" % repo)
output.write("#define MOZ_SOURCE_URL %s\n" % source)
def main(args):
if len(args):
func = globals().get(args[0])
if func:
return func(sys.stdout, *args[1:])
if __name__ == "__main__":
sys.exit(main(sys.argv[1:]))