#

# This is the mercurial setup script.

#

# 'python setup.py install', or

# 'python setup.py --help' for more options

import sys, platform

if sys.version_info < (2, 7, 0, 'final'):

    raise SystemExit('Mercurial requires Python 2.7 or later.')

if sys.version_info[0] >= 3:

    printf = eval('print')

    libdir_escape = 'unicode_escape'

else:

    libdir_escape = 'string_escape'

    def printf(*args, **kwargs):

        f = kwargs.get('file', sys.stdout)

        end = kwargs.get('end', '\n')

        f.write(b' '.join(args) + end)

# Solaris Python packaging brain damage

try:

    import hashlib

    sha = hashlib.sha1()

except ImportError:

    try:

        import sha

        sha.sha # silence unused import warning

    except ImportError:

        raise SystemExit(

            "Couldn't import standard hashlib (incomplete Python install).")

try:

    import zlib

    zlib.compressobj # silence unused import warning

except ImportError:

    raise SystemExit(

        "Couldn't import standard zlib (incomplete Python install).")

# The base IronPython distribution (as of 2.7.1) doesn't support bz2

isironpython = False

try:

    isironpython = (platform.python_implementation()

                    .lower().find("ironpython") != -1)

except AttributeError:

    pass

if isironpython:

    sys.stderr.write("warning: IronPython detected (no bz2 support)\n")

else:

    try:

        import bz2

        bz2.BZ2Compressor # silence unused import warning

    except ImportError:

        raise SystemExit(

            "Couldn't import standard bz2 (incomplete Python install).")

ispypy = "PyPy" in sys.version

import ctypes

import os, stat, subprocess, time

import re

import shutil

import tempfile

from distutils import log

# We have issues with setuptools on some platforms and builders. Until

# those are resolved, setuptools is opt-in except for platforms where

# we don't have issues.

if os.name == 'nt' or 'FORCE_SETUPTOOLS' in os.environ:

    from setuptools import setup

else:

    from distutils.core import setup

from distutils.ccompiler import new_compiler

from distutils.core import Command, Extension

from distutils.dist import Distribution

from distutils.command.build import build

from distutils.command.build_ext import build_ext

from distutils.command.build_py import build_py

from distutils.command.build_scripts import build_scripts

from distutils.command.install_lib import install_lib

from distutils.command.install_scripts import install_scripts

from distutils.spawn import spawn, find_executable

from distutils import file_util

from distutils.errors import (

    CCompilerError,

    DistutilsError,

    DistutilsExecError,

)

from distutils.sysconfig import get_python_inc, get_config_var

from distutils.version import StrictVersion

scripts = ['hg']

if os.name == 'nt':

    # We remove hg.bat if we are able to build hg.exe.

    scripts.append('contrib/win32/hg.bat')

def cancompile(cc, code):

    tmpdir = tempfile.mkdtemp(prefix='hg-install-')

    devnull = oldstderr = None

    try:

        fname = os.path.join(tmpdir, 'testcomp.c')

        f = open(fname, 'w')

        f.write(code)

        f.close()

        # Redirect stderr to /dev/null to hide any error messages

        # from the compiler.

        # This will have to be changed if we ever have to check

        # for a function on Windows.

        devnull = open('/dev/null', 'w')

        oldstderr = os.dup(sys.stderr.fileno())

        os.dup2(devnull.fileno(), sys.stderr.fileno())

        objects = cc.compile([fname], output_dir=tmpdir)

        cc.link_executable(objects, os.path.join(tmpdir, "a.out"))

        return True

    except Exception:

        return False

    finally:

        if oldstderr is not None:

            os.dup2(oldstderr, sys.stderr.fileno())

        if devnull is not None:

            devnull.close()

        shutil.rmtree(tmpdir)

# simplified version of distutils.ccompiler.CCompiler.has_function

# that actually removes its temporary files.

def hasfunction(cc, funcname):

    code = 'int main(void) { %s(); }\n' % funcname

    return cancompile(cc, code)

def hasheader(cc, headername):

    code = '#include <%s>\nint main(void) { return 0; }\n' % headername

    return cancompile(cc, code)

# py2exe needs to be installed to work

try:

    import py2exe

    py2exe.Distribution # silence unused import warning

    py2exeloaded = True

    # import py2exe's patched Distribution class

    from distutils.core import Distribution

