"""
Contains Builder Workflow Configs for different Runtimes
"""

import logging
import os
from typing import Dict, List, Optional, Union, cast

from samcli.lib.build.workflows import (
    CONFIG,
    DOTNET_CLIPACKAGE_CONFIG,
    GO_MOD_CONFIG,
    JAVA_GRADLE_CONFIG,
    JAVA_KOTLIN_GRADLE_CONFIG,
    JAVA_MAVEN_CONFIG,
    NODEJS_NPM_CONFIG,
    NODEJS_NPM_ESBUILD_CONFIG,
    PROVIDED_MAKE_CONFIG,
    PYTHON_PIP_CONFIG,
    RUBY_BUNDLER_CONFIG,
    RUST_CARGO_LAMBDA_CONFIG,
)
from samcli.lib.telemetry.event import EventTracker

LOG = logging.getLogger(__name__)


class UnsupportedRuntimeException(Exception):
    pass


class UnsupportedBuilderException(Exception):
    pass


WorkFlowSelector = Union["BasicWorkflowSelector", "ManifestWorkflowSelector"]


def get_selector(
    selector_list: List[Dict[str, WorkFlowSelector]],
    identifiers: List[Optional[str]],
    specified_workflow: Optional[str] = None,
) -> Optional[WorkFlowSelector]:
    """
    Determine the correct workflow selector from a list of selectors,
    series of identifiers and user specified workflow if defined.

    Parameters
    ----------
    selector_list list
        List of dictionaries, where the value of all dictionaries are workflow selectors.
    identifiers list
        List of identifiers specified in order of precedence that are to be looked up in selector_list.
    specified_workflow str
        User specified workflow for build.

    Returns
    -------
    selector(BasicWorkflowSelector)
        selector object which can specify a workflow configuration that can be passed to `aws-lambda-builders`

    """

    # Create a combined view of all the selectors
    all_selectors: Dict[str, WorkFlowSelector] = dict()
    for selector in selector_list:
        all_selectors = {**all_selectors, **selector}

    # Check for specified workflow being supported at all and if it's not, raise an UnsupportedBuilderException.
    if specified_workflow and specified_workflow not in all_selectors:
        raise UnsupportedBuilderException("'{}' does not have a supported builder".format(specified_workflow))

    # Loop through all identifiers to gather list of selectors with potential matches.
    selectors = [all_selectors.get(identifier) for identifier in identifiers if identifier]

    try:
        # Find first non-None selector.
        # Return the first selector with a match.
        return next(_selector for _selector in selectors if _selector)
    except StopIteration:
        pass

    return None


def get_layer_subfolder(build_workflow: str) -> str:
    subfolders_by_runtime = {
        "python3.8": "python",
        "python3.9": "python",
        "python3.10": "python",
        "python3.11": "python",
        "python3.12": "python",
        "python3.13": "python",
        "nodejs4.3": "nodejs",
        "nodejs6.10": "nodejs",
        "nodejs8.10": "nodejs",
        "nodejs16.x": "nodejs",
        "nodejs18.x": "nodejs",
        "nodejs20.x": "nodejs",
        "nodejs22.x": "nodejs",
        "ruby3.2": "ruby/lib",
        "ruby3.3": "ruby/lib",
        "ruby3.4": "ruby/lib",
        "java11": "java",
        "java8.al2": "java",
        "java17": "java",
        "java21": "java",
        "dotnet6": "dotnet",
        "dotnet8": "dotnet",
        # User is responsible for creating subfolder in these workflows
        "makefile": "",
    }

    if build_workflow not in subfolders_by_runtime:
        raise UnsupportedRuntimeException("'{}' runtime is not supported for layers".format(build_workflow))

    return subfolders_by_runtime[build_workflow]


