From 01ff49808cc6ccffcd73b15169c69b1305f422ea Mon Sep 17 00:00:00 2001
From: John Whitington <john@coherentgraphics.co.uk>
Date: Fri, 13 Jan 2023 14:30:46 +0800
Subject: [PATCH] more

---
 cpdfannot.mli  |  2 +-
 cpdfcommand.ml | 14 +++++++++++++-
 2 files changed, 14 insertions(+), 2 deletions(-)

diff --git a/cpdfannot.mli b/cpdfannot.mli
index 49d0437..3e95d6a 100644
--- a/cpdfannot.mli
+++ b/cpdfannot.mli
@@ -20,4 +20,4 @@ val remove_annotations : int list -> Pdf.t -> Pdf.t
 val get_annotations_json : Pdf.t -> Pdfio.bytes
 
 (** Set annotations from JSON. Existing annotations will be removed. *)
-val set_annotations_json : Pdf.t -> Pdfio.bytes -> unit
+val set_annotations_json : Pdf.t -> Pdfio.input -> unit
diff --git a/cpdfcommand.ml b/cpdfcommand.ml
index 33fb38b..fd0dfad 100644
--- a/cpdfcommand.ml
+++ b/cpdfcommand.ml
@@ -125,6 +125,7 @@ type op =
   | RemoveAnnotations
   | ListAnnotations
   | CopyAnnotations of string
+  | SetAnnotations of string
   | Merge
   | Split
   | SplitOnBookmarks of int
@@ -258,6 +259,7 @@ let string_of_op = function
   | RemoveAnnotations -> "RemoveAnnotations"
   | ListAnnotations -> "ListAnnotations"
   | CopyAnnotations _ -> "CopyAnnotations"
+  | SetAnnotations _ -> "SetAnnotations"
   | Merge -> "Merge"
   | Split -> "Split"
   | SplitOnBookmarks _ -> "SplitOnBookmarks"
@@ -841,7 +843,7 @@ let banned banlist = function
   | TwoUp|TwoUpStack|RemoveBookmarks|AddRectangle|RemoveText|
     Draft|Shift|Scale|ScaleToFit|RemoveAttachedFiles|
     RemoveAnnotations|RemoveFonts|Crop|RemoveCrop|Trim|RemoveTrim|Bleed|RemoveBleed|Art|RemoveArt|
-    CopyCropBoxToMediaBox|CopyBox|MediaBox|HardBox _|SetTrapped|SetUntrapped|Presentation|
+    CopyCropBoxToMediaBox|CopyBox|MediaBox|HardBox _|SetTrapped|SetAnnotations _|SetUntrapped|Presentation|
     BlackText|BlackLines|BlackFills|CopyFont _|StampOn _|StampUnder _|StampAsXObject _|
     AddText _|ScaleContents _|AttachFile _|CopyAnnotations _| ThinLines _ | RemoveClipping | RemoveAllText
     | Prepend _ | Postpend _ | Draw ->
@@ -1085,6 +1087,8 @@ let setthinlines s = setop (ThinLines (Cpdfcoord.parse_single_number empty s)) (
 
 let setcopyannotations s = setop (CopyAnnotations s) ()
 
+let setsetannotations s = setop (SetAnnotations s) ()
+
 let setshift s =
   setop Shift ();
   args.coord <- s
@@ -2479,6 +2483,9 @@ and specs =
    ("-remove-annotations",
       Arg.Unit (setop RemoveAnnotations),
       " Remove annotations");
+   ("-set-annotations",
+      Arg.String setsetannotations,
+      " Set annotations from JSON file");
    ("-list-fonts",
        Arg.Unit (setop Fonts),
        " Output font list");
@@ -3965,6 +3972,11 @@ let go () =
             write_pdf false pdf
       | _ -> error "copy-annotations: No input file specified"
       end
+  | Some (SetAnnotations json) ->
+      let data = Pdfio.input_of_channel (open_in_bin json) in
+      let pdf = get_single_pdf args.op false in
+        Cpdfannot.set_annotations_json pdf data;
+        write_pdf false pdf
   | Some ListAnnotations ->
       let pdf = get_single_pdf args.op true in
       let range = parse_pagespec_allow_empty pdf (get_pagespec ()) in