"""llm_prompt.py — Формирование промпта для LLM-анализа ДС."""


def build_prompt(current_spec: list, doc_text: str) -> str:
    is_first = len(current_spec) == 0
    if is_first:
        return _build_initial(doc_text)
    else:
        return _build_diff(current_spec, doc_text)


def _build_initial(doc_text: str) -> str:
    return f"""Ты — анализатор договоров облачного провайдера.

Ниже текст спецификации услуг из ПЕРВОГО документа (базовый договор).
Извлеки ВСЕ строки спецификации в JSON-массив.

Верни СТРОГО JSON без пояснений. Не используй markdown-блоки.

ФОРМАТ:
{{
  "mode": "partial",
  "ops": [
    {{"action": "ADD", "new_row": {{"name": "полное наименование", "price": число, "qty": число, "sum": число, "date_start": "YYYY-MM-DD"}}, "comment": ""}}
  ]
}}

ПРАВИЛА:
1. Извлеки КАЖДУЮ строку таблицы спецификации как отдельную ADD-операцию.
2. Пропускай итоговые строки ("Итого...") и строки с подписями.
3. Если ячейка пустая — ставь null (не пиши 0).
4. price, qty, sum — ЧИСЛА, не строки.

ТЕКСТ ДОКУМЕНТА:
---
{doc_text}
---"""


def _build_diff(current_spec: list, doc_text: str) -> str:
    lines = []
    for i, r in enumerate(current_spec):
        lines.append(
            f"[id: r{i+1}] {r.get('name', '?')} | "
            f"цена={r.get('price', '')} | объём={r.get('qty', '')} | "
            f"сумма={r.get('sum', '')} | начало={r.get('date_start', '')}"
        )
    spec_text = "\n".join(lines)

    return f"""Ты — анализатор допсоглашений к договорам облачного провайдера.

У тебя есть текущая спецификация услуг и текст нового допсоглашения (ДС).
Твоя задача — определить, какие изменения вносит ДС в текущую спецификацию.

Верни СТРОГО JSON без пояснений. Не используй markdown-блоки.

ФОРМАТ:
{{
  "mode": "partial" | "full_replace",
  "ops": [
    {{"action": "ADD", "new_row": {{"name": "...", "price": число, "qty": число, "sum": число, "date_start": "YYYY-MM-DD"}}, "comment": "..."}},
    {{"action": "UPDATE", "target_id": "rN", "new_values": {{"price": число}}, "comment": "..."}},
    {{"action": "DELETE", "target_id": "rN", "comment": "..."}},
    {{"action": "UNRESOLVED", "new_values": {{"name": "...", "price": число, ...}}, "reason": "почему не смог сопоставить"}}
  ]
}}

ПРАВИЛА:
1. mode = "full_replace" — если в тексте есть фразы: «в следующей редакции», «заменить приложение», «излагается в следующей редакции». При full_replace — опиши ВСЕ новые строки как ADD.
2. mode = "partial" — если ДС меняет только отдельные строки.
3. Для UPDATE/DELETE — укажи target_id (r1, r2...) из списка текущей спецификации. НЕ придумывай новые id.
4. new_values в UPDATE — только ИЗМЕНЁННЫЕ поля (не все).
5. Если не можешь однозначно сопоставить строку — action: UNRESOLVED с reason.
6. Пропускай итоговые строки и подписи.
7. price, qty, sum — ЧИСЛА (не строки).

ТЕКУЩАЯ СПЕЦИФИКАЦИЯ:
{spec_text}

ТЕКСТ ДОПСОГЛАШЕНИЯ:
---
{doc_text}
---"""