#!/usr/bin/env python3
# Copyright 2012 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
r"""Prints the lowest locally available SDK version greater than or equal to a
given minimum sdk version to standard output.

If --print_sdk_path is passed, then the script will also print the SDK path.
If --print_bin_path is passed, then the script will also print the path to the
toolchain bin dir.

Usage:
  python find_sdk.py     \
      [--print_sdk_path] \
      [--print_bin_path] \
      10.6  # Ignores SDKs < 10.6

Sample Output:
/Applications/Xcode.app/Contents/Developer/Platforms/MacOSX.platform/Developer/SDKs/MacOSX10.14.sdk
/Applications/Xcode.app/Contents/Developer/Toolchains/XcodeDefault.xctoolchain/usr/bin/
10.14
"""

import os
import plistlib
import re
import subprocess
import sys

from optparse import OptionParser


class SdkError(Exception):
    def __init__(self, value):
        self.value = value

    def __str__(self):
        return repr(self.value)


def parse_version(version_str):
    """'10.6' => [10, 6]"""
    return [int(s) for s in re.findall(r'(\d+)', version_str)]


def main():
    parser = OptionParser()
    parser.add_option(
        "--print_sdk_path",
        action="store_true",
        dest="print_sdk_path",
        default=False,
        help="Additionally print the path the SDK (appears first).",
    )
    parser.add_option(
        "--print_bin_path",
        action="store_true",
        dest="print_bin_path",
        default=False,
        help="Additionally print the path the toolchain bin dir.",
    )
    parser.add_option(
        "--print_sdk_build",
        action="store_true",
        dest="print_sdk_build",
        default=False,
        help="Additionally print the build version of the SDK.",
    )
    options, args = parser.parse_args()
    if len(args) != 1:
        parser.error('Please specify a minimum SDK version')
    min_sdk_version = args[0]

    job = subprocess.Popen(
        ['xcode-select', '-print-path'],
        stdout=subprocess.PIPE,
        stderr=subprocess.STDOUT,
    )
    out, err = job.communicate()
    if job.returncode != 0:
        print(out, file=sys.stderr)
        print(err, file=sys.stderr)
        raise Exception('Error %d running xcode-select' % job.returncode)
    dev_dir = out.decode('UTF-8').rstrip()
    sdk_dir = os.path.join(dev_dir, 'Platforms/MacOSX.platform/Developer/SDKs')

    if not os.path.isdir(sdk_dir):
        raise SdkError(
            'Install Xcode, launch it, accept the license '
            + 'agreement, and run `sudo xcode-select -s /path/to/Xcode.app` '
            + 'to continue.'
        )
    sdk_dir_list = os.listdir(sdk_dir)
    sdks = [re.findall('^MacOSX(\d+\.\d+)\.sdk$', s) for s in sdk_dir_list]
    sdks = [s[0] for s in sdks if s]  # [['10.5'], ['10.6']] => ['10.5', '10.6']
    sdks = [
        s
        for s in sdks  # ['10.5', '10.6'] => ['10.6']
        if parse_version(s) >= parse_version(min_sdk_version)
    ]
    if not sdks:
        raise Exception(
            f'No {min_sdk_version}+ SDK found. {sdk_dir} contains: {sdk_dir_list}'
        )
    best_sdk = sorted(sdks, key=parse_version)[0]
    sdk_name = 'MacOSX' + best_sdk + '.sdk'
    sdk_path = os.path.join(sdk_dir, sdk_name)

    if options.print_sdk_path:
        print(sdk_path)

    if options.print_bin_path:
        bin_path = 'Toolchains/XcodeDefault.xctoolchain/usr/bin/'
        print(os.path.join(dev_dir, bin_path))

    if options.print_sdk_build:
        system_version_plist = os.path.join(
            sdk_path, 'System/Library/CoreServices/SystemVersion.plist'
        )
        with open(system_version_plist, 'rb') as f:
            system_version_info = plistlib.load(f)
            if 'ProductBuildVersion' not in system_version_info:
                raise Exception(
                    'Failed to determine ProductBuildVersion'
                    + 'for SDK at path %s' % system_version_plist
                )
            print(system_version_info['ProductBuildVersion'])

    print(best_sdk)


if __name__ == '__main__':
    if sys.platform != 'darwin':
        raise Exception("This script only runs on Mac")
    sys.exit(main())