RemoveAt() added to array classes

[wxWidgets.git] / src / common / fs_zip.cpp

diff --git a/src/common/fs_zip.cpp b/src/common/fs_zip.cpp
index 56647fcd127e3e8a039999ac640c8bb1306b1f41..96985ddc656a339342c1053894855be242bae28c 100644 (file)
--- a/src/common/fs_zip.cpp
+++ b/src/common/fs_zip.cpp
@@ -11,27 +11,29 @@
 #pragma implementation
 #endif
 
-#include <wx/wxprec.h>
+#include "wx/wxprec.h"
 
 #ifdef __BORDLANDC__
 #pragma hdrstop
 #endif
 
-#if !wxUSE_SOCKETS
-    #undef wxUSE_FS_ZIP
-    #define wxUSE_FS_ZIP 0
-#endif
-
 #if wxUSE_FS_ZIP
 
 #ifndef WXPRECOMP
-#include <wx/wx.h>
+#include "wx/wx.h"
 #endif
 
 #include "wx/filesys.h"
-#include "wx/zipstream.h"
+#include "wx/zipstrm.h"
 #include "wx/fs_zip.h"
 
+/* Not the right solution (paths in makefiles) but... */
+#ifdef __BORLANDC__
+#include "../common/unzip.h"
+#else
+#include "unzip.h"
+#endif
+
 
 //--------------------------------------------------------------------------------
 // wxZipFSHandler
@@ -39,6 +41,23 @@
 
 
 
+wxZipFSHandler::wxZipFSHandler() : wxFileSystemHandler()
+{
+    m_Archive = NULL;
+    m_ZipFile = m_Pattern = m_BaseDir = wxEmptyString;
+    m_AllowDirs = m_AllowFiles = TRUE;
+}
+
+
+
+wxZipFSHandler::~wxZipFSHandler()
+{
+    if (m_Archive)         
+        unzClose((unzFile)m_Archive);
+}
+
+
+
 bool wxZipFSHandler::CanOpen(const wxString& location)
 {
     wxString p = GetProtocol(location);
@@ -70,8 +89,91 @@ wxFSFile* wxZipFSHandler::OpenFile(wxFileSystem& WXUNUSED(fs), const wxString& l
 
 
 
-wxZipFSHandler::~wxZipFSHandler()
+wxString wxZipFSHandler::FindFirst(const wxString& spec, int flags)
+{
+    wxString right = GetRightLocation(spec);
+    wxString left = GetLeftLocation(spec);
+    
+    if (right.Last() == wxT('/')) right.RemoveLast();
+
+    if (m_Archive) {
+        unzClose((unzFile)m_Archive);
+        m_Archive = NULL;
+    }
+
+    if (GetProtocol(left) != "file") {
+        return wxEmptyString;
+    }
+
+    switch (flags) {
+        case wxFILE  : m_AllowDirs = FALSE, m_AllowFiles = TRUE; break;
+        case wxDIR   : m_AllowDirs = TRUE, m_AllowFiles = FALSE; break;
+        default : m_AllowDirs = m_AllowFiles = TRUE; break;
+    }
+
+    m_ZipFile = left;
+    m_Archive = (void*) unzOpen(m_ZipFile.fn_str());
+    m_Pattern = right.AfterLast(wxT('/'));
+    m_BaseDir = right.BeforeLast(wxT('/'));
+
+    if (m_Archive) {
+        if (unzGoToFirstFile((unzFile)m_Archive) != UNZ_OK) {
+            unzClose((unzFile)m_Archive);
+            m_Archive = NULL;        
+        }
+    }
+    return DoFind();
+}
+
+
+
+wxString wxZipFSHandler::FindNext()
 {
+    if (!m_Archive) return wxEmptyString;
+    return DoFind();
 }
 
+
+
+wxString wxZipFSHandler::DoFind()
+{
+    static char namebuf[1024]; // char, not wxChar!
+    char *c;
+    wxString fn, dir, name;
+    wxString match = wxEmptyString;
+    bool wasdir;
+
+    while (match == wxEmptyString)
+    {
+        unzGetCurrentFileInfo((unzFile)m_Archive, NULL, namebuf, 1024, NULL, 0, NULL, 0);
+        for (c = namebuf; *c; c++) if (*c == '\\') *c = '/';
+        fn = namebuf;
+        if (fn.Last() == wxT('/')) {
+            fn.RemoveLast();
+            wasdir = TRUE;
+        }
+        else wasdir = FALSE;
+
+        name = fn.AfterLast(wxT('/'));
+        dir = fn.BeforeLast(wxT('/'));
+
+        if (dir == m_BaseDir) {
+            if (m_AllowFiles && !wasdir && wxMatchWild(m_Pattern, name, FALSE))
+                match = m_ZipFile + wxT("#zip:") + fn;
+            if (m_AllowDirs && wasdir && wxMatchWild(m_Pattern, name, FALSE))
+                match = m_ZipFile + wxT("#zip:") + fn;
+        }
+        
+        if (unzGoToNextFile((unzFile)m_Archive) != UNZ_OK) {
+            unzClose((unzFile)m_Archive);
+            m_Archive = NULL;
+            break;
+        }
+    }
+    
+    return match;
+}
+
+
+
 #endif // wxUSE_FS_ZIP