#!/usr/bin/python2.6

#

# CDDL HEADER START

#

# The contents of this file are subject to the terms of the

# Common Development and Distribution License (the "License").

# You may not use this file except in compliance with the License.

#

# You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE

# or http://www.opensolaris.org/os/licensing.

# See the License for the specific language governing permissions

# and limitations under the License.

#

# When distributing Covered Code, include this CDDL HEADER in each

# file and include the License file at usr/src/OPENSOLARIS.LICENSE.

# If applicable, add the following below this CDDL HEADER, with the

# fields enclosed by brackets "[]" replaced with your own identifying

# information: Portions Copyright [yyyy] [name of copyright owner]

#

# CDDL HEADER END

#

# Copyright (c) 2008, 2011, Oracle and/or its affiliates.  All rights reserved.

#

import errno

import fnmatch

import os

import platform

import stat

import sys

import shutil

import re

import subprocess

import tarfile

import tempfile

import urllib

import py_compile

import hashlib

import time

from distutils.core import setup, Extension

from distutils.cmd import Command

from distutils.command.install import install as _install

from distutils.command.build import build as _build

from distutils.command.build_ext import build_ext as _build_ext

from distutils.command.build_py import build_py as _build_py

from distutils.command.bdist import bdist as _bdist

from distutils.command.clean import clean as _clean

from distutils.dist import Distribution

from distutils.sysconfig import get_python_inc

import distutils.file_util as file_util

import distutils.util as util

import distutils.ccompiler

from distutils.unixccompiler import UnixCCompiler

osname = platform.uname()[0].lower()

ostype = arch = 'unknown'

if osname == 'sunos':

        arch = platform.processor()

        ostype = "posix"

elif osname == 'linux':

        arch = "linux_" + platform.machine()

        ostype = "posix"

elif osname == 'windows':

        arch = osname

        ostype = "windows"

elif osname == 'darwin':

        arch = osname

        ostype = "posix"

elif osname == 'aix':

        arch = "aix"

        ostype = "posix"

pwd = os.path.normpath(sys.path[0])

#

# Unbuffer stdout and stderr.  This helps to ensure that subprocess output

# is properly interleaved with output from this program.

#

sys.stdout = os.fdopen(sys.stdout.fileno(), "w", 0)

sys.stderr = os.fdopen(sys.stderr.fileno(), "w", 0)

dist_dir = os.path.normpath(os.path.join(pwd, os.pardir, "proto", "dist_" + arch))

build_dir = os.path.normpath(os.path.join(pwd, os.pardir, "proto", "build_" + arch))

if "ROOT" in os.environ and os.environ["ROOT"] != "":

        root_dir = os.environ["ROOT"]

else:

        root_dir = os.path.normpath(os.path.join(pwd, os.pardir, "proto", "root_" + arch))

pkgs_dir = os.path.normpath(os.path.join(pwd, os.pardir, "packages", arch))

extern_dir = os.path.normpath(os.path.join(pwd, "extern"))

py_install_dir = 'usr/lib/python2.6/vendor-packages'

scripts_dir = 'usr/bin'

lib_dir = 'usr/lib'

svc_method_dir = 'lib/svc/method'

man1_dir = 'usr/share/man/man1'

man1m_dir = 'usr/share/man/man1m'

man5_dir = 'usr/share/man/man5'

man1_ja_JP_dir = 'usr/share/man/ja_JP.UTF-8/man1'

man1m_ja_JP_dir = 'usr/share/man/ja_JP.UTF-8/man1m'

man5_ja_JP_dir = 'usr/share/man/ja_JP.UTF-8/man5'

man1_zh_CN_dir = 'usr/share/man/zh_CN.UTF-8/man1'

man1m_zh_CN_dir = 'usr/share/man/zh_CN.UTF-8/man1m'

man5_zh_CN_dir = 'usr/share/man/zh_CN.UTF-8/man5'

resource_dir = 'usr/share/lib/pkg'

transform_dir = 'usr/share/pkg/transforms'

smf_app_dir = 'lib/svc/manifest/application/pkg'

execattrd_dir = 'etc/security/exec_attr.d'

authattrd_dir = 'etc/security/auth_attr.d'

sysrepo_dir = 'etc/pkg/sysrepo'

sysrepo_logs_dir = 'var/log/pkg/sysrepo'

sysrepo_cache_dir = 'var/cache/pkg/sysrepo'

# A list of source, destination tuples of modules which should be hardlinked

# together if the os supports it and otherwise copied.

hardlink_modules = []

