# ingest_text.py
import os
import urllib.parse
from pathlib import Path

from flask import Blueprint, abort, jsonify, request

bp = Blueprint("ingest_text", __name__)
BASE = Path(os.getenv("INGEST_DIR", "/home/pi/tmp_git")).resolve()
BASE.mkdir(parents=True, exist_ok=True)
TOKEN = os.getenv("INGEST_TOKEN", "change-me")

# prosty bufor kawałków w RAM (na 1 proces)
chunks = {}


@bp.get("/api/ingest-text")
def ingest_text():
    if request.args.get("key") != TOKEN:
        return jsonify(error="unauthorized"), 401

    name = request.args.get("name", "").strip()
    index = int(request.args.get("index", "1"))
    total = int(request.args.get("total", "1"))
    chunk = request.args.get("chunk", "")

    if not name or "/" in name or ".." in name:
        return jsonify(error="bad name"), 400
    if not (1 <= index <= total <= 9999):
        return jsonify(error="bad indexing"), 400

    # gromadzimy w pamięci (możesz podmienić na Redis)
    key = f"{name}:{total}"
    entry = chunks.setdefault(key, {})
    entry[index] = urllib.parse.unquote_plus(chunk)

    if len(entry) == total:
        # składamy i zapisujemy
        data = "".join(entry[i] for i in range(1, total + 1))
        out = (BASE / name).resolve()
        if not str(out).startswith(str(BASE)):
            return jsonify(error="bad path"), 400
        out.write_text(data, encoding="utf-8")
        del chunks[key]
        return jsonify(ok=True, saved=str(out), bytes=len(data.encode("utf-8")))
    else:
        return jsonify(pending=True, got=len(entry), total=total)