#!/usr/bin/env python

#
# Licensed to the Apache Software Foundation (ASF) under one
# or more contributor license agreements. See the NOTICE file
# distributed with this work for additional information
# regarding copyright ownership. The ASF licenses this file
# to you under the Apache License, Version 2.0 (the
# "License"); you may not use this file except in compliance
# with the License. You may obtain a copy of the License at
#
#   http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
# KIND, either express or implied. See the License for the
# specific language governing permissions and limitations
# under the License.
#

from distutils.core import setup, Extension as _Extension
from distutils.command.build_ext import build_ext as _build_ext
from distutils.errors import CCompilerError, DistutilsError, CompileError

import sys


class Extension(_Extension):
    """Add an `optional` kwarg, to allow skipping on failure"""
    def __init__(self, name, sources, optional=False, **kwargs):
        _Extension.__init__(self, name, sources, **kwargs)
        self.optional = optional


class build_ext(_build_ext):
    """Build extensions, but skip on failure if optional is set"""
    def build_extensions(self):
        self.check_extensions_list(self.extensions)

        for ext in self.extensions:
            try:
                self.build_extension(ext)
            except (CCompilerError, DistutilsError, CompileError) as e:
                if not ext.optional:
                    raise
                self.warn('building extension "%s" failed: %s' %
                          (ext.name, e))


fastprotomod = Extension(
    'thrift.protocol.fastproto',
    sources = ['protocol/fastproto.cpp'],
    libraries=['thriftcpp2', 'folly'],
    extra_compile_args=['-std=c++0x'],
    optional=True,
)

version_info = sys.version_info
boost_python = "boost_python-py{}{}".format(version_info[0], version_info[1])
cppservermod = Extension(
    'thrift.server.CppServerWrapper',
    sources = ['server/CppServerWrapper.cpp'],
    libraries=[boost_python, 'thriftcpp2', 'folly', 'wangle'],
    extra_compile_args=['-std=c++0x', '-fno-strict-aliasing'],
    optional=True,
)

setup(name = 'Thrift',
      version = '0.1',
      description = 'Thrift Python Libraries',
      author = 'Thrift Developers',
      author_email = 'thrift-dev@incubator.apache.org',
      url = 'http://incubator.apache.org/thrift/',
      license = 'Apache License 2.0',
      packages = [
        'thrift',
        'thrift.protocol',
        'thrift.transport',
        'thrift.server',
        'thrift.util',
      ],
      package_dir = {'thrift' : '.'},
      ext_modules = [fastprotomod, cppservermod],
      cmdclass={'build_ext': build_ext},
      )
