import os, uuid, shutil
from fastapi import APIRouter, HTTPException, UploadFile, File, Form
from fastapi.responses import JSONResponse
from pydantic import BaseModel
from typing import Optional
from ..database import get_db

router = APIRouter(prefix="/api/tools", tags=["tools"])
UPLOAD_DIR = os.environ.get("UPLOAD_DIR", "/uploads")

class ToolUpdate(BaseModel):
    name: Optional[str] = None
    description: Optional[str] = None
    category: Optional[str] = None
    daily_price: Optional[float] = None
    weekend_price: Optional[float] = None
    deposit: Optional[float] = None
    notes: Optional[str] = None

def tool_with_images(conn, row):
    t = dict(row)
    imgs = conn.execute("SELECT * FROM tool_images WHERE tool_id = ? ORDER BY is_main DESC, id", (t["id"],)).fetchall()
    t["images"] = [dict(i) for i in imgs]
    t["main_image"] = next((i["filename"] for i in t["images"] if i["is_main"]), (t["images"][0]["filename"] if t["images"] else None))
    return t

@router.get("")
def list_tools():
    with get_db() as conn:
        rows = conn.execute("SELECT * FROM tools ORDER BY name").fetchall()
        return [tool_with_images(conn, r) for r in rows]

@router.get("/{tool_id}")
def get_tool(tool_id: int):
    with get_db() as conn:
        row = conn.execute("SELECT * FROM tools WHERE id = ?", (tool_id,)).fetchone()
        if not row:
            raise HTTPException(404, "Outil introuvable")
        return tool_with_images(conn, row)

@router.post("", status_code=201)
async def create_tool(
    name: str = Form(...),
    description: str = Form(""),
    category: str = Form(""),
    daily_price: float = Form(0),
    weekend_price: float = Form(0),
    deposit: float = Form(0),
    notes: str = Form(""),
    images: list[UploadFile] = File(default=[])
):
    with get_db() as conn:
        cur = conn.execute(
            "INSERT INTO tools (name, description, category, daily_price, weekend_price, deposit, notes) VALUES (?,?,?,?,?,?,?)",
            (name.strip(), description.strip(), category.strip(), daily_price, weekend_price, deposit, notes.strip())
        )
        tool_id = cur.lastrowid
        for i, img in enumerate(images):
            if img.filename:
                ext = os.path.splitext(img.filename)[1].lower()
                fname = f"{uuid.uuid4().hex}{ext}"
                path = os.path.join(UPLOAD_DIR, "tools", fname)
                with open(path, "wb") as f:
                    shutil.copyfileobj(img.file, f)
                conn.execute("INSERT INTO tool_images (tool_id, filename, is_main) VALUES (?,?,?)", (tool_id, fname, 1 if i == 0 else 0))
        row = conn.execute("SELECT * FROM tools WHERE id = ?", (tool_id,)).fetchone()
        return tool_with_images(conn, row)

@router.put("/{tool_id}")
def update_tool(tool_id: int, data: ToolUpdate):
    with get_db() as conn:
        row = conn.execute("SELECT * FROM tools WHERE id = ?", (tool_id,)).fetchone()
        if not row:
            raise HTTPException(404, "Outil introuvable")
        fields, values = [], []
        for field, val in data.model_dump(exclude_none=True).items():
            fields.append(f"{field} = ?")
            values.append(val.strip() if isinstance(val, str) else val)
        if fields:
            values.append(tool_id)
            conn.execute(f"UPDATE tools SET {', '.join(fields)} WHERE id = ?", values)
        return tool_with_images(conn, conn.execute("SELECT * FROM tools WHERE id = ?", (tool_id,)).fetchone())

@router.delete("/{tool_id}", status_code=204)
def delete_tool(tool_id: int):
    with get_db() as conn:
        imgs = conn.execute("SELECT filename FROM tool_images WHERE tool_id = ?", (tool_id,)).fetchall()
        for img in imgs:
            path = os.path.join(UPLOAD_DIR, "tools", img["filename"])
            if os.path.exists(path):
                os.remove(path)
        conn.execute("DELETE FROM tools WHERE id = ?", (tool_id,))

@router.post("/{tool_id}/images", status_code=201)
async def add_image(tool_id: int, image: UploadFile = File(...), is_main: bool = Form(False)):
    with get_db() as conn:
        if not conn.execute("SELECT id FROM tools WHERE id = ?", (tool_id,)).fetchone():
            raise HTTPException(404, "Outil introuvable")
        ext = os.path.splitext(image.filename)[1].lower()
        fname = f"{uuid.uuid4().hex}{ext}"
        path = os.path.join(UPLOAD_DIR, "tools", fname)
        with open(path, "wb") as f:
            shutil.copyfileobj(image.file, f)
        if is_main:
            conn.execute("UPDATE tool_images SET is_main = 0 WHERE tool_id = ?", (tool_id,))
        conn.execute("INSERT INTO tool_images (tool_id, filename, is_main) VALUES (?,?,?)", (tool_id, fname, 1 if is_main else 0))
        return {"filename": fname}

@router.delete("/{tool_id}/images/{image_id}", status_code=204)
def delete_image(tool_id: int, image_id: int):
    with get_db() as conn:
        img = conn.execute("SELECT * FROM tool_images WHERE id = ? AND tool_id = ?", (image_id, tool_id)).fetchone()
        if not img:
            raise HTTPException(404, "Image introuvable")
        path = os.path.join(UPLOAD_DIR, "tools", img["filename"])
        if os.path.exists(path):
            os.remove(path)
        conn.execute("DELETE FROM tool_images WHERE id = ?", (image_id,))
        if img["is_main"]:
            first = conn.execute("SELECT id FROM tool_images WHERE tool_id = ? LIMIT 1", (tool_id,)).fetchone()
            if first:
                conn.execute("UPDATE tool_images SET is_main = 1 WHERE id = ?", (first["id"],))

@router.post("/{tool_id}/images/{image_id}/main", status_code=204)
def set_main_image(tool_id: int, image_id: int):
    with get_db() as conn:
        conn.execute("UPDATE tool_images SET is_main = 0 WHERE tool_id = ?", (tool_id,))
        conn.execute("UPDATE tool_images SET is_main = 1 WHERE id = ? AND tool_id = ?", (image_id, tool_id))