src/common/filesys.cpp

   1 /////////////////////////////////////////////////////////////////////////////
   2 // Name:        filesys.cpp
   3 // Purpose:     wxFileSystem class - interface for opening files
   4 // Author:      Vaclav Slavik
   5 // Copyright:   (c) 1999 Vaclav Slavik
   6 // CVS-ID:      $Id$
   7 // Licence:     wxWindows Licence
   8 /////////////////////////////////////////////////////////////////////////////
   9
  10 #ifdef __GNUG__
  11 #pragma implementation
  12 #endif
  13
  14 #include "wx/wxprec.h"
  15
  16 #ifdef __BORDLANDC__
  17 #pragma hdrstop
  18 #endif
  19
  20
  21 #if wxUSE_FILESYSTEM
  22
  23 #include "wx/wfstream.h"
  24 #include "wx/module.h"
  25 #include "wx/filesys.h"
  26 #include "wx/mimetype.h"
  27
  28
  29
  30
  31 //--------------------------------------------------------------------------------
  32 // wxFileSystemHandler
  33 //--------------------------------------------------------------------------------
  34
  35 IMPLEMENT_ABSTRACT_CLASS(wxFileSystemHandler, wxObject)
  36
  37
  38 static wxFileTypeInfo *gs_FSMimeFallbacks = NULL;
  39
  40 wxString wxFileSystemHandler::GetMimeTypeFromExt(const wxString& location)
  41 {
  42     wxString ext = wxEmptyString, mime = wxEmptyString;
  43     wxString loc = GetRightLocation(location);
  44     char c;
  45     int l = loc.Length(), l2;
  46     wxFileType *ft;
  47
  48     l2 = l;
  49     for (int i = l-1; i >= 0; i--) {
  50         c = loc[(unsigned int) i];
  51         if (c == wxT('#')) l2 = i + 1;
  52         if (c == wxT('.')) {ext = loc.Right(l2-i-1); break;}
  53         if ((c == wxT('/')) || (c == wxT('\\')) || (c == wxT(':'))) {return wxEmptyString;}
  54     }
  55
  56     static bool s_MinimalMimeEnsured = FALSE;
  57     if (!s_MinimalMimeEnsured) {
  58         wxTheMimeTypesManager -> AddFallbacks(gs_FSMimeFallbacks);
  59         s_MinimalMimeEnsured = TRUE;
  60     }
  61
  62     ft = wxTheMimeTypesManager -> GetFileTypeFromExtension(ext);
  63     if ( !ft || !ft -> GetMimeType(&mime) ) {
  64         mime = wxEmptyString;
  65     }
  66
  67     delete ft;
  68
  69     return mime;
  70 }
  71
  72
  73
  74 wxString wxFileSystemHandler::GetProtocol(const wxString& location) const
  75 {
  76     wxString s = wxEmptyString;
  77     int i, l = location.Length();
  78     bool fnd;
  79
  80     fnd = FALSE;
  81     for (i = l-1; (i >= 0) && ((location[i] != wxT('#')) || (!fnd)); i--) {
  82         if ((location[i] == wxT(':')) && (i != 1 /*win: C:\path*/)) fnd = TRUE;
  83     }
  84     if (!fnd) return wxT("file");
  85     for (++i; (i < l) && (location[i] != wxT(':')); i++) s << location[i];
  86     return s;
  87 }
  88
  89
  90 wxString wxFileSystemHandler::GetLeftLocation(const wxString& location) const
  91 {
  92     int i;
  93     bool fnd;
  94
  95     fnd = FALSE;
  96     for (i = location.Length()-1; i >= 0; i--) {
  97         if ((location[i] == wxT(':')) && (i != 1 /*win: C:\path*/)) fnd = TRUE;
  98         else if (fnd && (location[i] == wxT('#'))) return location.Left(i);
  99     }
 100     return wxEmptyString;
 101 }
 102
 103 wxString wxFileSystemHandler::GetRightLocation(const wxString& location) const
 104 {
 105     int i, l = location.Length();
 106     int l2 = l + 1;
 107     for (i = l-1; (i >= 0) && ((location[i] != wxT(':')) || (i == 1) || (location[i-2] == wxT(':'))); i--) {if (location[i] == wxT('#')) l2 = i + 1;}
 108     if (i == 0) return wxEmptyString;
 109     else return location.Mid(i + 1, l2 - i - 2);
 110 }
 111
 112 wxString wxFileSystemHandler::GetAnchor(const wxString& location) const
 113 {
 114     char c;
 115     int l = location.Length();
 116
 117     for (int i = l-1; i >= 0; i--) {
 118         c = location[i];
 119         if (c == wxT('#')) return location.Right(l-i-1);
 120         else if ((c == wxT('.')) || (c == wxT('/')) || (c == wxT('\\')) || (c == wxT(':'))) return wxEmptyString;
 121     }
 122     return wxEmptyString;
 123 }
 124
 125
 126 wxString wxFileSystemHandler::FindFirst(const wxString& WXUNUSED(spec),
 127                                         int WXUNUSED(flags))
 128 {
 129     return wxEmptyString;
 130 }
 131
 132 wxString wxFileSystemHandler::FindNext()
 133 {
 134     return wxEmptyString;
 135 }
 136
 137 //--------------------------------------------------------------------------------
 138 // wxLocalFSHandler
 139 //--------------------------------------------------------------------------------
 140
 141 class wxLocalFSHandler : public wxFileSystemHandler
 142 {
 143     public:
 144         virtual bool CanOpen(const wxString& location);
 145         virtual wxFSFile* OpenFile(wxFileSystem& fs, const wxString& location);
 146         virtual wxString FindFirst(const wxString& spec, int flags = 0);
 147         virtual wxString FindNext();
 148 };
 149
 150
 151 bool wxLocalFSHandler::CanOpen(const wxString& location)
 152 {
 153     return GetProtocol(location) == wxT("file");
 154 }
 155
 156 wxFSFile* wxLocalFSHandler::OpenFile(wxFileSystem& WXUNUSED(fs), const wxString& location)
 157 {
 158     wxString right = GetRightLocation(location);
 159     if (!wxFileExists(right))
 160         return (wxFSFile*) NULL;
 161
 162     return new wxFSFile(new wxFileInputStream(right),
 163                         right,
 164                         GetMimeTypeFromExt(location),
 165                         GetAnchor(location),
 166                         wxDateTime(wxFileModificationTime(right)));
 167
 168 }
 169
 170 wxString wxLocalFSHandler::FindFirst(const wxString& spec, int flags)
 171 {
 172     wxString right = GetRightLocation(spec);
 173     return wxFindFirstFile(right, flags);
 174 }
 175
 176 wxString wxLocalFSHandler::FindNext()
 177 {
 178     return wxFindNextFile();
 179 }
 180
 181
 182
 183 //-----------------------------------------------------------------------------
 184 // wxFileSystem
 185 //-----------------------------------------------------------------------------
 186
 187 IMPLEMENT_DYNAMIC_CLASS(wxFileSystem, wxObject)
 188
 189
 190 wxList wxFileSystem::m_Handlers;
 191
 192
 193 static wxString MakeCorrectPath(const wxString& path)
 194 {
 195     wxString p(path);
 196     wxString r;
 197     int i, j, cnt;
 198
 199     cnt = p.Length();
 200     for (i = 0; i < cnt; i++)
 201       if (p.GetChar(i) == wxT('\\')) p.GetWritableChar(i) = wxT('/'); // Want to be windows-safe
 202
 203     if (p.Left(2) == wxT("./")) { p = p.Mid(2); cnt -= 2; }
 204
 205     if (cnt < 3) return p;
 206
 207     r << p.GetChar(0) << p.GetChar(1);
 208
 209     // skip trailing ../.., if any
 210     for (i = 2; i < cnt && (p.GetChar(i) == wxT('/') || p.GetChar(i) == wxT('.')); i++) r << p.GetChar(i);
 211
 212     // remove back references: translate dir1/../dir2 to dir2
 213     for (; i < cnt; i++)
 214     {
 215         r << p.GetChar(i);
 216         if (p.GetChar(i) == wxT('/') && p.GetChar(i-1) == wxT('.') && p.GetChar(i-2) == wxT('.'))
 217         {
 218             for (j = r.Length() - 2; j >= 0 && r.GetChar(j) != wxT('/') && r.GetChar(j) != wxT(':'); j--) {}
 219             if (j >= 0 && r.GetChar(j) != wxT(':'))
 220             {
 221                 for (j = j - 1; j >= 0 && r.GetChar(j) != wxT('/') && r.GetChar(j) != wxT(':'); j--) {}
 222                 r.Remove(j + 1);
 223             }
 224         }
 225     }
 226
 227     for (; i < cnt; i++) r << p.GetChar(i);
 228
 229     return r;
 230 }
 231
 232
 233 void wxFileSystem::ChangePathTo(const wxString& location, bool is_dir)
 234 {
 235     int i, pathpos = -1;
 236
 237     m_Path = MakeCorrectPath(location);
 238
 239     if (is_dir)
 240     {
 241         if (m_Path.Length() > 0 && m_Path.Last() != wxT('/') && m_Path.Last() != wxT(':'))
 242                 m_Path << wxT('/');
 243     }
 244
 245     else
 246     {
 247         for (i = m_Path.Length()-1; i >= 0; i--)
 248         {
 249             if (m_Path[(unsigned int) i] == wxT('/'))
 250                 {
 251                 if ((i > 1) && (m_Path[(unsigned int) (i-1)] == wxT('/')) && (m_Path[(unsigned int) (i-2)] == wxT(':')))
 252                         {
 253                     i -= 2;
 254                     continue;
 255                 }
 256                 else
 257                         {
 258                     pathpos = i;
 259                     break;
 260                 }
 261             }
 262             else if (m_Path[(unsigned int) i] == wxT(':')) {
 263                 pathpos = i;
 264                 break;
 265             }
 266         }
 267         if (pathpos == -1)
 268         {
 269             for (i = 0; i < (int) m_Path.Length(); i++)
 270             {
 271                 if (m_Path[(unsigned int) i] == wxT(':'))
 272                         {
 273                     m_Path.Remove(i+1);
 274                     break;
 275                 }
 276             }
 277             if (i == (int) m_Path.Length())
 278                 m_Path = wxEmptyString;
 279         }
 280         else
 281         {
 282             m_Path.Remove(pathpos+1);
 283         }
 284     }
 285 }
 286
 287
 288
 289 wxFSFile* wxFileSystem::OpenFile(const wxString& location)
 290 {
 291     wxString loc = MakeCorrectPath(location);
 292     unsigned i, ln;
 293     char meta;
 294     wxFSFile *s = NULL;
 295     wxNode *node;
 296
 297     ln = loc.Length();
 298     meta = 0;
 299     for (i = 0; i < ln; i++)
 300     {
 301         switch (loc[i])
 302         {
 303             case wxT('/') : case wxT(':') : case wxT('#') :
 304                 meta = loc[i];
 305                 break;
 306         }
 307         if (meta != 0) break;
 308     }
 309     m_LastName = wxEmptyString;
 310
 311     // try relative paths first :
 312     if (meta != wxT(':'))
 313     {
 314         node = m_Handlers.GetFirst();
 315         while (node)
 316         {
 317             wxFileSystemHandler *h = (wxFileSystemHandler*) node -> GetData();
 318             if (h->CanOpen(m_Path + loc))
 319                 {
 320                 s = h->OpenFile(*this, m_Path + loc);
 321                 if (s) { m_LastName = m_Path + loc; break; }
 322             }
 323             node = node->GetNext();
 324         }
 325     }
 326
 327     // if failed, try absolute paths :
 328     if (s == NULL)
 329     {
 330         node = m_Handlers.GetFirst();
 331         while (node)
 332         {
 333             wxFileSystemHandler *h = (wxFileSystemHandler*) node->GetData();
 334             if (h->CanOpen(loc))
 335             {
 336                 s = h->OpenFile(*this, loc);
 337                 if (s) { m_LastName = loc; break; }
 338             }
 339             node = node->GetNext();
 340         }
 341     }
 342     return (s);
 343 }
 344
 345
 346
 347 wxString wxFileSystem::FindFirst(const wxString& spec, int flags)
 348 {
 349     wxNode *node;
 350     wxString spec2(spec);
 351
 352     m_FindFileHandler = NULL;
 353
 354     for (int i = spec2.Length()-1; i >= 0; i--)
 355         if (spec2[(unsigned int) i] == wxT('\\')) spec2.GetWritableChar(i) = wxT('/'); // Want to be windows-safe
 356
 357     node = m_Handlers.GetFirst();
 358     while (node)
 359     {
 360         m_FindFileHandler = (wxFileSystemHandler*) node -> GetData();
 361         if (m_FindFileHandler -> CanOpen(m_Path + spec2))
 362             return m_FindFileHandler -> FindFirst(m_Path + spec2, flags);
 363         node = node->GetNext();
 364     }
 365
 366     node = m_Handlers.GetFirst();
 367     while (node)
 368     {
 369         m_FindFileHandler = (wxFileSystemHandler*) node -> GetData();
 370         if (m_FindFileHandler -> CanOpen(spec2))
 371             return m_FindFileHandler -> FindFirst(spec2, flags);
 372         node = node->GetNext();
 373     }
 374
 375     return wxEmptyString;
 376 }
 377
 378
 379
 380 wxString wxFileSystem::FindNext()
 381 {
 382     if (m_FindFileHandler == NULL) return wxEmptyString;
 383     else return m_FindFileHandler -> FindNext();
 384 }
 385
 386
 387
 388 void wxFileSystem::AddHandler(wxFileSystemHandler *handler)
 389 {
 390     m_Handlers.Append(handler);
 391 }
 392
 393
 394 void wxFileSystem::CleanUpHandlers()
 395 {
 396     m_Handlers.DeleteContents(TRUE);
 397     m_Handlers.Clear();
 398 }
 399
 400
 401
 402
 403 ///// Module:
 404
 405 class wxFileSystemModule : public wxModule
 406 {
 407     DECLARE_DYNAMIC_CLASS(wxFileSystemModule)
 408
 409     public:
 410         virtual bool OnInit()
 411         {
 412             wxFileSystem::AddHandler(new wxLocalFSHandler);
 413
 414             gs_FSMimeFallbacks = new wxFileTypeInfo[6];
 415             gs_FSMimeFallbacks[0] =
 416             wxFileTypeInfo("image/jpeg",
 417                            "",
 418                            "",
 419                            "JPEG image (from fallback)",
 420                            "jpg", "jpeg", NULL);
 421             gs_FSMimeFallbacks[1] =
 422             wxFileTypeInfo("image/gif",
 423                            "",
 424                            "",
 425                            "GIF image (from fallback)",
 426                            "gif", NULL);
 427             gs_FSMimeFallbacks[2] =
 428             wxFileTypeInfo("image/png",
 429                            "",
 430                            "",
 431                            "PNG image (from fallback)",
 432                            "png", NULL);
 433             gs_FSMimeFallbacks[3] =
 434             wxFileTypeInfo("image/bmp",
 435                            "",
 436                            "",
 437                            "windows bitmap image (from fallback)",
 438                            "bmp", NULL);
 439             gs_FSMimeFallbacks[4] =
 440             wxFileTypeInfo("text/html",
 441                            "",
 442                            "",
 443                            "HTML document (from fallback)",
 444                            "htm", "html", NULL);
 445             gs_FSMimeFallbacks[5] =
 446             // must terminate the table with this!
 447             wxFileTypeInfo();
 448
 449             return TRUE;
 450         }
 451         virtual void OnExit()
 452         {
 453             delete [] gs_FSMimeFallbacks;
 454             wxFileSystem::CleanUpHandlers();
 455         }
 456 };
 457
 458 IMPLEMENT_DYNAMIC_CLASS(wxFileSystemModule, wxModule)
 459
 460 #endif
 461   // wxUSE_FILESYSTEM
 462
 463
 464