"""Tests for blob deep-link infrastructure: symbol→lineno enrichment and #S: fragments.

Covers:
  _enrich_with_linenos (unit)
  - test_enrich_adds_lineno_for_known_function
  - test_enrich_adds_lineno_for_class
  - test_enrich_skips_non_python_files
  - test_enrich_does_not_overwrite_existing_lineno
  - test_enrich_tolerates_syntax_error
  - test_enrich_ignores_symbol_not_in_ast

  _symbol_line_map (unit)
  - test_symbol_line_map_returns_display_name_to_lineno
  - test_symbol_line_map_excludes_symbols_without_lineno
  - test_symbol_line_map_excludes_falsy_lineno

  Issue detail SSR — #S: deep links
  - test_issue_detail_symbol_anchor_link_contains_hash_fragment
  - test_issue_detail_symbol_anchor_plain_file_no_fragment
  - test_issue_detail_multiple_symbol_anchors_all_linked

  Blob page SSR — symbolLines in page_json
  - test_blob_page_json_contains_symbol_lines_key
"""
from __future__ import annotations

import pytest
from httpx import AsyncClient
from sqlalchemy.ext.asyncio import AsyncSession

from datetime import datetime, timezone

from musehub.api.routes.musehub.ui_blob import _enrich_with_linenos, _symbol_line_map
from muse.core.types import now_utc_iso
from musehub.core.genesis import compute_identity_id, compute_issue_id, compute_repo_id
from musehub.db.musehub_repo_models import MusehubRepo
from musehub.db.musehub_social_models import MusehubIssue
from musehub.types.json_types import JSONObject, StrDict


# ---------------------------------------------------------------------------
# Helpers
# ---------------------------------------------------------------------------

_PY_SRC = """\
class MyService:
    def run(self) -> None:
        pass

async def compute(x: int) -> int:
    return x * 2
"""


async def _make_repo(db: AsyncSession, owner: str = "blober", slug: str = "blobby") -> str:
    owner_id = compute_identity_id(owner.encode())
    created_at = datetime.now(tz=timezone.utc)
    repo = MusehubRepo(
        repo_id=compute_repo_id(owner_id, slug, "code", created_at.isoformat()),
        name=slug,
        owner=owner,
        slug=slug,
        visibility="public",
        owner_user_id=owner_id,
        created_at=created_at,
        updated_at=created_at,
    )
    db.add(repo)
    await db.commit()
    await db.refresh(repo)
    return str(repo.repo_id)


async def _make_issue(
    db: AsyncSession,
    repo_id: str,
    *,
    number: int = 1,
    title: str = "Deep link test issue",
    symbol_anchors: list[str] | None = None,
) -> MusehubIssue:
    author_id = compute_identity_id(b"blober")
    issue = MusehubIssue(
        issue_id=compute_issue_id(repo_id, author_id, now_utc_iso()),
        repo_id=repo_id,
        number=number,
        title=title,
        body="",
        state="open",
        labels=[],
        author="blober",
        symbol_anchors=symbol_anchors or [],
    )
    db.add(issue)
    await db.commit()
    await db.refresh(issue)
    return issue


# ---------------------------------------------------------------------------
# Unit: _enrich_with_linenos
# ---------------------------------------------------------------------------


def test_enrich_adds_lineno_for_known_function() -> None:
    syms: list[JSONObject] = [{"display_name": "compute"}]
    _enrich_with_linenos(syms, "service.py", _PY_SRC)
    assert syms[0]["lineno"] == 5


def test_enrich_adds_lineno_for_class() -> None:
    syms: list[JSONObject] = [{"display_name": "MyService"}]
    _enrich_with_linenos(syms, "service.py", _PY_SRC)
    assert syms[0]["lineno"] == 1


def test_enrich_skips_non_python_files() -> None:
    syms: list[JSONObject] = [{"display_name": "compute"}]
    _enrich_with_linenos(syms, "service.ts", _PY_SRC)
    assert "lineno" not in syms[0]


