"""Helpers for resolving and persisting note references across commands."""

from __future__ import annotations

import click

from .cookies import get_note_by_index, save_note_index
from .formatter import parse_note_reference


def resolve_note_reference(id_or_url: str, *, xsec_token: str = "") -> tuple[str, str, str]:
    """Resolve a note reference from URL/ID or the last listing index."""
    if id_or_url.isdigit():
        entry = get_note_by_index(int(id_or_url))
        if entry is None:
            raise click.UsageError(
                f"Index {id_or_url} not found — run a listing command first "
                "(search / feed / hot / user-posts / favorites / my-notes)"
            )
        return (
            entry["note_id"],
            xsec_token or entry.get("xsec_token", ""),
            entry.get("xsec_source", ""),
        )

    note_id, url_token, url_source = parse_note_reference(id_or_url)
    return note_id, xsec_token or url_token, url_source


def save_index_from_items(data: dict, *, xsec_source: str) -> None:
    """Persist ordered note references from list-style responses."""
    entries = []
    for item in data.get("items", []):
        note_card = item.get("note_card", {})
        note_id = item.get("id", note_card.get("note_id", ""))
        token = item.get("xsec_token", note_card.get("xsec_token", ""))
        if note_id:
            entries.append({
                "note_id": note_id,
                "xsec_token": token,
                "xsec_source": xsec_source if token else "",
            })
    save_note_index(entries)


def save_index_from_notes(notes: list[dict]) -> None:
    """Persist ordered note references from paged note payloads."""
    save_note_index([
        {
            "note_id": str(note.get("note_id", note.get("id", ""))).strip(),
            "xsec_token": str(note.get("xsec_token", "")).strip(),
            "xsec_source": "",
        }
        for note in notes
        if str(note.get("note_id", note.get("id", ""))).strip()
    ])