#
# Copyright (c) 2024 Airbyte, Inc., all rights reserved.
#

import os
import sys
from pathlib import Path
from typing import Any, Mapping

from pytest import fixture

from airbyte_cdk.sources.declarative.yaml_declarative_source import YamlDeclarativeSource
from airbyte_cdk.test.catalog_builder import CatalogBuilder
from airbyte_cdk.test.state_builder import StateBuilder


pytest_plugins = ["airbyte_cdk.test.utils.manifest_only_fixtures"]


def _get_manifest_path() -> Path:
    """Get path to manifest.yaml, handling both CI and local environments."""
    # CI path (inside Docker container)
    ci_path = Path("/airbyte/integration_code/source_declarative_manifest")
    if ci_path.exists():
        return ci_path
    # Local development path
    return Path(__file__).parent.parent


_SOURCE_FOLDER_PATH = _get_manifest_path()
_MANIFEST_PATH = _SOURCE_FOLDER_PATH / "manifest.yaml"

sys.path.append(str(_SOURCE_FOLDER_PATH))


def get_source(config: Mapping[str, Any], state=None) -> YamlDeclarativeSource:
    """Create a YamlDeclarativeSource instance with the given config."""
    catalog = CatalogBuilder().build()
    state = StateBuilder().build() if not state else state
    return YamlDeclarativeSource(
        path_to_yaml=str(_MANIFEST_PATH),
        catalog=catalog,
        config=config,
        state=state,
    )


@fixture(autouse=True)
def clear_cache_before_each_test():
    """Clear HTTP request cache between tests to ensure isolation."""
    cache_path = os.getenv("REQUEST_CACHE_PATH")
    if cache_path:
        cache_dir = Path(cache_path)
        if cache_dir.exists():
            for file_path in cache_dir.glob("*.sqlite"):
                file_path.unlink()
    yield