Logic/Modules/Backend/external_api_adapter/mock_loader.py

import json
from dataclasses import dataclass
from pathlib import Path

from .exceptions import MockDirectoryNotFound, MockFileNotFound


@dataclass
class LoadedMockResponse:
    data: object
    status_code: int
    file_path: str


class MockLoader:
    def __init__(self, base_path=None):
        self.base_path = Path(base_path or Path(__file__).resolve().parent / "json")

    def load(self, service_name, path, method):
        mock_files = self._find_mock_files(service_name=service_name, path=path, method=method)
        if not mock_files:
            raise MockFileNotFound(
                f"No mock file found for service='{service_name}' path='{path}' method='{method}'."
            )

        selected_file = sorted(mock_files, key=self._mock_file_priority)[0]
        with selected_file.open("r", encoding="utf-8") as file:
            service_root = self.base_path / service_name
            return LoadedMockResponse(
                data=self._resolve_references(
                    json.load(file),
                    current_file=selected_file,
                    service_root=service_root,
                ),
                status_code=self._extract_status_code(selected_file),
                file_path=str(selected_file),
            )

    def _find_mock_files(self, service_name, path, method):
        service_root = self.base_path / service_name
        directory_path = service_root / self._build_directory_path(path)
        pattern = f"{method.lower()}_*.json"

        if directory_path.exists() and directory_path.is_dir():
            return list(directory_path.glob(pattern))

        leaf_name = self._extract_leaf_name(path)
        parent_directory = directory_path.parent
        if parent_directory.exists() and parent_directory.is_dir():
            flat_pattern = f"{leaf_name}-{method.lower()}_*.json"
            flat_files = list(parent_directory.glob(flat_pattern))
            if flat_files:
                return flat_files

        normalized_parts = [part for part in str(path).strip().strip("/").split("/") if part]
        for index in range(len(normalized_parts) - 1, -1, -1):
            candidate_parts = normalized_parts[:index] + normalized_parts[index + 1 :]
            if not candidate_parts:
                continue

            candidate_directory = service_root / Path(*candidate_parts)
            if candidate_directory.exists() and candidate_directory.is_dir():
                candidate_files = list(candidate_directory.glob(pattern))
                if candidate_files:
                    return candidate_files

        raise MockDirectoryNotFound(
            f"Mock directory not found for service='{service_name}' path='{path}': {directory_path}"
        )

    @staticmethod
    def _build_directory_path(path):
        normalized = str(path).strip().strip("/")
        if not normalized:
            return Path(".")
        return Path(*normalized.split("/"))

    @staticmethod
    def _extract_status_code(file_path):
        parts = file_path.stem.split("_")
        if len(parts) < 2:
            return 200
        try:
            return int(parts[1])
        except ValueError:
            return 200

    @staticmethod
    def _extract_leaf_name(path):
        normalized = str(path).strip().strip("/")
        if not normalized:
            return ""
        return normalized.split("/")[-1]

    @staticmethod
    def _mock_file_priority(file_path):
        stem = file_path.stem.lower()
        status_code = MockLoader._extract_status_code(file_path)
        keyword_rank = 2
        if "success" in stem:
            keyword_rank = 0
        elif "stream" in stem:
            keyword_rank = 0
        elif "pending" in stem or "progress" in stem:
            keyword_rank = 1
        return (status_code >= 400, keyword_rank, stem)

    def _resolve_references(self, value, current_file, service_root, seen=None):
        current_file = current_file.resolve()
        service_root = service_root.resolve()
        seen = seen or {current_file}

        if isinstance(value, list):
            return [
                self._resolve_references(item, current_file=current_file, service_root=service_root, seen=seen)
                for item in value
            ]

        if not isinstance(value, dict):
            return value

        ref_path = value.get("$ref")
        if isinstance(ref_path, str) and len(value) == 1:
            referenced_file = self._resolve_reference_path(
                ref_path=ref_path,
                current_file=current_file,
                service_root=service_root,
            )

            if referenced_file in seen:
                raise MockFileNotFound(f"Circular mock reference detected for '{referenced_file}'.")

            with referenced_file.open("r", encoding="utf-8") as file:
                return self._resolve_references(
                    json.load(file),
                    current_file=referenced_file,
                    service_root=service_root,
                    seen=seen | {referenced_file},
                )

        return {
            key: self._resolve_references(item, current_file=current_file, service_root=service_root, seen=seen)
            for key, item in value.items()
        }

    @staticmethod
    def _resolve_reference_path(ref_path, current_file, service_root):
        candidates = [
            current_file.parent / ref_path,
            service_root / ref_path,
        ]

        service_root_resolved = service_root.resolve()
        for candidate in candidates:
            candidate_resolved = candidate.resolve()
            try:
                candidate_resolved.relative_to(service_root_resolved)
            except ValueError:
                continue
            if candidate_resolved.is_file():
                return candidate_resolved

        raise MockFileNotFound(f"Referenced mock file '{ref_path}' was not found.")