def test_enrich_does_not_overwrite_existing_lineno() -> None:
    syms: list[JSONObject] = [{"display_name": "compute", "lineno": 99}]
    _enrich_with_linenos(syms, "service.py", _PY_SRC)
    assert syms[0]["lineno"] == 99


def test_enrich_tolerates_syntax_error() -> None:
    syms: list[JSONObject] = [{"display_name": "compute"}]
    _enrich_with_linenos(syms, "bad.py", "def compute(: pass")
    assert "lineno" not in syms[0]


def test_enrich_ignores_symbol_not_in_ast() -> None:
    syms: list[JSONObject] = [{"display_name": "ghost_fn"}]
    _enrich_with_linenos(syms, "service.py", _PY_SRC)
    assert "lineno" not in syms[0]


# ---------------------------------------------------------------------------
# Unit: _symbol_line_map
# ---------------------------------------------------------------------------


def test_symbol_line_map_returns_display_name_to_lineno() -> None:
    syms: list[JSONObject] = [
        {"display_name": "foo", "lineno": 3, "end_lineno": 8},
        {"display_name": "bar", "lineno": 10, "end_lineno": 20},
    ]
    result = _symbol_line_map(syms)
    assert result == {"foo": [3, 8], "bar": [10, 20]}


def test_symbol_line_map_excludes_symbols_without_lineno() -> None:
    syms: list[JSONObject] = [
        {"display_name": "foo", "lineno": 3, "end_lineno": 5},
        {"display_name": "no_line"},
    ]
    result = _symbol_line_map(syms)
    assert "no_line" not in result
    assert result["foo"] == [3, 5]


def test_symbol_line_map_excludes_falsy_lineno() -> None:
    syms: list[JSONObject] = [{"display_name": "zero", "lineno": 0}]
    result = _symbol_line_map(syms)
    assert "zero" not in result


# ---------------------------------------------------------------------------
# Issue detail SSR — #S: deep links
# ---------------------------------------------------------------------------


async def test_issue_detail_symbol_anchor_link_contains_hash_fragment(
    client: AsyncClient, db_session: AsyncSession, auth_headers: StrDict
) -> None:
    repo_id = await _make_repo(db_session)
    await _make_issue(db_session, repo_id, symbol_anchors=["musehub/services/foo.py::compute"])
    r = await client.get("/blober/blobby/issues/1")
    assert r.status_code == 200
    assert "#S:compute" in r.text


async def test_issue_detail_symbol_anchor_plain_file_no_fragment(
    client: AsyncClient, db_session: AsyncSession, auth_headers: StrDict
) -> None:
    repo_id = await _make_repo(db_session, slug="blobby2")
    await _make_issue(db_session, repo_id, number=1, symbol_anchors=["musehub/services/bar.py"])
    r = await client.get("/blober/blobby2/issues/1")
    assert r.status_code == 200
    # plain file anchor — no #S: fragment, just the blob URL
    assert "blob/main/musehub/services/bar.py" in r.text
    assert "#S:" not in r.text


async def test_issue_detail_multiple_symbol_anchors_all_linked(
    client: AsyncClient, db_session: AsyncSession, auth_headers: StrDict
) -> None:
    repo_id = await _make_repo(db_session, slug="blobby3")
    await _make_issue(
        db_session,
        repo_id,
        symbol_anchors=[
            "musehub/services/a.py::Alpha",
            "musehub/services/b.py::Beta",
        ],
    )
    r = await client.get("/blober/blobby3/issues/1")
    assert r.status_code == 200
    assert "#S:Alpha" in r.text
    assert "#S:Beta" in r.text


# ---------------------------------------------------------------------------
# Cross-repo symbol anchors — SSR rendering (issue #38 motivating case)
# ---------------------------------------------------------------------------


