from pathlib import Path
from io import BytesIO

from pypdf import PdfReader


def extract_pdf_pages(file_path: Path, max_pages: int | None = None) -> tuple[list[tuple[int, str]], int]:
    reader = PdfReader(str(file_path))
    total_pages = len(reader.pages)
    page_limit = min(total_pages, max_pages) if max_pages else total_pages
    pages: list[tuple[int, str]] = []
    for index, page in enumerate(reader.pages[:page_limit], start=1):
        pages.append((index, (page.extract_text() or "").strip()))
    return pages, total_pages


def extract_pdf_text(file_path: Path, max_pages: int | None = None) -> tuple[str, int]:
    pages, total_pages = extract_pdf_pages(file_path, max_pages=max_pages)
    text = "\n\n".join(page_text for _, page_text in pages).strip()
    return text, total_pages


def extract_pdf_pages_from_bytes(content: bytes, max_pages: int | None = None) -> tuple[list[tuple[int, str]], int]:
    reader = PdfReader(BytesIO(content))
    total_pages = len(reader.pages)
    page_limit = min(total_pages, max_pages) if max_pages else total_pages
    pages: list[tuple[int, str]] = []
    for index, page in enumerate(reader.pages[:page_limit], start=1):
        pages.append((index, (page.extract_text() or "").strip()))
    return pages, total_pages


def extract_pdf_text_from_bytes(content: bytes, max_pages: int | None = None) -> tuple[str, int]:
    pages, total_pages = extract_pdf_pages_from_bytes(content, max_pages=max_pages)
    text = "\n\n".join(page_text for _, page_text in pages).strip()
    return text, total_pages


def count_pdf_pages_from_bytes(content: bytes) -> int:
    reader = PdfReader(BytesIO(content))
    return len(reader.pages)