from typing import cast

from PyPDF2 import PdfReader, PdfWriter
from PyPDF2.generic import ArrayObject

writer = PdfWriter()
reader = PdfReader("commented.pdf")

# print how many pages input1 has:
print(f"document1.pdf has {len(reader.pages)} pages.")

# add page 1 from input1 to output document, unchanged
page = reader.pages[0]

annots = cast(ArrayObject, page["/Annots"])
annot0 = annots[0].getObject()
annot1 = annots[1].getObject()
annot2 = annots[2].getObject()
annot3 = annots[3].getObject()
annot4 = annots[4].getObject()

# TEST: changing postition of the Popup's rectangle (Works!)
from PyPDF2.generic import *

popup_size = (180, 120)

rect = RectangleObject(annot0["/Rect"])
rect_x0 = rect.left
rect_y0 = rect.top
annot2.update(
    {
        NameObject("/Open"): BooleanObject(False),
        NameObject("/Rect"): RectangleObject(
            [rect_x0, rect_y0, rect_x0 + popup_size[0], rect_y0 + popup_size[1]]
        ),
    }
)

# Test make new Popup Annotation
rect = RectangleObject(annot1["/Rect"])
rect_x0 = rect.left
rect_y0 = rect.top

popup = DictionaryObject()
popup.update(
    {
        NameObject("/Type"): NameObject("/Annot"),
        NameObject("/Subtype"): NameObject("/Popup"),
        NameObject("/Parent"): IndirectObject(
            15, 0, reader
        ),  # TODO: How to find the parent's idnum? Manually: this is 15 for this object
        NameObject("/Open"): BooleanObject(False),
        NameObject("/Rect"): RectangleObject(
            [rect_x0, rect_y0, rect_x0 + popup_size[0], rect_y0 + popup_size[1]]
        ),
        NameObject("/F"): NumberObject(28),  # The type of object? A popup?
    }
)

popup_ref = writer._add_object(popup)

if "/Annots" in page:
    page["/Annots"].append(popup_ref)
else:
    page[NameObject("/Annots")] = ArrayObject([popup_ref])

# Adding the reference to its new popup:
annot1.update(
    {NameObject("/Popup"): popup_ref}  # TODO: put the number for the new reference
)

annots_ref = writer._add_object(annots)

writer.add_page(page)  # FIXME: not adding the new Popup annotation
# finally, write "output" to document-output.pdf
with open("PyPDF2-output.pdf", "wb") as fp:
    writer.write(fp)