from __future__ import annotations
from pathlib import Path
import fitz
from pypdf import PdfReader, PdfWriter

def open_pdf(path: Path) -> fitz.Document:
    return fitz.open(path)

def save_pdf(doc: fitz.Document, out_path: Path) -> Path:
    out_path.parent.mkdir(parents=True, exist_ok=True)
    doc.save(out_path, garbage=4, deflate=True, clean=True)
    return out_path

def rebuild_pdf(input_path: Path, output_path: Path) -> Path:
    reader = PdfReader(str(input_path))
    writer = PdfWriter()
    for page in reader.pages:
        writer.add_page(page)
    output_path.parent.mkdir(parents=True, exist_ok=True)
    with output_path.open("wb") as f:
        writer.write(f)
    return output_path