async def test_cross_repo_anchor_url_points_to_other_repo(
    client: AsyncClient, db_session: AsyncSession, auth_headers: StrDict
) -> None:
    """A cross-repo anchor 'gabriel/muse::path/file.py::Symbol' must generate
    a blob URL for gabriel/muse, not the issue's own repo (blober/blobby5)."""
    repo_id = await _make_repo(db_session, slug="blobby5")
    await _make_issue(
        db_session,
        repo_id,
        symbol_anchors=["gabriel/muse::muse/cli/commands/bridge.py::GitExporter.fix_file_modes"],
    )
    r = await client.get("/blober/blobby5/issues/1")
    assert r.status_code == 200
    # Must link to gabriel/muse, not blober/blobby5.
    assert "/gabriel/muse/blob/main/muse/cli/commands/bridge.py#S:GitExporter.fix_file_modes" in r.text
    assert "/blober/blobby5/blob/" not in r.text


async def test_cross_repo_anchor_with_ref_in_url(
    client: AsyncClient, db_session: AsyncSession, auth_headers: StrDict
) -> None:
    """@ref suffix must appear in the blob URL path."""
    repo_id = await _make_repo(db_session, slug="blobby6")
    await _make_issue(
        db_session,
        repo_id,
        symbol_anchors=["gabriel/muse::muse/cli/commands/bridge.py::GitExporter._has_shebang@dev"],
    )
    r = await client.get("/blober/blobby6/issues/1")
    assert r.status_code == 200
    assert "/gabriel/muse/blob/dev/muse/cli/commands/bridge.py#S:GitExporter._has_shebang" in r.text


async def test_same_repo_anchor_still_resolves_to_own_repo(
    client: AsyncClient, db_session: AsyncSession, auth_headers: StrDict
) -> None:
    """Existing same-repo anchor behaviour must be unchanged."""
    repo_id = await _make_repo(db_session, slug="blobby7")
    await _make_issue(
        db_session,
        repo_id,
        symbol_anchors=["musehub/services/billing.py::compute_total"],
    )
    r = await client.get("/blober/blobby7/issues/1")
    assert r.status_code == 200
    assert "/blober/blobby7/blob/main/musehub/services/billing.py#S:compute_total" in r.text


async def test_cross_repo_anchor_shows_repo_prefix_in_label(
    client: AsyncClient, db_session: AsyncSession, auth_headers: StrDict
) -> None:
    """Cross-repo anchors must display 'owner/repo::' before the file path."""
    repo_id = await _make_repo(db_session, slug="blobby8")
    await _make_issue(
        db_session,
        repo_id,
        symbol_anchors=["gabriel/muse::muse/cli/commands/bridge.py::GitExporter.fix_file_modes"],
    )
    r = await client.get("/blober/blobby8/issues/1")
    assert r.status_code == 200
    assert "gabriel/muse" in r.text
    assert "isd-anchor--cross-repo" in r.text


async def test_same_repo_anchor_with_ref_in_url(
    client: AsyncClient, db_session: AsyncSession, auth_headers: StrDict
) -> None:
    """Same-repo anchor with @ref uses the ref in the blob URL."""
    repo_id = await _make_repo(db_session, slug="blobby9")
    await _make_issue(
        db_session,
        repo_id,
        symbol_anchors=["src/main.py::build@feat/my-branch"],
    )
    r = await client.get("/blober/blobby9/issues/1")
    assert r.status_code == 200
    assert "/blober/blobby9/blob/feat/my-branch/src/main.py#S:build" in r.text


# ---------------------------------------------------------------------------
# Blob page SSR — symbolLines key present in page_json
# ---------------------------------------------------------------------------


async def test_blob_page_json_contains_symbol_lines_key(
    client: AsyncClient, db_session: AsyncSession
) -> None:
    """Even when no file is found the blob template must emit symbolLines in page_json."""
    await _make_repo(db_session, owner="blober", slug="blobby4")
    r = await client.get("/blober/blobby4/blob/main/some/file.py")
    # May 200 or 404 depending on whether the file exists, but the template always renders.
    # We only care that the response contains "symbolLines" in the page_json script block.
    assert r.status_code in (200, 404)
    assert '"symbolLines"' in r.text