// 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 search text in all pages in PDF document.

// Include Foxit SDK header files.
#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/pdf/fs_pdfdoc.h"
#include "../../../include/pdf/fs_pdfpage.h"
#include "../../../include/pdf/fs_search.h"

using namespace std;
using namespace foxit;
using namespace foxit::common;
using foxit::common::Library;
using namespace pdf;

static const char* sn = "fpj42y0uOGt8cNp2iJUJdW5LNQGpJDawgJ5niOHYLl+0Emd0mMDhtA==";
static const char* key = "8f3gFsNtvW8N+Vea402IXXgHXIciB/Uw21sIKl1s0+MWvuGiCwz8VVTEwrNsIr1uDlREGorTmUmz7lEC6UcgWUsLrWhG6ezQc5PcgosYOpKsVTaOCyFdxIVCIj7QY9pjCyDgAV1S24QKmBz/ayQSbCepuRh2MBANca27kAN9Frb1qZXYI6Bmrr08YesEz5YKDHFyC04BbHhHiCKBeoQQXzf3O8KST9cFlArWUdIIz2/c3cMjj8uuxFiqVskJrKA1wxvsW4TzZEnDldm9Aye/mZIgHOhIlf8Hg1gIvnA0GCnWhUOs2UpRSFd8YuSnm0UPI7WR1K6vRdgeT/1sYh5bctEzdQh/0bMoN0LpciklqYPwpIFR8HU/zxfleXVwkxWKYkaPW7HxBAE6apzSd18nwRZ8fTsZjcYhTo64q4Jzukyr7m4ODp3tnWgrTgJKnTQYDEEEqK2cRA0ZIsKYTJoQ45gbVMpB7u7wLZ8kAIJbCmXrRPRh72fobCrlh67yJeH98JF8tB7pljmdGX7AD/0IDqtbcXcTUBxLskyaFPQHCuTzzMpLdoCxzGIsqVyobq7E3XKBC7kYcmdH79GwcsHl3BrYma1jTnRobtVOAWg9AgdI2yIhcZJSPrCn7SiQ4+73kbKHpQ2ekoZCF7Cwy6mQ3I+45yg1SLNB0K4YTtoz0t7FMDU1sLf0WsZ4GGlI67WdrOCrYKiXmHK/xjO5+20eR1nSYQlm3H/gyfCLHGdJtEzCB0K/ZwncTJ8ChQTdMHjtU5/rGK7BpDwbIewU2QItE935S9+BHfTQqcSlIvdiDlMtakKkfTCFkr5DUJPCD7XTxoaFN2s8fr9FvQPhhduKufiyl4Ds//jjiT0gRGSN6zDPfrEfJehqsSJgdqDOPlPHDCCj0BK34I8khlWiE8GVxb8qOWUXUfHY9eGLJgMkDiC6UtODXYCwEwNnrWpXGsQjuwtFmUBX7rtNrFr1uVqC1hmvo17a+cnvkRvKMuW9AxiFyak0WMVKT5zvzm9zBsBpAslZP17COOVYIhsDWnmVKt/i5AGFTCwlTlyqxi4RABXOhXHGRpTog+9wV6R2Z9VFjPSLDogFTcO5SkChkmQIKml+anIAnbLjll5Tgwgs7pqsAiCjetSJN0v3fLfcEeyonSnyN3s4VZrxYS1LE4S604EcW0O9gOXAVC6cmZDZkUvSfMh+UVj8pBhF7BSL6qUd5HEfObkL1SZkhh8EHFfe48elqVidbY3Lj9pq52rBnR03oWtUeMNkTYWncYTSzeTTP3jEnlbduUkKQOLVPdXAzPAfQ0FlB31RXm3a1y6VZcgNMsHF/WmPmd6Hwgc48+TUFY7XzLaBlEwcfolyAvFCfVOxO/m0b9pxEfCBYN7lhaMU9+Vjyujj15uCNDSpn4wcs32F2+wfY03pvsv+XMCrYZxvYYDHtw==";

#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

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 TextOutput{
public:
  TextOutput(const String& file_name, const String& fill_mode);
  TextOutput(const WString& file_name, const WString& fill_mode);
  TextOutput() : file_(NULL){};
  ~TextOutput();

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

private:
  FILE* file_;
};

TextOutput::TextOutput(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);
}

TextOutput::TextOutput(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);
}

TextOutput::~TextOutput() {
  if(file_)
    fclose(file_);
  file_ = NULL;
}

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

void TextOutput::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;
  wcout << (FX_LPCWSTR)wide_str;
  int length = utf16le_str.GetLength();
  fwrite((const char*)utf16le_str, sizeof(char), length, file_);
}

