#!/usr/bin/env python
# -*- coding: utf-8 -*-

"""
Setup script for building clr.pyd and dependencies using mono and into
an egg or wheel.
"""

import collections
import fnmatch
import glob
import os
import subprocess
import sys
import sysconfig
from distutils import spawn
from distutils.command import install, build, build_ext, install_data, install_lib

from setuptools import Extension, setup

try:
    from wheel import bdist_wheel
except ImportError:
    bdist_wheel = None

# Allow config/verbosity to be set from cli
# http://stackoverflow.com/a/4792601/5208670
CONFIG = "Release"  # Release or Debug
VERBOSITY = "normal"  # quiet, minimal, normal, detailed, diagnostic

is_64bits = sys.maxsize > 2 ** 32
DEVTOOLS = "MsDev" if sys.platform == "win32" else "Mono"
ARCH = "x64" if is_64bits else "x86"
PY_MAJOR = sys.version_info[0]
PY_MINOR = sys.version_info[1]

###############################################################################
# Windows Keys Constants for MSBUILD tools
RegKey = collections.namedtuple("RegKey", "sdk_name key value_name suffix")
vs_python = "Programs\\Common\\Microsoft\\Visual C++ for Python\\9.0\\WinSDK"
vs_root = "SOFTWARE\\Microsoft\\MSBuild\\ToolsVersions\\{0}"
sdks_root = "SOFTWARE\\Microsoft\\Microsoft SDKs\\Windows\\v{0}Win32Tools"
kits_root = "SOFTWARE\\Microsoft\\Windows Kits\\Installed Roots"
kits_suffix = os.path.join("bin", ARCH)

WIN_SDK_KEYS = [
    RegKey(
        sdk_name="Windows Kit 10.0",
        key=kits_root,
        value_name="KitsRoot10",
        suffix=os.path.join("bin", "10.0.16299.0", ARCH),
    ),
    RegKey(
        sdk_name="Windows Kit 10.0",
        key=kits_root,
        value_name="KitsRoot10",
        suffix=os.path.join("bin", "10.0.15063.0", ARCH),
    ),
    RegKey(
        sdk_name="Windows Kit 10.0",
        key=kits_root,
        value_name="KitsRoot10",
        suffix=kits_suffix,
    ),
    RegKey(
        sdk_name="Windows Kit 8.1",
        key=kits_root,
        value_name="KitsRoot81",
        suffix=kits_suffix,
    ),
    RegKey(
        sdk_name="Windows Kit 8.0",
        key=kits_root,
        value_name="KitsRoot",
        suffix=kits_suffix,
    ),
    RegKey(
        sdk_name="Windows SDK 7.1A",
        key=sdks_root.format("7.1A\\WinSDK-"),
        value_name="InstallationFolder",
        suffix="",
    ),
    RegKey(
        sdk_name="Windows SDK 7.1",
        key=sdks_root.format("7.1\\WinSDK"),
        value_name="InstallationFolder",
        suffix="",
    ),
    RegKey(
        sdk_name="Windows SDK 7.0A",
        key=sdks_root.format("7.0A\\WinSDK-"),
        value_name="InstallationFolder",
        suffix="",
    ),
    RegKey(
        sdk_name="Windows SDK 7.0",
        key=sdks_root.format("7.0\\WinSDK"),
        value_name="InstallationFolder",
        suffix="",
    ),
    RegKey(
        sdk_name="Windows SDK 6.0A",
        key=sdks_root.format("6.0A\\WinSDK"),
        value_name="InstallationFolder",
        suffix="",
    ),
]

VS_KEYS = (
    RegKey(
        sdk_name="MSBuild 15",
        key=vs_root.format("15.0"),
        value_name="MSBuildToolsPath",
        suffix="",
    ),
    RegKey(
        sdk_name="MSBuild 14",
        key=vs_root.format("14.0"),
        value_name="MSBuildToolsPath",
        suffix="",
    ),
    RegKey(
        sdk_name="MSBuild 12",
        key=vs_root.format("12.0"),
        value_name="MSBuildToolsPath",
        suffix="",
    ),
    RegKey(
        sdk_name="MSBuild 4",
        key=vs_root.format("4.0"),
        value_name="MSBuildToolsPath",
        suffix="",
    ),
    RegKey(
        sdk_name="MSBuild 3.5",
        key=vs_root.format("3.5"),
        value_name="MSBuildToolsPath",
        suffix="",
    ),
    RegKey(
        sdk_name="MSBuild 2.0",
        key=vs_root.format("2.0"),
        value_name="MSBuildToolsPath",
        suffix="",
    ),
)


