src/os2/dir.cpp

   1 /////////////////////////////////////////////////////////////////////////////
   2 // Name:        src/os2/dir.cpp
   3 // Purpose:     wxDir implementation for OS/2
   4 // Author:      Vadim Zeitlin
   5 // Modified by: Stefan Neis
   6 // Created:     08.12.99
   7 // RCS-ID:      $Id$
   8 // Copyright:   (c) 1999 Vadim Zeitlin <zeitlin@dptmaths.ens-cachan.fr>
   9 // Licence:     wxWindows licence
  10 /////////////////////////////////////////////////////////////////////////////
  11
  12 // ============================================================================
  13 // declarations
  14 // ============================================================================
  15
  16 // ----------------------------------------------------------------------------
  17 // headers
  18 // ----------------------------------------------------------------------------
  19
  20 // For compilers that support precompilation, includes "wx.h".
  21 #include "wx/wxprec.h"
  22
  23 #ifndef WX_PRECOMP
  24     #include "wx/os2/private.h"
  25     #include "wx/intl.h"
  26     #include "wx/log.h"
  27 #endif // PCH
  28
  29 #include "wx/dir.h"
  30 #include "wx/filefn.h"          // for wxMatchWild
  31
  32 #include <sys/types.h>
  33
  34 #define INCL_DOSFILEMGR
  35 #define INCL_DOSERRORS
  36 #include <os2.h>
  37
  38 #ifdef __EMX__
  39 #include <dirent.h>
  40 #endif
  41
  42 // ----------------------------------------------------------------------------
  43 // define the types and functions used for file searching
  44 // ----------------------------------------------------------------------------
  45
  46 typedef FILEFINDBUF3                FIND_STRUCT;
  47 typedef HDIR                        FIND_DATA;
  48 typedef ULONG                       FIND_ATTR;
  49
  50 static inline FIND_DATA InitFindData() { return ERROR_INVALID_HANDLE; }
  51
  52 static inline bool IsFindDataOk(
  53   FIND_DATA                     vFd
  54 )
  55 {
  56     return vFd != ERROR_INVALID_HANDLE;
  57 }
  58
  59 static inline void FreeFindData(
  60   FIND_DATA                     vFd
  61 )
  62 {
  63     if (!::DosFindClose(vFd))
  64     {
  65         wxLogLastError(wxT("DosFindClose"));
  66     }
  67 }
  68
  69 static inline FIND_DATA FindFirst(
  70   const wxString&               rsSpec
  71 , FIND_STRUCT*                  pFinddata
  72 )
  73 {
  74     ULONG                       ulFindCount = 1;
  75     FIND_DATA                   hDir = HDIR_CREATE;
  76     FIND_ATTR                   rc;
  77
  78     rc = ::DosFindFirst( rsSpec.c_str()
  79                         ,&hDir
  80                         ,0x37 // was: FILE_NORMAL
  81                         ,pFinddata
  82                         ,sizeof(FILEFINDBUF3)
  83                         ,&ulFindCount
  84                         ,FIL_STANDARD
  85                        );
  86     if (rc != 0)
  87         return InitFindData();
  88     return hDir;
  89 }
  90
  91 static inline bool FindNext(
  92   FIND_DATA                         vFd
  93 , FIND_STRUCT*                      pFinddata
  94 )
  95 {
  96     ULONG                           ulFindCount = 1;
  97
  98     return ::DosFindNext( vFd
  99                          ,pFinddata
 100                          ,sizeof(FILEFINDBUF3)
 101                          ,&ulFindCount
 102                          ) == 0;
 103 }
 104
 105 static const wxChar* GetNameFromFindData(
 106   FIND_STRUCT*                      pFinddata
 107 )
 108 {
 109     return (wxChar*)pFinddata->achName;
 110 }
 111
 112 static const FIND_ATTR GetAttrFromFindData(
 113   FIND_STRUCT*                      pFinddata
 114 )
 115 {
 116     return pFinddata->attrFile;
 117 }
 118
 119 static inline bool IsDir(
 120   FIND_ATTR                         vAttr
 121 )
 122 {
 123     return (vAttr & FILE_DIRECTORY) != 0;
 124 }
 125
 126 static inline bool IsHidden(
 127   FIND_ATTR                         vAttr
 128 )
 129 {
 130     return (vAttr & (FILE_HIDDEN | FILE_SYSTEM)) != 0;
 131 }
 132
 133 // ----------------------------------------------------------------------------
 134 // constants
 135 // ----------------------------------------------------------------------------
 136
 137 #ifndef MAX_PATH
 138     #define MAX_PATH 260        // from PM++ headers
 139 #endif
 140
 141 // ----------------------------------------------------------------------------
 142 // macros
 143 // ----------------------------------------------------------------------------
 144
 145 #define M_DIR       ((wxDirData *)m_data)
 146
 147 // ----------------------------------------------------------------------------
 148 // private classes
 149 // ----------------------------------------------------------------------------
 150
 151 // this class stores everything we need to enumerate the files
 152 class wxDirData
 153 {
 154 public:
 155     wxDirData(const wxString& rsDirname);
 156     ~wxDirData();
 157
 158     void SetFileSpec(const wxString& rsFilespec) { m_sFilespec = rsFilespec; }
 159     void SetFlags(int nFlags) { m_nFlags = nFlags; }
 160
 161     const wxString& GetName() const { return m_sDirname; }
 162     void Close();
 163     void Rewind();
 164     bool Read(wxString* rsFilename);
 165
 166 private:
 167     FIND_DATA                       m_vFinddata;
 168     wxString                        m_sDirname;
 169     wxString                        m_sFilespec;
 170     int                             m_nFlags;
 171 }; // end of CLASS wxDirData
 172
 173 // ============================================================================
 174 // implementation
 175 // ============================================================================
 176
 177 // ----------------------------------------------------------------------------
 178 // wxDirData
 179 // ----------------------------------------------------------------------------
 180
 181 wxDirData::wxDirData(
 182   const wxString&                   rsDirname
 183 )
 184 : m_sDirname(rsDirname)
 185 {
 186     m_vFinddata = InitFindData();
 187 } // end of wxDirData::wxDirData
 188
 189 wxDirData::~wxDirData()
 190 {
 191     Close();
 192 } // end of wxDirData::~wxDirData
 193
 194 void wxDirData::Close()
 195 {
 196     if ( IsFindDataOk(m_vFinddata) )
 197     {
 198         FreeFindData(m_vFinddata);
 199         m_vFinddata = InitFindData();
 200     }
 201 } // end of wxDirData::Close
 202
 203 void wxDirData::Rewind()
 204 {
 205     Close();
 206 } // end of wxDirData::Rewind
 207
 208 bool wxDirData::Read(
 209   wxString*                         psFilename
 210 )
 211 {
 212     bool                            bFirst = false;
 213
 214     FILEFINDBUF3                    vFinddata;
 215     #define PTR_TO_FINDDATA (&vFinddata)
 216
 217     if (!IsFindDataOk(m_vFinddata))
 218     {
 219         //
 220         // Open first
 221         //
 222         wxString                    sFilespec = m_sDirname;
 223
 224         if ( !wxEndsWithPathSeparator(sFilespec) )
 225         {
 226             sFilespec += wxT('\\');
 227         }
 228         sFilespec += (!m_sFilespec ? wxT("*.*") : m_sFilespec.c_str());
 229
 230         m_vFinddata = FindFirst( sFilespec
 231                                 ,PTR_TO_FINDDATA
 232                                );
 233         bFirst = true;
 234     }
 235
 236     if ( !IsFindDataOk(m_vFinddata) )
 237     {
 238         return false;
 239     }
 240
 241     const wxChar*                   zName;
 242     FIND_ATTR                       vAttr;
 243
 244     for ( ;; )
 245     {
 246         if (bFirst)
 247         {
 248             bFirst = false;
 249         }
 250         else
 251         {
 252             if (!FindNext( m_vFinddata
 253                           ,PTR_TO_FINDDATA
 254                          ))
 255             {
 256                 return false;
 257             }
 258         }
 259
 260         zName = GetNameFromFindData(PTR_TO_FINDDATA);
 261         vAttr = GetAttrFromFindData(PTR_TO_FINDDATA);
 262
 263         //
 264         // Don't return "." and ".." unless asked for
 265         //
 266         if ( zName[0] == wxT('.') &&
 267              ((zName[1] == wxT('.') && zName[2] == wxT('\0')) ||
 268               (zName[1] == wxT('\0'))) )
 269         {
 270             if (!(m_nFlags & wxDIR_DOTDOT))
 271                 continue;
 272         }
 273
 274         //
 275         // Check the type now
 276         //
 277         if (!(m_nFlags & wxDIR_FILES) && !IsDir(vAttr))
 278         {
 279             //
 280             // It's a file, but we don't want them
 281             //
 282             continue;
 283         }
 284         else if (!(m_nFlags & wxDIR_DIRS) && IsDir(vAttr) )
 285         {
 286             //
 287             // It's a dir, and we don't want it
 288             //
 289             continue;
 290         }
 291
 292         //
 293         // Finally, check whether it's a hidden file
 294         //
 295         if (!(m_nFlags & wxDIR_HIDDEN))
 296         {
 297             if (IsHidden(vAttr))
 298             {
 299                 //
 300                 // It's a hidden file, skip it
 301                 //
 302                 continue;
 303             }
 304         }
 305         *psFilename = zName;
 306         break;
 307     }
 308     return true;
 309 } // end of wxDirData::Read
 310
 311 // ----------------------------------------------------------------------------
 312 // wxDir construction/destruction
 313 // ----------------------------------------------------------------------------
 314
 315 wxDir::wxDir(
 316   const wxString&                   rsDirname
 317 )
 318 {
 319     m_data = NULL;
 320
 321     (void)Open(rsDirname);
 322 } // end of wxDir::wxDir
 323
 324 bool wxDir::Open(
 325   const wxString&                   rsDirname
 326 )
 327 {
 328     delete M_DIR;
 329     m_data = new wxDirData(rsDirname);
 330     return true;
 331 } // end of wxDir::Open
 332
 333 bool wxDir::IsOpened() const
 334 {
 335     return m_data != NULL;
 336 } // end of wxDir::IsOpen
 337
 338 wxString wxDir::GetName() const
 339 {
 340     wxString name;
 341     if ( m_data )
 342     {
 343         name = M_DIR->GetName();
 344         if ( !name.empty() )
 345         {
 346             // bring to canonical Windows form
 347             name.Replace(wxT("/"), wxT("\\"));
 348
 349             if ( name.Last() == wxT('\\') )
 350             {
 351                 // chop off the last (back)slash
 352                 name.Truncate(name.length() - 1);
 353             }
 354         }
 355     }
 356
 357     return name;
 358 }
 359
 360 wxDir::~wxDir()
 361 {
 362     delete M_DIR;
 363 } // end of wxDir::~wxDir
 364
 365 // ----------------------------------------------------------------------------
 366 // wxDir enumerating
 367 // ----------------------------------------------------------------------------
 368
 369 bool wxDir::GetFirst(
 370   wxString*                         psFilename
 371 , const wxString&                   rsFilespec
 372 , int                               nFlags
 373 ) const
 374 {
 375     wxCHECK_MSG( IsOpened(), false, wxT("must wxDir::Open() first") );
 376     M_DIR->Rewind();
 377     M_DIR->SetFileSpec(rsFilespec);
 378     M_DIR->SetFlags(nFlags);
 379     return GetNext(psFilename);
 380 } // end of wxDir::GetFirst
 381
 382 bool wxDir::GetNext(
 383   wxString*                         psFilename
 384 ) const
 385 {
 386     wxCHECK_MSG( IsOpened(), false, wxT("must wxDir::Open() first") );
 387     wxCHECK_MSG( psFilename, false, wxT("bad pointer in wxDir::GetNext()") );
 388     return M_DIR->Read(psFilename);
 389 } // end of wxDir::GetNext
 390