# Copyright 2019 Dirk Thomas
# Licensed under the Apache License, Version 2.0

import os

from colcon_core.logging import colcon_logger
from colcon_core.plugin_system import satisfies_version
from colcon_core.prefix_path import PrefixPathExtensionPoint

logger = colcon_logger.getChild(__name__)

_get_ament_prefix_path_warnings = set()


class AmentPrefixPath(PrefixPathExtensionPoint):
    """Prefix path defined in the `AMENT_PREFIX_PATH` environment variable."""

    # the priority needs to be lower than the colcon prefix path extension
    PRIORITY = 90

    def __init__(self):  # noqa: D107
        super().__init__()
        satisfies_version(
            PrefixPathExtensionPoint.EXTENSION_POINT_VERSION, '^1.0')

    def extend_prefix_path(self, paths):  # noqa: D102
        global _get_ament_prefix_path_warnings
        ament_prefix_path = os.environ.get('AMENT_PREFIX_PATH', '')
        for path in ament_prefix_path.split(os.pathsep):
            if not path:
                continue
            if not os.path.exists(path):
                if path not in _get_ament_prefix_path_warnings:
                    logger.warning(
                        f"The path '{path}' in the environment variable "
                        "AMENT_PREFIX_PATH doesn't exist")
                _get_ament_prefix_path_warnings.add(path)
                continue

            for filename in os.listdir(path):
                if filename.startswith('local_setup.'):
                    break
            else:
                parent_path = os.path.dirname(path)
                marker_file = os.path.join(
                    parent_path, '.colcon_install_layout')
                if not os.path.exists(marker_file):
                    if path not in _get_ament_prefix_path_warnings:
                        logger.warning(
                            f"The path '{path}' in the environment variable "
                            "AMENT_PREFIX_PATH doesn't contain any "
                            "'local_setup.*' files.")
                        _get_ament_prefix_path_warnings.add(path)
                    continue
                with open(marker_file, 'r') as h:
                    install_layout = h.read().rstrip()
                if install_layout != 'isolated':
                    if path not in _get_ament_prefix_path_warnings:
                        logger.warning(
                            f"The path '{path}' in the environment variable "
                            "AMENT_PREFIX_PATH doesn't use the expected "
                            "install layout 'isolated'.")
                        _get_ament_prefix_path_warnings.add(path)
                    continue
                path = parent_path

            paths.append(path)
