]> git.saurik.com Git - wxWidgets.git/blob - src/common/fs_zip.cpp
warning in Unicode compilation fixed
[wxWidgets.git] / src / common / fs_zip.cpp
1 /////////////////////////////////////////////////////////////////////////////
2 // Name: fs_zip.cpp
3 // Purpose: ZIP file system
4 // Author: Vaclav Slavik
5 // Copyright: (c) 1999 Vaclav Slavik
6 // CVS-ID: $Id$
7 // Licence: wxWindows Licence
8 /////////////////////////////////////////////////////////////////////////////
9
10
11
12 #ifdef __GNUG__
13 #pragma implementation "fs_zip.h"
14 #endif
15
16 #include "wx/wxprec.h"
17
18 #ifdef __BORDLANDC__
19 #pragma hdrstop
20 #endif
21
22 #if wxUSE_FILESYSTEM && wxUSE_FS_ZIP && wxUSE_ZIPSTREAM
23
24 #ifndef WXPRECOMP
25 #include "wx/intl.h"
26 #include "wx/log.h"
27 #endif
28
29 #include "wx/hash.h"
30 #include "wx/filesys.h"
31 #include "wx/zipstrm.h"
32 #include "wx/fs_zip.h"
33
34 /* Not the right solution (paths in makefiles) but... */
35 #ifdef __BORLANDC__
36 #include "../common/unzip.h"
37 #else
38 #include "unzip.h"
39 #endif
40
41
42 //--------------------------------------------------------------------------------
43 // wxZipFSHandler
44 //--------------------------------------------------------------------------------
45
46
47
48 wxZipFSHandler::wxZipFSHandler() : wxFileSystemHandler()
49 {
50 m_Archive = NULL;
51 m_ZipFile = m_Pattern = m_BaseDir = wxEmptyString;
52 m_AllowDirs = m_AllowFiles = TRUE;
53 m_DirsFound = NULL;
54 }
55
56
57
58 wxZipFSHandler::~wxZipFSHandler()
59 {
60 if (m_Archive)
61 unzClose((unzFile)m_Archive);
62 if (m_DirsFound)
63 delete m_DirsFound;
64 }
65
66
67
68 bool wxZipFSHandler::CanOpen(const wxString& location)
69 {
70 wxString p = GetProtocol(location);
71 return (p == wxT("zip")) &&
72 (GetProtocol(GetLeftLocation(location)) == wxT("file"));
73 }
74
75
76
77
78 wxFSFile* wxZipFSHandler::OpenFile(wxFileSystem& WXUNUSED(fs), const wxString& location)
79 {
80 wxString right = GetRightLocation(location);
81 wxString left = GetLeftLocation(location);
82 wxInputStream *s;
83
84 if (GetProtocol(left) != wxT("file"))
85 {
86 wxLogError(_("ZIP handler currently supports only local files!"));
87 return NULL;
88 }
89
90 if (right.GetChar(0) == wxT('/')) right = right.Mid(1);
91
92 s = new wxZipInputStream(left, right);
93 if (s && (s->LastError() == wxStream_NOERROR))
94 {
95 return new wxFSFile(s,
96 left + wxT("#zip:") + right,
97 GetMimeTypeFromExt(location),
98 GetAnchor(location),
99 wxDateTime(wxFileModificationTime(left)));
100 }
101
102 delete s;
103 return NULL;
104 }
105
106
107
108 wxString wxZipFSHandler::FindFirst(const wxString& spec, int flags)
109 {
110 wxString right = GetRightLocation(spec);
111 wxString left = GetLeftLocation(spec);
112
113 if (right.Last() == wxT('/')) right.RemoveLast();
114
115 if (m_Archive)
116 {
117 unzClose((unzFile)m_Archive);
118 m_Archive = NULL;
119 }
120
121 if (GetProtocol(left) != wxT("file"))
122 {
123 wxLogError(_("ZIP handler currently supports only local files!"));
124 return wxEmptyString;
125 }
126
127 switch (flags)
128 {
129 case wxFILE:
130 m_AllowDirs = FALSE, m_AllowFiles = TRUE; break;
131 case wxDIR:
132 m_AllowDirs = TRUE, m_AllowFiles = FALSE; break;
133 default:
134 m_AllowDirs = m_AllowFiles = TRUE; break;
135 }
136
137 m_ZipFile = left;
138 m_Archive = (void*) unzOpen(m_ZipFile.mb_str());
139 m_Pattern = right.AfterLast(wxT('/'));
140 m_BaseDir = right.BeforeLast(wxT('/'));
141
142 if (m_Archive)
143 {
144 if (unzGoToFirstFile((unzFile)m_Archive) != UNZ_OK)
145 {
146 unzClose((unzFile)m_Archive);
147 m_Archive = NULL;
148 }
149 else
150 {
151 if (m_AllowDirs)
152 {
153 delete m_DirsFound;
154 m_DirsFound = new wxHashTableLong();
155 }
156 return DoFind();
157 }
158 }
159 return wxEmptyString;
160 }
161
162
163
164 wxString wxZipFSHandler::FindNext()
165 {
166 if (!m_Archive) return wxEmptyString;
167 return DoFind();
168 }
169
170
171
172 wxString wxZipFSHandler::DoFind()
173 {
174 static char namebuf[1024]; // char, not wxChar!
175 char *c;
176 wxString namestr, dir, filename;
177 wxString match = wxEmptyString;
178
179 while (match == wxEmptyString)
180 {
181 unzGetCurrentFileInfo((unzFile)m_Archive, NULL, namebuf, 1024, NULL, 0, NULL, 0);
182 for (c = namebuf; *c; c++) if (*c == wxT('\\')) *c = wxT('/');
183 namestr = namebuf;
184
185 if (m_AllowDirs)
186 {
187 dir = namestr.BeforeLast(wxT('/'));
188 while (!dir.IsEmpty())
189 {
190 long key = 0;
191 for (size_t i = 0; i < dir.Length(); i++) key += (wxUChar)dir[i];
192 if (m_DirsFound->Get(key) == wxNOT_FOUND)
193 {
194 m_DirsFound->Put(key, 1);
195 filename = dir.AfterLast(wxT('/'));
196 dir = dir.BeforeLast(wxT('/'));
197 if (!filename.IsEmpty() && m_BaseDir == dir &&
198 wxMatchWild(m_Pattern, filename, FALSE))
199 match = m_ZipFile + wxT("#zip:") + dir + wxT("/") + filename;
200 }
201 else
202 break; // already tranversed
203 }
204 }
205
206 filename = namestr.AfterLast(wxT('/'));
207 dir = namestr.BeforeLast(wxT('/'));
208 if (m_AllowFiles && !filename.IsEmpty() && m_BaseDir == dir &&
209 wxMatchWild(m_Pattern, filename, FALSE))
210 match = m_ZipFile + wxT("#zip:") + namestr;
211
212 if (unzGoToNextFile((unzFile)m_Archive) != UNZ_OK)
213 {
214 unzClose((unzFile)m_Archive);
215 m_Archive = NULL;
216 break;
217 }
218 }
219
220 return match;
221 }
222
223
224
225 #endif
226 //wxUSE_FILESYSTEM && wxUSE_FS_ZIP && wxUSE_ZIPSTREAM