from __future__ import annotations
from dataclasses import dataclass, field, asdict
from pathlib import Path
from typing import Any, Literal
import json, time, uuid

OperationType = Literal[
    "insert_text","whiteout_region","rotate_page","delete_page",
    "reorder_pages","highlight","comment","rectangle","redact"
]

@dataclass
class PDFOperation:
    type: OperationType
    page: int | None = None
    rect: tuple | None = None
    text: str | None = None
    position: tuple | None = None
    color: tuple | None = None
    pages: list | None = None
    rotation: int | None = None
    author: str = "Local PDF AI Orchestrator"
    metadata: dict = field(default_factory=dict)

@dataclass
class InspectionReport:
    page_count: int
    metadata: dict
    pages: list
    ocr_required_pages: list
    has_text: bool
    has_images: bool

@dataclass
class JobConfig:
    input_pdf: Path
    output_dir: Path
    run_ocr: bool = False
    force_ocr: bool = False
    ocr_language: str = "eng"
    pipeline: str = "full_document"
    operations: list = field(default_factory=list)
    job_id: str = field(default_factory=lambda: uuid.uuid4().hex)
    keep_intermediate: bool = True

@dataclass
class JobResult:
    job_id: str
    success: bool
    final_pdf: Path | None
    log_file: Path
    manifest_file: Path
    stages: list
    errors: list = field(default_factory=list)

    def to_json(self) -> str:
        return json.dumps({
            "job_id": self.job_id,
            "success": self.success,
            "final_pdf": str(self.final_pdf) if self.final_pdf else None,
            "log_file": str(self.log_file),
            "manifest_file": str(self.manifest_file),
            "stages": self.stages,
            "errors": self.errors,
        }, indent=2)

def dataclass_to_dict(obj: Any) -> dict:
    def convert(v):
        if isinstance(v, Path): return str(v)
        if isinstance(v, dict): return {k: convert(x) for k, x in v.items()}
        if isinstance(v, list): return [convert(x) for x in v]
        return v
    return {k: convert(v) for k, v in asdict(obj).items()}