// 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 get preflight keys and analyze or fixup PDF file.

#include <time.h>
#include <iostream>
#if defined(_WIN32) || defined(_WIN64)
#include<direct.h>
#else
#include <sys/stat.h>
#endif

#include "../../../include/common/fs_common.h"
#include "../../../include/addon/compliance/fs_compliancecommon.h"
#include "../../../include/addon/compliance/fs_pdfa.h"
#include "../../../include/addon/compliance/fs_pdfcompliance.h"

using namespace std;
using namespace foxit;
using namespace foxit::common;
using foxit::common::Library;
using namespace addon;
using namespace compliance;

static const char* sn = "Rew1vB0TqoMIxX444UcpScARWdhgJpsJfzmyMLR32a0+EgrCjp2NaQ==";
static const char* 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=";

#if defined(_WIN32) || defined(_WIN64)
static WString output_path = WString::FromLocal("../output_files/");
static WString input_path = WString::FromLocal("../input_files/");
#else
static WString output_path = WString::FromLocal("./output_files/");
static WString input_path = WString::FromLocal("./input_files/");
#endif
static WString output_directory = output_path + L"preflight/";

class SdkLibMgr {
public:
  SdkLibMgr() : is_initialize_(false){};
  ErrorCode Initialize() {
    ErrorCode error_code = Library::Initialize(sn, key);
    if (error_code != foxit::e_ErrSuccess) {
      printf("Library Initialize Error: %d\n", error_code);
    } else {
      is_initialize_ = true;
    }
    return error_code;

  }
  ~SdkLibMgr(){
    if(is_initialize_)
      Library::Release();
  }
private:
  bool is_initialize_;
};

class TextDoc {
public:
  TextDoc(const String& file_name, const String& fill_mode);
  TextDoc(const WString& file_name, const WString& fill_mode);
  ~TextDoc();

  void Write(const char* text_content);
  void Write(const wchar_t* text_content);

private:
  FILE* file_;
};

TextDoc::TextDoc(const String& file_name, const String& file_mode) throw(Exception) : file_(NULL) {
#if defined(_WIN32) || defined(_WIN64)
  fopen_s(&file_, file_name, (const char*)file_mode);
#else
  file_ = fopen((const char*)file_name, (const char*)file_mode);
#endif

  if (!file_)
    throw Exception(__FILE__, __LINE__, __FUNCTION__, foxit::e_ErrFile);

  uint8 temp[] = {0xFF,0xFE};
  fwrite(temp, sizeof(uint8), 2, file_);
  fseek(file_, 0, SEEK_END);

}

TextDoc::TextDoc(const WString& file_name, const WString& file_mode) throw(Exception)  : file_(NULL) {
  String s_file_name = String::FromUnicode(file_name);
  String s_file_mode = String::FromUnicode(file_mode);

#if defined(_WIN32) || defined(_WIN64)
  fopen_s(&file_, (const char*)s_file_name, (const char*)s_file_mode);
#else
  file_ = fopen((const char*)s_file_name, (const char*)s_file_mode);
#endif
  if (!file_)
    throw Exception(__FILE__, __LINE__, __FUNCTION__, foxit::e_ErrFile);

  uint8 temp[] = {0xFF,0xFE};
  fwrite(temp, sizeof(uint8), 2, file_);
  fseek(file_, 0, SEEK_END);

}

TextDoc::~TextDoc() {
  fclose(file_);
  file_ = NULL;
}

void TextDoc::Write(const char* text_content) {
  WString wide_str = WString::FromLocal(text_content);
  Write(wide_str);
}

void TextDoc::Write(const wchar_t* text_content) {
  WString wide_str(text_content);
  if (wide_str.IsEmpty()) return ;
  String utf16le_str = wide_str.UTF16LE_Encode(false);
  if (utf16le_str.IsEmpty()) return;
  int length = utf16le_str.GetLength();
  fwrite((const char*)utf16le_str, sizeof(char), length, file_);
}

