/* 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;

  nsresult 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