// 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/set page label information
// in a PDF document.
#include <time.h>
#include<direct.h>
#include<stdio.h>

#include "../../../include/fs_basictypes_c.h"
#include "../../../include/fs_common_c.h"
#include "../../../include/fs_pdfdoc_c.h"
#include "../../../include/fs_pdfgraphicsobject_c.h"
#include "../../../include/fs_pdfpage_c.h"
#include "../../../include/fs_pdfpagelabel_c.h"

#ifndef TRUE
#define TRUE 1
#endif

#ifndef FALSE
#define FALSE 0
#endif

#ifndef NULL
#define NULL (void*)0
#endif

#define MAX_FILE_PATH 260
#define MAX_FILE_BUFFER 1024

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

static const wchar_t* output_path = L"../output_files/";
static const wchar_t* input_path = L"../input_files/";

wchar_t* StyleCodeToString(FSPDFPageLabelStyle style) {
  switch (style) {
  case e_FSNone: {
    return L"None";
  }
  case e_FSDecimalNums: {
    return L"DecimalNums";
  }
  case e_FSUpperRomanNums: {
    return L"UpperRomanNums";
  }
  case e_FSLowerRomanNums: {
    return L"LowerRomanNums";
  }
  case e_FSUpperLetters: {
    return L"UpperLetters";
  }
  case e_FSLowerLetters: {
    return L"LowerLetters";
  }
  }
  return L"";
}

void WriteUnicode(FILE* file, const wchar_t* text_content) {
  FS_WSTR str;
  FS_BSTR dst_bstr;

  str.str = (wchar_t*)text_content;
  str.len = wcslen(text_content);
  FSDK_WStr_UTF16LEEncode(&str, &dst_bstr);

  if (dst_bstr.len == 0) return;
  fwrite((const char*)dst_bstr.str, sizeof(char), dst_bstr.len, file);
}

void ShowPageLabelsInfo(FS_PDFDOC_HANDLE doc, const wchar_t* text_doc_path) {
  FS_PAGELABELS_HANDLE page_labels;
  int count, i = 0;
  FS_WSTR return_title;
  FS_WSTR return_prefix;
  FSPDFPageLabelStyle return_style;
  FILE* file = NULL;

  _wfopen_s(&file, text_doc_path, L"w+b");

  FSDK_PageLabels_Create(doc, &page_labels);
  FSDK_PDFDoc_GetPageCount(doc, &count);

  for (i = 0; i < count; i++) {
    wchar_t file_buffer[MAX_FILE_BUFFER];
    FSDK_PageLabels_GetPageLabelStyle(page_labels, i, &return_style);
    FSDK_WStr_Init(&return_title);
    FSDK_PageLabels_GetPageLabelTitle(page_labels, i, &return_title);
    FSDK_WStr_Init(&return_prefix);
    FSDK_PageLabels_GetPageLabelPrefix(page_labels, i, &return_prefix);
    swprintf_s(file_buffer, MAX_FILE_BUFFER, L"page index: %d\tstyle: %ls\ttitle: %ls\tprefix: %ls\r\n", i,
      StyleCodeToString(return_style), return_title.str, return_prefix.str);
    WriteUnicode(file, file_buffer);
  }

  FSDK_PageLabels_Release(page_labels);
  fclose(file);
}

int main(int argc, char *argv[]) {
  FSErrorCode error_code;
  int err_ret = 0;
  wchar_t output_directory[MAX_FILE_PATH];
  wchar_t input_file[MAX_FILE_PATH];
  wchar_t text_path[MAX_FILE_PATH];
  wchar_t text_path_new[MAX_FILE_PATH];
  FS_PDFDOC_HANDLE doc = NULL;
  FS_PAGELABELS_HANDLE page_labels = NULL;
  wchar_t pdf_removeall[MAX_FILE_PATH];
  FS_BOOL return_result = FALSE;
  int count;
  wchar_t pdf_new[MAX_FILE_PATH];

  swprintf_s(output_directory, MAX_FILE_PATH, L"%lspage_labels/", output_path);
  _wmkdir(output_directory);

  error_code = FSDK_Library_Initialize(sn, key);
  if (error_code != e_FSErrSuccess) {
    printf("Library Initialize Error: %d\n", error_code);
    return 1;
  }
  swprintf_s(input_file, MAX_FILE_PATH, L"%lspage_labels.pdf", input_path);

  error_code = FSDK_Library_Initialize(sn, key);
  if (error_code != e_FSErrSuccess) {
    printf("Library Initialize Error: %d\n", error_code);
    return 1;
  }

  error_code = FSDK_PDFDoc_Create0(input_file, &doc);
  if (error_code != e_FSErrSuccess) return 1;
  error_code = FSDK_PDFDoc_Load(doc, NULL);
  if (error_code != e_FSErrSuccess) {
    FSDK_PDFDoc_Release(doc);
    wprintf(L"The Doc [%ls] Error: %d\n", input_file, error_code);
    return 1;
  }
  swprintf_s(text_path, MAX_FILE_PATH, L"%lspage_labels_original_page_labels_info.txt", output_directory);

  ShowPageLabelsInfo(doc, text_path);
  // Add new page labels

  FSDK_PageLabels_Create(doc, &page_labels);
  FSDK_PageLabels_RemoveAll(page_labels);
  swprintf_s(pdf_removeall, MAX_FILE_PATH, L"%lsremoveall_page_labels.pdf", output_directory);

  error_code = FSDK_PDFDoc_SaveAs(doc, pdf_removeall, e_FSSaveFlagNoOriginal, &return_result);
  if (error_code != e_FSErrSuccess) {
    printf("Save doc error!\n");
    FSDK_PDFDoc_Release(doc);
    return 1;
  }

  FSDK_PDFDoc_GetPageCount(doc, &count);
  if (count > 0) {
    FSDK_PageLabels_SetPageLabel(page_labels, 0, e_FSUpperRomanNums, 1, L"Cover-");
  }
  if (count > 1) {
    FSDK_PageLabels_SetPageLabel(page_labels, 1, e_FSLowerRomanNums, 1, L"Catolog-");
  }
  if (count > 2) {
    FSDK_PageLabels_SetPageLabel(page_labels, 2, e_FSDecimalNums, 1, L"Contents-");
  }
  if (count > 5) {
    FSDK_PageLabels_SetPageLabel(page_labels, count - 1, e_FSUpperLetters, 1, L"Appendix-");
  }

  swprintf_s(text_path_new, MAX_FILE_PATH, L"%lspage_labels_new_page_labels_info.txt", output_directory);

  ShowPageLabelsInfo(doc, text_path_new);
  swprintf_s(pdf_new, MAX_FILE_PATH, L"%lsnew_page_labels.pdf", output_directory);

  error_code = FSDK_PDFDoc_SaveAs(doc, pdf_new, e_FSSaveFlagNoOriginal, &return_result);
  if (error_code != e_FSErrSuccess) {
    printf("Save doc error!\n");
    FSDK_PDFDoc_Release(doc);
    return 1;
  }

  FSDK_PageLabels_Release(page_labels);
  FSDK_PDFDoc_Release(doc);
  FSDK_Library_Release();
  printf("Page label test.\n");

  return err_ret;
}

