// 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("@fuxinsoft/foxit-pdf-sdk-node");
const fs = require('fs');
const os = require('os');
const path = require('path');
const util = require('util');


var sn = "Rew1vB0TqoMIxX444UcpScARWdhgJpsJfzmyMLR32a0+EgrCjp2NaQ==";
var key = "8f3g1sFttW8NAgfSNv+BvaJWy4Pp8flcm7ncNDkdvCCO4JwUFbGLyZdujReMaiplbYJwwxam3lr7uHtQaEf0xTSu8j2ZBdrZ8FS4OQkEUjOI9UEGb3Ic0GqmlP9T9OnSItOYqoZZA/OnYg5EaVP+c/ha32nUquOk+5riEdhYgINAZDdgJkPc+YSWsrfixOIROf6t34j3eRcyFnOY0MS7tSXi1AEPE0VZdyu+5bfmBSGvLI+WR5lQ+oOQR5hRGwqf990WEdBelNO7dxhZELVhEZ+/L60PpDf5y9A8qmrY6PiEU8nhqIyLpks0BxeIsYyZ3Cr82jjfbItPGyEARsIVzluBSgVMD4oGjojz8T6wGDEXUbmZH/pra94XAs8wZ28mzbmIEppWnhBqX0c6Ov8iXCfFfUZr4KNuPxMEgk4VOYTMOd70GEjGnylivw18hbn7ZpkWh2Zql+uU2HejvOgn+WHvT0HF9NhK0Xb5vH8aFkgBP92PJx8obIbxBVeeExczxvsqWufuPdXS2sPNIgMYZ2TtvqVVsiUToeGB2FQo09fzZOSH5Vi18u/siyvC3FmjeMf4+OYWPuDPNCcrb21NgcHam7XI8AAoGLTGtLLWR+fxynbBBcg2d2ZPcZ5CMFIfOBHRWCkcA0sDT3cl9nsBppHNJze/jVxHy+lREeQEA/etSIgohTGAWW3ZlpibRXqMxlmVWEPifCubbWTzQ6U54mRMpW2UghH3gvCNFNgdday3Av7EtlIJOdUuiMq3zmmd5J4Jt4C82CKnVewA0DtZ4EEtwRG9IYuLrfpuZlnjBAtoeXRV1lqUZ6BcCe0E3m9jO28sGkALuIeJQ74XG7X8lMC7lMWWQYiYD7D6G9QT6X3qV4vXJwNKRKpYc/rI+amEawTSnDYAMEaYPwtZ0rb/7BL+atsZZaI+ncebdfVfjDJZbCZAc+JnkuNlUn2xrNRuObw8h/G0lm2pNOEJ10S8R4+yplo5B1QmC2dCDhaTaW67QJPqnJPKbVmjw7i6KGBi+CDP3JnAKlOX/YUu8uk56JfeQUuGcQg/KaBMGctnchH2n1i3+oGbibhurQj22+OO66Q6mijKoCRP/QJm5WTGcO4c0iVX0B8dIlai/uv/pGU8Wps9MxysH5ViVIj56LWKuNfy1FNO9+wx4iHYnUxfW0Iccq3YJYAlOBkMv8VZIJl9aTEEFJ626LJEw/UbEBdR9CdlwnylPf4RuNW/lcGYMgq3xNVpp/bzYKwZivvQOFgMdX1tv8D33SHvAm3zrRCU+v94BJXnBi9k25fnKsc705IMwv+I6bueXA+mDnagj1oTqsWXZ0WnPAYI3WUO7sBf7XUH230wfYXnTSVhujJxiEP/2oWqi2813gtQkKwz9N2p6yT8Vsh3+BaAwwDvtOZMJKUFcabTBSL6NoouzmzwItJ4mIboT6KXaef66rs8jCU=";

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();

