#pragma once

#include "common/mach-o.h"
#include "openssl.h"
#include <set>

class ZArchO {
public:
    ZArchO();

    bool Init(uint8_t *pBase, uint32_t uLength);

public:
    bool Sign(ZSignAsset *pSignAsset, bool bForce, const string &strBundleId, const string &strInfoPlistSHA1,
              const string &strInfoPlistSHA256, const string &strCodeResourcesData);

    void PrintInfo();

    bool IsExecute();

    bool RemoveDyLib(const char *szDyLibName, bool showLog);

    bool InjectDyLib(bool bWeakInject, const char *szDyLibPath, bool &bCreate, bool showLog);

    uint32_t ReallocCodeSignSpace(const string &strNewFile);

    void uninstallDylibs(set<string> dylibNames);

private:
    uint32_t BO(uint32_t uVal);

    const char *GetFileType(uint32_t uFileType);

    const char *GetArch(int cpuType, int cpuSubType);

    bool
    BuildCodeSignature(ZSignAsset *pSignAsset, bool bForce, const string &strBundleId, const string &strInfoPlistSHA1,
                       const string &strInfoPlistSHA256, const string &strCodeResourcesSHA1,
                       const string &strCodeResourcesSHA256, string &strOutput);

public:
    uint8_t *m_pBase;
    uint32_t m_uLength;
    uint32_t m_uCodeLength;
    uint8_t *m_pSignBase;
    uint32_t m_uSignLength;
    string m_strInfoPlist;
    bool m_bEncrypted;
    bool m_b64;
    bool m_bBigEndian;
    bool m_bEnoughSpace;
    uint8_t *m_pCodeSignSegment;
    uint8_t *m_pLinkEditSegment;
    uint32_t m_uLoadCommandsFreeSpace;
    mach_header *m_pHeader;
    uint32_t m_uHeaderSize;
};