##############################################################################

#

# Copyright (c) 2006 Zope Corporation and Contributors.

# All Rights Reserved.

#

# This software is subject to the provisions of the Zope Public License,

# Version 2.1 (ZPL).  A copy of the ZPL should accompany this distribution.

# THIS SOFTWARE IS PROVIDED "AS IS" AND ANY AND ALL EXPRESS OR IMPLIED

# WARRANTIES ARE DISCLAIMED, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED

# WARRANTIES OF TITLE, MERCHANTABILITY, AGAINST INFRINGEMENT, AND FITNESS

# FOR A PARTICULAR PURPOSE.

#

##############################################################################

"""Install packages as eggs

$Id: custom.py 84019 2008-02-18 12:41:04Z zagy $

"""

import logging, os, re, zipfile

import zc.buildout.easy_install

logger = logging.getLogger(__name__)

class Base:

    def __init__(self, buildout, name, options):

        self.name, self.options = name, options

        options['_d'] = buildout['buildout']['develop-eggs-directory']

        python = options.get('python', buildout['buildout']['python'])

        options['executable'] = buildout[python]['executable']

        self.build_ext = build_ext(buildout, options)

    def update(self):

        return self.install()

class Custom(Base):

    def __init__(self, buildout, name, options):

        Base.__init__(self, buildout, name, options)

        links = options.get('find-links',

                            buildout['buildout'].get('find-links'))

        if links:

            links = links.split()

            options['find-links'] = '\n'.join(links)

        else:

            links = ()

        self.links = links

        index = options.get('index', buildout['buildout'].get('index'))

        if index is not None:

            options['index'] = index

        self.index = index

        environment_section = options.get('environment')

        if environment_section:

            self.environment = buildout[environment_section]

        else:

            self.environment = {}

        environment_data = self.environment.items()

        environment_data.sort()

        options['_environment-data'] = repr(environment_data)

        options['_e'] = buildout['buildout']['eggs-directory']

        assert options.get('unzip') in ('true', 'false', None)

        if buildout['buildout'].get('offline') == 'true':

            self.install = lambda: ()

        self.newest = buildout['buildout'].get('newest') == 'true'

    def install(self):

        options = self.options

        distribution = options.get('egg')

        if distribution is None:

            distribution = options.get('eggs')

            if distribution is None:

                distribution = self.name

            else:

                logger.warn("The eggs option is deprecated. Use egg instead")

        distribution = options.get('egg', options.get('eggs', self.name)

                                   ).strip()

        self._set_environment()

        try:

            return zc.buildout.easy_install.build(

                distribution, options['_d'], self.build_ext,

                self.links, self.index, options['executable'], [options['_e']],

                newest=self.newest,

                )

        finally:

            self._restore_environment()

    def _set_environment(self):

        self._saved_environment = {}

        for key, value in self.environment.items():

            if key in os.environ:

                self._saved_environment[key] = os.environ[key]

            # Interpolate value with variables from environment. Maybe there

            # should be a general way of doing this in buildout with something

            # like ${environ:foo}:

            os.environ[key] = value % os.environ

    def _restore_environment(self):

        for key in self.environment:

            if key in self._saved_environment:

                os.environ[key] = self._saved_environment[key]

            else:

                try:

                    del os.environ[key]

                except KeyError:

                    pass

class Develop(Base):

    def __init__(self, buildout, name, options):

        Base.__init__(self, buildout, name, options)

        options['setup'] = os.path.join(buildout['buildout']['directory'],

                                        options['setup'])

    def install(self):

        options = self.options

        return zc.buildout.easy_install.develop(

            options['setup'], options['_d'], self.build_ext,

            options['executable'],

            )

def build_ext(buildout, options):

    result = {}

    for be_option in ('include-dirs', 'library-dirs', 'rpath'):

        value = options.get(be_option)

        if value is None:

            continue

        value = [

            os.path.join(

                buildout['buildout']['directory'],

                v.strip()

                )

            for v in value.strip().split('\n')

            if v.strip()

        ]

        result[be_option] = os.pathsep.join(value)

        options[be_option] = os.pathsep.join(value)

    swig = options.get('swig')

    if swig:

        options['swig'] = result['swig'] = os.path.join(

            buildout['buildout']['directory'],

            swig,

            )

    for be_option in ('define', 'undef', 'libraries', 'link-objects',

                      'debug', 'force', 'compiler', 'swig-cpp', 'swig-opts',

                      ):

        value = options.get(be_option)

        if value is None:

            continue

        result[be_option] = value

    return result