1 files changed, 79 insertions, 0 deletions
diff --git a/modules/libjar/zipwriter/nsZipWriter.h b/modules/libjar/zipwriter/nsZipWriter.h
new file mode 100644
index 0000000000..8ba04c72c7
--- /dev/null
+++ b/modules/libjar/zipwriter/nsZipWriter.h
@@ -0,0 +1,79 @@
+/* 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