import pytest

from haiku.rag.client import HaikuRAG
from haiku.rag.config import AppConfig


async def test_database_not_created_without_create_flag(tmp_path):
    """Test that database is not created without create=True."""
    db_path = tmp_path / "test.lancedb"

    config = AppConfig()

    with pytest.raises(FileNotFoundError, match="Database does not exist"):
        async with HaikuRAG(db_path=db_path, config=config):
            pass


async def test_database_created_with_create_flag(tmp_path):
    """Test that database is created with create=True."""
    db_path = tmp_path / "test.lancedb"

    config = AppConfig()

    async with HaikuRAG(db_path=db_path, config=config, create=True):
        assert db_path.exists()


@pytest.mark.vcr()
async def test_operations_work_after_database_created(tmp_path):
    """Test that operations work after DB is created."""
    db_path = tmp_path / "test.lancedb"

    config = AppConfig()

    # First, create DB with create=True and add document
    async with HaikuRAG(db_path=db_path, config=config, create=True) as client:
        await client.create_document("Test content", uri="test://doc1")

    # Re-open without create flag and verify we can read the document
    async with HaikuRAG(db_path=db_path, config=config) as client:
        docs = await client.list_documents()
        assert len(docs) == 1
        doc = await client.get_document_by_id(docs[0].id)
        assert doc is not None
        assert doc.content == "Test content"


@pytest.mark.asyncio
async def test_default_db_path_comes_from_storage_data_dir(tmp_path):
    """Omitting db_path places the database under the configured data dir."""
    from haiku.rag.client import HaikuRAG
    from haiku.rag.config import AppConfig

    config = AppConfig()
    config.storage.data_dir = tmp_path

    client = HaikuRAG(config=config)

    assert client._db_path == tmp_path / "haiku.rag.lancedb"


@pytest.mark.asyncio
async def test_vacuum_is_callable_on_the_client(temp_db_path):
    """The public vacuum() delegates to the store."""
    from haiku.rag.client import HaikuRAG

    async with HaikuRAG(temp_db_path, create=True) as client:
        await client.vacuum()