// 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 operate attachments in OFD documents.

// Include Foxit SDK header files.
#include <iostream>
#if defined(_WIN32) || defined(_WIN64)
#include <direct.h>
#else
#include <sys/stat.h>
#endif

#include "common/fs_common.h"
#include "common/fs_basictypes.h"
#include "ofd/fs_ofdpackage.h"
#include "ofd/fs_ofddoc.h"
#include "ofd/fs_ofdattachment.h"

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

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

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

  // Enumerate attachments in OFD document
  try {
    cout << "Start: Enumerate attachments in OFD document." << endl;
    {
      WString input_file = input_path + L"ofd/attachment.ofd";

      OFDPackage pkg(input_file);
      if (pkg.IsEmpty()) {
        cout << "Failed to open OFD package." << endl;
        return 1;
      }

      OFDDoc doc = pkg.LoadDocument(0, "");
      if (doc.IsEmpty()) {
        cout << "Failed to load OFD document." << endl;
        return 1;
      }

      Attachments attachments = doc.GetAttachments();
      if (attachments.IsEmpty()) {
        cout << "No attachments found in document." << endl;
      } else {
        int count = attachments.GetAttachmentCount();
        cout << "Attachment count: " << count << endl;

        for (int i = 0; i < count; i++) {
          Attachment att = attachments.GetAttachment(i);
          if (att.IsEmpty()) continue;

          AttachmentInfo info = att.GetAttachmentInfo();
          cout << "Attachment[" << i << "]:" << endl;
          cout << "  Name: " << (FX_LPCSTR)info.name << endl;
          cout << "  Format: " << (FX_LPCSTR)info.format << endl;
          cout << "  Usage: " << (FX_LPCSTR)info.usage << endl;
          cout << "  ID: " << att.GetID() << endl;
          cout << "  Visible: " << (att.IsVisible() ? "true" : "false") << endl;
          cout << "  Deletable: " << (att.IsDeletable() ? "true" : "false") << endl;
          cout << "  Editable: " << (att.IsEditable() ? "true" : "false") << endl;
          cout << "  Exportable: " << (att.IsExportable() ? "true" : "false") << endl;

          String modifier = att.GetModifier();
          if (!modifier.IsEmpty()) {
            cout << "  Modifier: " << (FX_LPCSTR)modifier << endl;
          }

          String description = att.GetDescription();
          if (!description.IsEmpty()) {
            cout << "  Description: " << (FX_LPCSTR)description << endl;
          }
        }
      }
    }
    cout << "End: Enumerate attachments." << endl;
  } catch (const Exception& e) {
    cout << e.GetMessage() << endl;
    err_ret = 1;
  }

  // Export attachment to file
  try {
    cout << "Start: Export attachment to file." << endl;
    {
      WString input_file = input_path + L"sample.ofd";

      OFDPackage pkg(input_file);
      if (pkg.IsEmpty()) {
        cout << "Failed to open OFD package." << endl;
        err_ret = 1;
      } else {
        OFDDoc doc = pkg.LoadDocument(0, "");
        if (doc.IsEmpty()) {
          cout << "Failed to load OFD document." << endl;
          err_ret = 1;
        } else {
          Attachments attachments = doc.GetAttachments();
          if (!attachments.IsEmpty()) {
            int count = attachments.GetAttachmentCount();
            if (count > 0) {
              Attachment att = attachments.GetAttachment(0);
              AttachmentInfo info = att.GetAttachmentInfo();

              WString export_file = output_directory + L"exported_" + WString::FromLocal(info.name);
              bool result = att.GetFile(export_file);
              if (result) {
                cout << "Attachment exported successfully: " << (FX_LPCSTR)info.name << endl;
              } else {
                cout << "Failed to export attachment." << endl;
                err_ret = 1;
              }
            } else {
              cout << "No attachments to export." << endl;
            }
          } else {
            cout << "No attachments collection in document." << endl;
          }
        }
      }
    }
    cout << "End: Export attachment." << endl;
  } catch (const Exception& e) {
    cout << e.GetMessage() << endl;
    err_ret = 1;
  }

  // Add attachment to OFD document
  try {
    cout << "Start: Add attachment to OFD document." << endl;
    {
      WString input_file = input_path + L"sample.ofd";
      WString output_file = output_directory + L"add_attachment.ofd";

      OFDPackage pkg(input_file);
      if (pkg.IsEmpty()) {
        cout << "Failed to open OFD package." << endl;
        err_ret = 1;
      } else {
        OFDDoc doc = pkg.LoadDocument(0, "");
        if (doc.IsEmpty()) {
          cout << "Failed to load OFD document." << endl;
          err_ret = 1;
        } else {
          Attachments attachments = doc.GetAttachments();
          if (attachments.IsEmpty()) {
            attachments = doc.CreateAttachments();
          }

          Attachment new_att = attachments.AddAttachment();
          if (new_att.IsEmpty()) {
            cout << "Failed to add attachment." << endl;
            err_ret = 1;
          } else {
            // Set attachment info
            DateTime creation_date;
            creation_date.year = 2026;
            creation_date.month = 4;
            creation_date.day = 17;
            creation_date.hour = 10;
            creation_date.minute = 0;
            creation_date.second = 0;

            DateTime mod_date = creation_date;

            new_att.SetName("sample_attachment.txt");
            new_att.SetFormat("txt");
            new_att.SetUsage("test");
            new_att.SetCreationDate(creation_date);
            new_att.SetModDate(mod_date);
            new_att.SetModifier("Foxit SDK Example");
            new_att.SetDescription("This is a sample attachment added by OFD attachment example.");

            // Set attachment properties
            new_att.EnableVisible(true);
            new_att.EnableDeletable(true);
            new_att.EnableEditable(true);
            new_att.EnableExportable(true);

            // Set file content from an existing file
            WString att_file = input_path + L"AboutFoxit.pdf";
            bool result = new_att.SetFile(att_file, "AboutFoxit.pdf");
            if (result) {
              cout << "Attachment file set successfully." << endl;
            } else {
              cout << "Failed to set attachment file." << endl;
              err_ret = 1;
            }

            cout << "New attachment added, ID: " << new_att.GetID() << endl;
          }

          pkg.SaveAs(output_file);
          cout << "Output file: add_attachment.ofd" << endl;
        }
      }
    }
    cout << "End: Add attachment." << endl;
  } catch (const Exception& e) {
    cout << e.GetMessage() << endl;
    err_ret = 1;
  }

  // Remove attachment from OFD document
  try {
    cout << "Start: Remove attachment from OFD document." << endl;
    {
      WString input_file = input_path + L"sample.ofd";
      WString output_file = output_directory + L"remove_attachment.ofd";

      OFDPackage pkg(input_file);
      if (pkg.IsEmpty()) {
        cout << "Failed to open OFD package." << endl;
        err_ret = 1;
      } else {
        OFDDoc doc = pkg.LoadDocument(0, "");
        if (doc.IsEmpty()) {
          cout << "Failed to load OFD document." << endl;
          err_ret = 1;
        } else {
          Attachments attachments = doc.GetAttachments();
          if (!attachments.IsEmpty()) {
            int count = attachments.GetAttachmentCount();
            cout << "Attachment count before removal: " << count << endl;

            if (count > 0) {
              bool result = attachments.RemoveAttachment(0);
              if (result) {
                cout << "First attachment removed successfully." << endl;
                cout << "Attachment count after removal: " << attachments.GetAttachmentCount() << endl;
              } else {
                cout << "Failed to remove attachment." << endl;
                err_ret = 1;
              }
            }
          } else {
            cout << "No attachments to remove." << endl;
          }

          pkg.SaveAs(output_file);
          cout << "Output file: remove_attachment.ofd" << endl;
        }
      }
    }
    cout << "End: Remove attachment." << endl;
  } catch (const Exception& e) {
    cout << e.GetMessage() << endl;
    err_ret = 1;
  }

  return err_ret;
}
