// Copyright (C) 2003-2026, Foxit Software Inc..
// All Rights Reserved.
//
// http://www.foxitsoftware.com
//
// The following code is copyrighted and contains proprietary information and trade secrets of Foxit Software Inc..
// You cannot distribute any part of Foxit PDF SDK to any third party or general public,
// unless there is a separate license agreement with Foxit Software Inc. which explicitly grants you such rights.
//
// This file contains an example to demonstrate how to use Foxit PDF SDK to save a PDF document as a wrapper file
// and then open the wrapper file.


const FSDK = require("@foxitsoftware/foxit-pdf-sdk-node");
const fs = require('fs');
const os = require('os');
const path = require('path');
const util = require('util');

var sn = "bPxThFlBAYifO3yiW0C51bKbjPap1zHaf5aG5sX378TjFvB7Q0802A==";
var key = "8f3g1sGNtR8NAgfSNv+BvaJWy2fnrw36NZvzh8GUxYXvs+/YczAk4e34yqlOnmL4asgxGMO7+rlOcSHUSSKagx5g4G7VEXfkYLXnDDx77FOOedZPrFFcOCu1vT4+wOKzy1zkORq9cKtiFf44byH5sfRqf+OvELjep2teT9eF9yPCsSw2vRKljXlPy40DMP9XX1O5OTe+mSG3RvwKUMb0+RpG3YobM87j+/ynTDlJMMXw4vgwuTb1ZuAtlvkTdFz7HDLV81C6TYnn73ddHeGD3qYWBVsD4uJkDwyyH/SWr7QX0HSgVg8HSZqnab1VGdAUYxKY2vGWeOTJdu6qDGb5ZEb9O3kZZg+9LYAmUKzDfCYwfIsYZ8Y9LPC4wMx70kIq7JpNnzPvigrdshV8/ew1F0+qP/P5Z1EB53vDVrp2LVSuL873Wv4g3PzjZlNH+1qgT87hC+Buto4qsSUfP5Y2z0uApu+c/TfttCXbeMLSsw9TY1HQD3mkrZExMaQqRrAuXAcZWdvBnVuG3DCZmm8q0DLxJiCfYfViAU/AkbeKMIGaTJdrvIXBcssfde4nmU6qr0M4x4+Hs2ESddtY2QRtLAUBq/7h7GJlJHkEZr8ueaUy/6D/2vvAWz6wMLgmjlPbDTLqzfnCJylHCPtcJmu1SPPT9v1VreBX3+zNWcIT8HueyHEiUniqzwWkRKaotrnNb/zhnHLPpgaBSwKjyi8rOzzVwu/nY84G9ye4jsVjmkIRItKDrW7z22iS+QQuTwfqLVuMB0OD4jMkPCC2slzPfEeZnuh4+Rewln2qLRJVgI72zJ20OBWAelGSpO1Ii2Y3gOvojdmtUwx6R3Z1qFNWTTB25o9ytcux8BLOtTCkA8T8xaT4N339OTCsu/TmfahIDixKvr8cDa6F2WkBT8SOfv64W9PgiNv5B/BMOD2VWYaAl0N7yD0NkYYg/cDWRC4dfpdeip1jrr0k3mND0oSpIkxC+FicZbaYYwRdtJ9OSVWJ5OE0voLv8HgLze8mw6s4TrN3z+a2NQCveMa2VRN7YcEF7agPrvb/qU+otMwHae/wM7+f+HFAaOW+WTfzfxIVFB8/A1DN0NiEX+em4G7nO5F5ZcvfYPpaaioe8Nn33BbVLtMjXqsOBeM8AXiVfpD6YO75j7BqnBUPnivvD8BxJaaIQZvVSuQmvGm0ynt03IYwE8Rq+jYsunvsZ86jh5BiUR+Nm7sK0Qe9BdZn/7jdqohMv9czfMGUAscX+xBMunFbDpcRbWVpUADcm209L8Ndek2XY7YlCMxPgPVNvfWRF2jxehib9Tpd7zB5ZIDfAEH9zrpLfp29DLvCh2Yx8mH7xxdKG1O9Q0GdHphYHhej6smAcPm68R0GUQaTOLfXqNpueGJC1GAOI3Y4swjyt7jMTWcibnKLzwukWE5rnEDhn5n64JnyUM0ZFkRmtOiV";

const input_path = "../input_files/";
const output_path = "../output_files/";


