# SPDX-FileCopyrightText: 2026 geisserml <geisserml@gmail.com>
# SPDX-License-Identifier: Apache-2.0 OR BSD-3-Clause

import atexit
import logging
import pypdfium2.raw as pdfium_c
import pypdfium2.internal as pdfium_i
from pypdfium2_cfg import DEBUG_AUTOCLOSE

logger = logging.getLogger("pypdfium2")


def init_lib():
    assert not pdfium_i.LIBRARY_AVAILABLE
    if DEBUG_AUTOCLOSE:  # pragma: no cover
        logger.log(DEBUG_AUTOCLOSE.value, "Initialize PDFium")
    
    # PDFium init API may change in the future: https://crbug.com/pdfium/1446
    # NOTE Technically, FPDF_InitLibrary() would be sufficient for our purposes, but pdfium docs say "This will be deprecated in the future", so don't use it to be on the safe side. Also, avoid experimental config versions that might not be promoted to stable.
    config = pdfium_c.FPDF_LIBRARY_CONFIG(
        version = 2,
        m_pUserFontPaths = None,
        m_pIsolate = None,
        m_v8EmbedderSlot = 0,
        # m_pPlatform = None,  # v3
        # m_RendererType = pdfium_c.FPDF_RENDERERTYPE_AGG,  # v4
    )
    pdfium_c.FPDF_InitLibraryWithConfig(config)
    pdfium_i.LIBRARY_AVAILABLE.value = True


def _close_objects():
    
    need_close = []
    for cls, obj_wrefs in pdfium_i.ObjectTracker.items():
        # pdfium_i._debug_close(f"{cls and cls.__name__}: {obj_wrefs}")
        for wref in obj_wrefs:
            obj = wref()
            if obj is None:
                pdfium_i._warn_close(f"Weakref {wref} was not cleaned up from ObjectTracker.")
            else:
                # outsource actual closing to avoid "RuntimeError: Set changed size during iteration" (because closing removes the object from the set of weakrefs)
                need_close.append(obj)
    
    if need_close:
        pdfium_i._warn_close(f"The following objects are still open and will now be closed: {need_close}")
        for obj in need_close:
            obj.close()


def destroy_lib():  # pragma: no cover
    assert pdfium_i.LIBRARY_AVAILABLE
    try:
        _close_objects()
    finally:
        pdfium_i._debug_close("Destroy PDFium")
        pdfium_c.FPDF_DestroyLibrary()
        pdfium_i.LIBRARY_AVAILABLE.value = False


# Load pdfium
init_lib()

# Register an exit handler that will free pdfium
atexit.register(destroy_lib)