import pytest


async def setup_project_and_get_token(client):
    await client.post("/api/v1/auth/register", json={"username": "docuser", "password": "pass123"})
    login = await client.post("/api/v1/auth/login", json={"username": "docuser", "password": "pass123"})
    token = login.json()["access_token"]
    proj_resp = await client.post(
        "/api/v1/projects",
        json={"name": "Doc Test Project"},
        headers={"Authorization": f"Bearer {token}"}
    )
    return token, proj_resp.json()["id"]


@pytest.mark.asyncio
async def test_create_document(client):
    token, proj_id = await setup_project_and_get_token(client)
    response = await client.post(
        f"/api/v1/projects/{proj_id}/documents",
        json={"title": "My Document", "content": "# Hello\n\nWorld"},
        headers={"Authorization": f"Bearer {token}"}
    )
    assert response.status_code == 201
    data = response.json()
    assert data["title"] == "My Document"
    assert data["content"] == "# Hello\n\nWorld"
    assert data["project_id"] == proj_id


@pytest.mark.asyncio
async def test_get_document(client):
    token, proj_id = await setup_project_and_get_token(client)
    create_resp = await client.post(
        f"/api/v1/projects/{proj_id}/documents",
        json={"title": "Get Doc Test"},
        headers={"Authorization": f"Bearer {token}"}
    )
    doc_id = create_resp.json()["id"]
    response = await client.get(
        f"/api/v1/documents/{doc_id}",
        headers={"Authorization": f"Bearer {token}"}
    )
    assert response.status_code == 200
    assert response.json()["title"] == "Get Doc Test"


@pytest.mark.asyncio
async def test_update_document_content(client):
    token, proj_id = await setup_project_and_get_token(client)
    create_resp = await client.post(
        f"/api/v1/projects/{proj_id}/documents",
        json={"title": "Original Title", "content": "Original Content"},
        headers={"Authorization": f"Bearer {token}"}
    )
    doc_id = create_resp.json()["id"]
    response = await client.put(
        f"/api/v1/documents/{doc_id}/content",
        json={"content": "Updated Content"},
        headers={"Authorization": f"Bearer {token}"}
    )
    assert response.status_code == 200
    assert response.json()["content"] == "Updated Content"


@pytest.mark.asyncio
async def test_soft_delete_document(client):
    token, proj_id = await setup_project_and_get_token(client)
    create_resp = await client.post(
        f"/api/v1/projects/{proj_id}/documents",
        json={"title": "To Delete"},
        headers={"Authorization": f"Bearer {token}"}
    )
    doc_id = create_resp.json()["id"]
    del_resp = await client.delete(
        f"/api/v1/documents/{doc_id}",
        headers={"Authorization": f"Bearer {token}"}
    )
    assert del_resp.status_code == 204


@pytest.mark.asyncio
async def test_assign_tag(client):
    token, proj_id = await setup_project_and_get_token(client)
    doc_resp = await client.post(
        f"/api/v1/projects/{proj_id}/documents",
        json={"title": "Tagged Doc"},
        headers={"Authorization": f"Bearer {token}"}
    )
    doc_id = doc_resp.json()["id"]
    tag_resp = await client.post(
        "/api/v1/tags",
        json={"name": "important", "color": "#ff0000"},
        headers={"Authorization": f"Bearer {token}"}
    )
    tag_id = tag_resp.json()["id"]
    assign_resp = await client.post(
        f"/api/v1/documents/{doc_id}/tags",
        json={"tag_ids": [tag_id]},
        headers={"Authorization": f"Bearer {token}"}
    )
    assert assign_resp.status_code == 204

    get_resp = await client.get(
        f"/api/v1/documents/{doc_id}",
        headers={"Authorization": f"Bearer {token}"}
    )
    assert len(get_resp.json()["tags"]) == 1
    assert get_resp.json()["tags"][0]["name"] == "important"


@pytest.mark.asyncio
async def test_remove_tag(client):
    token, proj_id = await setup_project_and_get_token(client)
    doc_resp = await client.post(
        f"/api/v1/projects/{proj_id}/documents",
        json={"title": "Tagged Doc 2"},
        headers={"Authorization": f"Bearer {token}"}
    )
    doc_id = doc_resp.json()["id"]
    tag_resp = await client.post(
        "/api/v1/tags",
        json={"name": "temp", "color": "#00ff00"},
        headers={"Authorization": f"Bearer {token}"}
    )
    tag_id = tag_resp.json()["id"]
    await client.post(
        f"/api/v1/documents/{doc_id}/tags",
        json={"tag_ids": [tag_id]},
        headers={"Authorization": f"Bearer {token}"}
    )
    remove_resp = await client.delete(
        f"/api/v1/documents/{doc_id}/tags/{tag_id}",
        headers={"Authorization": f"Bearer {token}"}
    )
    assert remove_resp.status_code == 204