]> git.saurik.com Git - wxWidgets.git/blob - src/common/fs_zip.cpp
Tweaks for demos on MacOSX
[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 #include "wx/wxprec.h"
11
12 #ifdef __BORLANDC__
13 #pragma hdrstop
14 #endif
15
16 #if wxUSE_FILESYSTEM && wxUSE_FS_ZIP && wxUSE_ZIPSTREAM && wxUSE_ZLIB
17
18 #ifndef WXPRECOMP
19 #include "wx/intl.h"
20 #include "wx/log.h"
21 #endif
22
23 #include "wx/filesys.h"
24 #include "wx/wfstream.h"
25 #include "wx/zipstrm.h"
26 #include "wx/fs_zip.h"
27
28
29 //---------------------------------------------------------------------------
30 // wxZipFSInputStream
31 //---------------------------------------------------------------------------
32 // Helper class for wxZipFSHandler
33
34 class wxZipFSInputStream : public wxZipInputStream
35 {
36 public:
37 wxZipFSInputStream(wxFSFile *file)
38 : wxZipInputStream(*file->GetStream())
39 {
40 m_file = file;
41 #if 1 //WXWIN_COMPATIBILITY_2_6
42 m_allowSeeking = true;
43 #endif
44 }
45
46 virtual ~wxZipFSInputStream() { delete m_file; }
47
48 private:
49 wxFSFile *m_file;
50 };
51
52 //----------------------------------------------------------------------------
53 // wxZipFSHandler
54 //----------------------------------------------------------------------------
55
56 wxZipFSHandler::wxZipFSHandler() : wxFileSystemHandler()
57 {
58 m_Archive = NULL;
59 m_ZipFile = m_Pattern = m_BaseDir = wxEmptyString;
60 m_AllowDirs = m_AllowFiles = true;
61 m_DirsFound = NULL;
62 }
63
64
65
66 wxZipFSHandler::~wxZipFSHandler()
67 {
68 Cleanup();
69 }
70
71
72 void wxZipFSHandler::Cleanup()
73 {
74 wxDELETE(m_Archive);
75 wxDELETE(m_DirsFound);
76 }
77
78
79
80 bool wxZipFSHandler::CanOpen(const wxString& location)
81 {
82 wxString p = GetProtocol(location);
83 return (p == wxT("zip"));
84 }
85
86
87 wxFSFile* wxZipFSHandler::OpenFile(wxFileSystem& WXUNUSED(fs), const wxString& location)
88 {
89 wxString right = GetRightLocation(location);
90 wxString left = GetLeftLocation(location);
91 wxZipInputStream *s;
92
93 if (right.Contains(wxT("./")))
94 {
95 if (right.GetChar(0) != wxT('/')) right = wxT('/') + right;
96 wxFileName rightPart(right, wxPATH_UNIX);
97 rightPart.Normalize(wxPATH_NORM_DOTS, wxT("/"), wxPATH_UNIX);
98 right = rightPart.GetFullPath(wxPATH_UNIX);
99 }
100
101 if (right.GetChar(0) == wxT('/')) right = right.Mid(1);
102
103 // a new wxFileSystem object is needed here to avoid infinite recursion
104 wxFSFile *leftFile = wxFileSystem().OpenFile(left);
105 if (!leftFile)
106 return NULL;
107
108 s = new wxZipFSInputStream(leftFile);
109 if (s && s->IsOk())
110 {
111 bool found = false;
112 while (!found)
113 {
114 wxZipEntry *ent = s->GetNextEntry();
115 if (!ent)
116 break;
117 if (ent->GetInternalName() == right)
118 found = true;
119 delete ent;
120 }
121 if (found)
122 return new wxFSFile(s,
123 left + wxT("#zip:") + right,
124 GetMimeTypeFromExt(location),
125 GetAnchor(location)
126 #if wxUSE_DATETIME
127 , wxDateTime(wxFileModificationTime(left))
128 #endif // wxUSE_DATETIME
129 );
130 }
131
132 delete s;
133 return NULL;
134 }
135
136
137
138 wxString wxZipFSHandler::FindFirst(const wxString& spec, int flags)
139 {
140 wxString right = GetRightLocation(spec);
141 wxString left = GetLeftLocation(spec);
142
143 if (!right.empty() && right.Last() == wxT('/')) right.RemoveLast();
144
145 if (m_Archive)
146 {
147 delete m_Archive;
148 m_Archive = NULL;
149 }
150
151 switch (flags)
152 {
153 case wxFILE:
154 m_AllowDirs = false, m_AllowFiles = true; break;
155 case wxDIR:
156 m_AllowDirs = true, m_AllowFiles = false; break;
157 default:
158 m_AllowDirs = m_AllowFiles = true; break;
159 }
160
161 m_ZipFile = left;
162
163 wxFSFile *leftFile = wxFileSystem().OpenFile(left);
164 if (leftFile)
165 m_Archive = new wxZipFSInputStream(leftFile);
166
167 m_Pattern = right.AfterLast(wxT('/'));
168 m_BaseDir = right.BeforeLast(wxT('/'));
169 if (m_BaseDir.StartsWith(wxT("/")))
170 m_BaseDir = m_BaseDir.Mid(1);
171
172 if (m_Archive)
173 {
174 if (m_AllowDirs)
175 {
176 delete m_DirsFound;
177 m_DirsFound = new wxZipFilenameHashMap();
178 if (right.empty()) // allow "/" to match the archive root
179 return spec;
180 }
181 return DoFind();
182 }
183 return wxEmptyString;
184 }
185
186
187
188 wxString wxZipFSHandler::FindNext()
189 {
190 if (!m_Archive) return wxEmptyString;
191 return DoFind();
192 }
193
194
195
196 wxString wxZipFSHandler::DoFind()
197 {
198 wxString namestr, dir, filename;
199 wxString match = wxEmptyString;
200
201 while (match == wxEmptyString)
202 {
203 wxZipEntry *entry = m_Archive->GetNextEntry();
204 if (!entry)
205 {
206 delete m_Archive;
207 m_Archive = NULL;
208 break;
209 }
210 namestr = entry->GetName(wxPATH_UNIX);
211 delete entry;
212
213 if (m_AllowDirs)
214 {
215 dir = namestr.BeforeLast(wxT('/'));
216 while (!dir.empty())
217 {
218 if( m_DirsFound->find(dir) == m_DirsFound->end() )
219 {
220 (*m_DirsFound)[dir] = 1;
221 filename = dir.AfterLast(wxT('/'));
222 dir = dir.BeforeLast(wxT('/'));
223 if (!filename.empty() && m_BaseDir == dir &&
224 wxMatchWild(m_Pattern, filename, false))
225 match = m_ZipFile + wxT("#zip:") + dir + wxT("/") + filename;
226 }
227 else
228 break; // already tranversed
229 }
230 }
231
232 filename = namestr.AfterLast(wxT('/'));
233 dir = namestr.BeforeLast(wxT('/'));
234 if (m_AllowFiles && !filename.empty() && m_BaseDir == dir &&
235 wxMatchWild(m_Pattern, filename, false))
236 match = m_ZipFile + wxT("#zip:") + namestr;
237 }
238
239 return match;
240 }
241
242
243
244 #endif
245 //wxUSE_FILESYSTEM && wxUSE_FS_ZIP && wxUSE_ZIPSTREAM