// 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 do encryption and decryption for a PDF file
// by using password security method, custom security method, Foxit DRM security method, certificate security method
// or Microsoft RMS security method.


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/";
const output_directory = output_path + "security/";

class DRMSecurityEvent{
  constructor(file_id, initialize_key) {
    this.file_id_ = file_id;
    this.initialize_key_ = initialize_key;
  }    
  Release() { }

  IsOwner(doc, subFilter) {
    return true;
  }

  GetUserPermissions(doc, subFilter) {
    return 0xfffffffc;
  }

  GetCipherType(doc, subFilter) {
    return FSDK.SecurityHandler.e_CipherAES;
  }

  GetKeyLength(doc, subFilter) {
    return 16;
  }

  GetFileID(doc, subFilter) {
    return this.file_id_;
  }
  
  GetInitialKey(doc, subFilter) {
    return this.initialize_key_;
  }  
};
 
function FoxitDRMDecrypt(input_file) {
  try {
    let doc = new FSDK.PDFDoc(input_file);
    let error_code = doc.Load("");
    if (error_code != FSDK.e_ErrSuccess) {
      console.log("The Doc [%s] Error: %d\n", input_file, error_code);
      return false;
    }

    // Do encryption.
    let handler = new FSDK.DRMSecurityHandler();
    let file_id = "Simple-DRM-file-ID";
    let initialize_key = "Simple-DRM-initialize-key";
    let encrypt_data = new FSDK.DRMEncryptData(true, "Simple-DRM-filter", FSDK.SecurityHandler.e_CipherAES, 16, true, 0xfffffffc);
    handler.Initialize(encrypt_data, file_id, initialize_key);
    doc.SetSecurityHandler(handler);

    let output_file = output_directory + "foxit_drm_encrypt.pdf";
    doc.SaveAs(output_file, FSDK.PDFDoc.e_SaveFlagNoOriginal);

    // Do decryption.
    let callback = new DRMSecurityEvent(file_id, initialize_key);
    let drm_security_callback = new FSDK.DRMSecurityCallback(callback);
    FSDK.Library.RegisterSecurityCallback("FoxitDRM", drm_security_callback);
    let encrypted_doc = new FSDK.PDFDoc(output_file);
    error_code = encrypted_doc.Load("");
    if (error_code != FSDK.e_ErrSuccess) {
      console.log("The Doc [%s] Error: %d\n", output_file, error_code);
      return false;
    }

    output_file = output_directory + "foxit_drm_decrypt.pdf";

    encrypted_doc.RemoveSecurity();
    encrypted_doc.SaveAs(output_file, FSDK.PDFDoc.e_SaveFlagNoOriginal);

    FSDK.Library.UnregisterSecurityCallback("FoxitDRM");
  } catch (e) {
    console.log(e);
    return false;
  }
  return true;
}

class CustomSecurityEvent{
  constructor() { }     
  Release() { }  
  
  CreateContext(filter, sub_filter, ncrypt_info) {    
    return null;
  }
  
  ReleaseContext(context) {
    return true;
  }
  
  GetUserPermissions(context, user_permission) {
    return 0xffffffc;
  }
  
  IsOwner(context) {
    return true;
  }

  GetCipher(context) {
    return FSDK.SecurityHandler.e_CipherAES;
  }
  
  GetEncryptKey(context) {
    return "key";
  }
  
  GetDecryptedSize(context, data_size) {
    return data_size;
  }
  
  StartDecryptor(context, obj_number, gen_number) {
    return null;
  }
  
  DecryptData(decryptor, enc_data_buf, enc_data_length) {
    let decrypted_data = Buffer.alloc(enc_data_length);
    for (let i = 0; i < enc_data_length; i++)
      decrypted_data[i] = enc_data_buf[i] + 1;
    return decrypted_data;      
  }
  
  FinishDecryptor(decryptor) {
    return "";
  }
  
