##// END OF EJS Templates
packaging: leverage os.path.relpath() in setup.py...
Martin von Zweigbergk -
r44539:4e05272d default
parent child Browse files
Show More
@@ -1,1731 +1,1727 b''
1 #
1 #
2 # This is the mercurial setup script.
2 # This is the mercurial setup script.
3 #
3 #
4 # 'python setup.py install', or
4 # 'python setup.py install', or
5 # 'python setup.py --help' for more options
5 # 'python setup.py --help' for more options
6
6
7 import os
7 import os
8
8
9 # Mercurial will never work on Python 3 before 3.5 due to a lack
9 # Mercurial will never work on Python 3 before 3.5 due to a lack
10 # of % formatting on bytestrings, and can't work on 3.6.0 or 3.6.1
10 # of % formatting on bytestrings, and can't work on 3.6.0 or 3.6.1
11 # due to a bug in % formatting in bytestrings.
11 # due to a bug in % formatting in bytestrings.
12 # We cannot support Python 3.5.0, 3.5.1, 3.5.2 because of bug in
12 # We cannot support Python 3.5.0, 3.5.1, 3.5.2 because of bug in
13 # codecs.escape_encode() where it raises SystemError on empty bytestring
13 # codecs.escape_encode() where it raises SystemError on empty bytestring
14 # bug link: https://bugs.python.org/issue25270
14 # bug link: https://bugs.python.org/issue25270
15 supportedpy = ','.join(
15 supportedpy = ','.join(
16 [
16 [
17 '>=2.7',
17 '>=2.7',
18 '!=3.0.*',
18 '!=3.0.*',
19 '!=3.1.*',
19 '!=3.1.*',
20 '!=3.2.*',
20 '!=3.2.*',
21 '!=3.3.*',
21 '!=3.3.*',
22 '!=3.4.*',
22 '!=3.4.*',
23 '!=3.5.0',
23 '!=3.5.0',
24 '!=3.5.1',
24 '!=3.5.1',
25 '!=3.5.2',
25 '!=3.5.2',
26 '!=3.6.0',
26 '!=3.6.0',
27 '!=3.6.1',
27 '!=3.6.1',
28 ]
28 ]
29 )
29 )
30
30
31 import sys, platform
31 import sys, platform
32 import sysconfig
32 import sysconfig
33
33
34 if sys.version_info[0] >= 3:
34 if sys.version_info[0] >= 3:
35 printf = eval('print')
35 printf = eval('print')
36 libdir_escape = 'unicode_escape'
36 libdir_escape = 'unicode_escape'
37
37
38 def sysstr(s):
38 def sysstr(s):
39 return s.decode('latin-1')
39 return s.decode('latin-1')
40
40
41
41
42 else:
42 else:
43 libdir_escape = 'string_escape'
43 libdir_escape = 'string_escape'
44
44
45 def printf(*args, **kwargs):
45 def printf(*args, **kwargs):
46 f = kwargs.get('file', sys.stdout)
46 f = kwargs.get('file', sys.stdout)
47 end = kwargs.get('end', '\n')
47 end = kwargs.get('end', '\n')
48 f.write(b' '.join(args) + end)
48 f.write(b' '.join(args) + end)
49
49
50 def sysstr(s):
50 def sysstr(s):
51 return s
51 return s
52
52
53
53
54 # Attempt to guide users to a modern pip - this means that 2.6 users
54 # Attempt to guide users to a modern pip - this means that 2.6 users
55 # should have a chance of getting a 4.2 release, and when we ratchet
55 # should have a chance of getting a 4.2 release, and when we ratchet
56 # the version requirement forward again hopefully everyone will get
56 # the version requirement forward again hopefully everyone will get
57 # something that works for them.
57 # something that works for them.
58 if sys.version_info < (2, 7, 0, 'final'):
58 if sys.version_info < (2, 7, 0, 'final'):
59 pip_message = (
59 pip_message = (
60 'This may be due to an out of date pip. '
60 'This may be due to an out of date pip. '
61 'Make sure you have pip >= 9.0.1.'
61 'Make sure you have pip >= 9.0.1.'
62 )
62 )
63 try:
63 try:
64 import pip
64 import pip
65
65
66 pip_version = tuple([int(x) for x in pip.__version__.split('.')[:3]])
66 pip_version = tuple([int(x) for x in pip.__version__.split('.')[:3]])
67 if pip_version < (9, 0, 1):
67 if pip_version < (9, 0, 1):
68 pip_message = (
68 pip_message = (
69 'Your pip version is out of date, please install '
69 'Your pip version is out of date, please install '
70 'pip >= 9.0.1. pip {} detected.'.format(pip.__version__)
70 'pip >= 9.0.1. pip {} detected.'.format(pip.__version__)
71 )
71 )
72 else:
72 else:
73 # pip is new enough - it must be something else
73 # pip is new enough - it must be something else
74 pip_message = ''
74 pip_message = ''
75 except Exception:
75 except Exception:
76 pass
76 pass
77 error = """
77 error = """
78 Mercurial does not support Python older than 2.7.
78 Mercurial does not support Python older than 2.7.
79 Python {py} detected.
79 Python {py} detected.
80 {pip}
80 {pip}
81 """.format(
81 """.format(
82 py=sys.version_info, pip=pip_message
82 py=sys.version_info, pip=pip_message
83 )
83 )
84 printf(error, file=sys.stderr)
84 printf(error, file=sys.stderr)
85 sys.exit(1)
85 sys.exit(1)
86
86
87 if sys.version_info[0] >= 3:
87 if sys.version_info[0] >= 3:
88 DYLIB_SUFFIX = sysconfig.get_config_vars()['EXT_SUFFIX']
88 DYLIB_SUFFIX = sysconfig.get_config_vars()['EXT_SUFFIX']
89 else:
89 else:
90 # deprecated in Python 3
90 # deprecated in Python 3
91 DYLIB_SUFFIX = sysconfig.get_config_vars()['SO']
91 DYLIB_SUFFIX = sysconfig.get_config_vars()['SO']
92
92
93 # Solaris Python packaging brain damage
93 # Solaris Python packaging brain damage
94 try:
94 try:
95 import hashlib
95 import hashlib
96
96
97 sha = hashlib.sha1()
97 sha = hashlib.sha1()
98 except ImportError:
98 except ImportError:
99 try:
99 try:
100 import sha
100 import sha
101
101
102 sha.sha # silence unused import warning
102 sha.sha # silence unused import warning
103 except ImportError:
103 except ImportError:
104 raise SystemExit(
104 raise SystemExit(
105 "Couldn't import standard hashlib (incomplete Python install)."
105 "Couldn't import standard hashlib (incomplete Python install)."
106 )
106 )
107
107
108 try:
108 try:
109 import zlib
109 import zlib
110
110
111 zlib.compressobj # silence unused import warning
111 zlib.compressobj # silence unused import warning
112 except ImportError:
112 except ImportError:
113 raise SystemExit(
113 raise SystemExit(
114 "Couldn't import standard zlib (incomplete Python install)."
114 "Couldn't import standard zlib (incomplete Python install)."
115 )
115 )
116
116
117 # The base IronPython distribution (as of 2.7.1) doesn't support bz2
117 # The base IronPython distribution (as of 2.7.1) doesn't support bz2
118 isironpython = False
118 isironpython = False
119 try:
119 try:
120 isironpython = (
120 isironpython = (
121 platform.python_implementation().lower().find("ironpython") != -1
121 platform.python_implementation().lower().find("ironpython") != -1
122 )
122 )
123 except AttributeError:
123 except AttributeError:
124 pass
124 pass
125
125
126 if isironpython:
126 if isironpython:
127 sys.stderr.write("warning: IronPython detected (no bz2 support)\n")
127 sys.stderr.write("warning: IronPython detected (no bz2 support)\n")
128 else:
128 else:
129 try:
129 try:
130 import bz2
130 import bz2
131
131
132 bz2.BZ2Compressor # silence unused import warning
132 bz2.BZ2Compressor # silence unused import warning
133 except ImportError:
133 except ImportError:
134 raise SystemExit(
134 raise SystemExit(
135 "Couldn't import standard bz2 (incomplete Python install)."
135 "Couldn't import standard bz2 (incomplete Python install)."
136 )
136 )
137
137
138 ispypy = "PyPy" in sys.version
138 ispypy = "PyPy" in sys.version
139
139
140 hgrustext = os.environ.get('HGWITHRUSTEXT')
140 hgrustext = os.environ.get('HGWITHRUSTEXT')
141 # TODO record it for proper rebuild upon changes
141 # TODO record it for proper rebuild upon changes
142 # (see mercurial/__modulepolicy__.py)
142 # (see mercurial/__modulepolicy__.py)
143 if hgrustext != 'cpython' and hgrustext is not None:
143 if hgrustext != 'cpython' and hgrustext is not None:
144 hgrustext = 'direct-ffi'
144 hgrustext = 'direct-ffi'
145
145
146 import ctypes
146 import ctypes
147 import errno
147 import errno
148 import stat, subprocess, time
148 import stat, subprocess, time
149 import re
149 import re
150 import shutil
150 import shutil
151 import tempfile
151 import tempfile
152 from distutils import log
152 from distutils import log
153
153
154 # We have issues with setuptools on some platforms and builders. Until
154 # We have issues with setuptools on some platforms and builders. Until
155 # those are resolved, setuptools is opt-in except for platforms where
155 # those are resolved, setuptools is opt-in except for platforms where
156 # we don't have issues.
156 # we don't have issues.
157 issetuptools = os.name == 'nt' or 'FORCE_SETUPTOOLS' in os.environ
157 issetuptools = os.name == 'nt' or 'FORCE_SETUPTOOLS' in os.environ
158 if issetuptools:
158 if issetuptools:
159 from setuptools import setup
159 from setuptools import setup
160 else:
160 else:
161 from distutils.core import setup
161 from distutils.core import setup
162 from distutils.ccompiler import new_compiler
162 from distutils.ccompiler import new_compiler
163 from distutils.core import Command, Extension
163 from distutils.core import Command, Extension
164 from distutils.dist import Distribution
164 from distutils.dist import Distribution
165 from distutils.command.build import build
165 from distutils.command.build import build
166 from distutils.command.build_ext import build_ext
166 from distutils.command.build_ext import build_ext
167 from distutils.command.build_py import build_py
167 from distutils.command.build_py import build_py
168 from distutils.command.build_scripts import build_scripts
168 from distutils.command.build_scripts import build_scripts
169 from distutils.command.install import install
169 from distutils.command.install import install
170 from distutils.command.install_lib import install_lib
170 from distutils.command.install_lib import install_lib
171 from distutils.command.install_scripts import install_scripts
171 from distutils.command.install_scripts import install_scripts
172 from distutils.spawn import spawn, find_executable
172 from distutils.spawn import spawn, find_executable
173 from distutils import file_util
173 from distutils import file_util
174 from distutils.errors import (
174 from distutils.errors import (
175 CCompilerError,
175 CCompilerError,
176 DistutilsError,
176 DistutilsError,
177 DistutilsExecError,
177 DistutilsExecError,
178 )
178 )
179 from distutils.sysconfig import get_python_inc, get_config_var
179 from distutils.sysconfig import get_python_inc, get_config_var
180 from distutils.version import StrictVersion
180 from distutils.version import StrictVersion
181
181
182 # Explain to distutils.StrictVersion how our release candidates are versionned
182 # Explain to distutils.StrictVersion how our release candidates are versionned
183 StrictVersion.version_re = re.compile(r'^(\d+)\.(\d+)(\.(\d+))?-?(rc(\d+))?$')
183 StrictVersion.version_re = re.compile(r'^(\d+)\.(\d+)(\.(\d+))?-?(rc(\d+))?$')
184
184
185
185
186 def write_if_changed(path, content):
186 def write_if_changed(path, content):
187 """Write content to a file iff the content hasn't changed."""
187 """Write content to a file iff the content hasn't changed."""
188 if os.path.exists(path):
188 if os.path.exists(path):
189 with open(path, 'rb') as fh:
189 with open(path, 'rb') as fh:
190 current = fh.read()
190 current = fh.read()
191 else:
191 else:
192 current = b''
192 current = b''
193
193
194 if current != content:
194 if current != content:
195 with open(path, 'wb') as fh:
195 with open(path, 'wb') as fh:
196 fh.write(content)
196 fh.write(content)
197
197
198
198
199 scripts = ['hg']
199 scripts = ['hg']
200 if os.name == 'nt':
200 if os.name == 'nt':
201 # We remove hg.bat if we are able to build hg.exe.
201 # We remove hg.bat if we are able to build hg.exe.
202 scripts.append('contrib/win32/hg.bat')
202 scripts.append('contrib/win32/hg.bat')
203
203
204
204
205 def cancompile(cc, code):
205 def cancompile(cc, code):
206 tmpdir = tempfile.mkdtemp(prefix='hg-install-')
206 tmpdir = tempfile.mkdtemp(prefix='hg-install-')
207 devnull = oldstderr = None
207 devnull = oldstderr = None
208 try:
208 try:
209 fname = os.path.join(tmpdir, 'testcomp.c')
209 fname = os.path.join(tmpdir, 'testcomp.c')
210 f = open(fname, 'w')
210 f = open(fname, 'w')
211 f.write(code)
211 f.write(code)
212 f.close()
212 f.close()
213 # Redirect stderr to /dev/null to hide any error messages
213 # Redirect stderr to /dev/null to hide any error messages
214 # from the compiler.
214 # from the compiler.
215 # This will have to be changed if we ever have to check
215 # This will have to be changed if we ever have to check
216 # for a function on Windows.
216 # for a function on Windows.
217 devnull = open('/dev/null', 'w')
217 devnull = open('/dev/null', 'w')
218 oldstderr = os.dup(sys.stderr.fileno())
218 oldstderr = os.dup(sys.stderr.fileno())
219 os.dup2(devnull.fileno(), sys.stderr.fileno())
219 os.dup2(devnull.fileno(), sys.stderr.fileno())
220 objects = cc.compile([fname], output_dir=tmpdir)
220 objects = cc.compile([fname], output_dir=tmpdir)
221 cc.link_executable(objects, os.path.join(tmpdir, "a.out"))
221 cc.link_executable(objects, os.path.join(tmpdir, "a.out"))
222 return True
222 return True
223 except Exception:
223 except Exception:
224 return False
224 return False
225 finally:
225 finally:
226 if oldstderr is not None:
226 if oldstderr is not None:
227 os.dup2(oldstderr, sys.stderr.fileno())
227 os.dup2(oldstderr, sys.stderr.fileno())
228 if devnull is not None:
228 if devnull is not None:
229 devnull.close()
229 devnull.close()
230 shutil.rmtree(tmpdir)
230 shutil.rmtree(tmpdir)
231
231
232
232
233 # simplified version of distutils.ccompiler.CCompiler.has_function
233 # simplified version of distutils.ccompiler.CCompiler.has_function
234 # that actually removes its temporary files.
234 # that actually removes its temporary files.
235 def hasfunction(cc, funcname):
235 def hasfunction(cc, funcname):
236 code = 'int main(void) { %s(); }\n' % funcname
236 code = 'int main(void) { %s(); }\n' % funcname
237 return cancompile(cc, code)
237 return cancompile(cc, code)
238
238
239
239
240 def hasheader(cc, headername):
240 def hasheader(cc, headername):
241 code = '#include <%s>\nint main(void) { return 0; }\n' % headername
241 code = '#include <%s>\nint main(void) { return 0; }\n' % headername
242 return cancompile(cc, code)
242 return cancompile(cc, code)
243
243
244
244
245 # py2exe needs to be installed to work
245 # py2exe needs to be installed to work
246 try:
246 try:
247 import py2exe
247 import py2exe
248
248
249 py2exe.Distribution # silence unused import warning
249 py2exe.Distribution # silence unused import warning
250 py2exeloaded = True
250 py2exeloaded = True
251 # import py2exe's patched Distribution class
251 # import py2exe's patched Distribution class
252 from distutils.core import Distribution
252 from distutils.core import Distribution
253 except ImportError:
253 except ImportError:
254 py2exeloaded = False
254 py2exeloaded = False
255
255
256
256
257 def runcmd(cmd, env, cwd=None):
257 def runcmd(cmd, env, cwd=None):
258 p = subprocess.Popen(
258 p = subprocess.Popen(
259 cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE, env=env, cwd=cwd
259 cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE, env=env, cwd=cwd
260 )
260 )
261 out, err = p.communicate()
261 out, err = p.communicate()
262 return p.returncode, out, err
262 return p.returncode, out, err
263
263
264
264
265 class hgcommand(object):
265 class hgcommand(object):
266 def __init__(self, cmd, env):
266 def __init__(self, cmd, env):
267 self.cmd = cmd
267 self.cmd = cmd
268 self.env = env
268 self.env = env
269
269
270 def run(self, args):
270 def run(self, args):
271 cmd = self.cmd + args
271 cmd = self.cmd + args
272 returncode, out, err = runcmd(cmd, self.env)
272 returncode, out, err = runcmd(cmd, self.env)
273 err = filterhgerr(err)
273 err = filterhgerr(err)
274 if err or returncode != 0:
274 if err or returncode != 0:
275 printf("stderr from '%s':" % (' '.join(cmd)), file=sys.stderr)
275 printf("stderr from '%s':" % (' '.join(cmd)), file=sys.stderr)
276 printf(err, file=sys.stderr)
276 printf(err, file=sys.stderr)
277 return ''
277 return ''
278 return out
278 return out
279
279
280
280
281 def filterhgerr(err):
281 def filterhgerr(err):
282 # If root is executing setup.py, but the repository is owned by
282 # If root is executing setup.py, but the repository is owned by
283 # another user (as in "sudo python setup.py install") we will get
283 # another user (as in "sudo python setup.py install") we will get
284 # trust warnings since the .hg/hgrc file is untrusted. That is
284 # trust warnings since the .hg/hgrc file is untrusted. That is
285 # fine, we don't want to load it anyway. Python may warn about
285 # fine, we don't want to load it anyway. Python may warn about
286 # a missing __init__.py in mercurial/locale, we also ignore that.
286 # a missing __init__.py in mercurial/locale, we also ignore that.
287 err = [
287 err = [
288 e
288 e
289 for e in err.splitlines()
289 for e in err.splitlines()
290 if (
290 if (
291 not e.startswith(b'not trusting file')
291 not e.startswith(b'not trusting file')
292 and not e.startswith(b'warning: Not importing')
292 and not e.startswith(b'warning: Not importing')
293 and not e.startswith(b'obsolete feature not enabled')
293 and not e.startswith(b'obsolete feature not enabled')
294 and not e.startswith(b'*** failed to import extension')
294 and not e.startswith(b'*** failed to import extension')
295 and not e.startswith(b'devel-warn:')
295 and not e.startswith(b'devel-warn:')
296 and not (
296 and not (
297 e.startswith(b'(third party extension')
297 e.startswith(b'(third party extension')
298 and e.endswith(b'or newer of Mercurial; disabling)')
298 and e.endswith(b'or newer of Mercurial; disabling)')
299 )
299 )
300 )
300 )
301 ]
301 ]
302 return b'\n'.join(b' ' + e for e in err)
302 return b'\n'.join(b' ' + e for e in err)
303
303
304
304
305 def findhg():
305 def findhg():
306 """Try to figure out how we should invoke hg for examining the local
306 """Try to figure out how we should invoke hg for examining the local
307 repository contents.
307 repository contents.
308
308
309 Returns an hgcommand object."""
309 Returns an hgcommand object."""
310 # By default, prefer the "hg" command in the user's path. This was
310 # By default, prefer the "hg" command in the user's path. This was
311 # presumably the hg command that the user used to create this repository.
311 # presumably the hg command that the user used to create this repository.
312 #
312 #
313 # This repository may require extensions or other settings that would not
313 # This repository may require extensions or other settings that would not
314 # be enabled by running the hg script directly from this local repository.
314 # be enabled by running the hg script directly from this local repository.
315 hgenv = os.environ.copy()
315 hgenv = os.environ.copy()
316 # Use HGPLAIN to disable hgrc settings that would change output formatting,
316 # Use HGPLAIN to disable hgrc settings that would change output formatting,
317 # and disable localization for the same reasons.
317 # and disable localization for the same reasons.
318 hgenv['HGPLAIN'] = '1'
318 hgenv['HGPLAIN'] = '1'
319 hgenv['LANGUAGE'] = 'C'
319 hgenv['LANGUAGE'] = 'C'
320 hgcmd = ['hg']
320 hgcmd = ['hg']
321 # Run a simple "hg log" command just to see if using hg from the user's
321 # Run a simple "hg log" command just to see if using hg from the user's
322 # path works and can successfully interact with this repository. Windows
322 # path works and can successfully interact with this repository. Windows
323 # gives precedence to hg.exe in the current directory, so fall back to the
323 # gives precedence to hg.exe in the current directory, so fall back to the
324 # python invocation of local hg, where pythonXY.dll can always be found.
324 # python invocation of local hg, where pythonXY.dll can always be found.
325 check_cmd = ['log', '-r.', '-Ttest']
325 check_cmd = ['log', '-r.', '-Ttest']
326 if os.name != 'nt':
326 if os.name != 'nt':
327 try:
327 try:
328 retcode, out, err = runcmd(hgcmd + check_cmd, hgenv)
328 retcode, out, err = runcmd(hgcmd + check_cmd, hgenv)
329 except EnvironmentError:
329 except EnvironmentError:
330 retcode = -1
330 retcode = -1
331 if retcode == 0 and not filterhgerr(err):
331 if retcode == 0 and not filterhgerr(err):
332 return hgcommand(hgcmd, hgenv)
332 return hgcommand(hgcmd, hgenv)
333
333
334 # Fall back to trying the local hg installation.
334 # Fall back to trying the local hg installation.
335 hgenv = localhgenv()
335 hgenv = localhgenv()
336 hgcmd = [sys.executable, 'hg']
336 hgcmd = [sys.executable, 'hg']
337 try:
337 try:
338 retcode, out, err = runcmd(hgcmd + check_cmd, hgenv)
338 retcode, out, err = runcmd(hgcmd + check_cmd, hgenv)
339 except EnvironmentError:
339 except EnvironmentError:
340 retcode = -1
340 retcode = -1
341 if retcode == 0 and not filterhgerr(err):
341 if retcode == 0 and not filterhgerr(err):
342 return hgcommand(hgcmd, hgenv)
342 return hgcommand(hgcmd, hgenv)
343
343
344 raise SystemExit(
344 raise SystemExit(
345 'Unable to find a working hg binary to extract the '
345 'Unable to find a working hg binary to extract the '
346 'version from the repository tags'
346 'version from the repository tags'
347 )
347 )
348
348
349
349
350 def localhgenv():
350 def localhgenv():
351 """Get an environment dictionary to use for invoking or importing
351 """Get an environment dictionary to use for invoking or importing
352 mercurial from the local repository."""
352 mercurial from the local repository."""
353 # Execute hg out of this directory with a custom environment which takes
353 # Execute hg out of this directory with a custom environment which takes
354 # care to not use any hgrc files and do no localization.
354 # care to not use any hgrc files and do no localization.
355 env = {
355 env = {
356 'HGMODULEPOLICY': 'py',
356 'HGMODULEPOLICY': 'py',
357 'HGRCPATH': '',
357 'HGRCPATH': '',
358 'LANGUAGE': 'C',
358 'LANGUAGE': 'C',
359 'PATH': '',
359 'PATH': '',
360 } # make pypi modules that use os.environ['PATH'] happy
360 } # make pypi modules that use os.environ['PATH'] happy
361 if 'LD_LIBRARY_PATH' in os.environ:
361 if 'LD_LIBRARY_PATH' in os.environ:
362 env['LD_LIBRARY_PATH'] = os.environ['LD_LIBRARY_PATH']
362 env['LD_LIBRARY_PATH'] = os.environ['LD_LIBRARY_PATH']
363 if 'SystemRoot' in os.environ:
363 if 'SystemRoot' in os.environ:
364 # SystemRoot is required by Windows to load various DLLs. See:
364 # SystemRoot is required by Windows to load various DLLs. See:
365 # https://bugs.python.org/issue13524#msg148850
365 # https://bugs.python.org/issue13524#msg148850
366 env['SystemRoot'] = os.environ['SystemRoot']
366 env['SystemRoot'] = os.environ['SystemRoot']
367 return env
367 return env
368
368
369
369
370 version = ''
370 version = ''
371
371
372 if os.path.isdir('.hg'):
372 if os.path.isdir('.hg'):
373 hg = findhg()
373 hg = findhg()
374 cmd = ['log', '-r', '.', '--template', '{tags}\n']
374 cmd = ['log', '-r', '.', '--template', '{tags}\n']
375 numerictags = [t for t in sysstr(hg.run(cmd)).split() if t[0:1].isdigit()]
375 numerictags = [t for t in sysstr(hg.run(cmd)).split() if t[0:1].isdigit()]
376 hgid = sysstr(hg.run(['id', '-i'])).strip()
376 hgid = sysstr(hg.run(['id', '-i'])).strip()
377 if not hgid:
377 if not hgid:
378 # Bail out if hg is having problems interacting with this repository,
378 # Bail out if hg is having problems interacting with this repository,
379 # rather than falling through and producing a bogus version number.
379 # rather than falling through and producing a bogus version number.
380 # Continuing with an invalid version number will break extensions
380 # Continuing with an invalid version number will break extensions
381 # that define minimumhgversion.
381 # that define minimumhgversion.
382 raise SystemExit('Unable to determine hg version from local repository')
382 raise SystemExit('Unable to determine hg version from local repository')
383 if numerictags: # tag(s) found
383 if numerictags: # tag(s) found
384 version = numerictags[-1]
384 version = numerictags[-1]
385 if hgid.endswith('+'): # propagate the dirty status to the tag
385 if hgid.endswith('+'): # propagate the dirty status to the tag
386 version += '+'
386 version += '+'
387 else: # no tag found
387 else: # no tag found
388 ltagcmd = ['parents', '--template', '{latesttag}']
388 ltagcmd = ['parents', '--template', '{latesttag}']
389 ltag = sysstr(hg.run(ltagcmd))
389 ltag = sysstr(hg.run(ltagcmd))
390 changessincecmd = ['log', '-T', 'x\n', '-r', "only(.,'%s')" % ltag]
390 changessincecmd = ['log', '-T', 'x\n', '-r', "only(.,'%s')" % ltag]
391 changessince = len(hg.run(changessincecmd).splitlines())
391 changessince = len(hg.run(changessincecmd).splitlines())
392 version = '%s+%s-%s' % (ltag, changessince, hgid)
392 version = '%s+%s-%s' % (ltag, changessince, hgid)
393 if version.endswith('+'):
393 if version.endswith('+'):
394 version += time.strftime('%Y%m%d')
394 version += time.strftime('%Y%m%d')
395 elif os.path.exists('.hg_archival.txt'):
395 elif os.path.exists('.hg_archival.txt'):
396 kw = dict(
396 kw = dict(
397 [[t.strip() for t in l.split(':', 1)] for l in open('.hg_archival.txt')]
397 [[t.strip() for t in l.split(':', 1)] for l in open('.hg_archival.txt')]
398 )
398 )
399 if 'tag' in kw:
399 if 'tag' in kw:
400 version = kw['tag']
400 version = kw['tag']
401 elif 'latesttag' in kw:
401 elif 'latesttag' in kw:
402 if 'changessincelatesttag' in kw:
402 if 'changessincelatesttag' in kw:
403 version = '%(latesttag)s+%(changessincelatesttag)s-%(node).12s' % kw
403 version = '%(latesttag)s+%(changessincelatesttag)s-%(node).12s' % kw
404 else:
404 else:
405 version = '%(latesttag)s+%(latesttagdistance)s-%(node).12s' % kw
405 version = '%(latesttag)s+%(latesttagdistance)s-%(node).12s' % kw
406 else:
406 else:
407 version = kw.get('node', '')[:12]
407 version = kw.get('node', '')[:12]
408
408
409 if version:
409 if version:
410 versionb = version
410 versionb = version
411 if not isinstance(versionb, bytes):
411 if not isinstance(versionb, bytes):
412 versionb = versionb.encode('ascii')
412 versionb = versionb.encode('ascii')
413
413
414 write_if_changed(
414 write_if_changed(
415 'mercurial/__version__.py',
415 'mercurial/__version__.py',
416 b''.join(
416 b''.join(
417 [
417 [
418 b'# this file is autogenerated by setup.py\n'
418 b'# this file is autogenerated by setup.py\n'
419 b'version = b"%s"\n' % versionb,
419 b'version = b"%s"\n' % versionb,
420 ]
420 ]
421 ),
421 ),
422 )
422 )
423
423
424 try:
424 try:
425 oldpolicy = os.environ.get('HGMODULEPOLICY', None)
425 oldpolicy = os.environ.get('HGMODULEPOLICY', None)
426 os.environ['HGMODULEPOLICY'] = 'py'
426 os.environ['HGMODULEPOLICY'] = 'py'
427 from mercurial import __version__
427 from mercurial import __version__
428
428
429 version = __version__.version
429 version = __version__.version
430 except ImportError:
430 except ImportError:
431 version = b'unknown'
431 version = b'unknown'
432 finally:
432 finally:
433 if oldpolicy is None:
433 if oldpolicy is None:
434 del os.environ['HGMODULEPOLICY']
434 del os.environ['HGMODULEPOLICY']
435 else:
435 else:
436 os.environ['HGMODULEPOLICY'] = oldpolicy
436 os.environ['HGMODULEPOLICY'] = oldpolicy
437
437
438
438
439 class hgbuild(build):
439 class hgbuild(build):
440 # Insert hgbuildmo first so that files in mercurial/locale/ are found
440 # Insert hgbuildmo first so that files in mercurial/locale/ are found
441 # when build_py is run next.
441 # when build_py is run next.
442 sub_commands = [('build_mo', None)] + build.sub_commands
442 sub_commands = [('build_mo', None)] + build.sub_commands
443
443
444
444
445 class hgbuildmo(build):
445 class hgbuildmo(build):
446
446
447 description = "build translations (.mo files)"
447 description = "build translations (.mo files)"
448
448
449 def run(self):
449 def run(self):
450 if not find_executable('msgfmt'):
450 if not find_executable('msgfmt'):
451 self.warn(
451 self.warn(
452 "could not find msgfmt executable, no translations "
452 "could not find msgfmt executable, no translations "
453 "will be built"
453 "will be built"
454 )
454 )
455 return
455 return
456
456
457 podir = 'i18n'
457 podir = 'i18n'
458 if not os.path.isdir(podir):
458 if not os.path.isdir(podir):
459 self.warn("could not find %s/ directory" % podir)
459 self.warn("could not find %s/ directory" % podir)
460 return
460 return
461
461
462 join = os.path.join
462 join = os.path.join
463 for po in os.listdir(podir):
463 for po in os.listdir(podir):
464 if not po.endswith('.po'):
464 if not po.endswith('.po'):
465 continue
465 continue
466 pofile = join(podir, po)
466 pofile = join(podir, po)
467 modir = join('locale', po[:-3], 'LC_MESSAGES')
467 modir = join('locale', po[:-3], 'LC_MESSAGES')
468 mofile = join(modir, 'hg.mo')
468 mofile = join(modir, 'hg.mo')
469 mobuildfile = join('mercurial', mofile)
469 mobuildfile = join('mercurial', mofile)
470 cmd = ['msgfmt', '-v', '-o', mobuildfile, pofile]
470 cmd = ['msgfmt', '-v', '-o', mobuildfile, pofile]
471 if sys.platform != 'sunos5':
471 if sys.platform != 'sunos5':
472 # msgfmt on Solaris does not know about -c
472 # msgfmt on Solaris does not know about -c
473 cmd.append('-c')
473 cmd.append('-c')
474 self.mkpath(join('mercurial', modir))
474 self.mkpath(join('mercurial', modir))
475 self.make_file([pofile], mobuildfile, spawn, (cmd,))
475 self.make_file([pofile], mobuildfile, spawn, (cmd,))
476
476
477
477
478 class hgdist(Distribution):
478 class hgdist(Distribution):
479 pure = False
479 pure = False
480 rust = hgrustext is not None
480 rust = hgrustext is not None
481 cffi = ispypy
481 cffi = ispypy
482
482
483 global_options = Distribution.global_options + [
483 global_options = Distribution.global_options + [
484 ('pure', None, "use pure (slow) Python code instead of C extensions"),
484 ('pure', None, "use pure (slow) Python code instead of C extensions"),
485 ('rust', None, "use Rust extensions additionally to C extensions"),
485 ('rust', None, "use Rust extensions additionally to C extensions"),
486 ]
486 ]
487
487
488 def has_ext_modules(self):
488 def has_ext_modules(self):
489 # self.ext_modules is emptied in hgbuildpy.finalize_options which is
489 # self.ext_modules is emptied in hgbuildpy.finalize_options which is
490 # too late for some cases
490 # too late for some cases
491 return not self.pure and Distribution.has_ext_modules(self)
491 return not self.pure and Distribution.has_ext_modules(self)
492
492
493
493
494 # This is ugly as a one-liner. So use a variable.
494 # This is ugly as a one-liner. So use a variable.
495 buildextnegops = dict(getattr(build_ext, 'negative_options', {}))
495 buildextnegops = dict(getattr(build_ext, 'negative_options', {}))
496 buildextnegops['no-zstd'] = 'zstd'
496 buildextnegops['no-zstd'] = 'zstd'
497 buildextnegops['no-rust'] = 'rust'
497 buildextnegops['no-rust'] = 'rust'
498
498
499
499
500 class hgbuildext(build_ext):
500 class hgbuildext(build_ext):
501 user_options = build_ext.user_options + [
501 user_options = build_ext.user_options + [
502 ('zstd', None, 'compile zstd bindings [default]'),
502 ('zstd', None, 'compile zstd bindings [default]'),
503 ('no-zstd', None, 'do not compile zstd bindings'),
503 ('no-zstd', None, 'do not compile zstd bindings'),
504 (
504 (
505 'rust',
505 'rust',
506 None,
506 None,
507 'compile Rust extensions if they are in use '
507 'compile Rust extensions if they are in use '
508 '(requires Cargo) [default]',
508 '(requires Cargo) [default]',
509 ),
509 ),
510 ('no-rust', None, 'do not compile Rust extensions'),
510 ('no-rust', None, 'do not compile Rust extensions'),
511 ]
511 ]
512
512
513 boolean_options = build_ext.boolean_options + ['zstd', 'rust']
513 boolean_options = build_ext.boolean_options + ['zstd', 'rust']
514 negative_opt = buildextnegops
514 negative_opt = buildextnegops
515
515
516 def initialize_options(self):
516 def initialize_options(self):
517 self.zstd = True
517 self.zstd = True
518 self.rust = True
518 self.rust = True
519
519
520 return build_ext.initialize_options(self)
520 return build_ext.initialize_options(self)
521
521
522 def finalize_options(self):
522 def finalize_options(self):
523 # Unless overridden by the end user, build extensions in parallel.
523 # Unless overridden by the end user, build extensions in parallel.
524 # Only influences behavior on Python 3.5+.
524 # Only influences behavior on Python 3.5+.
525 if getattr(self, 'parallel', None) is None:
525 if getattr(self, 'parallel', None) is None:
526 self.parallel = True
526 self.parallel = True
527
527
528 return build_ext.finalize_options(self)
528 return build_ext.finalize_options(self)
529
529
530 def build_extensions(self):
530 def build_extensions(self):
531 ruststandalones = [
531 ruststandalones = [
532 e for e in self.extensions if isinstance(e, RustStandaloneExtension)
532 e for e in self.extensions if isinstance(e, RustStandaloneExtension)
533 ]
533 ]
534 self.extensions = [
534 self.extensions = [
535 e for e in self.extensions if e not in ruststandalones
535 e for e in self.extensions if e not in ruststandalones
536 ]
536 ]
537 # Filter out zstd if disabled via argument.
537 # Filter out zstd if disabled via argument.
538 if not self.zstd:
538 if not self.zstd:
539 self.extensions = [
539 self.extensions = [
540 e for e in self.extensions if e.name != 'mercurial.zstd'
540 e for e in self.extensions if e.name != 'mercurial.zstd'
541 ]
541 ]
542
542
543 # Build Rust standalon extensions if it'll be used
543 # Build Rust standalon extensions if it'll be used
544 # and its build is not explictely disabled (for external build
544 # and its build is not explictely disabled (for external build
545 # as Linux distributions would do)
545 # as Linux distributions would do)
546 if self.distribution.rust and self.rust and hgrustext != 'direct-ffi':
546 if self.distribution.rust and self.rust and hgrustext != 'direct-ffi':
547 for rustext in ruststandalones:
547 for rustext in ruststandalones:
548 rustext.build('' if self.inplace else self.build_lib)
548 rustext.build('' if self.inplace else self.build_lib)
549
549
550 return build_ext.build_extensions(self)
550 return build_ext.build_extensions(self)
551
551
552 def build_extension(self, ext):
552 def build_extension(self, ext):
553 if (
553 if (
554 self.distribution.rust
554 self.distribution.rust
555 and self.rust
555 and self.rust
556 and isinstance(ext, RustExtension)
556 and isinstance(ext, RustExtension)
557 ):
557 ):
558 ext.rustbuild()
558 ext.rustbuild()
559 try:
559 try:
560 build_ext.build_extension(self, ext)
560 build_ext.build_extension(self, ext)
561 except CCompilerError:
561 except CCompilerError:
562 if not getattr(ext, 'optional', False):
562 if not getattr(ext, 'optional', False):
563 raise
563 raise
564 log.warn(
564 log.warn(
565 "Failed to build optional extension '%s' (skipping)", ext.name
565 "Failed to build optional extension '%s' (skipping)", ext.name
566 )
566 )
567
567
568
568
569 class hgbuildscripts(build_scripts):
569 class hgbuildscripts(build_scripts):
570 def run(self):
570 def run(self):
571 if os.name != 'nt' or self.distribution.pure:
571 if os.name != 'nt' or self.distribution.pure:
572 return build_scripts.run(self)
572 return build_scripts.run(self)
573
573
574 exebuilt = False
574 exebuilt = False
575 try:
575 try:
576 self.run_command('build_hgexe')
576 self.run_command('build_hgexe')
577 exebuilt = True
577 exebuilt = True
578 except (DistutilsError, CCompilerError):
578 except (DistutilsError, CCompilerError):
579 log.warn('failed to build optional hg.exe')
579 log.warn('failed to build optional hg.exe')
580
580
581 if exebuilt:
581 if exebuilt:
582 # Copying hg.exe to the scripts build directory ensures it is
582 # Copying hg.exe to the scripts build directory ensures it is
583 # installed by the install_scripts command.
583 # installed by the install_scripts command.
584 hgexecommand = self.get_finalized_command('build_hgexe')
584 hgexecommand = self.get_finalized_command('build_hgexe')
585 dest = os.path.join(self.build_dir, 'hg.exe')
585 dest = os.path.join(self.build_dir, 'hg.exe')
586 self.mkpath(self.build_dir)
586 self.mkpath(self.build_dir)
587 self.copy_file(hgexecommand.hgexepath, dest)
587 self.copy_file(hgexecommand.hgexepath, dest)
588
588
589 # Remove hg.bat because it is redundant with hg.exe.
589 # Remove hg.bat because it is redundant with hg.exe.
590 self.scripts.remove('contrib/win32/hg.bat')
590 self.scripts.remove('contrib/win32/hg.bat')
591
591
592 return build_scripts.run(self)
592 return build_scripts.run(self)
593
593
594
594
595 class hgbuildpy(build_py):
595 class hgbuildpy(build_py):
596 def finalize_options(self):
596 def finalize_options(self):
597 build_py.finalize_options(self)
597 build_py.finalize_options(self)
598
598
599 if self.distribution.pure:
599 if self.distribution.pure:
600 self.distribution.ext_modules = []
600 self.distribution.ext_modules = []
601 elif self.distribution.cffi:
601 elif self.distribution.cffi:
602 from mercurial.cffi import (
602 from mercurial.cffi import (
603 bdiffbuild,
603 bdiffbuild,
604 mpatchbuild,
604 mpatchbuild,
605 )
605 )
606
606
607 exts = [
607 exts = [
608 mpatchbuild.ffi.distutils_extension(),
608 mpatchbuild.ffi.distutils_extension(),
609 bdiffbuild.ffi.distutils_extension(),
609 bdiffbuild.ffi.distutils_extension(),
610 ]
610 ]
611 # cffi modules go here
611 # cffi modules go here
612 if sys.platform == 'darwin':
612 if sys.platform == 'darwin':
613 from mercurial.cffi import osutilbuild
613 from mercurial.cffi import osutilbuild
614
614
615 exts.append(osutilbuild.ffi.distutils_extension())
615 exts.append(osutilbuild.ffi.distutils_extension())
616 self.distribution.ext_modules = exts
616 self.distribution.ext_modules = exts
617 else:
617 else:
618 h = os.path.join(get_python_inc(), 'Python.h')
618 h = os.path.join(get_python_inc(), 'Python.h')
619 if not os.path.exists(h):
619 if not os.path.exists(h):
620 raise SystemExit(
620 raise SystemExit(
621 'Python headers are required to build '
621 'Python headers are required to build '
622 'Mercurial but weren\'t found in %s' % h
622 'Mercurial but weren\'t found in %s' % h
623 )
623 )
624
624
625 def run(self):
625 def run(self):
626 basepath = os.path.join(self.build_lib, 'mercurial')
626 basepath = os.path.join(self.build_lib, 'mercurial')
627 self.mkpath(basepath)
627 self.mkpath(basepath)
628
628
629 rust = self.distribution.rust
629 rust = self.distribution.rust
630 if self.distribution.pure:
630 if self.distribution.pure:
631 modulepolicy = 'py'
631 modulepolicy = 'py'
632 elif self.build_lib == '.':
632 elif self.build_lib == '.':
633 # in-place build should run without rebuilding and Rust extensions
633 # in-place build should run without rebuilding and Rust extensions
634 modulepolicy = 'rust+c-allow' if rust else 'allow'
634 modulepolicy = 'rust+c-allow' if rust else 'allow'
635 else:
635 else:
636 modulepolicy = 'rust+c' if rust else 'c'
636 modulepolicy = 'rust+c' if rust else 'c'
637
637
638 content = b''.join(
638 content = b''.join(
639 [
639 [
640 b'# this file is autogenerated by setup.py\n',
640 b'# this file is autogenerated by setup.py\n',
641 b'modulepolicy = b"%s"\n' % modulepolicy.encode('ascii'),
641 b'modulepolicy = b"%s"\n' % modulepolicy.encode('ascii'),
642 ]
642 ]
643 )
643 )
644 write_if_changed(os.path.join(basepath, '__modulepolicy__.py'), content)
644 write_if_changed(os.path.join(basepath, '__modulepolicy__.py'), content)
645
645
646 build_py.run(self)
646 build_py.run(self)
647
647
648
648
649 class buildhgextindex(Command):
649 class buildhgextindex(Command):
650 description = 'generate prebuilt index of hgext (for frozen package)'
650 description = 'generate prebuilt index of hgext (for frozen package)'
651 user_options = []
651 user_options = []
652 _indexfilename = 'hgext/__index__.py'
652 _indexfilename = 'hgext/__index__.py'
653
653
654 def initialize_options(self):
654 def initialize_options(self):
655 pass
655 pass
656
656
657 def finalize_options(self):
657 def finalize_options(self):
658 pass
658 pass
659
659
660 def run(self):
660 def run(self):
661 if os.path.exists(self._indexfilename):
661 if os.path.exists(self._indexfilename):
662 with open(self._indexfilename, 'w') as f:
662 with open(self._indexfilename, 'w') as f:
663 f.write('# empty\n')
663 f.write('# empty\n')
664
664
665 # here no extension enabled, disabled() lists up everything
665 # here no extension enabled, disabled() lists up everything
666 code = (
666 code = (
667 'import pprint; from mercurial import extensions; '
667 'import pprint; from mercurial import extensions; '
668 'pprint.pprint(extensions.disabled())'
668 'pprint.pprint(extensions.disabled())'
669 )
669 )
670 returncode, out, err = runcmd(
670 returncode, out, err = runcmd(
671 [sys.executable, '-c', code], localhgenv()
671 [sys.executable, '-c', code], localhgenv()
672 )
672 )
673 if err or returncode != 0:
673 if err or returncode != 0:
674 raise DistutilsExecError(err)
674 raise DistutilsExecError(err)
675
675
676 with open(self._indexfilename, 'wb') as f:
676 with open(self._indexfilename, 'wb') as f:
677 f.write(b'# this file is autogenerated by setup.py\n')
677 f.write(b'# this file is autogenerated by setup.py\n')
678 f.write(b'docs = ')
678 f.write(b'docs = ')
679 f.write(out)
679 f.write(out)
680
680
681
681
682 class buildhgexe(build_ext):
682 class buildhgexe(build_ext):
683 description = 'compile hg.exe from mercurial/exewrapper.c'
683 description = 'compile hg.exe from mercurial/exewrapper.c'
684 user_options = build_ext.user_options + [
684 user_options = build_ext.user_options + [
685 (
685 (
686 'long-paths-support',
686 'long-paths-support',
687 None,
687 None,
688 'enable support for long paths on '
688 'enable support for long paths on '
689 'Windows (off by default and '
689 'Windows (off by default and '
690 'experimental)',
690 'experimental)',
691 ),
691 ),
692 ]
692 ]
693
693
694 LONG_PATHS_MANIFEST = """
694 LONG_PATHS_MANIFEST = """
695 <?xml version="1.0" encoding="UTF-8" standalone="yes"?>
695 <?xml version="1.0" encoding="UTF-8" standalone="yes"?>
696 <assembly xmlns="urn:schemas-microsoft-com:asm.v1" manifestVersion="1.0">
696 <assembly xmlns="urn:schemas-microsoft-com:asm.v1" manifestVersion="1.0">
697 <application>
697 <application>
698 <windowsSettings
698 <windowsSettings
699 xmlns:ws2="http://schemas.microsoft.com/SMI/2016/WindowsSettings">
699 xmlns:ws2="http://schemas.microsoft.com/SMI/2016/WindowsSettings">
700 <ws2:longPathAware>true</ws2:longPathAware>
700 <ws2:longPathAware>true</ws2:longPathAware>
701 </windowsSettings>
701 </windowsSettings>
702 </application>
702 </application>
703 </assembly>"""
703 </assembly>"""
704
704
705 def initialize_options(self):
705 def initialize_options(self):
706 build_ext.initialize_options(self)
706 build_ext.initialize_options(self)
707 self.long_paths_support = False
707 self.long_paths_support = False
708
708
709 def build_extensions(self):
709 def build_extensions(self):
710 if os.name != 'nt':
710 if os.name != 'nt':
711 return
711 return
712 if isinstance(self.compiler, HackedMingw32CCompiler):
712 if isinstance(self.compiler, HackedMingw32CCompiler):
713 self.compiler.compiler_so = self.compiler.compiler # no -mdll
713 self.compiler.compiler_so = self.compiler.compiler # no -mdll
714 self.compiler.dll_libraries = [] # no -lmsrvc90
714 self.compiler.dll_libraries = [] # no -lmsrvc90
715
715
716 pythonlib = None
716 pythonlib = None
717
717
718 if getattr(sys, 'dllhandle', None):
718 if getattr(sys, 'dllhandle', None):
719 # Different Python installs can have different Python library
719 # Different Python installs can have different Python library
720 # names. e.g. the official CPython distribution uses pythonXY.dll
720 # names. e.g. the official CPython distribution uses pythonXY.dll
721 # and MinGW uses libpythonX.Y.dll.
721 # and MinGW uses libpythonX.Y.dll.
722 _kernel32 = ctypes.windll.kernel32
722 _kernel32 = ctypes.windll.kernel32
723 _kernel32.GetModuleFileNameA.argtypes = [
723 _kernel32.GetModuleFileNameA.argtypes = [
724 ctypes.c_void_p,
724 ctypes.c_void_p,
725 ctypes.c_void_p,
725 ctypes.c_void_p,
726 ctypes.c_ulong,
726 ctypes.c_ulong,
727 ]
727 ]
728 _kernel32.GetModuleFileNameA.restype = ctypes.c_ulong
728 _kernel32.GetModuleFileNameA.restype = ctypes.c_ulong
729 size = 1000
729 size = 1000
730 buf = ctypes.create_string_buffer(size + 1)
730 buf = ctypes.create_string_buffer(size + 1)
731 filelen = _kernel32.GetModuleFileNameA(
731 filelen = _kernel32.GetModuleFileNameA(
732 sys.dllhandle, ctypes.byref(buf), size
732 sys.dllhandle, ctypes.byref(buf), size
733 )
733 )
734
734
735 if filelen > 0 and filelen != size:
735 if filelen > 0 and filelen != size:
736 dllbasename = os.path.basename(buf.value)
736 dllbasename = os.path.basename(buf.value)
737 if not dllbasename.lower().endswith(b'.dll'):
737 if not dllbasename.lower().endswith(b'.dll'):
738 raise SystemExit(
738 raise SystemExit(
739 'Python DLL does not end with .dll: %s' % dllbasename
739 'Python DLL does not end with .dll: %s' % dllbasename
740 )
740 )
741 pythonlib = dllbasename[:-4]
741 pythonlib = dllbasename[:-4]
742
742
743 if not pythonlib:
743 if not pythonlib:
744 log.warn(
744 log.warn(
745 'could not determine Python DLL filename; assuming pythonXY'
745 'could not determine Python DLL filename; assuming pythonXY'
746 )
746 )
747
747
748 hv = sys.hexversion
748 hv = sys.hexversion
749 pythonlib = b'python%d%d' % (hv >> 24, (hv >> 16) & 0xFF)
749 pythonlib = b'python%d%d' % (hv >> 24, (hv >> 16) & 0xFF)
750
750
751 log.info('using %s as Python library name' % pythonlib)
751 log.info('using %s as Python library name' % pythonlib)
752 with open('mercurial/hgpythonlib.h', 'wb') as f:
752 with open('mercurial/hgpythonlib.h', 'wb') as f:
753 f.write(b'/* this file is autogenerated by setup.py */\n')
753 f.write(b'/* this file is autogenerated by setup.py */\n')
754 f.write(b'#define HGPYTHONLIB "%s"\n' % pythonlib)
754 f.write(b'#define HGPYTHONLIB "%s"\n' % pythonlib)
755
755
756 macros = None
756 macros = None
757 if sys.version_info[0] >= 3:
757 if sys.version_info[0] >= 3:
758 macros = [('_UNICODE', None), ('UNICODE', None)]
758 macros = [('_UNICODE', None), ('UNICODE', None)]
759
759
760 objects = self.compiler.compile(
760 objects = self.compiler.compile(
761 ['mercurial/exewrapper.c'],
761 ['mercurial/exewrapper.c'],
762 output_dir=self.build_temp,
762 output_dir=self.build_temp,
763 macros=macros,
763 macros=macros,
764 )
764 )
765 dir = os.path.dirname(self.get_ext_fullpath('dummy'))
765 dir = os.path.dirname(self.get_ext_fullpath('dummy'))
766 self.hgtarget = os.path.join(dir, 'hg')
766 self.hgtarget = os.path.join(dir, 'hg')
767 self.compiler.link_executable(
767 self.compiler.link_executable(
768 objects, self.hgtarget, libraries=[], output_dir=self.build_temp
768 objects, self.hgtarget, libraries=[], output_dir=self.build_temp
769 )
769 )
770 if self.long_paths_support:
770 if self.long_paths_support:
771 self.addlongpathsmanifest()
771 self.addlongpathsmanifest()
772
772
773 def addlongpathsmanifest(self):
773 def addlongpathsmanifest(self):
774 r"""Add manifest pieces so that hg.exe understands long paths
774 r"""Add manifest pieces so that hg.exe understands long paths
775
775
776 This is an EXPERIMENTAL feature, use with care.
776 This is an EXPERIMENTAL feature, use with care.
777 To enable long paths support, one needs to do two things:
777 To enable long paths support, one needs to do two things:
778 - build Mercurial with --long-paths-support option
778 - build Mercurial with --long-paths-support option
779 - change HKLM\SYSTEM\CurrentControlSet\Control\FileSystem\
779 - change HKLM\SYSTEM\CurrentControlSet\Control\FileSystem\
780 LongPathsEnabled to have value 1.
780 LongPathsEnabled to have value 1.
781
781
782 Please ignore 'warning 81010002: Unrecognized Element "longPathAware"';
782 Please ignore 'warning 81010002: Unrecognized Element "longPathAware"';
783 it happens because Mercurial uses mt.exe circa 2008, which is not
783 it happens because Mercurial uses mt.exe circa 2008, which is not
784 yet aware of long paths support in the manifest (I think so at least).
784 yet aware of long paths support in the manifest (I think so at least).
785 This does not stop mt.exe from embedding/merging the XML properly.
785 This does not stop mt.exe from embedding/merging the XML properly.
786
786
787 Why resource #1 should be used for .exe manifests? I don't know and
787 Why resource #1 should be used for .exe manifests? I don't know and
788 wasn't able to find an explanation for mortals. But it seems to work.
788 wasn't able to find an explanation for mortals. But it seems to work.
789 """
789 """
790 exefname = self.compiler.executable_filename(self.hgtarget)
790 exefname = self.compiler.executable_filename(self.hgtarget)
791 fdauto, manfname = tempfile.mkstemp(suffix='.hg.exe.manifest')
791 fdauto, manfname = tempfile.mkstemp(suffix='.hg.exe.manifest')
792 os.close(fdauto)
792 os.close(fdauto)
793 with open(manfname, 'w') as f:
793 with open(manfname, 'w') as f:
794 f.write(self.LONG_PATHS_MANIFEST)
794 f.write(self.LONG_PATHS_MANIFEST)
795 log.info("long paths manifest is written to '%s'" % manfname)
795 log.info("long paths manifest is written to '%s'" % manfname)
796 inputresource = '-inputresource:%s;#1' % exefname
796 inputresource = '-inputresource:%s;#1' % exefname
797 outputresource = '-outputresource:%s;#1' % exefname
797 outputresource = '-outputresource:%s;#1' % exefname
798 log.info("running mt.exe to update hg.exe's manifest in-place")
798 log.info("running mt.exe to update hg.exe's manifest in-place")
799 # supplying both -manifest and -inputresource to mt.exe makes
799 # supplying both -manifest and -inputresource to mt.exe makes
800 # it merge the embedded and supplied manifests in the -outputresource
800 # it merge the embedded and supplied manifests in the -outputresource
801 self.spawn(
801 self.spawn(
802 [
802 [
803 'mt.exe',
803 'mt.exe',
804 '-nologo',
804 '-nologo',
805 '-manifest',
805 '-manifest',
806 manfname,
806 manfname,
807 inputresource,
807 inputresource,
808 outputresource,
808 outputresource,
809 ]
809 ]
810 )
810 )
811 log.info("done updating hg.exe's manifest")
811 log.info("done updating hg.exe's manifest")
812 os.remove(manfname)
812 os.remove(manfname)
813
813
814 @property
814 @property
815 def hgexepath(self):
815 def hgexepath(self):
816 dir = os.path.dirname(self.get_ext_fullpath('dummy'))
816 dir = os.path.dirname(self.get_ext_fullpath('dummy'))
817 return os.path.join(self.build_temp, dir, 'hg.exe')
817 return os.path.join(self.build_temp, dir, 'hg.exe')
818
818
819
819
820 class hgbuilddoc(Command):
820 class hgbuilddoc(Command):
821 description = 'build documentation'
821 description = 'build documentation'
822 user_options = [
822 user_options = [
823 ('man', None, 'generate man pages'),
823 ('man', None, 'generate man pages'),
824 ('html', None, 'generate html pages'),
824 ('html', None, 'generate html pages'),
825 ]
825 ]
826
826
827 def initialize_options(self):
827 def initialize_options(self):
828 self.man = None
828 self.man = None
829 self.html = None
829 self.html = None
830
830
831 def finalize_options(self):
831 def finalize_options(self):
832 # If --man or --html are set, only generate what we're told to.
832 # If --man or --html are set, only generate what we're told to.
833 # Otherwise generate everything.
833 # Otherwise generate everything.
834 have_subset = self.man is not None or self.html is not None
834 have_subset = self.man is not None or self.html is not None
835
835
836 if have_subset:
836 if have_subset:
837 self.man = True if self.man else False
837 self.man = True if self.man else False
838 self.html = True if self.html else False
838 self.html = True if self.html else False
839 else:
839 else:
840 self.man = True
840 self.man = True
841 self.html = True
841 self.html = True
842
842
843 def run(self):
843 def run(self):
844 def normalizecrlf(p):
844 def normalizecrlf(p):
845 with open(p, 'rb') as fh:
845 with open(p, 'rb') as fh:
846 orig = fh.read()
846 orig = fh.read()
847
847
848 if b'\r\n' not in orig:
848 if b'\r\n' not in orig:
849 return
849 return
850
850
851 log.info('normalizing %s to LF line endings' % p)
851 log.info('normalizing %s to LF line endings' % p)
852 with open(p, 'wb') as fh:
852 with open(p, 'wb') as fh:
853 fh.write(orig.replace(b'\r\n', b'\n'))
853 fh.write(orig.replace(b'\r\n', b'\n'))
854
854
855 def gentxt(root):
855 def gentxt(root):
856 txt = 'doc/%s.txt' % root
856 txt = 'doc/%s.txt' % root
857 log.info('generating %s' % txt)
857 log.info('generating %s' % txt)
858 res, out, err = runcmd(
858 res, out, err = runcmd(
859 [sys.executable, 'gendoc.py', root], os.environ, cwd='doc'
859 [sys.executable, 'gendoc.py', root], os.environ, cwd='doc'
860 )
860 )
861 if res:
861 if res:
862 raise SystemExit(
862 raise SystemExit(
863 'error running gendoc.py: %s' % '\n'.join([out, err])
863 'error running gendoc.py: %s' % '\n'.join([out, err])
864 )
864 )
865
865
866 with open(txt, 'wb') as fh:
866 with open(txt, 'wb') as fh:
867 fh.write(out)
867 fh.write(out)
868
868
869 def gengendoc(root):
869 def gengendoc(root):
870 gendoc = 'doc/%s.gendoc.txt' % root
870 gendoc = 'doc/%s.gendoc.txt' % root
871
871
872 log.info('generating %s' % gendoc)
872 log.info('generating %s' % gendoc)
873 res, out, err = runcmd(
873 res, out, err = runcmd(
874 [sys.executable, 'gendoc.py', '%s.gendoc' % root],
874 [sys.executable, 'gendoc.py', '%s.gendoc' % root],
875 os.environ,
875 os.environ,
876 cwd='doc',
876 cwd='doc',
877 )
877 )
878 if res:
878 if res:
879 raise SystemExit(
879 raise SystemExit(
880 'error running gendoc: %s' % '\n'.join([out, err])
880 'error running gendoc: %s' % '\n'.join([out, err])
881 )
881 )
882
882
883 with open(gendoc, 'wb') as fh:
883 with open(gendoc, 'wb') as fh:
884 fh.write(out)
884 fh.write(out)
885
885
886 def genman(root):
886 def genman(root):
887 log.info('generating doc/%s' % root)
887 log.info('generating doc/%s' % root)
888 res, out, err = runcmd(
888 res, out, err = runcmd(
889 [
889 [
890 sys.executable,
890 sys.executable,
891 'runrst',
891 'runrst',
892 'hgmanpage',
892 'hgmanpage',
893 '--halt',
893 '--halt',
894 'warning',
894 'warning',
895 '--strip-elements-with-class',
895 '--strip-elements-with-class',
896 'htmlonly',
896 'htmlonly',
897 '%s.txt' % root,
897 '%s.txt' % root,
898 root,
898 root,
899 ],
899 ],
900 os.environ,
900 os.environ,
901 cwd='doc',
901 cwd='doc',
902 )
902 )
903 if res:
903 if res:
904 raise SystemExit(
904 raise SystemExit(
905 'error running runrst: %s' % '\n'.join([out, err])
905 'error running runrst: %s' % '\n'.join([out, err])
906 )
906 )
907
907
908 normalizecrlf('doc/%s' % root)
908 normalizecrlf('doc/%s' % root)
909
909
910 def genhtml(root):
910 def genhtml(root):
911 log.info('generating doc/%s.html' % root)
911 log.info('generating doc/%s.html' % root)
912 res, out, err = runcmd(
912 res, out, err = runcmd(
913 [
913 [
914 sys.executable,
914 sys.executable,
915 'runrst',
915 'runrst',
916 'html',
916 'html',
917 '--halt',
917 '--halt',
918 'warning',
918 'warning',
919 '--link-stylesheet',
919 '--link-stylesheet',
920 '--stylesheet-path',
920 '--stylesheet-path',
921 'style.css',
921 'style.css',
922 '%s.txt' % root,
922 '%s.txt' % root,
923 '%s.html' % root,
923 '%s.html' % root,
924 ],
924 ],
925 os.environ,
925 os.environ,
926 cwd='doc',
926 cwd='doc',
927 )
927 )
928 if res:
928 if res:
929 raise SystemExit(
929 raise SystemExit(
930 'error running runrst: %s' % '\n'.join([out, err])
930 'error running runrst: %s' % '\n'.join([out, err])
931 )
931 )
932
932
933 normalizecrlf('doc/%s.html' % root)
933 normalizecrlf('doc/%s.html' % root)
934
934
935 # This logic is duplicated in doc/Makefile.
935 # This logic is duplicated in doc/Makefile.
936 sources = set(
936 sources = set(
937 f
937 f
938 for f in os.listdir('mercurial/helptext')
938 for f in os.listdir('mercurial/helptext')
939 if re.search(r'[0-9]\.txt$', f)
939 if re.search(r'[0-9]\.txt$', f)
940 )
940 )
941
941
942 # common.txt is a one-off.
942 # common.txt is a one-off.
943 gentxt('common')
943 gentxt('common')
944
944
945 for source in sorted(sources):
945 for source in sorted(sources):
946 assert source[-4:] == '.txt'
946 assert source[-4:] == '.txt'
947 root = source[:-4]
947 root = source[:-4]
948
948
949 gentxt(root)
949 gentxt(root)
950 gengendoc(root)
950 gengendoc(root)
951
951
952 if self.man:
952 if self.man:
953 genman(root)
953 genman(root)
954 if self.html:
954 if self.html:
955 genhtml(root)
955 genhtml(root)
956
956
957
957
958 class hginstall(install):
958 class hginstall(install):
959
959
960 user_options = install.user_options + [
960 user_options = install.user_options + [
961 (
961 (
962 'old-and-unmanageable',
962 'old-and-unmanageable',
963 None,
963 None,
964 'noop, present for eggless setuptools compat',
964 'noop, present for eggless setuptools compat',
965 ),
965 ),
966 (
966 (
967 'single-version-externally-managed',
967 'single-version-externally-managed',
968 None,
968 None,
969 'noop, present for eggless setuptools compat',
969 'noop, present for eggless setuptools compat',
970 ),
970 ),
971 ]
971 ]
972
972
973 # Also helps setuptools not be sad while we refuse to create eggs.
973 # Also helps setuptools not be sad while we refuse to create eggs.
974 single_version_externally_managed = True
974 single_version_externally_managed = True
975
975
976 def get_sub_commands(self):
976 def get_sub_commands(self):
977 # Screen out egg related commands to prevent egg generation. But allow
977 # Screen out egg related commands to prevent egg generation. But allow
978 # mercurial.egg-info generation, since that is part of modern
978 # mercurial.egg-info generation, since that is part of modern
979 # packaging.
979 # packaging.
980 excl = set(['bdist_egg'])
980 excl = set(['bdist_egg'])
981 return filter(lambda x: x not in excl, install.get_sub_commands(self))
981 return filter(lambda x: x not in excl, install.get_sub_commands(self))
982
982
983
983
984 class hginstalllib(install_lib):
984 class hginstalllib(install_lib):
985 '''
985 '''
986 This is a specialization of install_lib that replaces the copy_file used
986 This is a specialization of install_lib that replaces the copy_file used
987 there so that it supports setting the mode of files after copying them,
987 there so that it supports setting the mode of files after copying them,
988 instead of just preserving the mode that the files originally had. If your
988 instead of just preserving the mode that the files originally had. If your
989 system has a umask of something like 027, preserving the permissions when
989 system has a umask of something like 027, preserving the permissions when
990 copying will lead to a broken install.
990 copying will lead to a broken install.
991
991
992 Note that just passing keep_permissions=False to copy_file would be
992 Note that just passing keep_permissions=False to copy_file would be
993 insufficient, as it might still be applying a umask.
993 insufficient, as it might still be applying a umask.
994 '''
994 '''
995
995
996 def run(self):
996 def run(self):
997 realcopyfile = file_util.copy_file
997 realcopyfile = file_util.copy_file
998
998
999 def copyfileandsetmode(*args, **kwargs):
999 def copyfileandsetmode(*args, **kwargs):
1000 src, dst = args[0], args[1]
1000 src, dst = args[0], args[1]
1001 dst, copied = realcopyfile(*args, **kwargs)
1001 dst, copied = realcopyfile(*args, **kwargs)
1002 if copied:
1002 if copied:
1003 st = os.stat(src)
1003 st = os.stat(src)
1004 # Persist executable bit (apply it to group and other if user
1004 # Persist executable bit (apply it to group and other if user
1005 # has it)
1005 # has it)
1006 if st[stat.ST_MODE] & stat.S_IXUSR:
1006 if st[stat.ST_MODE] & stat.S_IXUSR:
1007 setmode = int('0755', 8)
1007 setmode = int('0755', 8)
1008 else:
1008 else:
1009 setmode = int('0644', 8)
1009 setmode = int('0644', 8)
1010 m = stat.S_IMODE(st[stat.ST_MODE])
1010 m = stat.S_IMODE(st[stat.ST_MODE])
1011 m = (m & ~int('0777', 8)) | setmode
1011 m = (m & ~int('0777', 8)) | setmode
1012 os.chmod(dst, m)
1012 os.chmod(dst, m)
1013
1013
1014 file_util.copy_file = copyfileandsetmode
1014 file_util.copy_file = copyfileandsetmode
1015 try:
1015 try:
1016 install_lib.run(self)
1016 install_lib.run(self)
1017 finally:
1017 finally:
1018 file_util.copy_file = realcopyfile
1018 file_util.copy_file = realcopyfile
1019
1019
1020
1020
1021 class hginstallscripts(install_scripts):
1021 class hginstallscripts(install_scripts):
1022 '''
1022 '''
1023 This is a specialization of install_scripts that replaces the @LIBDIR@ with
1023 This is a specialization of install_scripts that replaces the @LIBDIR@ with
1024 the configured directory for modules. If possible, the path is made relative
1024 the configured directory for modules. If possible, the path is made relative
1025 to the directory for scripts.
1025 to the directory for scripts.
1026 '''
1026 '''
1027
1027
1028 def initialize_options(self):
1028 def initialize_options(self):
1029 install_scripts.initialize_options(self)
1029 install_scripts.initialize_options(self)
1030
1030
1031 self.install_lib = None
1031 self.install_lib = None
1032
1032
1033 def finalize_options(self):
1033 def finalize_options(self):
1034 install_scripts.finalize_options(self)
1034 install_scripts.finalize_options(self)
1035 self.set_undefined_options('install', ('install_lib', 'install_lib'))
1035 self.set_undefined_options('install', ('install_lib', 'install_lib'))
1036
1036
1037 def run(self):
1037 def run(self):
1038 install_scripts.run(self)
1038 install_scripts.run(self)
1039
1039
1040 # It only makes sense to replace @LIBDIR@ with the install path if
1040 # It only makes sense to replace @LIBDIR@ with the install path if
1041 # the install path is known. For wheels, the logic below calculates
1041 # the install path is known. For wheels, the logic below calculates
1042 # the libdir to be "../..". This is because the internal layout of a
1042 # the libdir to be "../..". This is because the internal layout of a
1043 # wheel archive looks like:
1043 # wheel archive looks like:
1044 #
1044 #
1045 # mercurial-3.6.1.data/scripts/hg
1045 # mercurial-3.6.1.data/scripts/hg
1046 # mercurial/__init__.py
1046 # mercurial/__init__.py
1047 #
1047 #
1048 # When installing wheels, the subdirectories of the "<pkg>.data"
1048 # When installing wheels, the subdirectories of the "<pkg>.data"
1049 # directory are translated to system local paths and files therein
1049 # directory are translated to system local paths and files therein
1050 # are copied in place. The mercurial/* files are installed into the
1050 # are copied in place. The mercurial/* files are installed into the
1051 # site-packages directory. However, the site-packages directory
1051 # site-packages directory. However, the site-packages directory
1052 # isn't known until wheel install time. This means we have no clue
1052 # isn't known until wheel install time. This means we have no clue
1053 # at wheel generation time what the installed site-packages directory
1053 # at wheel generation time what the installed site-packages directory
1054 # will be. And, wheels don't appear to provide the ability to register
1054 # will be. And, wheels don't appear to provide the ability to register
1055 # custom code to run during wheel installation. This all means that
1055 # custom code to run during wheel installation. This all means that
1056 # we can't reliably set the libdir in wheels: the default behavior
1056 # we can't reliably set the libdir in wheels: the default behavior
1057 # of looking in sys.path must do.
1057 # of looking in sys.path must do.
1058
1058
1059 if (
1059 if (
1060 os.path.splitdrive(self.install_dir)[0]
1060 os.path.splitdrive(self.install_dir)[0]
1061 != os.path.splitdrive(self.install_lib)[0]
1061 != os.path.splitdrive(self.install_lib)[0]
1062 ):
1062 ):
1063 # can't make relative paths from one drive to another, so use an
1063 # can't make relative paths from one drive to another, so use an
1064 # absolute path instead
1064 # absolute path instead
1065 libdir = self.install_lib
1065 libdir = self.install_lib
1066 else:
1066 else:
1067 common = os.path.commonprefix((self.install_dir, self.install_lib))
1067 libdir = os.path.relpath(self.install_lib, self.install_dir)
1068 rest = self.install_dir[len(common) :]
1069 uplevel = len([n for n in os.path.split(rest) if n])
1070
1071 libdir = uplevel * ('..' + os.sep) + self.install_lib[len(common) :]
1072
1068
1073 for outfile in self.outfiles:
1069 for outfile in self.outfiles:
1074 with open(outfile, 'rb') as fp:
1070 with open(outfile, 'rb') as fp:
1075 data = fp.read()
1071 data = fp.read()
1076
1072
1077 # skip binary files
1073 # skip binary files
1078 if b'\0' in data:
1074 if b'\0' in data:
1079 continue
1075 continue
1080
1076
1081 # During local installs, the shebang will be rewritten to the final
1077 # During local installs, the shebang will be rewritten to the final
1082 # install path. During wheel packaging, the shebang has a special
1078 # install path. During wheel packaging, the shebang has a special
1083 # value.
1079 # value.
1084 if data.startswith(b'#!python'):
1080 if data.startswith(b'#!python'):
1085 log.info(
1081 log.info(
1086 'not rewriting @LIBDIR@ in %s because install path '
1082 'not rewriting @LIBDIR@ in %s because install path '
1087 'not known' % outfile
1083 'not known' % outfile
1088 )
1084 )
1089 continue
1085 continue
1090
1086
1091 data = data.replace(b'@LIBDIR@', libdir.encode(libdir_escape))
1087 data = data.replace(b'@LIBDIR@', libdir.encode(libdir_escape))
1092 with open(outfile, 'wb') as fp:
1088 with open(outfile, 'wb') as fp:
1093 fp.write(data)
1089 fp.write(data)
1094
1090
1095
1091
1096 # virtualenv installs custom distutils/__init__.py and
1092 # virtualenv installs custom distutils/__init__.py and
1097 # distutils/distutils.cfg files which essentially proxy back to the
1093 # distutils/distutils.cfg files which essentially proxy back to the
1098 # "real" distutils in the main Python install. The presence of this
1094 # "real" distutils in the main Python install. The presence of this
1099 # directory causes py2exe to pick up the "hacked" distutils package
1095 # directory causes py2exe to pick up the "hacked" distutils package
1100 # from the virtualenv and "import distutils" will fail from the py2exe
1096 # from the virtualenv and "import distutils" will fail from the py2exe
1101 # build because the "real" distutils files can't be located.
1097 # build because the "real" distutils files can't be located.
1102 #
1098 #
1103 # We work around this by monkeypatching the py2exe code finding Python
1099 # We work around this by monkeypatching the py2exe code finding Python
1104 # modules to replace the found virtualenv distutils modules with the
1100 # modules to replace the found virtualenv distutils modules with the
1105 # original versions via filesystem scanning. This is a bit hacky. But
1101 # original versions via filesystem scanning. This is a bit hacky. But
1106 # it allows us to use virtualenvs for py2exe packaging, which is more
1102 # it allows us to use virtualenvs for py2exe packaging, which is more
1107 # deterministic and reproducible.
1103 # deterministic and reproducible.
1108 #
1104 #
1109 # It's worth noting that the common StackOverflow suggestions for this
1105 # It's worth noting that the common StackOverflow suggestions for this
1110 # problem involve copying the original distutils files into the
1106 # problem involve copying the original distutils files into the
1111 # virtualenv or into the staging directory after setup() is invoked.
1107 # virtualenv or into the staging directory after setup() is invoked.
1112 # The former is very brittle and can easily break setup(). Our hacking
1108 # The former is very brittle and can easily break setup(). Our hacking
1113 # of the found modules routine has a similar result as copying the files
1109 # of the found modules routine has a similar result as copying the files
1114 # manually. But it makes fewer assumptions about how py2exe works and
1110 # manually. But it makes fewer assumptions about how py2exe works and
1115 # is less brittle.
1111 # is less brittle.
1116
1112
1117 # This only catches virtualenvs made with virtualenv (as opposed to
1113 # This only catches virtualenvs made with virtualenv (as opposed to
1118 # venv, which is likely what Python 3 uses).
1114 # venv, which is likely what Python 3 uses).
1119 py2exehacked = py2exeloaded and getattr(sys, 'real_prefix', None) is not None
1115 py2exehacked = py2exeloaded and getattr(sys, 'real_prefix', None) is not None
1120
1116
1121 if py2exehacked:
1117 if py2exehacked:
1122 from distutils.command.py2exe import py2exe as buildpy2exe
1118 from distutils.command.py2exe import py2exe as buildpy2exe
1123 from py2exe.mf import Module as py2exemodule
1119 from py2exe.mf import Module as py2exemodule
1124
1120
1125 class hgbuildpy2exe(buildpy2exe):
1121 class hgbuildpy2exe(buildpy2exe):
1126 def find_needed_modules(self, mf, files, modules):
1122 def find_needed_modules(self, mf, files, modules):
1127 res = buildpy2exe.find_needed_modules(self, mf, files, modules)
1123 res = buildpy2exe.find_needed_modules(self, mf, files, modules)
1128
1124
1129 # Replace virtualenv's distutils modules with the real ones.
1125 # Replace virtualenv's distutils modules with the real ones.
1130 modules = {}
1126 modules = {}
1131 for k, v in res.modules.items():
1127 for k, v in res.modules.items():
1132 if k != 'distutils' and not k.startswith('distutils.'):
1128 if k != 'distutils' and not k.startswith('distutils.'):
1133 modules[k] = v
1129 modules[k] = v
1134
1130
1135 res.modules = modules
1131 res.modules = modules
1136
1132
1137 import opcode
1133 import opcode
1138
1134
1139 distutilsreal = os.path.join(
1135 distutilsreal = os.path.join(
1140 os.path.dirname(opcode.__file__), 'distutils'
1136 os.path.dirname(opcode.__file__), 'distutils'
1141 )
1137 )
1142
1138
1143 for root, dirs, files in os.walk(distutilsreal):
1139 for root, dirs, files in os.walk(distutilsreal):
1144 for f in sorted(files):
1140 for f in sorted(files):
1145 if not f.endswith('.py'):
1141 if not f.endswith('.py'):
1146 continue
1142 continue
1147
1143
1148 full = os.path.join(root, f)
1144 full = os.path.join(root, f)
1149
1145
1150 parents = ['distutils']
1146 parents = ['distutils']
1151
1147
1152 if root != distutilsreal:
1148 if root != distutilsreal:
1153 rel = os.path.relpath(root, distutilsreal)
1149 rel = os.path.relpath(root, distutilsreal)
1154 parents.extend(p for p in rel.split(os.sep))
1150 parents.extend(p for p in rel.split(os.sep))
1155
1151
1156 modname = '%s.%s' % ('.'.join(parents), f[:-3])
1152 modname = '%s.%s' % ('.'.join(parents), f[:-3])
1157
1153
1158 if modname.startswith('distutils.tests.'):
1154 if modname.startswith('distutils.tests.'):
1159 continue
1155 continue
1160
1156
1161 if modname.endswith('.__init__'):
1157 if modname.endswith('.__init__'):
1162 modname = modname[: -len('.__init__')]
1158 modname = modname[: -len('.__init__')]
1163 path = os.path.dirname(full)
1159 path = os.path.dirname(full)
1164 else:
1160 else:
1165 path = None
1161 path = None
1166
1162
1167 res.modules[modname] = py2exemodule(
1163 res.modules[modname] = py2exemodule(
1168 modname, full, path=path
1164 modname, full, path=path
1169 )
1165 )
1170
1166
1171 if 'distutils' not in res.modules:
1167 if 'distutils' not in res.modules:
1172 raise SystemExit('could not find distutils modules')
1168 raise SystemExit('could not find distutils modules')
1173
1169
1174 return res
1170 return res
1175
1171
1176
1172
1177 cmdclass = {
1173 cmdclass = {
1178 'build': hgbuild,
1174 'build': hgbuild,
1179 'build_doc': hgbuilddoc,
1175 'build_doc': hgbuilddoc,
1180 'build_mo': hgbuildmo,
1176 'build_mo': hgbuildmo,
1181 'build_ext': hgbuildext,
1177 'build_ext': hgbuildext,
1182 'build_py': hgbuildpy,
1178 'build_py': hgbuildpy,
1183 'build_scripts': hgbuildscripts,
1179 'build_scripts': hgbuildscripts,
1184 'build_hgextindex': buildhgextindex,
1180 'build_hgextindex': buildhgextindex,
1185 'install': hginstall,
1181 'install': hginstall,
1186 'install_lib': hginstalllib,
1182 'install_lib': hginstalllib,
1187 'install_scripts': hginstallscripts,
1183 'install_scripts': hginstallscripts,
1188 'build_hgexe': buildhgexe,
1184 'build_hgexe': buildhgexe,
1189 }
1185 }
1190
1186
1191 if py2exehacked:
1187 if py2exehacked:
1192 cmdclass['py2exe'] = hgbuildpy2exe
1188 cmdclass['py2exe'] = hgbuildpy2exe
1193
1189
1194 packages = [
1190 packages = [
1195 'mercurial',
1191 'mercurial',
1196 'mercurial.cext',
1192 'mercurial.cext',
1197 'mercurial.cffi',
1193 'mercurial.cffi',
1198 'mercurial.defaultrc',
1194 'mercurial.defaultrc',
1199 'mercurial.helptext',
1195 'mercurial.helptext',
1200 'mercurial.helptext.internals',
1196 'mercurial.helptext.internals',
1201 'mercurial.hgweb',
1197 'mercurial.hgweb',
1202 'mercurial.interfaces',
1198 'mercurial.interfaces',
1203 'mercurial.pure',
1199 'mercurial.pure',
1204 'mercurial.thirdparty',
1200 'mercurial.thirdparty',
1205 'mercurial.thirdparty.attr',
1201 'mercurial.thirdparty.attr',
1206 'mercurial.thirdparty.zope',
1202 'mercurial.thirdparty.zope',
1207 'mercurial.thirdparty.zope.interface',
1203 'mercurial.thirdparty.zope.interface',
1208 'mercurial.utils',
1204 'mercurial.utils',
1209 'mercurial.revlogutils',
1205 'mercurial.revlogutils',
1210 'mercurial.testing',
1206 'mercurial.testing',
1211 'hgext',
1207 'hgext',
1212 'hgext.convert',
1208 'hgext.convert',
1213 'hgext.fsmonitor',
1209 'hgext.fsmonitor',
1214 'hgext.fastannotate',
1210 'hgext.fastannotate',
1215 'hgext.fsmonitor.pywatchman',
1211 'hgext.fsmonitor.pywatchman',
1216 'hgext.highlight',
1212 'hgext.highlight',
1217 'hgext.infinitepush',
1213 'hgext.infinitepush',
1218 'hgext.largefiles',
1214 'hgext.largefiles',
1219 'hgext.lfs',
1215 'hgext.lfs',
1220 'hgext.narrow',
1216 'hgext.narrow',
1221 'hgext.remotefilelog',
1217 'hgext.remotefilelog',
1222 'hgext.zeroconf',
1218 'hgext.zeroconf',
1223 'hgext3rd',
1219 'hgext3rd',
1224 'hgdemandimport',
1220 'hgdemandimport',
1225 ]
1221 ]
1226 if sys.version_info[0] == 2:
1222 if sys.version_info[0] == 2:
1227 packages.extend(
1223 packages.extend(
1228 [
1224 [
1229 'mercurial.thirdparty.concurrent',
1225 'mercurial.thirdparty.concurrent',
1230 'mercurial.thirdparty.concurrent.futures',
1226 'mercurial.thirdparty.concurrent.futures',
1231 ]
1227 ]
1232 )
1228 )
1233
1229
1234 if 'HG_PY2EXE_EXTRA_INSTALL_PACKAGES' in os.environ:
1230 if 'HG_PY2EXE_EXTRA_INSTALL_PACKAGES' in os.environ:
1235 # py2exe can't cope with namespace packages very well, so we have to
1231 # py2exe can't cope with namespace packages very well, so we have to
1236 # install any hgext3rd.* extensions that we want in the final py2exe
1232 # install any hgext3rd.* extensions that we want in the final py2exe
1237 # image here. This is gross, but you gotta do what you gotta do.
1233 # image here. This is gross, but you gotta do what you gotta do.
1238 packages.extend(os.environ['HG_PY2EXE_EXTRA_INSTALL_PACKAGES'].split(' '))
1234 packages.extend(os.environ['HG_PY2EXE_EXTRA_INSTALL_PACKAGES'].split(' '))
1239
1235
1240 common_depends = [
1236 common_depends = [
1241 'mercurial/bitmanipulation.h',
1237 'mercurial/bitmanipulation.h',
1242 'mercurial/compat.h',
1238 'mercurial/compat.h',
1243 'mercurial/cext/util.h',
1239 'mercurial/cext/util.h',
1244 ]
1240 ]
1245 common_include_dirs = ['mercurial']
1241 common_include_dirs = ['mercurial']
1246
1242
1247 osutil_cflags = []
1243 osutil_cflags = []
1248 osutil_ldflags = []
1244 osutil_ldflags = []
1249
1245
1250 # platform specific macros
1246 # platform specific macros
1251 for plat, func in [('bsd', 'setproctitle')]:
1247 for plat, func in [('bsd', 'setproctitle')]:
1252 if re.search(plat, sys.platform) and hasfunction(new_compiler(), func):
1248 if re.search(plat, sys.platform) and hasfunction(new_compiler(), func):
1253 osutil_cflags.append('-DHAVE_%s' % func.upper())
1249 osutil_cflags.append('-DHAVE_%s' % func.upper())
1254
1250
1255 for plat, macro, code in [
1251 for plat, macro, code in [
1256 (
1252 (
1257 'bsd|darwin',
1253 'bsd|darwin',
1258 'BSD_STATFS',
1254 'BSD_STATFS',
1259 '''
1255 '''
1260 #include <sys/param.h>
1256 #include <sys/param.h>
1261 #include <sys/mount.h>
1257 #include <sys/mount.h>
1262 int main() { struct statfs s; return sizeof(s.f_fstypename); }
1258 int main() { struct statfs s; return sizeof(s.f_fstypename); }
1263 ''',
1259 ''',
1264 ),
1260 ),
1265 (
1261 (
1266 'linux',
1262 'linux',
1267 'LINUX_STATFS',
1263 'LINUX_STATFS',
1268 '''
1264 '''
1269 #include <linux/magic.h>
1265 #include <linux/magic.h>
1270 #include <sys/vfs.h>
1266 #include <sys/vfs.h>
1271 int main() { struct statfs s; return sizeof(s.f_type); }
1267 int main() { struct statfs s; return sizeof(s.f_type); }
1272 ''',
1268 ''',
1273 ),
1269 ),
1274 ]:
1270 ]:
1275 if re.search(plat, sys.platform) and cancompile(new_compiler(), code):
1271 if re.search(plat, sys.platform) and cancompile(new_compiler(), code):
1276 osutil_cflags.append('-DHAVE_%s' % macro)
1272 osutil_cflags.append('-DHAVE_%s' % macro)
1277
1273
1278 if sys.platform == 'darwin':
1274 if sys.platform == 'darwin':
1279 osutil_ldflags += ['-framework', 'ApplicationServices']
1275 osutil_ldflags += ['-framework', 'ApplicationServices']
1280
1276
1281 xdiff_srcs = [
1277 xdiff_srcs = [
1282 'mercurial/thirdparty/xdiff/xdiffi.c',
1278 'mercurial/thirdparty/xdiff/xdiffi.c',
1283 'mercurial/thirdparty/xdiff/xprepare.c',
1279 'mercurial/thirdparty/xdiff/xprepare.c',
1284 'mercurial/thirdparty/xdiff/xutils.c',
1280 'mercurial/thirdparty/xdiff/xutils.c',
1285 ]
1281 ]
1286
1282
1287 xdiff_headers = [
1283 xdiff_headers = [
1288 'mercurial/thirdparty/xdiff/xdiff.h',
1284 'mercurial/thirdparty/xdiff/xdiff.h',
1289 'mercurial/thirdparty/xdiff/xdiffi.h',
1285 'mercurial/thirdparty/xdiff/xdiffi.h',
1290 'mercurial/thirdparty/xdiff/xinclude.h',
1286 'mercurial/thirdparty/xdiff/xinclude.h',
1291 'mercurial/thirdparty/xdiff/xmacros.h',
1287 'mercurial/thirdparty/xdiff/xmacros.h',
1292 'mercurial/thirdparty/xdiff/xprepare.h',
1288 'mercurial/thirdparty/xdiff/xprepare.h',
1293 'mercurial/thirdparty/xdiff/xtypes.h',
1289 'mercurial/thirdparty/xdiff/xtypes.h',
1294 'mercurial/thirdparty/xdiff/xutils.h',
1290 'mercurial/thirdparty/xdiff/xutils.h',
1295 ]
1291 ]
1296
1292
1297
1293
1298 class RustCompilationError(CCompilerError):
1294 class RustCompilationError(CCompilerError):
1299 """Exception class for Rust compilation errors."""
1295 """Exception class for Rust compilation errors."""
1300
1296
1301
1297
1302 class RustExtension(Extension):
1298 class RustExtension(Extension):
1303 """Base classes for concrete Rust Extension classes.
1299 """Base classes for concrete Rust Extension classes.
1304 """
1300 """
1305
1301
1306 rusttargetdir = os.path.join('rust', 'target', 'release')
1302 rusttargetdir = os.path.join('rust', 'target', 'release')
1307
1303
1308 def __init__(
1304 def __init__(
1309 self, mpath, sources, rustlibname, subcrate, py3_features=None, **kw
1305 self, mpath, sources, rustlibname, subcrate, py3_features=None, **kw
1310 ):
1306 ):
1311 Extension.__init__(self, mpath, sources, **kw)
1307 Extension.__init__(self, mpath, sources, **kw)
1312 srcdir = self.rustsrcdir = os.path.join('rust', subcrate)
1308 srcdir = self.rustsrcdir = os.path.join('rust', subcrate)
1313 self.py3_features = py3_features
1309 self.py3_features = py3_features
1314
1310
1315 # adding Rust source and control files to depends so that the extension
1311 # adding Rust source and control files to depends so that the extension
1316 # gets rebuilt if they've changed
1312 # gets rebuilt if they've changed
1317 self.depends.append(os.path.join(srcdir, 'Cargo.toml'))
1313 self.depends.append(os.path.join(srcdir, 'Cargo.toml'))
1318 cargo_lock = os.path.join(srcdir, 'Cargo.lock')
1314 cargo_lock = os.path.join(srcdir, 'Cargo.lock')
1319 if os.path.exists(cargo_lock):
1315 if os.path.exists(cargo_lock):
1320 self.depends.append(cargo_lock)
1316 self.depends.append(cargo_lock)
1321 for dirpath, subdir, fnames in os.walk(os.path.join(srcdir, 'src')):
1317 for dirpath, subdir, fnames in os.walk(os.path.join(srcdir, 'src')):
1322 self.depends.extend(
1318 self.depends.extend(
1323 os.path.join(dirpath, fname)
1319 os.path.join(dirpath, fname)
1324 for fname in fnames
1320 for fname in fnames
1325 if os.path.splitext(fname)[1] == '.rs'
1321 if os.path.splitext(fname)[1] == '.rs'
1326 )
1322 )
1327
1323
1328 @staticmethod
1324 @staticmethod
1329 def rustdylibsuffix():
1325 def rustdylibsuffix():
1330 """Return the suffix for shared libraries produced by rustc.
1326 """Return the suffix for shared libraries produced by rustc.
1331
1327
1332 See also: https://doc.rust-lang.org/reference/linkage.html
1328 See also: https://doc.rust-lang.org/reference/linkage.html
1333 """
1329 """
1334 if sys.platform == 'darwin':
1330 if sys.platform == 'darwin':
1335 return '.dylib'
1331 return '.dylib'
1336 elif os.name == 'nt':
1332 elif os.name == 'nt':
1337 return '.dll'
1333 return '.dll'
1338 else:
1334 else:
1339 return '.so'
1335 return '.so'
1340
1336
1341 def rustbuild(self):
1337 def rustbuild(self):
1342 env = os.environ.copy()
1338 env = os.environ.copy()
1343 if 'HGTEST_RESTOREENV' in env:
1339 if 'HGTEST_RESTOREENV' in env:
1344 # Mercurial tests change HOME to a temporary directory,
1340 # Mercurial tests change HOME to a temporary directory,
1345 # but, if installed with rustup, the Rust toolchain needs
1341 # but, if installed with rustup, the Rust toolchain needs
1346 # HOME to be correct (otherwise the 'no default toolchain'
1342 # HOME to be correct (otherwise the 'no default toolchain'
1347 # error message is issued and the build fails).
1343 # error message is issued and the build fails).
1348 # This happens currently with test-hghave.t, which does
1344 # This happens currently with test-hghave.t, which does
1349 # invoke this build.
1345 # invoke this build.
1350
1346
1351 # Unix only fix (os.path.expanduser not really reliable if
1347 # Unix only fix (os.path.expanduser not really reliable if
1352 # HOME is shadowed like this)
1348 # HOME is shadowed like this)
1353 import pwd
1349 import pwd
1354
1350
1355 env['HOME'] = pwd.getpwuid(os.getuid()).pw_dir
1351 env['HOME'] = pwd.getpwuid(os.getuid()).pw_dir
1356
1352
1357 cargocmd = ['cargo', 'rustc', '-vv', '--release']
1353 cargocmd = ['cargo', 'rustc', '-vv', '--release']
1358 if sys.version_info[0] == 3 and self.py3_features is not None:
1354 if sys.version_info[0] == 3 and self.py3_features is not None:
1359 cargocmd.extend(
1355 cargocmd.extend(
1360 ('--features', self.py3_features, '--no-default-features')
1356 ('--features', self.py3_features, '--no-default-features')
1361 )
1357 )
1362 cargocmd.append('--')
1358 cargocmd.append('--')
1363 if sys.platform == 'darwin':
1359 if sys.platform == 'darwin':
1364 cargocmd.extend(
1360 cargocmd.extend(
1365 ("-C", "link-arg=-undefined", "-C", "link-arg=dynamic_lookup")
1361 ("-C", "link-arg=-undefined", "-C", "link-arg=dynamic_lookup")
1366 )
1362 )
1367 try:
1363 try:
1368 subprocess.check_call(cargocmd, env=env, cwd=self.rustsrcdir)
1364 subprocess.check_call(cargocmd, env=env, cwd=self.rustsrcdir)
1369 except OSError as exc:
1365 except OSError as exc:
1370 if exc.errno == errno.ENOENT:
1366 if exc.errno == errno.ENOENT:
1371 raise RustCompilationError("Cargo not found")
1367 raise RustCompilationError("Cargo not found")
1372 elif exc.errno == errno.EACCES:
1368 elif exc.errno == errno.EACCES:
1373 raise RustCompilationError(
1369 raise RustCompilationError(
1374 "Cargo found, but permisssion to execute it is denied"
1370 "Cargo found, but permisssion to execute it is denied"
1375 )
1371 )
1376 else:
1372 else:
1377 raise
1373 raise
1378 except subprocess.CalledProcessError:
1374 except subprocess.CalledProcessError:
1379 raise RustCompilationError(
1375 raise RustCompilationError(
1380 "Cargo failed. Working directory: %r, "
1376 "Cargo failed. Working directory: %r, "
1381 "command: %r, environment: %r"
1377 "command: %r, environment: %r"
1382 % (self.rustsrcdir, cargocmd, env)
1378 % (self.rustsrcdir, cargocmd, env)
1383 )
1379 )
1384
1380
1385
1381
1386 class RustEnhancedExtension(RustExtension):
1382 class RustEnhancedExtension(RustExtension):
1387 """A C Extension, conditionally enhanced with Rust code.
1383 """A C Extension, conditionally enhanced with Rust code.
1388
1384
1389 If the HGRUSTEXT environment variable is set to something else
1385 If the HGRUSTEXT environment variable is set to something else
1390 than 'cpython', the Rust sources get compiled and linked within the
1386 than 'cpython', the Rust sources get compiled and linked within the
1391 C target shared library object.
1387 C target shared library object.
1392 """
1388 """
1393
1389
1394 def __init__(self, mpath, sources, rustlibname, subcrate, **kw):
1390 def __init__(self, mpath, sources, rustlibname, subcrate, **kw):
1395 RustExtension.__init__(
1391 RustExtension.__init__(
1396 self, mpath, sources, rustlibname, subcrate, **kw
1392 self, mpath, sources, rustlibname, subcrate, **kw
1397 )
1393 )
1398 if hgrustext != 'direct-ffi':
1394 if hgrustext != 'direct-ffi':
1399 return
1395 return
1400 self.extra_compile_args.append('-DWITH_RUST')
1396 self.extra_compile_args.append('-DWITH_RUST')
1401 self.libraries.append(rustlibname)
1397 self.libraries.append(rustlibname)
1402 self.library_dirs.append(self.rusttargetdir)
1398 self.library_dirs.append(self.rusttargetdir)
1403
1399
1404 def rustbuild(self):
1400 def rustbuild(self):
1405 if hgrustext == 'direct-ffi':
1401 if hgrustext == 'direct-ffi':
1406 RustExtension.rustbuild(self)
1402 RustExtension.rustbuild(self)
1407
1403
1408
1404
1409 class RustStandaloneExtension(RustExtension):
1405 class RustStandaloneExtension(RustExtension):
1410 def __init__(self, pydottedname, rustcrate, dylibname, **kw):
1406 def __init__(self, pydottedname, rustcrate, dylibname, **kw):
1411 RustExtension.__init__(
1407 RustExtension.__init__(
1412 self, pydottedname, [], dylibname, rustcrate, **kw
1408 self, pydottedname, [], dylibname, rustcrate, **kw
1413 )
1409 )
1414 self.dylibname = dylibname
1410 self.dylibname = dylibname
1415
1411
1416 def build(self, target_dir):
1412 def build(self, target_dir):
1417 self.rustbuild()
1413 self.rustbuild()
1418 target = [target_dir]
1414 target = [target_dir]
1419 target.extend(self.name.split('.'))
1415 target.extend(self.name.split('.'))
1420 target[-1] += DYLIB_SUFFIX
1416 target[-1] += DYLIB_SUFFIX
1421 shutil.copy2(
1417 shutil.copy2(
1422 os.path.join(
1418 os.path.join(
1423 self.rusttargetdir, self.dylibname + self.rustdylibsuffix()
1419 self.rusttargetdir, self.dylibname + self.rustdylibsuffix()
1424 ),
1420 ),
1425 os.path.join(*target),
1421 os.path.join(*target),
1426 )
1422 )
1427
1423
1428
1424
1429 extmodules = [
1425 extmodules = [
1430 Extension(
1426 Extension(
1431 'mercurial.cext.base85',
1427 'mercurial.cext.base85',
1432 ['mercurial/cext/base85.c'],
1428 ['mercurial/cext/base85.c'],
1433 include_dirs=common_include_dirs,
1429 include_dirs=common_include_dirs,
1434 depends=common_depends,
1430 depends=common_depends,
1435 ),
1431 ),
1436 Extension(
1432 Extension(
1437 'mercurial.cext.bdiff',
1433 'mercurial.cext.bdiff',
1438 ['mercurial/bdiff.c', 'mercurial/cext/bdiff.c'] + xdiff_srcs,
1434 ['mercurial/bdiff.c', 'mercurial/cext/bdiff.c'] + xdiff_srcs,
1439 include_dirs=common_include_dirs,
1435 include_dirs=common_include_dirs,
1440 depends=common_depends + ['mercurial/bdiff.h'] + xdiff_headers,
1436 depends=common_depends + ['mercurial/bdiff.h'] + xdiff_headers,
1441 ),
1437 ),
1442 Extension(
1438 Extension(
1443 'mercurial.cext.mpatch',
1439 'mercurial.cext.mpatch',
1444 ['mercurial/mpatch.c', 'mercurial/cext/mpatch.c'],
1440 ['mercurial/mpatch.c', 'mercurial/cext/mpatch.c'],
1445 include_dirs=common_include_dirs,
1441 include_dirs=common_include_dirs,
1446 depends=common_depends,
1442 depends=common_depends,
1447 ),
1443 ),
1448 RustEnhancedExtension(
1444 RustEnhancedExtension(
1449 'mercurial.cext.parsers',
1445 'mercurial.cext.parsers',
1450 [
1446 [
1451 'mercurial/cext/charencode.c',
1447 'mercurial/cext/charencode.c',
1452 'mercurial/cext/dirs.c',
1448 'mercurial/cext/dirs.c',
1453 'mercurial/cext/manifest.c',
1449 'mercurial/cext/manifest.c',
1454 'mercurial/cext/parsers.c',
1450 'mercurial/cext/parsers.c',
1455 'mercurial/cext/pathencode.c',
1451 'mercurial/cext/pathencode.c',
1456 'mercurial/cext/revlog.c',
1452 'mercurial/cext/revlog.c',
1457 ],
1453 ],
1458 'hgdirectffi',
1454 'hgdirectffi',
1459 'hg-direct-ffi',
1455 'hg-direct-ffi',
1460 include_dirs=common_include_dirs,
1456 include_dirs=common_include_dirs,
1461 depends=common_depends
1457 depends=common_depends
1462 + [
1458 + [
1463 'mercurial/cext/charencode.h',
1459 'mercurial/cext/charencode.h',
1464 'mercurial/cext/revlog.h',
1460 'mercurial/cext/revlog.h',
1465 'rust/hg-core/src/ancestors.rs',
1461 'rust/hg-core/src/ancestors.rs',
1466 'rust/hg-core/src/lib.rs',
1462 'rust/hg-core/src/lib.rs',
1467 ],
1463 ],
1468 ),
1464 ),
1469 Extension(
1465 Extension(
1470 'mercurial.cext.osutil',
1466 'mercurial.cext.osutil',
1471 ['mercurial/cext/osutil.c'],
1467 ['mercurial/cext/osutil.c'],
1472 include_dirs=common_include_dirs,
1468 include_dirs=common_include_dirs,
1473 extra_compile_args=osutil_cflags,
1469 extra_compile_args=osutil_cflags,
1474 extra_link_args=osutil_ldflags,
1470 extra_link_args=osutil_ldflags,
1475 depends=common_depends,
1471 depends=common_depends,
1476 ),
1472 ),
1477 Extension(
1473 Extension(
1478 'mercurial.thirdparty.zope.interface._zope_interface_coptimizations',
1474 'mercurial.thirdparty.zope.interface._zope_interface_coptimizations',
1479 [
1475 [
1480 'mercurial/thirdparty/zope/interface/_zope_interface_coptimizations.c',
1476 'mercurial/thirdparty/zope/interface/_zope_interface_coptimizations.c',
1481 ],
1477 ],
1482 ),
1478 ),
1483 Extension(
1479 Extension(
1484 'mercurial.thirdparty.sha1dc',
1480 'mercurial.thirdparty.sha1dc',
1485 [
1481 [
1486 'mercurial/thirdparty/sha1dc/cext.c',
1482 'mercurial/thirdparty/sha1dc/cext.c',
1487 'mercurial/thirdparty/sha1dc/lib/sha1.c',
1483 'mercurial/thirdparty/sha1dc/lib/sha1.c',
1488 'mercurial/thirdparty/sha1dc/lib/ubc_check.c',
1484 'mercurial/thirdparty/sha1dc/lib/ubc_check.c',
1489 ],
1485 ],
1490 ),
1486 ),
1491 Extension(
1487 Extension(
1492 'hgext.fsmonitor.pywatchman.bser', ['hgext/fsmonitor/pywatchman/bser.c']
1488 'hgext.fsmonitor.pywatchman.bser', ['hgext/fsmonitor/pywatchman/bser.c']
1493 ),
1489 ),
1494 RustStandaloneExtension(
1490 RustStandaloneExtension(
1495 'mercurial.rustext', 'hg-cpython', 'librusthg', py3_features='python3'
1491 'mercurial.rustext', 'hg-cpython', 'librusthg', py3_features='python3'
1496 ),
1492 ),
1497 ]
1493 ]
1498
1494
1499
1495
1500 sys.path.insert(0, 'contrib/python-zstandard')
1496 sys.path.insert(0, 'contrib/python-zstandard')
1501 import setup_zstd
1497 import setup_zstd
1502
1498
1503 extmodules.append(
1499 extmodules.append(
1504 setup_zstd.get_c_extension(
1500 setup_zstd.get_c_extension(
1505 name='mercurial.zstd', root=os.path.abspath(os.path.dirname(__file__))
1501 name='mercurial.zstd', root=os.path.abspath(os.path.dirname(__file__))
1506 )
1502 )
1507 )
1503 )
1508
1504
1509 try:
1505 try:
1510 from distutils import cygwinccompiler
1506 from distutils import cygwinccompiler
1511
1507
1512 # the -mno-cygwin option has been deprecated for years
1508 # the -mno-cygwin option has been deprecated for years
1513 mingw32compilerclass = cygwinccompiler.Mingw32CCompiler
1509 mingw32compilerclass = cygwinccompiler.Mingw32CCompiler
1514
1510
1515 class HackedMingw32CCompiler(cygwinccompiler.Mingw32CCompiler):
1511 class HackedMingw32CCompiler(cygwinccompiler.Mingw32CCompiler):
1516 def __init__(self, *args, **kwargs):
1512 def __init__(self, *args, **kwargs):
1517 mingw32compilerclass.__init__(self, *args, **kwargs)
1513 mingw32compilerclass.__init__(self, *args, **kwargs)
1518 for i in 'compiler compiler_so linker_exe linker_so'.split():
1514 for i in 'compiler compiler_so linker_exe linker_so'.split():
1519 try:
1515 try:
1520 getattr(self, i).remove('-mno-cygwin')
1516 getattr(self, i).remove('-mno-cygwin')
1521 except ValueError:
1517 except ValueError:
1522 pass
1518 pass
1523
1519
1524 cygwinccompiler.Mingw32CCompiler = HackedMingw32CCompiler
1520 cygwinccompiler.Mingw32CCompiler = HackedMingw32CCompiler
1525 except ImportError:
1521 except ImportError:
1526 # the cygwinccompiler package is not available on some Python
1522 # the cygwinccompiler package is not available on some Python
1527 # distributions like the ones from the optware project for Synology
1523 # distributions like the ones from the optware project for Synology
1528 # DiskStation boxes
1524 # DiskStation boxes
1529 class HackedMingw32CCompiler(object):
1525 class HackedMingw32CCompiler(object):
1530 pass
1526 pass
1531
1527
1532
1528
1533 if os.name == 'nt':
1529 if os.name == 'nt':
1534 # Allow compiler/linker flags to be added to Visual Studio builds. Passing
1530 # Allow compiler/linker flags to be added to Visual Studio builds. Passing
1535 # extra_link_args to distutils.extensions.Extension() doesn't have any
1531 # extra_link_args to distutils.extensions.Extension() doesn't have any
1536 # effect.
1532 # effect.
1537 from distutils import msvccompiler
1533 from distutils import msvccompiler
1538
1534
1539 msvccompilerclass = msvccompiler.MSVCCompiler
1535 msvccompilerclass = msvccompiler.MSVCCompiler
1540
1536
1541 class HackedMSVCCompiler(msvccompiler.MSVCCompiler):
1537 class HackedMSVCCompiler(msvccompiler.MSVCCompiler):
1542 def initialize(self):
1538 def initialize(self):
1543 msvccompilerclass.initialize(self)
1539 msvccompilerclass.initialize(self)
1544 # "warning LNK4197: export 'func' specified multiple times"
1540 # "warning LNK4197: export 'func' specified multiple times"
1545 self.ldflags_shared.append('/ignore:4197')
1541 self.ldflags_shared.append('/ignore:4197')
1546 self.ldflags_shared_debug.append('/ignore:4197')
1542 self.ldflags_shared_debug.append('/ignore:4197')
1547
1543
1548 msvccompiler.MSVCCompiler = HackedMSVCCompiler
1544 msvccompiler.MSVCCompiler = HackedMSVCCompiler
1549
1545
1550 packagedata = {
1546 packagedata = {
1551 'mercurial': [
1547 'mercurial': [
1552 'locale/*/LC_MESSAGES/hg.mo',
1548 'locale/*/LC_MESSAGES/hg.mo',
1553 'defaultrc/*.rc',
1549 'defaultrc/*.rc',
1554 'dummycert.pem',
1550 'dummycert.pem',
1555 ],
1551 ],
1556 'mercurial.helptext': ['*.txt',],
1552 'mercurial.helptext': ['*.txt',],
1557 'mercurial.helptext.internals': ['*.txt',],
1553 'mercurial.helptext.internals': ['*.txt',],
1558 }
1554 }
1559
1555
1560
1556
1561 def ordinarypath(p):
1557 def ordinarypath(p):
1562 return p and p[0] != '.' and p[-1] != '~'
1558 return p and p[0] != '.' and p[-1] != '~'
1563
1559
1564
1560
1565 for root in ('templates',):
1561 for root in ('templates',):
1566 for curdir, dirs, files in os.walk(os.path.join('mercurial', root)):
1562 for curdir, dirs, files in os.walk(os.path.join('mercurial', root)):
1567 curdir = curdir.split(os.sep, 1)[1]
1563 curdir = curdir.split(os.sep, 1)[1]
1568 dirs[:] = filter(ordinarypath, dirs)
1564 dirs[:] = filter(ordinarypath, dirs)
1569 for f in filter(ordinarypath, files):
1565 for f in filter(ordinarypath, files):
1570 f = os.path.join(curdir, f)
1566 f = os.path.join(curdir, f)
1571 packagedata['mercurial'].append(f)
1567 packagedata['mercurial'].append(f)
1572
1568
1573 datafiles = []
1569 datafiles = []
1574
1570
1575 # distutils expects version to be str/unicode. Converting it to
1571 # distutils expects version to be str/unicode. Converting it to
1576 # unicode on Python 2 still works because it won't contain any
1572 # unicode on Python 2 still works because it won't contain any
1577 # non-ascii bytes and will be implicitly converted back to bytes
1573 # non-ascii bytes and will be implicitly converted back to bytes
1578 # when operated on.
1574 # when operated on.
1579 assert isinstance(version, bytes)
1575 assert isinstance(version, bytes)
1580 setupversion = version.decode('ascii')
1576 setupversion = version.decode('ascii')
1581
1577
1582 extra = {}
1578 extra = {}
1583
1579
1584 py2exepackages = [
1580 py2exepackages = [
1585 'hgdemandimport',
1581 'hgdemandimport',
1586 'hgext3rd',
1582 'hgext3rd',
1587 'hgext',
1583 'hgext',
1588 'email',
1584 'email',
1589 # implicitly imported per module policy
1585 # implicitly imported per module policy
1590 # (cffi wouldn't be used as a frozen exe)
1586 # (cffi wouldn't be used as a frozen exe)
1591 'mercurial.cext',
1587 'mercurial.cext',
1592 #'mercurial.cffi',
1588 #'mercurial.cffi',
1593 'mercurial.pure',
1589 'mercurial.pure',
1594 ]
1590 ]
1595
1591
1596 py2exeexcludes = []
1592 py2exeexcludes = []
1597 py2exedllexcludes = ['crypt32.dll']
1593 py2exedllexcludes = ['crypt32.dll']
1598
1594
1599 if issetuptools:
1595 if issetuptools:
1600 extra['python_requires'] = supportedpy
1596 extra['python_requires'] = supportedpy
1601
1597
1602 if py2exeloaded:
1598 if py2exeloaded:
1603 extra['console'] = [
1599 extra['console'] = [
1604 {
1600 {
1605 'script': 'hg',
1601 'script': 'hg',
1606 'copyright': 'Copyright (C) 2005-2019 Matt Mackall and others',
1602 'copyright': 'Copyright (C) 2005-2019 Matt Mackall and others',
1607 'product_version': version,
1603 'product_version': version,
1608 }
1604 }
1609 ]
1605 ]
1610 # Sub command of 'build' because 'py2exe' does not handle sub_commands.
1606 # Sub command of 'build' because 'py2exe' does not handle sub_commands.
1611 # Need to override hgbuild because it has a private copy of
1607 # Need to override hgbuild because it has a private copy of
1612 # build.sub_commands.
1608 # build.sub_commands.
1613 hgbuild.sub_commands.insert(0, ('build_hgextindex', None))
1609 hgbuild.sub_commands.insert(0, ('build_hgextindex', None))
1614 # put dlls in sub directory so that they won't pollute PATH
1610 # put dlls in sub directory so that they won't pollute PATH
1615 extra['zipfile'] = 'lib/library.zip'
1611 extra['zipfile'] = 'lib/library.zip'
1616
1612
1617 # We allow some configuration to be supplemented via environment
1613 # We allow some configuration to be supplemented via environment
1618 # variables. This is better than setup.cfg files because it allows
1614 # variables. This is better than setup.cfg files because it allows
1619 # supplementing configs instead of replacing them.
1615 # supplementing configs instead of replacing them.
1620 extrapackages = os.environ.get('HG_PY2EXE_EXTRA_PACKAGES')
1616 extrapackages = os.environ.get('HG_PY2EXE_EXTRA_PACKAGES')
1621 if extrapackages:
1617 if extrapackages:
1622 py2exepackages.extend(extrapackages.split(' '))
1618 py2exepackages.extend(extrapackages.split(' '))
1623
1619
1624 excludes = os.environ.get('HG_PY2EXE_EXTRA_EXCLUDES')
1620 excludes = os.environ.get('HG_PY2EXE_EXTRA_EXCLUDES')
1625 if excludes:
1621 if excludes:
1626 py2exeexcludes.extend(excludes.split(' '))
1622 py2exeexcludes.extend(excludes.split(' '))
1627
1623
1628 dllexcludes = os.environ.get('HG_PY2EXE_EXTRA_DLL_EXCLUDES')
1624 dllexcludes = os.environ.get('HG_PY2EXE_EXTRA_DLL_EXCLUDES')
1629 if dllexcludes:
1625 if dllexcludes:
1630 py2exedllexcludes.extend(dllexcludes.split(' '))
1626 py2exedllexcludes.extend(dllexcludes.split(' '))
1631
1627
1632 if os.name == 'nt':
1628 if os.name == 'nt':
1633 # Windows binary file versions for exe/dll files must have the
1629 # Windows binary file versions for exe/dll files must have the
1634 # form W.X.Y.Z, where W,X,Y,Z are numbers in the range 0..65535
1630 # form W.X.Y.Z, where W,X,Y,Z are numbers in the range 0..65535
1635 setupversion = setupversion.split(r'+', 1)[0]
1631 setupversion = setupversion.split(r'+', 1)[0]
1636
1632
1637 if sys.platform == 'darwin' and os.path.exists('/usr/bin/xcodebuild'):
1633 if sys.platform == 'darwin' and os.path.exists('/usr/bin/xcodebuild'):
1638 version = runcmd(['/usr/bin/xcodebuild', '-version'], {})[1].splitlines()
1634 version = runcmd(['/usr/bin/xcodebuild', '-version'], {})[1].splitlines()
1639 if version:
1635 if version:
1640 version = version[0]
1636 version = version[0]
1641 if sys.version_info[0] == 3:
1637 if sys.version_info[0] == 3:
1642 version = version.decode('utf-8')
1638 version = version.decode('utf-8')
1643 xcode4 = version.startswith('Xcode') and StrictVersion(
1639 xcode4 = version.startswith('Xcode') and StrictVersion(
1644 version.split()[1]
1640 version.split()[1]
1645 ) >= StrictVersion('4.0')
1641 ) >= StrictVersion('4.0')
1646 xcode51 = re.match(r'^Xcode\s+5\.1', version) is not None
1642 xcode51 = re.match(r'^Xcode\s+5\.1', version) is not None
1647 else:
1643 else:
1648 # xcodebuild returns empty on OS X Lion with XCode 4.3 not
1644 # xcodebuild returns empty on OS X Lion with XCode 4.3 not
1649 # installed, but instead with only command-line tools. Assume
1645 # installed, but instead with only command-line tools. Assume
1650 # that only happens on >= Lion, thus no PPC support.
1646 # that only happens on >= Lion, thus no PPC support.
1651 xcode4 = True
1647 xcode4 = True
1652 xcode51 = False
1648 xcode51 = False
1653
1649
1654 # XCode 4.0 dropped support for ppc architecture, which is hardcoded in
1650 # XCode 4.0 dropped support for ppc architecture, which is hardcoded in
1655 # distutils.sysconfig
1651 # distutils.sysconfig
1656 if xcode4:
1652 if xcode4:
1657 os.environ['ARCHFLAGS'] = ''
1653 os.environ['ARCHFLAGS'] = ''
1658
1654
1659 # XCode 5.1 changes clang such that it now fails to compile if the
1655 # XCode 5.1 changes clang such that it now fails to compile if the
1660 # -mno-fused-madd flag is passed, but the version of Python shipped with
1656 # -mno-fused-madd flag is passed, but the version of Python shipped with
1661 # OS X 10.9 Mavericks includes this flag. This causes problems in all
1657 # OS X 10.9 Mavericks includes this flag. This causes problems in all
1662 # C extension modules, and a bug has been filed upstream at
1658 # C extension modules, and a bug has been filed upstream at
1663 # http://bugs.python.org/issue21244. We also need to patch this here
1659 # http://bugs.python.org/issue21244. We also need to patch this here
1664 # so Mercurial can continue to compile in the meantime.
1660 # so Mercurial can continue to compile in the meantime.
1665 if xcode51:
1661 if xcode51:
1666 cflags = get_config_var('CFLAGS')
1662 cflags = get_config_var('CFLAGS')
1667 if cflags and re.search(r'-mno-fused-madd\b', cflags) is not None:
1663 if cflags and re.search(r'-mno-fused-madd\b', cflags) is not None:
1668 os.environ['CFLAGS'] = (
1664 os.environ['CFLAGS'] = (
1669 os.environ.get('CFLAGS', '') + ' -Qunused-arguments'
1665 os.environ.get('CFLAGS', '') + ' -Qunused-arguments'
1670 )
1666 )
1671
1667
1672 setup(
1668 setup(
1673 name='mercurial',
1669 name='mercurial',
1674 version=setupversion,
1670 version=setupversion,
1675 author='Matt Mackall and many others',
1671 author='Matt Mackall and many others',
1676 author_email='mercurial@mercurial-scm.org',
1672 author_email='mercurial@mercurial-scm.org',
1677 url='https://mercurial-scm.org/',
1673 url='https://mercurial-scm.org/',
1678 download_url='https://mercurial-scm.org/release/',
1674 download_url='https://mercurial-scm.org/release/',
1679 description=(
1675 description=(
1680 'Fast scalable distributed SCM (revision control, version '
1676 'Fast scalable distributed SCM (revision control, version '
1681 'control) system'
1677 'control) system'
1682 ),
1678 ),
1683 long_description=(
1679 long_description=(
1684 'Mercurial is a distributed SCM tool written in Python.'
1680 'Mercurial is a distributed SCM tool written in Python.'
1685 ' It is used by a number of large projects that require'
1681 ' It is used by a number of large projects that require'
1686 ' fast, reliable distributed revision control, such as '
1682 ' fast, reliable distributed revision control, such as '
1687 'Mozilla.'
1683 'Mozilla.'
1688 ),
1684 ),
1689 license='GNU GPLv2 or any later version',
1685 license='GNU GPLv2 or any later version',
1690 classifiers=[
1686 classifiers=[
1691 'Development Status :: 6 - Mature',
1687 'Development Status :: 6 - Mature',
1692 'Environment :: Console',
1688 'Environment :: Console',
1693 'Intended Audience :: Developers',
1689 'Intended Audience :: Developers',
1694 'Intended Audience :: System Administrators',
1690 'Intended Audience :: System Administrators',
1695 'License :: OSI Approved :: GNU General Public License (GPL)',
1691 'License :: OSI Approved :: GNU General Public License (GPL)',
1696 'Natural Language :: Danish',
1692 'Natural Language :: Danish',
1697 'Natural Language :: English',
1693 'Natural Language :: English',
1698 'Natural Language :: German',
1694 'Natural Language :: German',
1699 'Natural Language :: Italian',
1695 'Natural Language :: Italian',
1700 'Natural Language :: Japanese',
1696 'Natural Language :: Japanese',
1701 'Natural Language :: Portuguese (Brazilian)',
1697 'Natural Language :: Portuguese (Brazilian)',
1702 'Operating System :: Microsoft :: Windows',
1698 'Operating System :: Microsoft :: Windows',
1703 'Operating System :: OS Independent',
1699 'Operating System :: OS Independent',
1704 'Operating System :: POSIX',
1700 'Operating System :: POSIX',
1705 'Programming Language :: C',
1701 'Programming Language :: C',
1706 'Programming Language :: Python',
1702 'Programming Language :: Python',
1707 'Topic :: Software Development :: Version Control',
1703 'Topic :: Software Development :: Version Control',
1708 ],
1704 ],
1709 scripts=scripts,
1705 scripts=scripts,
1710 packages=packages,
1706 packages=packages,
1711 ext_modules=extmodules,
1707 ext_modules=extmodules,
1712 data_files=datafiles,
1708 data_files=datafiles,
1713 package_data=packagedata,
1709 package_data=packagedata,
1714 cmdclass=cmdclass,
1710 cmdclass=cmdclass,
1715 distclass=hgdist,
1711 distclass=hgdist,
1716 options={
1712 options={
1717 'py2exe': {
1713 'py2exe': {
1718 'bundle_files': 3,
1714 'bundle_files': 3,
1719 'dll_excludes': py2exedllexcludes,
1715 'dll_excludes': py2exedllexcludes,
1720 'excludes': py2exeexcludes,
1716 'excludes': py2exeexcludes,
1721 'packages': py2exepackages,
1717 'packages': py2exepackages,
1722 },
1718 },
1723 'bdist_mpkg': {
1719 'bdist_mpkg': {
1724 'zipdist': False,
1720 'zipdist': False,
1725 'license': 'COPYING',
1721 'license': 'COPYING',
1726 'readme': 'contrib/packaging/macosx/Readme.html',
1722 'readme': 'contrib/packaging/macosx/Readme.html',
1727 'welcome': 'contrib/packaging/macosx/Welcome.html',
1723 'welcome': 'contrib/packaging/macosx/Welcome.html',
1728 },
1724 },
1729 },
1725 },
1730 **extra
1726 **extra
1731 )
1727 )
General Comments 0
You need to be logged in to leave comments. Login now