except ImportError:

    py2exeloaded = False

def runcmd(cmd, env):

    if (sys.platform == 'plan9'

       and (sys.version_info[0] == 2 and sys.version_info[1] < 7)):

        # subprocess kludge to work around issues in half-baked Python

        # ports, notably bichued/python:

        _, out, err = os.popen3(cmd)

        return str(out), str(err)

    else:

        p = subprocess.Popen(cmd, stdout=subprocess.PIPE,

                             stderr=subprocess.PIPE, env=env)

        out, err = p.communicate()

        return out, err

def runhg(cmd, env):

    out, err = runcmd(cmd, env)

    # If root is executing setup.py, but the repository is owned by

    # another user (as in "sudo python setup.py install") we will get

    # trust warnings since the .hg/hgrc file is untrusted. That is

    # fine, we don't want to load it anyway.  Python may warn about

    # a missing __init__.py in mercurial/locale, we also ignore that.

    err = [e for e in err.splitlines()

           if not e.startswith(b'not trusting file') \

              and not e.startswith(b'warning: Not importing') \

              and not e.startswith(b'obsolete feature not enabled')]

    if err:

        printf("stderr from '%s':" % (' '.join(cmd)), file=sys.stderr)

        printf(b'\n'.join([b'  ' + e for e in err]), file=sys.stderr)

        return ''

    return out

version = ''

# Execute hg out of this directory with a custom environment which takes care

# to not use any hgrc files and do no localization.

env = {'HGMODULEPOLICY': 'py',

       'HGRCPATH': '',

       'LANGUAGE': 'C',

       'PATH': ''} # make pypi modules that use os.environ['PATH'] happy

if 'LD_LIBRARY_PATH' in os.environ:

    env['LD_LIBRARY_PATH'] = os.environ['LD_LIBRARY_PATH']

if 'SystemRoot' in os.environ:

    # Copy SystemRoot into the custom environment for Python 2.6

    # under Windows. Otherwise, the subprocess will fail with

    # error 0xc0150004. See: http://bugs.python.org/issue3440

    env['SystemRoot'] = os.environ['SystemRoot']

if os.path.isdir('.hg'):

    cmd = [sys.executable, 'hg', 'log', '-r', '.', '--template', '{tags}\n']

    numerictags = [t for t in runhg(cmd, env).split() if t[0].isdigit()]

    hgid = runhg([sys.executable, 'hg', 'id', '-i'], env).strip()

    if numerictags: # tag(s) found

        version = numerictags[-1]

        if hgid.endswith('+'): # propagate the dirty status to the tag

            version += '+'

    else: # no tag found

        ltagcmd = [sys.executable, 'hg', 'parents', '--template',

                   '{latesttag}']

        ltag = runhg(ltagcmd, env)

        changessincecmd = [sys.executable, 'hg', 'log', '-T', 'x\n', '-r',

                           "only(.,'%s')" % ltag]

        changessince = len(runhg(changessincecmd, env).splitlines())

        version = '%s+%s-%s' % (ltag, changessince, hgid)

    if version.endswith('+'):

        version += time.strftime('%Y%m%d')

elif os.path.exists('.hg_archival.txt'):

    kw = dict([[t.strip() for t in l.split(':', 1)]

               for l in open('.hg_archival.txt')])

    if 'tag' in kw:

        version = kw['tag']

    elif 'latesttag' in kw:

        if 'changessincelatesttag' in kw:

            version = '%(latesttag)s+%(changessincelatesttag)s-%(node).12s' % kw

        else:

            version = '%(latesttag)s+%(latesttagdistance)s-%(node).12s' % kw

    else:

        version = kw.get('node', '')[:12]

if version:

    with open("mercurial/__version__.py", "w") as f:

        f.write('# this file is autogenerated by setup.py\n')

        f.write('version = "%s"\n' % version)

try:

    oldpolicy = os.environ.get('HGMODULEPOLICY', None)

    os.environ['HGMODULEPOLICY'] = 'py'

    from mercurial import __version__

    version = __version__.version

except ImportError:

    version = 'unknown'

finally:

    if oldpolicy is None:

        del os.environ['HGMODULEPOLICY']

    else:

        os.environ['HGMODULEPOLICY'] = oldpolicy

