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 __BORLANDC__
  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     wxFileName leftFilename = wxFileSystem::URLToFileName(left);
  93
  94     s = new wxZipInputStream(leftFilename.GetFullPath(), right);
  95     if (s && s->IsOk() )
  96     {
  97         return new wxFSFile(s,
  98                             left + wxT("#zip:") + right,
  99                             GetMimeTypeFromExt(location),
 100                             GetAnchor(location)
 101 #if wxUSE_DATETIME
 102                             , wxDateTime(wxFileModificationTime(left))
 103 #endif // wxUSE_DATETIME
 104                             );
 105     }
 106
 107     delete s;
 108     return NULL;
 109 }
 110
 111
 112
 113 wxString wxZipFSHandler::FindFirst(const wxString& spec, int flags)
 114 {
 115     wxString right = GetRightLocation(spec);
 116     wxString left = GetLeftLocation(spec);
 117
 118     if (right.Last() == wxT('/')) right.RemoveLast();
 119
 120     if (m_Archive)
 121     {
 122         unzClose((unzFile)m_Archive);
 123         m_Archive = NULL;
 124     }
 125
 126     if (GetProtocol(left) != wxT("file"))
 127     {
 128         wxLogError(_("ZIP handler currently supports only local files!"));
 129         return wxEmptyString;
 130     }
 131
 132     switch (flags)
 133     {
 134         case wxFILE:
 135             m_AllowDirs = FALSE, m_AllowFiles = TRUE; break;
 136         case wxDIR:
 137             m_AllowDirs = TRUE, m_AllowFiles = FALSE; break;
 138         default:
 139             m_AllowDirs = m_AllowFiles = TRUE; break;
 140     }
 141
 142     m_ZipFile = left;
 143     wxString nativename = wxFileSystem::URLToFileName(m_ZipFile).GetFullPath();
 144     m_Archive = (void*) unzOpen(nativename.mb_str());
 145     m_Pattern = right.AfterLast(wxT('/'));
 146     m_BaseDir = right.BeforeLast(wxT('/'));
 147
 148     if (m_Archive)
 149     {
 150         if (unzGoToFirstFile((unzFile)m_Archive) != UNZ_OK)
 151         {
 152             unzClose((unzFile)m_Archive);
 153             m_Archive = NULL;
 154         }
 155         else
 156         {
 157             if (m_AllowDirs)
 158             {
 159                 delete m_DirsFound;
 160                 m_DirsFound = new wxHashTableLong();
 161             }
 162             return DoFind();
 163         }
 164     }
 165     return wxEmptyString;
 166 }
 167
 168
 169
 170 wxString wxZipFSHandler::FindNext()
 171 {
 172     if (!m_Archive) return wxEmptyString;
 173     return DoFind();
 174 }
 175
 176
 177
 178 wxString wxZipFSHandler::DoFind()
 179 {
 180     static char namebuf[1024]; // char, not wxChar!
 181     char *c;
 182     wxString namestr, dir, filename;
 183     wxString match = wxEmptyString;
 184
 185     while (match == wxEmptyString)
 186     {
 187         unzGetCurrentFileInfo((unzFile)m_Archive, NULL, namebuf, 1024, NULL, 0, NULL, 0);
 188         for (c = namebuf; *c; c++) if (*c == '\\') *c = '/';
 189         namestr = wxString::FromAscii( namebuf );    // TODO what encoding does ZIP use?
 190
 191         if (m_AllowDirs)
 192         {
 193             dir = namestr.BeforeLast(wxT('/'));
 194             while (!dir.IsEmpty())
 195             {
 196                 long key = 0;
 197                 for (size_t i = 0; i < dir.Length(); i++) key += (wxUChar)dir[i];
 198                 if (m_DirsFound->Get(key) == wxNOT_FOUND)
 199                 {
 200                     m_DirsFound->Put(key, 1);
 201                     filename = dir.AfterLast(wxT('/'));
 202                     dir = dir.BeforeLast(wxT('/'));
 203                     if (!filename.IsEmpty() && m_BaseDir == dir &&
 204                                 wxMatchWild(m_Pattern, filename, FALSE))
 205                         match = m_ZipFile + wxT("#zip:") + dir + wxT("/") + filename;
 206                 }
 207                 else
 208                     break; // already tranversed
 209             }
 210         }
 211
 212         filename = namestr.AfterLast(wxT('/'));
 213         dir = namestr.BeforeLast(wxT('/'));
 214         if (m_AllowFiles && !filename.IsEmpty() && m_BaseDir == dir &&
 215                             wxMatchWild(m_Pattern, filename, FALSE))
 216             match = m_ZipFile + wxT("#zip:") + namestr;
 217
 218         if (unzGoToNextFile((unzFile)m_Archive) != UNZ_OK)
 219         {
 220             unzClose((unzFile)m_Archive);
 221             m_Archive = NULL;
 222             break;
 223         }
 224     }
 225
 226     return match;
 227 }
 228
 229
 230
 231 #endif
 232       //wxUSE_FILESYSTEM && wxUSE_FS_ZIP && wxUSE_ZIPSTREAM