#

# This is the mercurial setup script.

#

# 'python setup.py install', or

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

import sys, platform

if getattr(sys, 'version_info', (0, 0, 0)) < (2, 6, 0, 'final'):

    raise SystemExit("Mercurial requires Python 2.6 or later.")

if sys.version_info[0] >= 3:

    def b(s):

        '''A helper function to emulate 2.6+ bytes literals using string

        literals.'''

        return s.encode('latin1')

    printf = eval('print')

    libdir_escape = 'unicode_escape'

else:

    libdir_escape = 'string_escape'

    def b(s):

        '''A helper function to emulate 2.6+ bytes literals using string

        literals.'''

        return s

    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 os, stat, subprocess, time

import re

import shutil

import tempfile

from distutils import log

if 'FORCE_SETUPTOOLS' in os.environ:

    from setuptools import setup

else:

    from distutils.core import setup

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.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.sysconfig import get_python_inc, get_config_var

from distutils.version import StrictVersion

convert2to3 = '--c2to3' in sys.argv

if convert2to3:

    try:

        from distutils.command.build_py import build_py_2to3 as build_py

        from lib2to3.refactor import get_fixers_from_package as getfixers

    except ImportError:

        if sys.version_info[0] < 3:

            raise SystemExit("--c2to3 is only compatible with python3.")

        raise

    sys.path.append('contrib')

elif sys.version_info[0] >= 3:

    raise SystemExit("setup.py with python3 needs --c2to3 (experimental)")

scripts = ['hg']

if os.name == 'nt':

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

# simplified version of distutils.ccompiler.CCompiler.has_function

# that actually removes its temporary files.

def hasfunction(cc, funcname):

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

    devnull = oldstderr = None

    try:

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

        f = open(fname, 'w')

        f.write('int main(void) {\n')

        f.write('    %s();\n' % funcname)

        f.write('}\n')

        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)

# 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'}

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:

    f = open("mercurial/__version__.py", "w")

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

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

    f.close()

try:

    from mercurial import __version__

    version = __version__.version

except ImportError:

    version = 'unknown'

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

    # We also need build_ext before build_py. Otherwise, when 2to3 is

    # called (in build_py), it will not find osutil & friends,

    # thinking that those modules are global and, consequently, making

    # a mess, now that all module imports are global.

                    ('build_ext', build.has_ext_modules),

                   ] + 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 = ispypy

    global_options = Distribution.global_options + \

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

                        "code instead of C extensions"),

                      ('c2to3', None, "(experimental!) convert "

                        "code with 2to3"),

                     ]

    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)

class hgbuildext(build_ext):

    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 hgbuildpy(build_py):

    if convert2to3:

        fixer_names = sorted(set(getfixers("lib2to3.fixes") +

                                 getfixers("hgfixes")))

    def finalize_options(self):

        build_py.finalize_options(self)

        if self.distribution.pure:

            self.distribution.ext_modules = []

        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)

    def copy_file(self, *args, **kwargs):

        dst, copied = build_py.copy_file(self, *args, **kwargs)

        if copied and dst.endswith('__init__.py'):

            if self.distribution.pure:

                modulepolicy = 'py'

            else:

                modulepolicy = 'c'

            content = open(dst, 'rb').read()

            content = content.replace(b'@MODULELOADPOLICY@',

                                      modulepolicy.encode(libdir_escape))

            with open(dst, 'wb') as fh:

                fh.write(content)

        return dst, copied

class buildhgextindex(Command):

    description = 'generate prebuilt index of hgext (for frozen package)'