Add acrobat reader compatibility for szipdf format1.4.1

2 files changed, 16 insertions, 5 deletions

diff --git a/PolyglotFile/polyglotszippdf.py b/PolyglotFile/polyglotszippdf.py
index 0796946..eadf5d0 100644
--- a/PolyglotFile/polyglotszippdf.py
+++ b/PolyglotFile/polyglotszippdf.py
@@ -28,8 +28,9 @@ from PdfFileTransformer import Pdf
 
 class PolyglotSZipPdf(PolyglotPdfZip):
 
-    def __init__(self, Pdf, Zip):
+    def __init__(self, Pdf, Zip, acrobat_compatibility):
         super().__init__(Pdf, Zip)
+        self.acrobat_compatibility = acrobat_compatibility
 
     def get_rebuild_zip_first_part_size(self):
 
@@ -82,7 +83,11 @@ class PolyglotSZipPdf(PolyglotPdfZip):
         new_pdf.file_offset = offset
         pdf_buffer = new_pdf.get_build_buffer()
         j2 = pdf_buffer[k2_stream_offset + size_k2_stream:]
-        new_zip.add_data_to_file(b'', j2, True)
+
+        if self.acrobat_compatibility:
+            new_zip.add_data_to_file(b'\x00', j2, True)
+        else:
+            new_zip.add_data_to_file(b'', j2, True)
 
         return new_zip.buffer
 
diff --git a/truepolyglot b/truepolyglot
index 2ff9269..4b4075f 100755
--- a/truepolyglot
+++ b/truepolyglot
@@ -19,7 +19,7 @@ def main():
                        ' The format is strictly a ZIP.' +
                        ' Archive is modified.')
     usage_str = '%(prog)s format [options] output-file'
-    epilog_str = 'TruePolyglot v1.3'
+    epilog_str = 'TruePolyglot v1.4.1'
     frm = argparse.RawTextHelpFormatter
     parser = argparse.ArgumentParser(description=description_str,
                                      epilog=epilog_str,
@@ -33,8 +33,12 @@ def main():
                         help='PDF input file')
     parser.add_argument('--zipfile', dest='zipfile',
                         help='ZIP input file')
+    parser.add_argument('--acrobat-compatibility',
+                        dest='acrobat_compatibility',
+                        help='Add a byte at start for Acrobat Reader compatibility with szippdf format',
+                        action='store_true')
     parser.add_argument('--verbose', dest='verbose',
-                        help='Verbosity level  (default: debug)',
+                        help='Verbosity level  (default: info)',
                         default="info",
                         choices=["none", "error", "info", "debug"])
     parser.add_argument('output_file', nargs='+',
@@ -43,6 +47,8 @@ def main():
     args = parser.parse_args()
 
     formats = ["pdfzip", "zippdf", "szippdf"]
+    if args.acrobat_compatibility and args.format[0] != "szippdf":
+        parser.error('--acrobat-compatibility is for szippdf only')
     if args.format[0] in formats:
         if args.pdffile is None:
             parser.error('pdffile is required')
@@ -65,7 +71,7 @@ def main():
     if args.format[0] == "zippdf":
         a = PolyglotZipPdf(p, z)
     if args.format[0] == "szippdf":
-        a = PolyglotSZipPdf(p, z)
+        a = PolyglotSZipPdf(p, z, args.acrobat_compatibility)
     a.generate()
     a.write(args.output_file[0])