From 9c5a905cc7381565a36967c8e3c0346cb8cb338c Mon Sep 17 00:00:00 2001
From: John Whitington <john@coherentgraphics.co.uk>
Date: Fri, 31 Dec 2021 19:28:42 +0000
Subject: [PATCH] more

---
 cpdfannot.ml     | 25 +++++++++++++++++++++++++
 cpdfbookmarks.ml |  8 ++------
 2 files changed, 27 insertions(+), 6 deletions(-)

diff --git a/cpdfannot.ml b/cpdfannot.ml
index c041ef0..e82de61 100644
--- a/cpdfannot.ml
+++ b/cpdfannot.ml
@@ -30,6 +30,31 @@ let make_direct pdf annot =
   | None -> annot
   | Some d -> Pdf.add_dict_entry annot "/A" d
 
+let rewrite_destination calculate_pagenumber d =
+  match d with
+  | Pdf.Array (Pdf.Indirect i::r) ->
+      Pdf.Array (Pdf.Indirect (calculate_pagenumber (Pdfdest.Fit (Pdfdest.PageObject i)))::r)
+  | x -> x
+
+let rewrite_destinations pdf annot =
+  let refnums = Pdf.page_reference_numbers pdf in
+  let fastrefnums = hashtable_of_dictionary (combine refnums (indx refnums)) in
+  let calculate_pagenumber =  Pdfpage.pagenumber_of_target ~fastrefnums pdf in
+    (* Deal with /Dest in annotation *)
+    match Pdf.lookup_direct pdf "/Dest" annot with
+    | Some d -> Pdf.add_dict_entry annot "/Dest" (rewrite_destination calculate_pagenumber d)
+    | None ->
+        (* Deal with /A --> /D dest when /A --> /S = /GoTo *)
+        match Pdf.lookup_direct pdf "/A" annot with
+        | Some action ->
+            begin match Pdf.lookup_direct pdf "/D" action with
+            | Some d ->
+                Pdf.add_dict_entry
+                  annot "/A" (Pdf.add_dict_entry action "/D" (rewrite_destination calculate_pagenumber d))
+            | None -> annot
+            end
+       | None -> annot
+
 let annotations_json_page pdf page pagenum =
   match Pdf.lookup_direct pdf "/Annots" page.Pdfpage.rest with
   | Some (Pdf.Array annots) ->
diff --git a/cpdfbookmarks.ml b/cpdfbookmarks.ml
index 04198bb..29038b5 100644
--- a/cpdfbookmarks.ml
+++ b/cpdfbookmarks.ml
@@ -1,7 +1,7 @@
 open Pdfutil
 open Cpdferror
 
-(* \section{Add bookmarks} *)
+(* Add bookmarks *)
 let read_lines input =
   let lines = ref [] in
    try
@@ -264,7 +264,7 @@ let list_bookmarks ~json encoding range pdf output =
 
 (* o is the stamp, u is the main pdf page *)
 
-(* \section{Split at bookmarks} *)
+(* Split at bookmarks *)
 
 let get_bookmark_name pdf marks splitlevel n _ =
   let refnums = Pdf.page_reference_numbers pdf in
@@ -273,10 +273,6 @@ let get_bookmark_name pdf marks splitlevel n _ =
   | {Pdfmarks.text = title}::_ -> Cpdfattach.remove_unsafe_characters Cpdfmetadata.UTF8 title
   | _ -> ""
 
-(* Find the stem of a filename *)
-(*let stem s =
-  implode (rev (tail_no_fail (dropwhile (neq '.') (rev (explode (Filename.basename s))))))*)
-
 (* Return list, in order, a *set* of page numbers of bookmarks at a given level *)
 let bookmark_pages level pdf =
   let refnums = Pdf.page_reference_numbers pdf in