class MyComplianceProgressCallback : public ProgressCallback {
public:
  MyComplianceProgressCallback(const WString& output_txt_file_path);
  ~MyComplianceProgressCallback();

  virtual void Release();
  virtual void UpdateCurrentStateData(int current_rate, const WString& current_state_string);

private:
  TextDoc text_doc_;
};

MyComplianceProgressCallback::MyComplianceProgressCallback(const WString& output_txt_file_path)
  : text_doc_(output_txt_file_path, L"w+b") {}

MyComplianceProgressCallback::~MyComplianceProgressCallback() {}

void MyComplianceProgressCallback::Release() {

}

void MyComplianceProgressCallback::UpdateCurrentStateData(int current_rate, const WString& current_state_string) {
  std::cout << "Current rate:" << current_rate << std::endl;

  char temp_string[32];
#if defined(_WIN32) || defined(_WIN64)
  sprintf_s(temp_string, 32, "Current rate:%d, state str:", current_rate);
#else
  sprintf(temp_string, "Current rate:%d, state str:", current_rate);
#endif  // defined(_WIN32) || defined(_WIN64)
  text_doc_.Write(temp_string);
  text_doc_.Write(current_state_string);
  text_doc_.Write("\r\n");
}

void ListPreflightKeys(Preflight preflight, const PreflightSettingData::LibraryType& library_type, const PreflightSettingData::OperateType& operate_type) {
  StringArray group_arr = preflight.GetGroupNamesArray(library_type, operate_type);
  for (int i = 0; i < group_arr.GetSize(); i++) {
    cout << "Group name: " << group_arr[i] << endl;
    StringArray item_array = preflight.GetItemKeysArray(group_arr[i]);
    cout << "    Item count: " << item_array.GetSize() << endl;
    for (int j = 0; j < item_array.GetSize(); j++) {
      cout << "    Item name: " << item_array[j] << endl;
    }
  }
}

void AnalyzeAndFixup(Preflight preflight, const PreflightSettingData::OperateType& operate_type, const WString& preflight_key, bool is_fixed) {
  PreflightSettingData setting_data;
  setting_data.operate_type = operate_type;
  setting_data.preflight_key = preflight_key;
  setting_data.first_page_index = 0;
  setting_data.src_pdf_path = input_path + L"AboutFoxit.pdf";
  WString fixpath = output_directory + L"AboutFoxit_" + preflight_key + L"_fixed.pdf";
  setting_data.saved_pdf_path = fixpath;
  MyComplianceProgressCallback* progress_callback = new MyComplianceProgressCallback(output_directory + L"AboutFoxit_" + preflight_key + L"_progress.txt");
  if(is_fixed)
    preflight.AnalyzeAndFixup(setting_data, progress_callback);
  else
    preflight.Analyze(setting_data, progress_callback);
}

void GenerateReport(Preflight preflight, const PreflightReportSettingData::ReportFileFormatType& format_type, const WString& preflight_key) {
  PreflightReportSettingData report_setting_data;
  report_setting_data.report_file_format_type = format_type;
  WString path = output_directory + L"AboutFoxit_" + preflight_key + L"_report.pdf";
  report_setting_data.report_file_path = path;
  report_setting_data.to_generate_overview = true;
  report_setting_data.to_highlight_problems = true;
  report_setting_data.problems_highlight_method = PreflightReportSettingData::e_ProblemsHighlightMethodTransparentMasks;
  preflight.GenerateReport(report_setting_data, NULL);
}