def get_workflow_config(
    runtime: Optional[str], code_dir: str, project_dir: str, specified_workflow: Optional[str] = None
) -> CONFIG:
    """
    Get a workflow config that corresponds to the runtime provided. This method examines contents of the project
    and code directories to determine the most appropriate workflow for the given runtime. Currently the decision is
    based on the presence of a supported manifest file. For runtimes that have more than one workflow, we choose a
    workflow by examining ``code_dir`` followed by ``project_dir`` for presence of a supported manifest.

    Parameters
    ----------
    runtime str
        The runtime of the config

    code_dir str
        Directory where Lambda function code is present

    project_dir str
        Root of the Serverless application project.

    specified_workflow str
        Workflow to be used, if directly specified. They are currently scoped to "makefile" and the official runtime
        identifier names themselves, eg: nodejs20.x. If a workflow is not directly specified,
        it is calculated by the current method based on the runtime.

    Returns
    -------
    namedtuple(Capability)
        namedtuple that represents the Builder Workflow Config
    """

    selectors_by_build_method = {
        "makefile": BasicWorkflowSelector(PROVIDED_MAKE_CONFIG),
        "dotnet7": BasicWorkflowSelector(DOTNET_CLIPACKAGE_CONFIG),
        "dotnet": BasicWorkflowSelector(DOTNET_CLIPACKAGE_CONFIG),
        "rust-cargolambda": BasicWorkflowSelector(RUST_CARGO_LAMBDA_CONFIG),
    }

    selectors_by_runtime = {
        "python3.8": BasicWorkflowSelector(PYTHON_PIP_CONFIG),
        "python3.9": BasicWorkflowSelector(PYTHON_PIP_CONFIG),
        "python3.10": BasicWorkflowSelector(PYTHON_PIP_CONFIG),
        "python3.11": BasicWorkflowSelector(PYTHON_PIP_CONFIG),
        "python3.12": BasicWorkflowSelector(PYTHON_PIP_CONFIG),
        "python3.13": BasicWorkflowSelector(PYTHON_PIP_CONFIG),
        "nodejs16.x": BasicWorkflowSelector(NODEJS_NPM_CONFIG),
        "nodejs18.x": BasicWorkflowSelector(NODEJS_NPM_CONFIG),
        "nodejs20.x": BasicWorkflowSelector(NODEJS_NPM_CONFIG),
        "nodejs22.x": BasicWorkflowSelector(NODEJS_NPM_CONFIG),
        "ruby3.2": BasicWorkflowSelector(RUBY_BUNDLER_CONFIG),
        "ruby3.3": BasicWorkflowSelector(RUBY_BUNDLER_CONFIG),
        "ruby3.4": BasicWorkflowSelector(RUBY_BUNDLER_CONFIG),
        "dotnet6": BasicWorkflowSelector(DOTNET_CLIPACKAGE_CONFIG),
        "dotnet8": BasicWorkflowSelector(DOTNET_CLIPACKAGE_CONFIG),
        "go1.x": BasicWorkflowSelector(GO_MOD_CONFIG),
        # When Maven builder exists, add to this list so we can automatically choose a builder based on the supported
        # manifest
        "java11": ManifestWorkflowSelector(
            [
                # Gradle builder needs custom executable paths to find `gradlew` binary
                JAVA_GRADLE_CONFIG._replace(executable_search_paths=[code_dir, project_dir]),
                JAVA_KOTLIN_GRADLE_CONFIG._replace(executable_search_paths=[code_dir, project_dir]),
                JAVA_MAVEN_CONFIG,
            ]
        ),
        "java8.al2": ManifestWorkflowSelector(
            [
                # Gradle builder needs custom executable paths to find `gradlew` binary
                JAVA_GRADLE_CONFIG._replace(executable_search_paths=[code_dir, project_dir]),
                JAVA_KOTLIN_GRADLE_CONFIG._replace(executable_search_paths=[code_dir, project_dir]),
                JAVA_MAVEN_CONFIG,
            ]
        ),
        "java17": ManifestWorkflowSelector(
            [
                # Gradle builder needs custom executable paths to find `gradlew` binary
                JAVA_GRADLE_CONFIG._replace(executable_search_paths=[code_dir, project_dir]),
                JAVA_KOTLIN_GRADLE_CONFIG._replace(executable_search_paths=[code_dir, project_dir]),
                JAVA_MAVEN_CONFIG,
            ]
        ),
        "java21": ManifestWorkflowSelector(
            [
                # Gradle builder needs custom executable paths to find `gradlew` binary
                JAVA_GRADLE_CONFIG._replace(executable_search_paths=[code_dir, project_dir]),
                JAVA_KOTLIN_GRADLE_CONFIG._replace(executable_search_paths=[code_dir, project_dir]),
                JAVA_MAVEN_CONFIG,
            ]
        ),
        "provided": BasicWorkflowSelector(PROVIDED_MAKE_CONFIG),
        "provided.al2": BasicWorkflowSelector(PROVIDED_MAKE_CONFIG),
        "provided.al2023": BasicWorkflowSelector(PROVIDED_MAKE_CONFIG),
    }

    selectors_by_builder = {
        "esbuild": BasicWorkflowSelector(NODEJS_NPM_ESBUILD_CONFIG),
    }

    # First check if the runtime is present and is buildable, if not raise an UnsupportedRuntimeException Error.
    # If runtime is present it should be in selectors_by_runtime, however for layers there will be no runtime
    # so in that case we move ahead and resolve to any matching workflow from both types.
    if runtime and runtime not in selectors_by_runtime:
        raise UnsupportedRuntimeException("'{}' runtime is not supported".format(runtime))

    try:
        # Identify appropriate workflow selector.
        selector = get_selector(
            selector_list=[selectors_by_build_method, selectors_by_runtime, selectors_by_builder],
            identifiers=[specified_workflow, runtime],
            specified_workflow=specified_workflow,
        )

        # pylint: disable=fixme
        # FIXME: selector could be None here, we should raise an exception if it is None.

        # Identify workflow configuration from the workflow selector.
        config = cast(WorkFlowSelector, selector).get_config(code_dir, project_dir)

        EventTracker.track_event("BuildWorkflowUsed", f"{config.language}-{config.dependency_manager}")

        return config
    except ValueError as ex:
        raise UnsupportedRuntimeException(
            "Unable to find a supported build workflow for runtime '{}'. Reason: {}".format(runtime, str(ex))
        ) from ex