class hgbuild(build):

    # Insert hgbuildmo first so that files in mercurial/locale/ are found

    # when build_py is run next.

    sub_commands = [('build_mo', None)] + build.sub_commands

class hgbuildmo(build):

    description = "build translations (.mo files)"

    def run(self):

        if not find_executable('msgfmt'):

            self.warn("could not find msgfmt executable, no translations "

                     "will be built")

            return

        podir = 'i18n'

        if not os.path.isdir(podir):

            self.warn("could not find %s/ directory" % podir)

            return

        join = os.path.join

        for po in os.listdir(podir):

            if not po.endswith('.po'):

                continue

            pofile = join(podir, po)

            modir = join('locale', po[:-3], 'LC_MESSAGES')

            mofile = join(modir, 'hg.mo')

            mobuildfile = join('mercurial', mofile)

            cmd = ['msgfmt', '-v', '-o', mobuildfile, pofile]

            if sys.platform != 'sunos5':

                # msgfmt on Solaris does not know about -c

                cmd.append('-c')

            self.mkpath(join('mercurial', modir))

            self.make_file([pofile], mobuildfile, spawn, (cmd,))

class hgdist(Distribution):

    pure = False

    cffi = ispypy

    global_options = Distribution.global_options + \

                     [('pure', None, "use pure (slow) Python "

                        "code instead of C extensions"),

                     ]

    def has_ext_modules(self):

        # self.ext_modules is emptied in hgbuildpy.finalize_options which is

        # too late for some cases

        return not self.pure and Distribution.has_ext_modules(self)

# This is ugly as a one-liner. So use a variable.

buildextnegops = dict(getattr(build_ext, 'negative_options', {}))

buildextnegops['no-zstd'] = 'zstd'

class hgbuildext(build_ext):

    user_options = build_ext.user_options + [

        ('zstd', None, 'compile zstd bindings [default]'),

        ('no-zstd', None, 'do not compile zstd bindings'),

    ]

    boolean_options = build_ext.boolean_options + ['zstd']

    negative_opt = buildextnegops

    def initialize_options(self):

        self.zstd = True

        return build_ext.initialize_options(self)

    def build_extensions(self):

        # Filter out zstd if disabled via argument.

        if not self.zstd:

            self.extensions = [e for e in self.extensions

                               if e.name != 'mercurial.zstd']

        return build_ext.build_extensions(self)

    def build_extension(self, ext):

        try:

            build_ext.build_extension(self, ext)

        except CCompilerError:

            if not getattr(ext, 'optional', False):

                raise

            log.warn("Failed to build optional extension '%s' (skipping)",

                     ext.name)

class hgbuildscripts(build_scripts):

    def run(self):

        if os.name != 'nt' or self.distribution.pure:

            return build_scripts.run(self)

        exebuilt = False

        try:

            self.run_command('build_hgexe')

            exebuilt = True

        except (DistutilsError, CCompilerError):

            log.warn('failed to build optional hg.exe')

        if exebuilt:

            # Copying hg.exe to the scripts build directory ensures it is

            # installed by the install_scripts command.

            hgexecommand = self.get_finalized_command('build_hgexe')

            dest = os.path.join(self.build_dir, 'hg.exe')

            self.mkpath(self.build_dir)

            self.copy_file(hgexecommand.hgexepath, dest)

            # Remove hg.bat because it is redundant with hg.exe.

            self.scripts.remove('contrib/win32/hg.bat')

        return build_scripts.run(self)

class hgbuildpy(build_py):

    def finalize_options(self):

        build_py.finalize_options(self)

        if self.distribution.pure:

            self.distribution.ext_modules = []

        elif self.distribution.cffi:

            from mercurial.cffi import (

                bdiffbuild,

                mpatchbuild,

            )

            exts = [mpatchbuild.ffi.distutils_extension(),

                    bdiffbuild.ffi.distutils_extension()]

            # cffi modules go here

            if sys.platform == 'darwin':

                from mercurial.cffi import osutilbuild

                exts.append(osutilbuild.ffi.distutils_extension())

            self.distribution.ext_modules = exts

        else:

            h = os.path.join(get_python_inc(), 'Python.h')

            if not os.path.exists(h):

                raise SystemExit('Python headers are required to build '

                                 'Mercurial but weren\'t found in %s' % h)