/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/.
 */

#ifndef _nsZipHeader_h_
#define _nsZipHeader_h_

#include "nsString.h"
#include "nsIOutputStream.h"
#include "nsIInputStream.h"
#include "nsIZipReader.h"
#include "mozilla/Attributes.h"
#include "mozilla/UniquePtr.h"

// High word is S_IFREG, low word is DOS file attribute
#define ZIP_ATTRS_FILE 0x80000000
// High word is S_IFDIR, low word is DOS dir attribute
#define ZIP_ATTRS_DIRECTORY 0x40000010
#define PERMISSIONS_FILE 0644
#define PERMISSIONS_DIR 0755

// Combine file type attributes with unix style permissions
#define ZIP_ATTRS(p, a) ((p & 0xfff) << 16) | a

class nsZipHeader final : public nsIZipEntry
{
    ~nsZipHeader()
    {
        mExtraField = nullptr;
        mLocalExtraField = nullptr;
    }

public:
    NS_DECL_ISUPPORTS
    NS_DECL_NSIZIPENTRY

    nsZipHeader() :
        mCRC(0),
        mCSize(0),
        mUSize(0),
        mEAttr(0),
        mOffset(0),
        mFieldLength(0),
        mLocalFieldLength(0),
        mVersionMade(0x0300 + 23), // Generated on Unix by v2.3 (matches infozip)
        mVersionNeeded(20), // Requires v2.0 to extract
        mFlags(0),
        mMethod(0),
        mTime(0),
        mDate(0),
        mDisk(0),
        mIAttr(0),
        mInited(false),
        mWriteOnClose(false),
        mExtraField(nullptr),
        mLocalExtraField(nullptr)
    {
    }

    uint32_t mCRC;
    uint32_t mCSize;
    uint32_t mUSize;
    uint32_t mEAttr;
    uint32_t mOffset;
    uint32_t mFieldLength;
    uint32_t mLocalFieldLength;
    uint16_t mVersionMade;
    uint16_t mVersionNeeded;
    uint16_t mFlags;
    uint16_t mMethod;
    uint16_t mTime;
    uint16_t mDate;
    uint16_t mDisk;
    uint16_t mIAttr;
    bool mInited;
    bool mWriteOnClose;
    nsCString mName;
    nsCString mComment;
    mozilla::UniquePtr<uint8_t[]> mExtraField;
    mozilla::UniquePtr<uint8_t[]> mLocalExtraField;

    void Init(const nsACString & aPath, PRTime aDate, uint32_t aAttr,
              uint32_t aOffset);
    uint32_t GetFileHeaderLength();
    nsresult WriteFileHeader(nsIOutputStream *aStream);
    uint32_t GetCDSHeaderLength();
    nsresult WriteCDSHeader(nsIOutputStream *aStream);
    nsresult ReadCDSHeader(nsIInputStream *aStream);
    const uint8_t * GetExtraField(uint16_t aTag, bool aLocal, uint16_t *aBlockSize);
    nsresult PadExtraField(uint32_t aOffset, uint16_t aAlignSize);
};

#endif