def supports_specified_workflow(specified_workflow: Optional[str]) -> bool:
    """
    Given a specified workflow, returns whether it is supported in container builds,
    can be used to overwrite runtime and get docker image or not

    Parameters
    ----------
    specified_workflow
        Workflow specified in the template

    Returns
    -------
    bool
        True, if this workflow is supported, can be used to overwrite runtime and get docker image
    """

    supported_specified_workflow = ["dotnet7"]

    return specified_workflow in supported_specified_workflow


class BasicWorkflowSelector:
    """
    Basic workflow selector that returns the first available configuration in the given list of configurations
    """

    def __init__(self, configs: Union[CONFIG, List[CONFIG]]) -> None:
        if not isinstance(configs, list):
            configs = [configs]

        self.configs: List[CONFIG] = configs

    def get_config(self, code_dir: str, project_dir: str) -> CONFIG:
        """
        Returns the first available configuration
        """
        return self.configs[0]


class ManifestWorkflowSelector(BasicWorkflowSelector):
    """
    Selects a workflow by examining the directories for presence of a supported manifest
    """

    def get_config(self, code_dir: str, project_dir: str) -> CONFIG:
        """
        Finds a configuration by looking for a manifest in the given directories.

        Returns
        -------
        samcli.lib.build.workflow_config.CONFIG
            A supported configuration if one is found

        Raises
        ------
        ValueError
            If none of the supported manifests files are found
        """

        # Search for manifest first in code directory and then in the project directory.
        # Search order is important here because we want to prefer the manifest present within the code directory over
        # a manifest present in project directory.
        search_dirs = [code_dir, project_dir]
        LOG.debug("Looking for a supported build workflow in following directories: %s", search_dirs)

        for config in self.configs:
            if any([self._has_manifest(config, directory) for directory in search_dirs]):
                return config

        raise ValueError(
            "None of the supported manifests '{}' were found in the following paths '{}'".format(
                [config.manifest_name for config in self.configs], search_dirs
            )
        )

    @staticmethod
    def _has_manifest(config: CONFIG, directory: str) -> bool:
        return os.path.exists(os.path.join(directory, config.manifest_name))