int main(int argc, char *argv[])
{
  WString output_directory = output_path + L"preflight/";
#if defined(_WIN32) || defined(_WIN64)
  _mkdir(String::FromUnicode(output_directory));
#else
  mkdir(String::FromUnicode(output_directory), S_IRWXU | S_IRWXG | S_IROTH | S_IXOTH);
#endif

  int err_ret = 0;
  WString input_file = input_path + L"AboutFoxit.pdf";
  SdkLibMgr sdk_lib_mgr;
  // Initialize library
  ErrorCode error_code = sdk_lib_mgr.Initialize();
  if (error_code != foxit::e_ErrSuccess) {
    return 1;
  }

  try {
    // "compliance_resource_folder_path" is the path of compliance resource folder. Please refer to Developer Guide for more details.
    WString compliance_resource_folder_path = L"";
    // If you use an authorization key for Foxit PDF SDK, please set a valid unlock code string to compliance_engine_unlockcode for ComplianceEngine.
    // If you use an trial key for Foxit PDF SDK, just keep compliance_engine_unlockcode as an empty string.
    const char* compliance_engine_unlockcode = "";

    if (compliance_resource_folder_path.IsEmpty()) {
      std::cout << "compliance_resource_folder_path is still empty. Please set it with a valid path to compliance resource folder path." << std::endl;
      return 1;
    }
    // Initialize compliance engine.
    ErrorCode error_code = ComplianceEngine::Initialize(compliance_resource_folder_path, compliance_engine_unlockcode);
    if (error_code != foxit::e_ErrSuccess) {
      switch (error_code) {
      case foxit::e_ErrInvalidLicense:
        printf("[Failed] Compliance module is not contained in current Foxit PDF SDK keys.\n");
        break;
      default:
        printf("[Failed] Fail to initialize compliance engine. Error: %d\n", error_code);
        break;
      }
      return 1;
    }
    cout << "ComplianceEngine is initialized." << endl;
    {
      Preflight preflight;
      // List all Foxit preflight keys.
      cout << "Library Type: Foxit, Operate Type: Profiles" << endl;
      ListPreflightKeys(preflight, PreflightSettingData::e_LibraryTypeFoxit, PreflightSettingData::e_OperateTypeProfiles);
      cout << "Library Type: Foxit, Operate Type: Checks" << endl;
      ListPreflightKeys(preflight, PreflightSettingData::e_LibraryTypeFoxit, PreflightSettingData::e_OperateTypeSingleChecks);
      cout << "Library Type: Foxit, Operate Type: Fixups" << endl;
      ListPreflightKeys(preflight, PreflightSettingData::e_LibraryTypeFoxit, PreflightSettingData::e_OperateTypeSingleFixups);
      
      bool is_fixup = false;
      // Analyze and fixup a PDF file.
      cout << "== Profiles. ==" << endl;
      is_fixup = preflight.CanFixup("pppp_ConverttoPDFA1a");
      AnalyzeAndFixup(preflight, PreflightSettingData::e_OperateTypeProfiles, L"pppp_ConverttoPDFA1a", is_fixup);
      GenerateReport(preflight, PreflightReportSettingData::e_ReportFileFormatTypePdf, L"pppp_ConverttoPDFA1a");
    
      // Analyze a PDF file.
      cout << "== Single Checks.==" << endl;
      AnalyzeAndFixup(preflight, PreflightSettingData::e_OperateTypeSingleChecks, L"R_6_Annotationinsidepagearea", false);
      GenerateReport(preflight, PreflightReportSettingData::e_ReportFileFormatTypePdf, L"R_6_Annotationinsidepagearea");
    
      //Fixup a PDF file.
      cout << "== Single Fixups. ==" << endl;
      AnalyzeAndFixup(preflight, PreflightSettingData::e_OperateTypeSingleFixups, L"F_9_RemoveXMPMetadataifnotcompliantwithPDFA1", true);
      GenerateReport(preflight, PreflightReportSettingData::e_ReportFileFormatTypePdf, L"F_9_RemoveXMPMetadataifnotcompliantwithPDFA1");
    }
    // Release compliance engine.
    ComplianceEngine::Release();

    cout << "== End: Preflight demo. ==" << endl;

  } catch (const Exception& e) {
    cout << e.GetMessage() << endl;
    err_ret = 1;
  }
  catch(...)
  {
    cout << "Unknown Exception" << endl;
    err_ret = 1;
  }

  return err_ret;
}

