# Copyright 2023 The Servo Project Developers. See the COPYRIGHT
# file at the top-level directory of this distribution.
#
# Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
# http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
# <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
# option. This file may not be copied, modified, or distributed
# except according to those terms.

import os
import subprocess
import tempfile
from typing import Optional

from .. import util
from .base import Base

URL_BASE = "https://github.com/servo/servo-build-deps/releases/download/macOS"
GSTREAMER_URL = f"{URL_BASE}/gstreamer-1.0-1.22.2-universal.pkg"
GSTREAMER_DEVEL_URL = f"{URL_BASE}/gstreamer-1.0-devel-1.22.2-universal.pkg"
GSTREAMER_ROOT = "/Library/Frameworks/GStreamer.framework/Versions/1.0"


class MacOS(Base):
    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.is_macos = True

    def library_path_variable_name(self):
        return "DYLD_LIBRARY_PATH"

    def gstreamer_root(self, cross_compilation_target: Optional[str]) -> Optional[str]:
        # We do not support building with gstreamer while cross-compiling on MacOS.
        if cross_compilation_target or not os.path.exists(GSTREAMER_ROOT):
            return None
        return GSTREAMER_ROOT

    def is_gstreamer_installed(self, cross_compilation_target: Optional[str]) -> bool:
        if not super().is_gstreamer_installed(cross_compilation_target):
            return False

        # Servo only supports the official GStreamer distribution on MacOS.
        env = os.environ.copy()
        self.set_gstreamer_environment_variables_if_necessary(
            env, cross_compilation_target, check_installation=False
        )
        gst_lib_dir = subprocess.check_output(
            ["pkg-config", "--variable=libdir", "gstreamer-1.0"], env=env
        )
        if not gst_lib_dir.startswith(bytes(GSTREAMER_ROOT, "utf-8")):
            print("GStreamer is installed, but not the official packages.\n"
                  "Run `./mach bootstrap-gtstreamer` or install packages from "
                  "https://gstreamer.freedesktop.org/")
            return False
        return True

    def _platform_bootstrap(self, _cache_dir: str, force: bool) -> bool:
        installed_something = False
        try:
            brewfile = os.path.join(util.SERVO_ROOT, "etc", "homebrew", "Brewfile")
            output = subprocess.check_output(
                ['brew', 'bundle', 'install', "--file", brewfile]
            ).decode("utf-8")
            print(output)
            installed_something = "Installing" in output
        except subprocess.CalledProcessError as e:
            print("Could not run homebrew. Is it installed?")
            raise e
        installed_something |= self._platform_bootstrap_gstreamer(False)
        return installed_something

    def _platform_bootstrap_gstreamer(self, force: bool) -> bool:
        if not force and self.is_gstreamer_installed(cross_compilation_target=None):
            return False

        with tempfile.TemporaryDirectory() as temp_dir:
            libs_pkg = os.path.join(temp_dir, GSTREAMER_URL.rsplit("/", maxsplit=1)[-1])
            devel_pkg = os.path.join(
                temp_dir, GSTREAMER_DEVEL_URL.rsplit("/", maxsplit=1)[-1]
            )

            util.download_file("GStreamer libraries", GSTREAMER_URL, libs_pkg)
            util.download_file(
                "GStreamer development support", GSTREAMER_DEVEL_URL, devel_pkg
            )

            print("Installing GStreamer packages...")
            subprocess.check_call(
                [
                    "sudo",
                    "sh",
                    "-c",
                    f"installer -pkg '{libs_pkg}' -target / &&"
                    f"installer -pkg '{devel_pkg}' -target /",
                ]
            )

            assert self.is_gstreamer_installed(cross_compilation_target=None)
            return True