void OutputMatchedInfo(TextOutput& text_out, TextSearch search, int matched_index) {
  int page_index = search.GetMatchPageIndex();
  WString format_str = L"";
  format_str.Format(L"Index of matched pattern:\t%d\r\n", matched_index);
  text_out.Write((FX_LPCWSTR)format_str);
  format_str.Format(L"\tpage:\t%d\r\n", page_index);
  text_out.Write((FX_LPCWSTR)format_str);
  format_str.Format(L"\tmatch char start index:\t%d\r\n", search.GetMatchStartCharIndex());
  text_out.Write((FX_LPCWSTR)format_str);
  format_str.Format(L"\tmatch char end index:\t%d\r\n", search.GetMatchEndCharIndex());
  text_out.Write((FX_LPCWSTR)format_str);
  format_str.Format(L"\tmatch sentence start index:\t%d\r\n", search.GetMatchSentenceStartIndex());
  text_out.Write((FX_LPCWSTR)format_str);
  format_str = L"\tmatch sentence:\t";
  format_str += search.GetMatchSentence();
  format_str += L"\r\n";
  text_out.Write((FX_LPCWSTR)format_str);
  RectFArray rect_array = search.GetMatchRects();
  int rect_count = rect_array.GetSize();
  format_str.Format(L"\tmatch rectangles count:\t%d\r\n", rect_count);
  text_out.Write((FX_LPCWSTR)format_str);
  for (int i = 0; i < rect_count; i++) {
    foxit::RectF rect = rect_array[i];
    format_str.Format(L"\trectangle(in PDF space) :%d\t[left = %.4f, bottom = %.4f, right = %.4f, top = %.4f]\r\n", i,
      rect.left, rect.bottom, rect.right, rect.top);
    text_out.Write((FX_LPCWSTR)format_str);
  }
}

int main(int argc, char *argv[])
{
  WString output_directory = output_path + L"search/";
#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";
  WString output_file = output_directory + L"search.txt";
  SdkLibMgr sdk_lib_mgr;
  // Initialize library
  ErrorCode error_code = sdk_lib_mgr.Initialize();
  if (error_code != foxit::e_ErrSuccess) {
    return 1;
  }

  try {
    PDFDoc doc(input_file);
    ErrorCode error_code = doc.Load();
    if (error_code != foxit::e_ErrSuccess) {
      printf("The Doc [%s] Error: %d\n", (const char*)String::FromUnicode(input_file), error_code);
      return 1;
    }
    TextOutput text_out(output_file, L"wb+");

    // sample 1: search for all pages of doc.
    TextSearch search(doc, NULL);

    int start_index = 0, end_index = doc.GetPageCount() - 1;
    search.SetStartPage(0);
    search.SetEndPage(doc.GetPageCount() - 1);

    WString pattern = L"Foxit";
    search.SetPattern(pattern);

    foxit::uint32 flags = TextSearch::e_SearchNormal;
    // If want to specify flags, you can do as followings:
    // flags |= TextSearch::e_SearchMatchCase;
    // flags |= TextSearch::e_SearchMatchWholeWord;
    // flags |= TextSearch::e_SearchConsecutive;
    search.SetSearchFlags(flags);
    WString format_str = L"";
    format_str = L"Begin search ";
    format_str += pattern;
    format_str += L" at ";
    format_str += input_file;
    format_str += L".\n";
    text_out.Write((FX_LPCWSTR)format_str);
    format_str.Format(L"Start index:\t%d\r\n", start_index);
    text_out.Write((FX_LPCWSTR)format_str);
    format_str.Format(L"End index:\t%d\r\n", end_index);
    text_out.Write((FX_LPCWSTR)format_str);
    format_str = L"Match key:\t";
    format_str += pattern;
    format_str += L"\r\n";
    text_out.Write((FX_LPCWSTR)format_str);
    WString match_case = flags & TextSearch::e_SearchMatchCase ? L"Yes" : L"No";
    format_str = L"Match Case\t";
    format_str += match_case;
    format_str += L"\r\n";
    text_out.Write((FX_LPCWSTR)format_str);
    WString match_whole_word = flags & TextSearch::e_SearchMatchWholeWord ? L"Yes" : L"No";
    format_str = L"Match whole word:\t";
    format_str += match_whole_word;
    format_str += L"\r\n";
    text_out.Write((FX_LPCWSTR)format_str);
    WString match_consecutive = flags & TextSearch::e_SearchConsecutive ? L"Yes" : L"No";
    format_str = L"Consecutive:\t";
    format_str += match_consecutive;
    format_str += L"\r\n";
    text_out.Write((FX_LPCWSTR)format_str);
    int match_count = 0;
    while (search.FindNext()) {
      RectFArray rect_array = search.GetMatchRects();
      OutputMatchedInfo(text_out, search, match_count);
      match_count ++;
    }
    wcout << L"Marched " << match_count << L" counts." << endl;
    // end of sample 1.

    wcout << L"Search demo finished." << endl;

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

  return err_ret;
}

