# -*- coding: utf-8 -*-
"""그림 코드 끼워 넣기 API — I 코드 목록 · 그림 보기(받기) · 그림 넣기(그림저장).

근거: img_code.py 머리(STmate.Cpm TwM_Boxa 0x514df0 · 0x514bc0 · 0x5155a8)
쓰기는 tools/stmate-test/work_dbf/ 아래만(server.WORK_DBF_ROOT) · 바뀌는 파일은 COSTN.DBF · COSTN.SMT 둘뿐.
"""

from __future__ import annotations

import base64
import pathlib
from urllib.parse import quote

from fastapi import APIRouter, HTTPException
from fastapi.responses import Response
from img_code import get_image, list_images, put_image
from pydantic import BaseModel
from stc_project import safe_name, write_output
from stc_read import Source

router = APIRouter()
EXT = {"TJPEGIMAGE": ".jpg", "TBITMAP": ".bmp", "TICON": ".ico", "TMETAFILE": ".emf"}


def _root() -> pathlib.Path:
    import server

    return server.WORK_DBF_ROOT


def _src(path: str) -> Source:
    try:
        return Source(path)
    except FileNotFoundError:
        raise HTTPException(404, f"경로 없음: {path}")
    except ValueError as e:
        raise HTTPException(400, str(e))


def _guard(fn, *a, **k):
    try:
        return fn(*a, **k)
    except (ValueError, KeyError) as e:
        raise HTTPException(400, str(e))


class ImgReq(BaseModel):
    source_path: str
    code: str = ""


class ImgPutReq(BaseModel):
    source_path: str
    code: str
    data_b64: str
    filename: str
    to_jpeg: bool = False
    target_folder_name: str = ""


@router.post("/api/img/list")
def img_list(req: ImgReq):
    return {"rows": _guard(list_images, _src(req.source_path).files)}


@router.post("/api/img/get")
def img_get(req: ImgReq):
    """그림 바이트 — 브라우저가 못 그리는 꼴(TMETAFILE)은 mime 없이(받기만)."""
    data, kind, mime = _guard(get_image, _src(req.source_path).files, req.code)
    if not data:
        raise HTTPException(404, f"{req.code} 그림 없음")
    name = quote(f"{req.code.strip()}{EXT.get(kind, '.bin')}")
    head = {"X-Img-Kind": kind or "?", "Content-Disposition": f"inline; filename*=UTF-8''{name}"}
    return Response(content=data, media_type=mime or "application/octet-stream", headers=head)


@router.post("/api/img/put")
def img_put(req: ImgPutReq):
    """그림 넣기 → work_dbf/<이름>/ 와 STC(COSTN.DBF · COSTN.SMT 만 바뀜)."""
    s = _src(req.source_path)
    try:
        data = base64.b64decode(req.data_b64.split(",", 1)[-1])
    except ValueError:
        raise HTTPException(400, "base64 아님")
    changed, rep = _guard(put_image, s.files, req.code, data, req.filename, req.to_jpeg)
    files = dict(s.files) | changed
    res = write_output(
        _root(), safe_name(req.target_folder_name or s.path.stem), files, s.names, s.entries
    )
    return {**res, "report": rep}