  GetEncryptedSize(context, obj_number, gen_number, src_buffer, src_length) {
    return src_length;
  }
  
  EncryptData(context, obj_number, gen_number, src_buffer, src_length) {
    let dst_bufffer = Buffer.alloc(src_length);
    for (let i = 0; i < src_length; i++)
      dst_bufffer[i] = src_buffer[i] - 1;       
    return [true, dst_bufffer];  
  }
};

function CustomSecurity(input_file) {
  try {
    let doc = new FSDK.PDFDoc(input_file);
    let error_code = doc.Load("");
    if (error_code != FSDK.e_ErrSuccess) {
      console.log("The Doc [%s] Error: %d\n", input_file, error_code);
      return false;
    }    
    // Do encryption.
    let callback = new CustomSecurityEvent();
    let custiom_security_callback = new FSDK.CustomSecurityCallback(callback);
    let handler = new FSDK.CustomSecurityHandler();
    let encrypt_info = "Foxit simple demo custom security";
    let encrypt_data = new FSDK.CustomEncryptData(true, "Custom", "foxit-simple-demo");
    handler.Initialize(encrypt_data, custiom_security_callback, encrypt_info);
    doc.SetSecurityHandler(handler);    
    let output_file = output_directory + "custom_encrypt.pdf";  
    doc.SaveAs(output_file, FSDK.PDFDoc.e_SaveFlagNoOriginal);    
    
    // Do decryption.
    FSDK.Library.RegisterSecurityCallback("Custom", custiom_security_callback);
    let encrypted_doc = new FSDK.PDFDoc(output_file);

    error_code = encrypted_doc.Load("");
    if (error_code != FSDK.e_ErrSuccess) {
      console.log("The Doc [%s] Error: %d\n", output_file, error_code);
      return false;
    }

    output_file = output_directory + "custom_decrypt.pdf";
    encrypted_doc.RemoveSecurity();
    encrypted_doc.SaveAs(output_file, FSDK.PDFDoc.e_SaveFlagNoOriginal);

    FSDK.Library.UnregisterSecurityCallback("Custom");
  } catch (e) {
    console.log(e); 
    return false;
  }
  return true;
}

class RMSSecurityEvent{
  constructor(key) {
    this.key_ = key;
  }     
  
  Release() { }  
  
  CreateContext(filter, sub_filter, ncrypt_info) {
    return null;
  }
  
  ReleaseContext(context) {
    return true;
  }
  
  GetUserPermissions(context, user_permission) {
    return 0xffffffc;
  }
  
  IsOwner(context) {
    return true;
  }

  GetCipher(context) {
    return FSDK.SecurityHandler.e_CipherAES;
  }
  
  GetEncryptKey(context) {
    return this.key_;
  }
  
  GetDecryptedSize(context, data_size) {
    return data_size;
  }
  
  StartDecryptor(context, obj_number, gen_number) {
    return null;
  }
  
  DecryptData(decryptor, enc_data_buf, enc_data_length) {
    let decrypted_data = Buffer.alloc(enc_data_length);
    for (let i = 0; i < enc_data_length; i++)
      decrypted_data[i] = enc_data_buf[i] + 5;    
    return decrypted_data;
  }
  
  FinishDecryptor(decryptor) {
    return "";
  }
  
  GetEncryptedSize(context, obj_number, gen_number, src_buffer, src_length) {
    return src_length;
  }
  
  EncryptData(context, obj_number, gen_number, src_buffer, src_length) {
    let dst_bufffer = Buffer.alloc(src_length);
    for (let i = 0; i < src_length; i++)
      dst_bufffer[i] = src_buffer[i] - 5;      
    return [true, dst_bufffer];  
  }
};