class TextDoc {
  constructor(file_name, file_mode) {
    this.file_name = file_name;
    this.file_mode = file_mode;
    this.file_stream = fs.openSync(file_name, file_mode);
    if (!this.file_stream) {
      throw new Error('Cannot open file');
    }
  }
  Write(format, ...args) {
    if(!this.file_stream) 
      throw new Error("File not opened");
    let formatted_string = util.format(format, ...args);
    let buffer = new Buffer.from(formatted_string, 'utf-8');
    fs.writeSync(this.file_stream, buffer, 0, buffer.length);
  }
  WriteWithPrefix(count, prefix, format, ...args){
    for (let i = 0; i < count; i++) {
      this.Write(prefix);
    }
    this.Write(format, ...args);
  }
  Close() {
    fs.closeSync(this.file_stream);
    this.file_stream = null;
  }
}

function RenderPDF2Img(doc) {
  let page_count = doc.GetPageCount();

  for (i = 0; i < page_count; i++) {
    let page = doc.GetPage(i);

    // Parse page.
    page.StartParse(FSDK.PDFPage.e_ParsePageNormal, null, false);

    let width = page.GetWidth();
    let height = page.GetHeight();
    let matrix = page.GetDisplayMatrix(0, 0, width, height, page.GetRotation());

    // Prepare a bitmap for rendering.
    let bitmap = new FSDK.Bitmap(width, height, FSDK.Bitmap.e_DIBArgb, null, 0);
    bitmap.FillRect(0xFFFFFFFF, null);

    // Render page.
    let render = new FSDK.Renderer(bitmap, false);
    render.StartRender(page, matrix, null);
    // Add the bitmap to an image and save the image.
    let image = new FSDK.Image();
    image.AddFrame(bitmap);
    let image_name = `AboutFoxit_${i}`;
    image_name = output_directory + image_name + "_wrapper.jpg";

    if (!image.SaveAs(image_name)) {
      console.log("[Failed] Cannot save image file.\r\nPage index: %d", i);
    }
    matrix.Delete();
    bitmap.Delete();
    image.Delete();
    render.Delete();
    page.Delete();
  }
}

class FileReader {
  constructor(offset) {
    this.file_ = null;
    this.file_path_ = "";
    this.hint_data_record_ = [];
    this.filesize_ = offset;
  }

  LoadFile(file_path) {
    try {
      this.file_path_ = file_path;
      this.file_ = fs.openSync(file_path, 'r');
      return true;
    } catch (err) {
      console.error(err);
      return false;
    }
  }

  CloseFile() {
    fs.closeSync(this.file_);
  }

  GetSize() {
    return this.filesize_;
  }

  ReadBlock(offset, size_t) {
    try {
      let buffer = Buffer.alloc(size_t);
      fs.readSync(this.file_, buffer, 0, size_t, offset);
      return [true, buffer]
    } catch (err) {
      return [false, null];
    }
  }

  Release() {  }
}

class FileWriter {
  constructor() {
    this.m_fileFP = null;
    this.file_path_ = "";
  }

  LoadFile(file_path) {
    try {
      this.file_path_ = file_path;
      this.m_fileFP = fs.openSync(file_path, 'w');
      if (!this.m_fileFP) {
        return false;
      }
      return true;
    } catch (err) {
      console.error(err);
      return false;
    }
  }

  GetSize() {
    try {
      let stats = fs.statSync(this.file_path_);
      return stats.size;
    } catch (err) {
      console.error(err);
      return 0;
    }
  }

  Flush() {
    return true;  
  }

  WriteBlock(buffer, offset, size) { 
    try {
      const write_size = fs.writeSync(this.m_fileFP, buffer, 0, size, offset);
      if (write_size == size) {
        return true;
      } else {
        return false;
      }
    } catch (err) {
      console.error(err);
      return false;
    }
  }

  Release() {
    try {
      fs.closeSync(this.m_fileFP);
      this.m_fileFP = null;
    } catch (err) {
      console.error(err);

    }
  }
}