###############################################################################
def _check_output(*args, **kwargs):
    """Check output wrapper for py2/py3 compatibility"""
    output = subprocess.check_output(*args, **kwargs)
    if PY_MAJOR == 2:
        return output
    return output.decode("ascii")


def _get_interop_filename():
    """interopXX.cs is auto-generated as part of the build.
    For common windows platforms pre-generated files are included
    as most windows users won't have Clang installed, which is
    required to generate the file.
    """
    interop_filename = "interop{0}{1}{2}.cs".format(
        PY_MAJOR, PY_MINOR, getattr(sys, "abiflags", "")
    )
    return os.path.join("src", "runtime", interop_filename)


def _get_source_files():
    """Walk project and collect the files needed for ext_module"""
    for ext in (".sln",):
        for path in glob.glob("*" + ext):
            yield path

    for root, dirnames, filenames in os.walk("src"):
        for ext in (".cs", ".csproj", ".snk", ".config", ".py", ".c", ".h", ".ico"):
            for filename in fnmatch.filter(filenames, "*" + ext):
                yield os.path.join(root, filename)

    for root, dirnames, filenames in os.walk("tools"):
        for ext in (".exe", ".py", ".c", ".h"):
            for filename in fnmatch.filter(filenames, "*" + ext):
                yield os.path.join(root, filename)


def _get_long_description():
    """Helper to populate long_description for pypi releases"""
    return open("README.rst").read()


def _update_xlat_devtools():
    global DEVTOOLS
    if DEVTOOLS == "MsDev":
        DEVTOOLS = "MsDev15"
    elif DEVTOOLS == "Mono":
        DEVTOOLS = "dotnet"


def _collect_installed_windows_kits_v10(winreg):
    """Adds the installed Windows 10 kits to WIN_SDK_KEYS """
    global WIN_SDK_KEYS
    installed_kits = []

    with winreg.OpenKey(
        winreg.HKEY_LOCAL_MACHINE, kits_root, 0, winreg.KEY_READ
    ) as key:
        i = 0
        while True:
            try:
                installed_kits.append(winreg.EnumKey(key, i))
                i += 1
            except WindowsError:
                break

    def make_reg_key(version):
        return RegKey(
            sdk_name="Windows Kit 10.0",
            key=kits_root,
            value_name="KitsRoot10",
            suffix=os.path.join("bin", version, ARCH),
        )

    WIN_SDK_KEYS += [make_reg_key(e) for e in installed_kits if e.startswith("10.")]

    # Make sure this function won't be called again
    _collect_installed_windows_kits_v10 = lambda: None


class BuildExtPythonnet(build_ext.build_ext):
    user_options = build_ext.build_ext.user_options + [("xplat", None, None)]

    def initialize_options(self):
        build_ext.build_ext.initialize_options(self)
        self.xplat = None

    def finalize_options(self):
        build_ext.build_ext.finalize_options(self)

    def build_extension(self, ext):
        if self.xplat:
            _update_xlat_devtools()

        """Builds the .pyd file using msbuild or xbuild"""
        if ext.name != "clr":
            return build_ext.build_ext.build_extension(self, ext)

        # install packages using nuget
        self._install_packages()

        dest_file = self.get_ext_fullpath(ext.name)
        dest_dir = os.path.dirname(dest_file)
        if not os.path.exists(dest_dir):
            os.makedirs(dest_dir)

        # Up to Python 3.2 sys.maxunicode is used to determine the size of
        # Py_UNICODE, but from 3.3 onwards Py_UNICODE is a typedef of wchar_t.
        # TODO: Is this doing the right check for Py27?
        if sys.version_info[:2]