/* 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 _nsZipWriter_h_
#define _nsZipWriter_h_

#include "nsIZipWriter.h"
#include "nsIRequestObserver.h"
#include "nsZipHeader.h"
#include "nsCOMPtr.h"
#include "nsCOMArray.h"
#include "nsTArray.h"
#include "nsTHashMap.h"
#include "mozilla/Attributes.h"

#define ZIPWRITER_CONTRACTID "@mozilla.org/zipwriter;1"
#define ZIPWRITER_CID                                \
  {                                                  \
    0x430d416c, 0xa722, 0x4ad1, {                    \
      0xbe, 0x98, 0xd9, 0xa4, 0x45, 0xf8, 0x5e, 0x3f \
    }                                                \
  }

#define OPERATION_ADD 0
#define OPERATION_REMOVE 1
struct nsZipQueueItem {
 public:
  uint32_t mOperation;
  nsCString mZipEntry;
  nsCOMPtr<nsIFile> mFile;
  nsCOMPtr<nsIChannel> mChannel;
  nsCOMPtr<nsIInputStream> mStream;
  PRTime mModTime;
  int32_t mCompression;
  uint32_t mPermissions;
};

class nsZipWriter final : public nsIZipWriter, public nsIRequestObserver {
 public:
  NS_DECL_ISUPPORTS
  NS_DECL_NSIZIPWRITER
  NS_DECL_NSIREQUESTOBSERVER

  nsZipWriter();
  nsresult EntryCompleteCallback(nsZipHeader* aHeader, nsresult aStatus);

 private:
  ~nsZipWriter();

  uint32_t mCDSOffset;
  bool mCDSDirty;
  bool mInQueue;

  nsCOMPtr<nsIFile> mFile;
  nsCOMPtr<nsIRequestObserver> mProcessObserver;
  nsCOMPtr<nsISupports> mProcessContext;
  nsCOMPtr<nsIOutputStream> mStream;
  nsCOMArray<nsZipHeader> mHeaders;
  nsTArray<nsZipQueueItem> mQueue;
  nsTHashMap<nsCStringHashKey, int32_t> mEntryHash;
  nsCString mComment;

  nsresult SeekCDS();
  void Cleanup();
  nsresult ReadFile(nsIFile* aFile);
  nsresult InternalAddEntryDirectory(const nsACString& aZipEntry,
                                     PRTime aModTime, uint32_t aPermissions);
  nsresult BeginProcessingAddition(nsZipQueueItem* aItem, bool* complete);
  nsresult BeginProcessingRemoval(int32_t aPos);
  nsresult AddEntryStream(const nsACString& aZipEntry, PRTime aModTime,
                          int32_t aCompression, nsIInputStream* aStream,
                          bool aQueue, uint32_t aPermissions);
  void BeginProcessingNextItem();
  void FinishQueue(nsresult aStatus);
};

#endif