# setup.py

# A distutils setup script to register TortoiseHg COM server

# To build stand-alone package, use 'python setup.py py2exe' then use

# InnoSetup to build the installer.  By default, the installer will be

# created as dist\Output\setup.exe.

import time

import sys

import os

from distutils.core import setup

_description = 'Graphical interface for Mercurial VCS'

_scripts = []

_packages = []

_data_files = []

_extra = {}

_options = {}

if os.name == 'nt':

    _description = 'Windows shell extension for Mercurial VCS'

    _packages = ['tortoise', 'hggtk', 'hggtk.vis', 'hggtk.iniparse']

    _data_files = [(root, [os.path.join(root, file_) for file_ in files])

                           for root, dirs, files in os.walk('icons')]

    # ModuleFinder can't handle runtime changes to __path__, but win32com uses them

    try:

        # if this doesn't work, try import modulefinder

        import py2exe.mf as modulefinder

        import win32com

        for p in win32com.__path__[1:]:

            modulefinder.AddPackagePath("win32com", p)

        for extra in ["win32com.shell"]: #,"win32com.mapi"

            __import__(extra)

            m = sys.modules[extra]

            for p in m.__path__[1:]:

                modulefinder.AddPackagePath(extra, p)

    except ImportError:

        # no build path setup, no worries.

        pass

    import py2exe

    hgextmods = []

    if 'py2exe' in sys.argv:

        # FIXME: quick hack to include installed hg extensions in py2exe binary

        import hgext

        hgextdir = os.path.dirname(hgext.__file__)

        hgextmods = set(["hgext." + os.path.splitext(f)[0]

                      for f in os.listdir(hgextdir)])

        _data_files = [(root, [os.path.join(root, file_) for file_ in files])

                            for root, dirs, files in os.walk('icons')]

        extra['windows'] = [

                {"script":"hgproc.py",

                 "icon_resources": [(1, "icons/tortoise/hg.ico")]},

                {"script":"hggtk/tracelog.py",

                 "icon_resources": [(1, "icons/tortoise/python.ico")]}

                ]

        extra['com_server'] = ["tortoisehg"]

        extra['console'] = ["contrib/hg"]

    _options = {

       "py2exe" : {

           # Don't pull in all this MFC stuff used by the makepy UI.

           "excludes" : "pywin,pywin.dialogs,pywin.dialogs.list",

           # add library files to support PyGtk-based dialogs/windows

           # Note:

           #    after py2exe build, copy GTK's etc and lib directories into

           #    the dist directory created by py2exe.

           #    also needed is the GTK's share/themes (as dist/share/themes), 

           #    for dialogs to display in MS-Windows XP theme.

           "includes" : "dbhash,pango,atk,pangocairo,cairo,gobject," + \

                        ",".join(hgextmods),

       }

    }

    _extra['options'] = _options

else: # non-NT; assume Unixish

    _scripts = ['contrib/hgtk']

    _packages = ['hggtk', 'hggtk.vis', 'hggtk.iniparse']

    _data_files = [(os.path.join('share/pixmaps/tortoisehg', root),

        [os.path.join(root, file_) for file_ in files])

        for root, dirs, files in os.walk('icons')]

    _data_files += [('lib/nautilus/extensions-2.0/python', ['contrib/nautilus-thg.py'])]

# specify version string, otherwise 'hg identify' will be used:

version = ''

import tortoise.version

tortoise.version.remember_version(version)

setup(name="TortoiseHg",

        version=tortoise.version.get_version(),

        author='TK Soh',

        author_email='teekaysoh@gmail.com',

        url='http://tortoisehg.sourceforge.net/',

        description=_description,

        license='GNU GPL2',

        scripts=_scripts,

        packages=_packages,

        data_files=_data_files,

        **_extra

    )