# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this
# file, You can obtain one at http://mozilla.org/MPL/2.0/.

from __future__ import print_function
import json
import subprocess


def get_build_variants():
    print("Fetching build variants from gradle")
    output = _run_gradle_process('printBuildVariants')
    content = _extract_content_from_command_output(output, prefix='variants: ')
    variants = json.loads(content)

    if len(variants) == 0:
        raise ValueError("Could not get build variants from gradle")

    print("Got variants: {}".format(' '.join(variants)))

    return variants


def get_geckoview_versions():
    print("Fetching geckoview version from gradle")
    output = _run_gradle_process('printGeckoviewVersions')

    versions = {}
    for version_type in ('nightly',):
        version = _extract_content_from_command_output(output, prefix='{}: '.format(version_type))
        version = version.strip('"')
        versions[version_type] = version
        print('Got {} version: "{}"'.format(version_type, version))

    return versions


def _run_gradle_process(gradle_command):
    process = subprocess.Popen(["./gradlew", "--no-daemon", "--quiet", gradle_command], stdout=subprocess.PIPE)
    output, err = process.communicate()
    exit_code = process.wait()

    if exit_code is not 0:
        print("Gradle command returned error: {}".format(exit_code))

    return output


def _extract_content_from_command_output(output, prefix):
    variants_line = [line for line in output.split('\n') if line.startswith(prefix)][0]
    return variants_line.split(' ', 1)[1]