home *** CD-ROM | disk | FTP | other *** search
/ OS/2 Shareware BBS: 10 Tools / 10-Tools.zip / wxos2233.zip / wxOS2-2_3_3.zip / wxWindows-2.3.3 / src / common / fs_zip.cpp < prev    next >
C/C++ Source or Header  |  2001-09-07  |  6KB  |  227 lines
  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: fs_zip.cpp,v 1.18 2001/09/07 20:08:05 KLB Exp $
  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
  227.