function RMSSecurity(input_file) {
  try {
    let doc = new FSDK.PDFDoc(input_file);
    let error_code = doc.Load("");
    if (error_code != FSDK.e_ErrSuccess) {
      console.log("The Doc [%s] Error: %d\n", input_file, error_code);
      return false;
    }

    // Do encryption.
    let callback = new RMSSecurityEvent("Simple-RMS-encrpty-key");
    let rms_security_callback = new FSDK.RMSSecurityCallback(callback);
    let handler = new FSDK.RMSSecurityHandler();
    server_eul_list = new FSDK.StringArray();
    server_eul_list.Add("WM-1");
    server_eul_list.Add("This document has been encrypted by RMS encryption.");
    server_eul_list.Add("WM-2");
    server_eul_list.Add("Just for simple demo rms security.");
    let encrypt_data = new FSDK.RMSEncryptData(true, "PubishLicense_0123", server_eul_list, 1);
    handler.Initialize(encrypt_data, rms_security_callback);
    doc.SetSecurityHandler(handler);
    let output_file = output_directory + "rms_encrypt.pdf";
    doc.SaveAs(output_file, FSDK.PDFDoc.e_SaveFlagNoOriginal);

    // Do decryption.
    FSDK.Library.RegisterSecurityCallback("MicrosoftIRMServices", rms_security_callback);
    let encrypted_doc = new FSDK.PDFDoc(output_file);
    error_code = encrypted_doc.Load("");
    if (error_code != FSDK.e_ErrSuccess) {
      console.log("The Doc [%s] Error: %d\n", output_file, error_code);
      return false;
    }
    output_file = output_directory + "rms_decrypt.pdf";
    encrypted_doc.RemoveSecurity();
    encrypted_doc.SaveAs(output_file, FSDK.PDFDoc.e_SaveFlagNoOriginal);

    FSDK.Library.UnregisterSecurityCallback("Simple-RMS-encrpty-key");
  } catch (e) {
    console.log(e);
    return false;
  }
  return true;
}

function StdSecurity(input_file) {
  try {
    let doc = new FSDK.PDFDoc(input_file);
    let error_code = doc.Load("");
    if (error_code != FSDK.e_ErrSuccess) {
      console.log("The Doc [%s] Error: %d\n", input_file, error_code);
      return false;
    }

    // Do encryption.
    let handler = new FSDK.StdSecurityHandler();
    let encrypt_data = new FSDK.StdEncryptData(true, 0xfffffffc, FSDK.SecurityHandler.e_CipherAES, 16);
    handler.Initialize(encrypt_data, String("user"), String("owner"));
    doc.SetSecurityHandler(handler);

    let output_file = output_directory + "std_encrypt_userpwd[user]_ownerpwd[owner].pdf";
    doc.SaveAs(output_file, FSDK.PDFDoc.e_SaveFlagNoOriginal);

    // Do decryption.
    let encrypted_doc = new FSDK.PDFDoc (output_file);
    error_code = encrypted_doc.Load("owner");
    if (error_code != FSDK.e_ErrSuccess) {
      console.log("The Doc [%s] Error: %d\n", output_file, error_code);
      return false;
    }
    encrypted_doc.RemoveSecurity();
    output_file = output_directory + "std_decrypt.pdf";
    encrypted_doc.SaveAs(output_file, FSDK.PDFDoc.e_SaveFlagNoOriginal);
  } catch (e) {
    console.log(e);
    return false;
  }
  return true;
}

function SecurityImpl(pdf_file) {
  let is_ok = true;

  if (!FoxitDRMDecrypt(pdf_file)) {
    is_ok = false;
  }
  
  if (!CustomSecurity(pdf_file)) {
    is_ok = false;
  }
  if (!RMSSecurity(pdf_file)) {
    is_ok = false;
  }
  if (!StdSecurity(pdf_file)) {
    is_ok = false;
  }

  return is_ok;
}

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;
}

if (!fs.existsSync(output_directory))  {
  fs.mkdirSync(output_directory);
}


let input_file = input_path + "AboutFoxit.pdf";
if (!SecurityImpl(input_file)) {
    return 1;
  }
console.log("Security demo test.");
FSDK.Library.Release();