function OpenWrapperFile(file_name) {
  let file_info = output_directory + "Wrapperinfo.txt";
  let text_doc = new TextDoc(file_info, "w");

  let doc = new FSDK.PDFDoc(file_name);
  let code = doc.Load("");
  if (code != FSDK.e_ErrSuccess) {
    console.log("[Failed] Cannot load PDF document: %s\r\nError Message: %d", file_name, code);
    return false;
  }
  if(!doc.IsWrapper()){
    console.log("[Failed] %s is not a wrapper file.", file_name);
    return false;
  }
  let offset = doc.GetWrapperOffset();
  text_doc.Write("offset: %d\r\n", offset);
  let wrapper_data = doc.GetWrapperData();
  text_doc.Write("version: %d\r\n", wrapper_data.version);
  text_doc.Write("type: %s\r\n", wrapper_data.type);
  text_doc.Write("app_id: %s\r\n", wrapper_data.app_id);
  text_doc.Write("uri: %s\r\n",wrapper_data.uri);
  text_doc.Write("description: %s\r\n", wrapper_data.description);

  doc.Delete();
  //"offset" can also indicate the end position of the document or the document size
  let file_reader = new FileReader(offset);
  file_reader.LoadFile(file_name);
  let reader_callback = new FSDK.FileReaderCallback(file_reader);

  let doc_real = new FSDK.PDFDoc(reader_callback, false);
  code = doc_real.Load("");
  if (code != FSDK.e_ErrSuccess) {
    console.log("[Failed] Cannot load real PDF document.\r\nError Message: %d", code);
    return false;
  }

  RenderPDF2Img(doc_real);
  doc_real.Delete();
  return true;
}


let error_code = FSDK.Library.Initialize(sn, key);
if (FSDK.e_ErrSuccess != error_code) {
  if (FSDK.e_ErrInvalidLicense == error_code)
    console.log("[Failed] Current used Foxit PDF SDK key information is invalid.");
  else
    console.log("Library Initialize Error: %d", FSDK.e_ErrSuccess);
  return;
}

let output_directory = output_path + "pdfwrapper/";
if (!fs.existsSync(output_directory))  {
  fs.mkdirSync(output_directory);
}


let input_file = input_path + "wrapper.pdf";
try {
   // Load PDF document.
  let doc_wrapper = new FSDK.PDFDoc(input_file);
  let error_code = doc_wrapper.Load("");
  if (error_code != FSDK.e_ErrSuccess) {
    console.log("[Failed] Cannot load PDF document: %s.\r\nError Code: %d", input_file, error_code);
   return;
  }
  // Save the PDF document as a wrapper file.
  let wrapperData = new FSDK.WrapperData(10, "Foxit", "Foxit", "www.foxitsoftware.com", "foxit");
  let output_file = output_directory + "AboutFoxit_wrapper.pdf";

  let doc_real = new FSDK.PDFDoc(input_path + "AboutFoxit.pdf");
  error_code = doc_real.Load("");
  if (error_code != FSDK.e_ErrSuccess) {
    console.log("[Failed] Cannot load PDF document: %s.\r\nError Code: %d", input_path + "SamplePDF.pdf", error_code);
    return;
  }
  doc_real.SaveAs(output_file, FSDK.PDFDoc.e_SaveFlagNormal);
  doc_real.Delete();

  // "0xFFFFFFFC" can be changed to other values defined in enumeration FSUserPermissions.
  doc_wrapper.SaveAsWrapperFile(output_file, wrapperData, 0xFFFFFFFC, "");

  // Open the wrapper file.
  if (!OpenWrapperFile(output_file)) {
    return;
  }

  console.log("Add wrapper to PDF file.");

  output_file = output_directory + "AboutFoxit_payloadfile.pdf";
  let progressive = doc_wrapper.StartSaveAsPayloadFile(output_file, input_path + "AboutFoxit.pdf", "Unknown", "no_description", 1.0, FSDK.PDFDoc.e_SaveFlagIncremental, null);
  progressive.Delete();
  wrapperData.Delete();
  doc_wrapper.Delete();
  console.log("Save as payload file.");

  let doc_rmsv2 = new FSDK.PDFDoc(input_path + "Microsoft IRM protection 2.pdf");

  error_code = doc_rmsv2.Load("");
  if (error_code != FSDK.e_ErrSuccess) {
    console.log("[Failed] Cannot load PDF document: %s.\r\nError Code: %d", input_path + "Microsoft IRM protection 2.pdf", error_code);
    return;
  }

  if (FSDK.PDFDoc.e_WrapperPDFV2 == doc_rmsv2.GetWrapperType()) {
    let payloadfile = new FileWriter();
    payloadfile.LoadFile(output_directory + "MicrosoftIRMServices Protected PDF.pdf");
    let writer_callback = new FSDK.FileWriterCallback(payloadfile);
    let progressive = doc_rmsv2.StartGetPayloadFile(writer_callback, null);
    progressive.Delete();
  }
  doc_rmsv2.Delete();
  console.log("Get RMS V2 payload file.");

} catch (e) {
 console.log(e);
}
FSDK.Library.Release();