scripts_sunos = {

        scripts_dir: [

                ['client.py', 'pkg'],

                ['pkgdep.py', 'pkgdepend'],

                ['pkgrepo.py', 'pkgrepo'],

                ['util/publish/pkgdiff.py', 'pkgdiff'],

                ['util/publish/pkgfmt.py', 'pkgfmt'],

                ['util/publish/pkglint.py', 'pkglint'],

                ['util/publish/pkgmerge.py', 'pkgmerge'],

                ['util/publish/pkgmogrify.py', 'pkgmogrify'],

                ['publish.py', 'pkgsend'],

                ['pull.py', 'pkgrecv'],

                ['sign.py', 'pkgsign'],

                ['packagemanager.py', 'packagemanager'],

                ['updatemanager.py', 'pm-updatemanager'],

                ],

        lib_dir: [

                ['depot.py', 'pkg.depotd'],

                ['checkforupdates.py', 'pm-checkforupdates'],

                ['updatemanagernotifier.py', 'updatemanagernotifier'],

                ['launch.py', 'pm-launch'],

                ['sysrepo.py', 'pkg.sysrepo'],

                ],

        svc_method_dir: [

                ['svc/svc-pkg-depot', 'svc-pkg-depot'],

                ['svc/svc-pkg-mdns', 'svc-pkg-mdns'],

                ['svc/svc-pkg-sysrepo', 'svc-pkg-sysrepo'],

                ],

        }

scripts_windows = {

        scripts_dir: [

                ['client.py', 'client.py'],

                ['pkgrepo.py', 'pkgrepo.py'],

                ['publish.py', 'publish.py'],

                ['pull.py', 'pull.py'],

                ['scripts/pkg.bat', 'pkg.bat'],

                ['scripts/pkgsend.bat', 'pkgsend.bat'],

                ['scripts/pkgrecv.bat', 'pkgrecv.bat'],

                ],

        lib_dir: [

                ['depot.py', 'depot.py'],

                ['scripts/pkg.depotd.bat', 'pkg.depotd.bat'],

                ],

        }

scripts_other_unix = {

        scripts_dir: [

                ['client.py', 'client.py'],

                ['pkgdep.py', 'pkgdep'],

                ['util/publish/pkgdiff.py', 'pkgdiff'],

                ['util/publish/pkgfmt.py', 'pkgfmt'],

                ['util/publish/pkgmogrify.py', 'pkgmogrify'],

                ['pull.py', 'pull.py'],

                ['publish.py', 'publish.py'],

                ['scripts/pkg.sh', 'pkg'],

                ['scripts/pkgsend.sh', 'pkgsend'],

                ['scripts/pkgrecv.sh', 'pkgrecv'],

                ],

        lib_dir: [

                ['depot.py', 'depot.py'],

                ['scripts/pkg.depotd.sh', 'pkg.depotd'],

                ],

        }

# indexed by 'osname'

scripts = {

        "sunos": scripts_sunos,

        "linux": scripts_other_unix,

        "windows": scripts_windows,

        "darwin": scripts_other_unix,

        "aix" : scripts_other_unix,

        "unknown": scripts_sunos,

        }

man1_files = [

        'man/packagemanager.1',

        'man/pkg.1',

        'man/pkgdepend.1',

        'man/pkgdiff.1',

        'man/pkgfmt.1',

        'man/pkglint.1',

        'man/pkgmerge.1',

        'man/pkgmogrify.1',

        'man/pkgsend.1',

        'man/pkgsign.1',

        'man/pkgrecv.1',

        'man/pkgrepo.1',

        'man/pm-updatemanager.1',

        ]

man1m_files = [

        'man/pkg.depotd.1m',

        'man/pkg.sysrepo.1m'

        ]

man5_files = [

        'man/pkg.5'

        ]

man1_ja_files = [

        'man/ja_JP/packagemanager.1',

        'man/ja_JP/pkg.1',

        'man/ja_JP/pkgdepend.1',

        'man/ja_JP/pkgdiff.1',

        'man/ja_JP/pkgfmt.1',

        'man/ja_JP/pkglint.1',

        'man/ja_JP/pkgmerge.1',

        'man/ja_JP/pkgmogrify.1',

        'man/ja_JP/pkgsend.1',

        'man/ja_JP/pkgsign.1',

        'man/ja_JP/pkgrecv.1',

        'man/ja_JP/pkgrepo.1',

        'man/ja_JP/pm-updatemanager.1',

        ]

man1m_ja_files = [

        'man/ja_JP/pkg.depotd.1m',

        'man/ja_JP/pkg.sysrepo.1m'

        ]

man5_ja_files = [

        'man/ja_JP/pkg.5'

        ]

man1_zh_CN_files = [

        'man/zh_CN/packagemanager.1',

        'man/zh_CN/pkg.1',

        'man/zh_CN/pkgdepend.1',

        'man/zh_CN/pkgdiff.1',

        'man/zh_CN/pkgfmt.1',

        'man/zh_CN/pkglint.1',

        'man/zh_CN/pkgmerge.1',

        'man/zh_CN/pkgmogrify.1',

        'man/zh_CN/pkgsend.1',

        'man/zh_CN/pkgsign.1',

        'man/zh_CN/pkgrecv.1',

        'man/zh_CN/pkgrepo.1',

        'man/zh_CN/pm-updatemanager.1',

        ]

