import pytest
import importlib
import yaml
from app.config import EmbeddingProvider


@pytest.fixture
def tmp_config_file(tmp_path):
    """
    Creates a temporary config.yaml file and returns its path.
    Corrected the 'provider' value to be lowercase 'mock' to match the Enum.
    Added database settings for testing.
    """
    config_content = {
        "application": {
            "project_name": "Test Project from YAML",
            "log_level": "WARNING"
        },
        "llm_providers": {"deepseek_model_name": "deepseek-from-yaml"},
        "embedding_provider": {
            # This value must be lowercase to match the Pydantic Enum member
            "provider": "mock",
            "model_name": "embedding-model-from-yaml"
        },
        "database": {
            "mode": "sqlite",
            "local_path": "custom_folder/test_ai_hub.db",
            "url": "postgresql://user:pass@host/dbname"  # Should be ignored for sqlite mode
        }
    }
    config_path = tmp_path / "test_config.yaml"
    with open(config_path, 'w') as f:
        yaml.dump(config_content, f)
    return str(config_path)


@pytest.fixture(autouse=True)
def mock_api_keys(monkeypatch):
    """
    Automatically sets mock API keys for all tests to prevent the
    ValueError from being raised in config.py.
    """
    monkeypatch.setenv("DEEPSEEK_API_KEY", "mock_deepseek_key")
    monkeypatch.setenv("GEMINI_API_KEY", "mock_gemini_key")
    monkeypatch.setenv("EMBEDDING_API_KEY", "mock_embedding_key")


def test_sqlite_db_url_from_yaml(monkeypatch, tmp_config_file):
    """Tests DATABASE_URL is constructed correctly from YAML sqlite local_path."""
    monkeypatch.setenv("CONFIG_PATH", tmp_config_file)
    monkeypatch.delenv("DB_MODE", raising=False)
    monkeypatch.delenv("LOCAL_DB_PATH", raising=False)
    monkeypatch.delenv("DATABASE_URL", raising=False)

    from app import config
    importlib.reload(config)

    expected_path = "sqlite:///./custom_folder/test_ai_hub.db"
    assert config.settings.DB_MODE == "sqlite"
    assert config.settings.DATABASE_URL == expected_path


def test_sqlite_db_url_from_env_local_path(monkeypatch, tmp_path):
    """Tests that LOCAL_DB_PATH env var overrides YAML for sqlite DATABASE_URL."""
    monkeypatch.setenv("DB_MODE", "sqlite")
    monkeypatch.setenv("LOCAL_DB_PATH", "env_folder/env_ai_hub.db")
    monkeypatch.delenv("CONFIG_PATH", raising=False)
    monkeypatch.delenv("DATABASE_URL", raising=False)

    from app import config
    importlib.reload(config)

    expected_path = "sqlite:///./env_folder/env_ai_hub.db"
    assert config.settings.DB_MODE == "sqlite"
    assert config.settings.DATABASE_URL == expected_path


def test_external_db_url_used_when_not_sqlite(monkeypatch, tmp_config_file):
    """Tests DATABASE_URL uses external URL when DB_MODE is not sqlite."""
    monkeypatch.setenv("CONFIG_PATH", tmp_config_file)
    monkeypatch.setenv("DB_MODE", "postgresql")
    monkeypatch.setenv("DATABASE_URL", "postgresql://env_user:env_pass@env_host/env_db")
    monkeypatch.delenv("LOCAL_DB_PATH", raising=False)

    from app import config
    importlib.reload(config)

    assert config.settings.DB_MODE == "postgresql"
    assert config.settings.DATABASE_URL == "postgresql://env_user:env_pass@env_host/env_db"


def test_external_db_url_from_yaml_when_not_sqlite(monkeypatch, tmp_path):
    """Tests DATABASE_URL uses YAML url when DB_MODE != sqlite and no env DATABASE_URL."""
    # Write YAML with postgresql mode and url
    config_content = {
        "database": {
            "mode": "postgresql",
            "url": "postgresql://yaml_user:yaml_pass@yaml_host/yaml_db",
            "local_path": "ignored_path_for_postgresql.db"
        }
    }
    config_path = tmp_path / "test_config_pg.yaml"
    with open(config_path, 'w') as f:
        yaml.dump(config_content, f)
    monkeypatch.setenv("CONFIG_PATH", str(config_path))
    monkeypatch.delenv("DATABASE_URL", raising=False)
    monkeypatch.delenv("DB_MODE", raising=False)
    monkeypatch.delenv("LOCAL_DB_PATH", raising=False)

    from app import config
    importlib.reload(config)

    assert config.settings.DB_MODE == "postgresql"
    assert config.settings.DATABASE_URL == "postgresql://yaml_user:yaml_pass@yaml_host/yaml_db"


def test_sqlite_db_url_defaults(monkeypatch):
    """Tests DATABASE_URL defaults to sqlite path if no env or YAML."""
    monkeypatch.setenv("DB_MODE", "sqlite")
    monkeypatch.delenv("LOCAL_DB_PATH", raising=False)
    monkeypatch.delenv("DATABASE_URL", raising=False)
    monkeypatch.delenv("CONFIG_PATH", raising=False)

    from app import config
    importlib.reload(config)

    assert config.settings.DB_MODE == "sqlite"
    assert config.settings.DATABASE_URL == "sqlite:///./data/ai_hub.db"