1 : /* ***** BEGIN LICENSE BLOCK *****
2 : * Version: MPL 1.1/GPL 2.0/LGPL 2.1
3 : *
4 : * The contents of this file are subject to the Mozilla Public License Version
5 : * 1.1 (the "License"); you may not use this file except in compliance with
6 : * the License. You may obtain a copy of the License at
7 : * http://www.mozilla.org/MPL/
8 : *
9 : * Software distributed under the License is distributed on an "AS IS" basis,
10 : * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
11 : * for the specific language governing rights and limitations under the
12 : * License.
13 : *
14 : * The Original Code is Zip Writer Component.
15 : *
16 : * The Initial Developer of the Original Code is
17 : * Dave Townsend <dtownsend@oxymoronical.com>.
18 : *
19 : * Portions created by the Initial Developer are Copyright (C) 2007
20 : * the Initial Developer. All Rights Reserved.
21 : *
22 : * Contributor(s):
23 : * Mook <mook.moz+random.code@gmail.com>
24 : *
25 : * Alternatively, the contents of this file may be used under the terms of
26 : * either the GNU General Public License Version 2 or later (the "GPL"), or
27 : * the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
28 : * in which case the provisions of the GPL or the LGPL are applicable instead
29 : * of those above. If you wish to allow use of your version of this file only
30 : * under the terms of either the GPL or the LGPL, and not to allow others to
31 : * use your version of this file under the terms of the MPL, indicate your
32 : * decision by deleting the provisions above and replace them with the notice
33 : * and other provisions required by the GPL or the LGPL. If you do not delete
34 : * the provisions above, a recipient may use your version of this file under
35 : * the terms of any one of the MPL, the GPL or the LGPL.
36 : *
37 : * ***** END LICENSE BLOCK *****
38 : */
39 :
40 : #ifndef _nsZipWriter_h_
41 : #define _nsZipWriter_h_
42 :
43 : #include "nsIZipWriter.h"
44 : #include "nsIFileStreams.h"
45 : #include "nsIBufferedStreams.h"
46 : #include "nsIRequestObserver.h"
47 : #include "nsZipHeader.h"
48 : #include "nsCOMPtr.h"
49 : #include "nsCOMArray.h"
50 : #include "nsTArray.h"
51 : #include "nsDataHashtable.h"
52 :
53 : #define ZIPWRITER_CONTRACTID "@mozilla.org/zipwriter;1"
54 : #define ZIPWRITER_CLASSNAME "Zip Writer"
55 : #define ZIPWRITER_CID { 0x430d416c, 0xa722, 0x4ad1, \
56 : { 0xbe, 0x98, 0xd9, 0xa4, 0x45, 0xf8, 0x5e, 0x3f } }
57 :
58 : #define OPERATION_ADD 0
59 : #define OPERATION_REMOVE 1
60 : struct nsZipQueueItem
61 60 : {
62 : public:
63 : PRUint32 mOperation;
64 : nsCString mZipEntry;
65 : nsCOMPtr<nsIFile> mFile;
66 : nsCOMPtr<nsIChannel> mChannel;
67 : nsCOMPtr<nsIInputStream> mStream;
68 : PRTime mModTime;
69 : PRInt32 mCompression;
70 : PRUint32 mPermissions;
71 : };
72 :
73 : class nsZipWriter : public nsIZipWriter,
74 : public nsIRequestObserver
75 : {
76 : public:
77 : NS_DECL_ISUPPORTS
78 : NS_DECL_NSIZIPWRITER
79 : NS_DECL_NSIREQUESTOBSERVER
80 :
81 : nsZipWriter();
82 : nsresult EntryCompleteCallback(nsZipHeader *aHeader, nsresult aStatus);
83 :
84 : private:
85 : ~nsZipWriter();
86 :
87 : PRUint32 mCDSOffset;
88 : bool mCDSDirty;
89 : bool mInQueue;
90 :
91 : nsCOMPtr<nsIFile> mFile;
92 : nsCOMPtr<nsIRequestObserver> mProcessObserver;
93 : nsCOMPtr<nsISupports> mProcessContext;
94 : nsCOMPtr<nsIOutputStream> mStream;
95 : nsCOMArray<nsZipHeader> mHeaders;
96 : nsTArray<nsZipQueueItem> mQueue;
97 : nsDataHashtable<nsCStringHashKey, PRInt32> mEntryHash;
98 : nsCString mComment;
99 :
100 : nsresult SeekCDS();
101 : void Cleanup();
102 : nsresult ReadFile(nsIFile *aFile);
103 : nsresult InternalAddEntryDirectory(const nsACString & aZipEntry,
104 : PRTime aModTime, PRUint32 aPermissions);
105 : nsresult BeginProcessingAddition(nsZipQueueItem* aItem, bool* complete);
106 : nsresult BeginProcessingRemoval(PRInt32 aPos);
107 : nsresult AddEntryStream(const nsACString & aZipEntry, PRTime aModTime,
108 : PRInt32 aCompression, nsIInputStream *aStream,
109 : bool aQueue, PRUint32 aPermissions);
110 : void BeginProcessingNextItem();
111 : void FinishQueue(nsresult aStatus);
112 : };
113 :
114 : #endif
|