man1m_zh_CN_files = [

        'man/zh_CN/pkg.depotd.1m',

        'man/zh_CN/pkg.sysrepo.1m'

        ]

man5_zh_CN_files = [

        'man/zh_CN/pkg.5'

        ]

packages = [

        'pkg',

        'pkg.actions',

        'pkg.bundle',

        'pkg.client',

        'pkg.client.linkedimage',

        'pkg.client.transport',

        'pkg.file_layout',

        'pkg.flavor',

        'pkg.lint',

        'pkg.portable',

        'pkg.publish',

        'pkg.server'

        ]

pylint_targets = [

        'pkg.altroot',

        'pkg.client.linkedimage',

        'pkg.client.pkgdefs',

        ]

web_files = []

for entry in os.walk("web"):

        web_dir, dirs, files = entry

        if not files:

                continue

        web_files.append((os.path.join(resource_dir, web_dir), [

            os.path.join(web_dir, f) for f in files

            if f != "Makefile"

            ]))

smf_app_files = [

        'svc/pkg-mdns.xml',

        'svc/pkg-server.xml',

        'svc/pkg-update.xml',

        'svc/pkg-system-repository.xml',

        'svc/zoneproxy-client.xml',

        'svc/zoneproxyd.xml'

        ]

resource_files = [

        'util/opensolaris.org.sections',

        'util/pkglintrc',

        ]

transform_files = [

        'util/publish/transforms/developer',

        'util/publish/transforms/documentation',

        'util/publish/transforms/locale',

        'util/publish/transforms/smf-manifests'

        ]

sysrepo_files = [

        'util/apache2/sysrepo/sysrepo_httpd.conf.mako',

        'util/apache2/sysrepo/sysrepo_publisher_response.mako',

        ]

sysrepo_log_stubs = [

        'util/apache2/sysrepo/logs/access_log',

        'util/apache2/sysrepo/logs/error_log',

        ]

execattrd_files = [

        'util/misc/exec_attr.d/package:pkg',

        'util/misc/exec_attr.d/package:pkg:package-manager'

]

authattrd_files = ['util/misc/auth_attr.d/package:pkg']

syscallat_srcs = [

        'modules/syscallat.c'

        ]

pspawn_srcs = [

        'modules/pspawn.c'

        ]

elf_srcs = [

        'modules/elf.c',

        'modules/elfextract.c',

        'modules/liblist.c',

        ]

arch_srcs = [

        'modules/arch.c'

        ]

_actions_srcs = [

        'modules/actions/_actions.c'

        ]

solver_srcs = [

        'modules/solver/solver.c',

        'modules/solver/py_solver.c'

        ]

solver_link_args = ["-lm", "-lc"]

if osname == 'sunos':

        solver_link_args = ["-ztext"] + solver_link_args

# Runs lint on the extension module source code

class pylint_func(Command):

        description = "Runs pylint tools over IPS python source code"

        user_options = []

        def initialize_options(self):

                pass

        def finalize_options(self):

                pass

        # Make string shell-friendly

        @staticmethod

        def escape(astring):

                return astring.replace(' ', '\\ ')

        def run(self, quiet=False):

                proto = os.path.join(root_dir, py_install_dir)

                sys.path.insert(0, proto)

                # Insert tests directory onto sys.path so any custom checkers

                # can be found.

                sys.path.insert(0, os.path.join(pwd, 'tests'))

                # assumes pylint is accessible on the sys.path

                from pylint import lint

                #

                # For some reason, the load-plugins option, when used in the

                # rcfile, does not work, so we put it here instead, to load

                # our custom checkers.

                #

                # Unfortunately, pylint seems pretty fragile and will crash if

                # we try to run it over all the current pkg source.  Hence for

                # now we only run it over a subset of the source.  As source

                # files are made pylint clean they should be added to the

                # pylint_targets list.

                #

                args = ['--load-plugins=multiplatform']

                if quiet:

                        args += ['--reports=no']

                args += ['--rcfile', os.path.join(pwd, 'tests', 'pylintrc')]

                args += pylint_targets

                lint.Run(args)

class pylint_func_quiet(pylint_func):

        def run(self, quiet=False):

                pylint_func.run(self, quiet=True)

include_dirs = [ 'modules' ]

lint_flags = [ '-u', '-axms', '-erroff=E_NAME_DEF_NOT_USED2' ]

# Runs lint on the extension module source code

class clint_func(Command):

        description = "Runs lint tools over IPS C extension source code"