"""
Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
SPDX-License-Identifier: MIT-0
"""

from __future__ import annotations

import fnmatch
import json
import os
import pathlib
from typing import Any, List
from unittest import TestCase

import pytest
import regex as re

import cfnlint
from cfnlint.helpers import ensure_list, load_plugins, load_resource
from cfnlint.jsonschema import StandardValidator, ValidationError
from cfnlint.schema.resolver import RefResolutionError, RefResolver


@pytest.mark.data
class TestSchemaFiles(TestCase):
    """Test schema files"""

    _found_keywords: List[str] = [
        "*",
        "Conditions",
        "Description",
        "Mappings",
        "Metadata",
        "Metadata/AWS::CloudFormation::Interface",
        "Metadata/cfn-lint",
        "Outputs",
        "Outputs/*",
        "Outputs/*/Condition",
        "Outputs/*/Export/Name",
        "Outputs/*/Value",
        "Parameters",
        "Parameters/*",
        "Parameters/*/AllowedPattern",
        "Parameters/*/Type",
        "Resources",
        "Resources/*",
        "Resources/*/Condition",
        "Resources/*/CreationPolicy",
        "Resources/*/DeletionPolicy",
        "Resources/*/DependsOn",
        "Resources/*/DependsOn/*",
        "Resources/*/Metadata",
        "Resources/*/Metadata/AWS::CloudFormation::Init",
        "Resources/*/Type",
        "Resources/*/UpdatePolicy",
        "Resources/*/UpdateReplacePolicy",
        "Rules",
        "Rules/*/Assertions/*/Assert",
        "Rules/*/RuleCondition",
        "Transform",
    ]

    def setUp(self) -> None:
        schema_path = os.path.join(os.path.dirname(cfnlint.__file__), "data", "schemas")
        self.paths = {
            "extensions": os.path.join(schema_path, "extensions"),
            "providers": os.path.join(schema_path, "providers"),
            "resources": os.path.join(schema_path, "resources"),
            "other": os.path.join(schema_path, "other"),
            "fixtures": os.path.join(
                os.path.dirname(__file__),
                "..",
                "fixtures",
                "schemas",
                "providers",
            ),
        }
        filename = os.path.join(
            self.paths["fixtures"], "..", "json_schema", "draft7.json"
        )
        with open(filename, "r") as fh:
            d = json.load(fh)
            self.schema_draft7 = d

        super().setUp()

    def get_files(self, dir):
        for dirpath, _, filenames in os.walk(dir):
            for filename in fnmatch.filter(filenames, "*.json"):
                yield dirpath, filename

    def pattern(self, validator, patrn, instance, schema):
        try:
            re.compile(patrn)
        except Exception:
            yield ValidationError(f"Pattern doesn't compile: {patrn}")

    def validate_basic_schema_details(
        self, schema_resolver: RefResolver, filepath: str
    ):
        """
        Validate that readOnly, writeOnly, etc are valid
        """
        sections = [
            "readOnlyProperties",
            "writeOnlyProperties",
            "conditionalCreateOnlyProperties",
            "nonPublicProperties",
            "nonPublicDefinitions",
            "createOnlyProperties",
            "deprecatedProperties",
            "primaryIdentifier",
        ]
        for section in sections:
            for prop in schema_resolver.referrer.get(section, []):
                try:
                    self.assertIsNotNone(schema_resolver.resolve_cfn_pointer(prop))
                except RefResolutionError:
                    self.fail(f"Can't find prop {prop} for {section} in {filepath}")

    def _build_keywords(self, obj: Any, schema_resolver: RefResolver, refs: list[str]):
        if not isinstance(obj, dict):
            yield []
            return

        if "$ref" in obj:
            ref = obj["$ref"]
            if ref in refs:
                yield []
                return
            _, resolved_schema = schema_resolver.resolve(ref)
            yield from self._build_keywords(
                resolved_schema, schema_resolver, refs + [ref]
            )

        if "type" in obj:
            if "object" in ensure_list(obj["type"]):
                if "properties" in obj:
                    for k, v in obj["properties"].items():
                        for item in self._build_keywords(v, schema_resolver, refs):
                            yield [k] + item
            if "array" in ensure_list(obj["type"]):
                if "items" in obj:
                    for item in self._build_keywords(
                        obj["items"], schema_resolver, refs
                    ):
                        yield ["*"] + item

        yield []

    def build_keywords(self, schema_resolver):
        self._found_keywords.append(
            "/".join(["Resources", schema_resolver.referrer["typeName"], "Properties"])
        )
        for k, v in schema_resolver.referrer.get("properties").items():
            for item in self._build_keywords(v, schema_resolver, []):
                self._found_keywords.append(
                    "/".join(
                        [
                            "Resources",
                            schema_resolver.referrer["typeName"],
                            "Properties",
                            k,
                        ]
                        + item
                    )
                )

    def test_data_module_specs(self):
        """Test data file formats"""

        draft7_schema = load_resource(
            "cfnlint.data.schemas.other.draft7", "schema.json"
        )
        store = {"http://json-schema.org/draft-07/schema": draft7_schema}
        dir = self.paths["fixtures"]
        for dirpath, filename in self.get_files(dir):
            with open(os.path.join(dirpath, filename), "r", encoding="utf8") as fh:
                store[filename] = json.load(fh)

        resolver = RefResolver.from_schema(
            store["provider.definition.schema.v1.json"], store=store
        )

        validator = (
            StandardValidator({})
            .extend(
                validators={
                    "cfnLint": self.cfn_lint,
                    "pattern": self.pattern,
                },
            )(schema=store["provider.definition.schema.v1.json"])
            .evolve(resolver=resolver)
        )

        # Only process us-east-1 region for keyword building (like original test)
        region = "us-east-1"
        provider_module = region.replace("-", "_")
        try:
            provider = __import__(
                f"cfnlint.data.schemas.providers.{provider_module}", fromlist=["types"]
            )
            resource_types = getattr(provider, "types", {})
        except (ImportError, AttributeError):
            self.fail(f"Could not load provider module for {region}")

        # Load each resource schema
        for resource_type, schema_hash in resource_types.items():
            schema_path = os.path.join(self.paths["resources"], f"{schema_hash}.json")

            if not os.path.exists(schema_path):
                continue

            with open(schema_path, "r", encoding="utf8") as fh:
                d = json.load(fh)
                # not allowed but true with this resource
                if resource_type == "AWS::CloudFormation::CustomResource":
                    d["additionalProperties"] = False
                # Skip synthetic types that don't follow standard schema
                if resource_type in ("Module", "AWS::CDK::Metadata"):
                    continue
                errs = list(validator.iter_errors(d))
                self.assertListEqual(
                    errs, [], f"Error with {resource_type} ({schema_hash}): {errs}"
                )
                schema_resolver = RefResolver(d)
                self.validate_basic_schema_details(
                    schema_resolver, f"{resource_type} ({schema_hash})"
                )

                # Build keywords for us-east-1 schemas
                self.build_keywords(schema_resolver)

    def cfn_lint(self, validator, _, keywords, schema):
        keywords = ensure_list(keywords)
        self._found_keywords.extend(keywords)

    def test_other_specs(self):
        """Test data file formats"""

        store = {}
        dir = self.paths["fixtures"]
        for dirpath, filename in self.get_files(dir):
            with open(os.path.join(dirpath, filename), "r", encoding="utf8") as fh:
                store[filename] = json.load(fh)

        resolver = RefResolver.from_schema(store["cfnlint.schema.v1.json"], store=store)

        validator = (
            StandardValidator({})
            .extend(
                validators={
                    "cfnLint": self.cfn_lint,
                    "pattern": self.pattern,
                },
            )(schema=store["cfnlint.schema.v1.json"])
            .evolve(resolver=resolver)
        )

        for dir_name in ["extensions", "other"]:
            dir = self.paths[dir_name]

            for dirpath, filename in self.get_files(dir):
                with open(os.path.join(dirpath, filename), "r", encoding="utf8") as fh:
                    d = json.load(fh)
                    errs = list(validator.iter_errors(d))
                    self.assertListEqual(
                        errs, [], f"Error with {dirpath}/{filename}: {errs}"
                    )
                    schema_resolver = RefResolver(d)
                    self.validate_basic_schema_details(
                        schema_resolver, f"{dirpath}/{filename}"
                    )

    def test_all_referenced_hashes_exist(self):
        """Test that all hashes referenced by provider files
        have corresponding schema files
        """
        import importlib
        from pathlib import Path

        # Collect all hashes referenced by provider files
        referenced_hashes = set()
        providers_dir = Path(self.paths["providers"])

        for provider_file in providers_dir.glob("*.py"):
            if provider_file.name == "__init__.py":
                continue

            region_name = provider_file.stem
            try:
                module_name = f"cfnlint.data.schemas.providers.{region_name}"
                provider = importlib.import_module(module_name)
                if hasattr(provider, "types"):
                    for resource_type, schema_hash in provider.types.items():
                        referenced_hashes.add((schema_hash, resource_type, region_name))
            except (ImportError, AttributeError):
                continue

        # Check all referenced hashes have corresponding files
        resources_dir = Path(self.paths["resources"])
        missing_files = []

        for schema_hash, resource_type, region_name in referenced_hashes:
            schema_file = resources_dir / f"{schema_hash}.json"
            if not schema_file.exists():
                missing_files.append(
                    f"{resource_type} ({schema_hash}) in {region_name}"
                )

        self.assertEqual(
            missing_files,
            [],
            f"Found {len(missing_files)} missing schema files: {missing_files[:10]}...",
        )

    def test_all_resource_files_referenced(self):
        """Test that all resource schema files are
        referenced by at least one provider
        """
        import importlib
        from pathlib import Path

        # Collect all hashes referenced by provider files
        referenced_hashes = set()
        providers_dir = Path(self.paths["providers"])

        for provider_file in providers_dir.glob("*.py"):
            if provider_file.name == "__init__.py":
                continue

            region_name = provider_file.stem
            try:
                module_name = f"cfnlint.data.schemas.providers.{region_name}"
                provider = importlib.import_module(module_name)
                if hasattr(provider, "types"):
                    referenced_hashes.update(provider.types.values())
            except (ImportError, AttributeError):
                continue

        # Check all resource files are referenced
        resources_dir = Path(self.paths["resources"])
        orphaned_files = []

        for schema_file in resources_dir.glob("*.json"):
            schema_hash = schema_file.stem
            if schema_hash not in referenced_hashes:
                orphaned_files.append(schema_hash)

        self.assertEqual(
            orphaned_files,
            [],
            (
                f"Found {len(orphaned_files)} orphaned "
                f"resource files: {orphaned_files[:10]}..."
            ),
        )

    def test_x_keywords(self):
        root_dir = pathlib.Path(__file__).parent.parent.parent / "src/cfnlint/rules"
        rules = load_plugins(
            str(root_dir),
            "CfnLintKeyword",
            "cfnlint.rules.jsonschema.CfnLintKeyword",
        )
        rules.extend(
            load_plugins(
                str(root_dir),
                "CfnLintJsonSchema",
                "cfnlint.rules.jsonschema.CfnLintJsonSchema",
            )
        )

        for rule in rules:
            for keyword in rule.keywords:
                self.assertIn(keyword, self._found_keywords, f"{keyword} not found")
