#

# 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 import install

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):

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

                         stderr=subprocess.PIPE, env=env)

    out, err = p.communicate()

    return p.returncode, out, err

class hgcommand(object):

    def __init__(self, cmd, env):

        self.cmd = cmd

        self.env = env

    def run(self, args):

        cmd = self.cmd + args

        returncode, out, err = runcmd(cmd, self.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 or returncode != 0:

            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

def findhg():

    """Try to figure out how we should invoke hg for examining the local

    repository contents.

    Returns an hgcommand object."""

    # By default, prefer the "hg" command in the user's path.  This was

    # presumably the hg command that the user used to create this repository.

    #

    # This repository may require extensions or other settings that would not

    # be enabled by running the hg script directly from this local repository.

    hgenv = os.environ.copy()

    # Use HGPLAIN to disable hgrc settings that would change output formatting,

    # and disable localization for the same reasons.

    hgenv['HGPLAIN'] = '1'

    hgenv['LANGUAGE'] = 'C'

    hgcmd = ['hg']

    # Run a simple "hg log" command just to see if using hg from the user's

    # path works and can successfully interact with this repository.

    check_cmd = ['log', '-r.', '-Ttest']

    try:

        retcode, out, err = runcmd(hgcmd + check_cmd, hgenv)

    except EnvironmentError:

        retcode = -1

    if retcode == 0:

        return hgcommand(hgcmd, hgenv)

    # Fall back to trying the local hg installation.

    hgenv = localhgenv()

    # Don't source any system hgrc files when using the local hg.

    hgenv['HGRCPATH'] = ''

    hgcmd = [sys.executable, 'hg']

    try:

        retcode, out, err = runcmd(hgcmd + check_cmd, hgenv)

    except EnvironmentError:

        retcode = -1

    if retcode == 0:

        return hgcommand(hgcmd, hgenv)

    raise SystemExit('Unable to find a working hg binary to extract the '

                     'version from the repository tags')

def localhgenv():

    """Get an environment dictionary to use for invoking or importing

    mercurial from the local repository."""

    # 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:

        # SystemRoot is required by Windows to load various DLLs.  See:

        # https://bugs.python.org/issue13524#msg148850

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

    return env

version = ''

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

    hg = findhg()

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

    numerictags = [t for t in hg.run(cmd).split() if t[0:1].isdigit()]

    hgid = hg.run(['id', '-i']).strip()

    if not hgid:

        # Bail out if hg is having problems interacting with this repository,

        # rather than falling through and producing a bogus version number.

        # Continuing with an invalid version number will break extensions

        # that define minimumhgversion.

        raise SystemExit('Unable to determine hg version from local repository')

    if numerictags: # tag(s) found

        version = numerictags[-1]

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

            version += '+'

    else: # no tag found

        ltagcmd = ['parents', '--template', '{latesttag}']

        ltag = hg.run(ltagcmd)

        changessincecmd = ['log', '-T', 'x\n', '-r', "only(.,'%s')" % ltag]

        changessince = len(hg.run(changessincecmd).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']