transex3/source/directory.cxx

   1 /*************************************************************************
   2  *
   3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   4  *
   5  * Copyright 2008 by Sun Microsystems, Inc.
   6  *
   7  * OpenOffice.org - a multi-platform office productivity suite
   8  *
   9  * $RCSfile: directory.cxx,v $
  10  * $Revision: 1.6 $
  11  *
  12  * This file is part of OpenOffice.org.
  13  *
  14  * OpenOffice.org is free software: you can redistribute it and/or modify
  15  * it under the terms of the GNU Lesser General Public License version 3
  16  * only, as published by the Free Software Foundation.
  17  *
  18  * OpenOffice.org is distributed in the hope that it will be useful,
  19  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  20  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  21  * GNU Lesser General Public License version 3 for more details
  22  * (a copy is included in the LICENSE file that accompanied this code).
  23  *
  24  * You should have received a copy of the GNU Lesser General Public License
  25  * version 3 along with OpenOffice.org.  If not, see
  26  * <http://www.openoffice.org/license.html>
  27  * for a copy of the LGPLv3 License.
  28  *
  29  ************************************************************************/
  30
  31
  32 // MARKER(update_precomp.py): autogen include statement, do not remove
  33 #include "precompiled_transex3.hxx"
  34 #include <transex3/directory.hxx>
  35 #include "tools/string.hxx"
  36 #include <iostream>
  37 #include <vector>
  38 #include <algorithm>
  39
  40 namespace transex
  41 {
  42
  43 Directory::Directory( const rtl::OUString sFullpath ) : bSkipLinks( false )
  44 {
  45     sFullName = sFullpath;
  46 }
  47
  48 Directory::Directory( const rtl::OUString sFullPath , const rtl::OUString sEntry ) : bSkipLinks( false )
  49 {
  50     sFullName       = sFullPath;
  51     sDirectoryName  = sEntry;
  52 }
  53
  54
  55 Directory::Directory( const ByteString sFullPath ) : bSkipLinks( false )
  56 {
  57     sDirectoryName = rtl::OUString( sFullPath.GetBuffer() , RTL_TEXTENCODING_UTF8 , sFullPath.Len() );
  58 }
  59
  60 bool Directory::lessDir ( const Directory& rKey1, const Directory& rKey2 )
  61 {
  62     rtl::OUString sName1( ( static_cast< Directory >( rKey1 ) ).getDirectoryName() );
  63     rtl::OUString sName2( ( static_cast< Directory >( rKey2 ) ).getDirectoryName() );
  64
  65     return sName1.compareTo( sName2 ) < 0 ;
  66 }
  67
  68
  69 void Directory::dump()
  70 {
  71
  72     for( std::vector< transex::File >::iterator iter = aFileVec.begin() ; iter != aFileVec.end() ; ++iter )
  73     {
  74         std::cout << "FILE " << rtl::OUStringToOString( (*iter).getFullName().getStr() , RTL_TEXTENCODING_UTF8 , (*iter).getFullName().getLength() ).getStr() << "\n";
  75     }
  76
  77     for( std::vector< transex::Directory >::iterator iter = aDirVec.begin() ; iter != aDirVec.end() ; ++iter )
  78     {
  79         std::cout << "DIR " << rtl::OUStringToOString( (*iter).getFullName().getStr() , RTL_TEXTENCODING_UTF8 , (*iter).getFullName().getLength() ).getStr() << "\n";
  80     }
  81
  82 }
  83
  84 void Directory::scanSubDir( int nLevels )
  85 {
  86     readDirectory( sFullName );
  87     dump();
  88     if( nLevels > 0 ) {
  89         for( std::vector< transex::Directory >::iterator iter = aDirVec.begin() ; iter != aDirVec.end() || nLevels > 0 ; ++iter , nLevels-- )
  90         {
  91             ( *iter ).scanSubDir();
  92         }
  93     }
  94 }
  95
  96 void Directory::setSkipLinks( bool is_skipped )
  97 {
  98     bSkipLinks = is_skipped;
  99 }
 100
 101 void Directory::readDirectory()
 102 {
 103     readDirectory( sFullName );
 104 }
 105
 106 #ifdef WNT
 107 #include <tools/prewin.h>
 108 #include <windows.h>
 109 #include <tools/postwin.h>
 110
 111 void Directory::readDirectory ( const rtl::OUString& sFullpath )
 112 {
 113     BOOL            fFinished;
 114     HANDLE          hList;
 115     TCHAR           szDir[MAX_PATH+1];
 116     TCHAR           szSubDir[MAX_PATH+1];
 117     WIN32_FIND_DATA FileData;
 118
 119     rtl::OString sFullpathext = rtl::OUStringToOString( sFullpath , RTL_TEXTENCODING_UTF8 , sFullpath.getLength() );
 120     const char *dirname = sFullpathext.getStr();
 121
 122     // Get the proper directory path
 123     sprintf(szDir, "%s\\*", dirname);
 124
 125     // Get the first file
 126     hList = FindFirstFile(szDir, &FileData);
 127     if (hList == INVALID_HANDLE_VALUE)
 128     {
 129         //FindClose(hList);
 130         //printf("No files found %s\n", szDir ); return;
 131     }
 132     else
 133     {
 134         fFinished = FALSE;
 135         while (!fFinished)
 136         {
 137
 138             sprintf(szSubDir, "%s\\%s", dirname, FileData.cFileName);
 139             rtl::OString myfile( FileData.cFileName );
 140             rtl::OString mydir( szSubDir );
 141
 142             if (FileData.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)
 143             {
 144                 if ( (strcmp(FileData.cFileName, ".") != 0 ) &&
 145                      (strcmp(FileData.cFileName, "..") != 0 ) )
 146                 {
 147                     //sprintf(szSubDir, "%s\\%s", dirname, FileData.cFileName);
 148                     transex::Directory aDir(    rtl::OStringToOUString( mydir , RTL_TEXTENCODING_UTF8 , mydir.getLength() ),
 149                                                 rtl::OStringToOUString( myfile , RTL_TEXTENCODING_UTF8 , myfile.getLength() ) );
 150                     aDirVec.push_back( aDir );
 151                 }
 152             }
 153             else
 154             {
 155                 transex::File aFile(    rtl::OStringToOUString( mydir , RTL_TEXTENCODING_UTF8 , mydir.getLength() ),
 156                                         rtl::OStringToOUString( myfile , RTL_TEXTENCODING_UTF8 , myfile.getLength() ) );
 157                 aFileVec.push_back( aFile );
 158             }
 159             if (!FindNextFile(hList, &FileData))
 160             {
 161                 if (GetLastError() == ERROR_NO_MORE_FILES)
 162                 {
 163                     fFinished = TRUE;
 164                 }
 165             }
 166         }
 167     }
 168
 169     FindClose(hList);
 170
 171     ::std::sort( aFileVec.begin() , aFileVec.end() , File::lessFile );
 172     ::std::sort( aDirVec.begin()  , aDirVec.end()  , Directory::lessDir  );
 173 }
 174
 175 #else
 176
 177 void Directory::readDirectory( const rtl::OUString& sFullpath )
 178 {
 179     struct stat     statbuf;
 180     struct stat     statbuf2;
 181     struct dirent   *dirp;
 182     DIR             *dir;
 183     //int             ret;
 184     //char            *ptr;
 185
 186     if( sFullpath.getLength() < 1 ) return;
 187
 188     rtl::OString   sFullpathext = rtl::OUStringToOString( sFullpath , RTL_TEXTENCODING_UTF8 , sFullpath.getLength() ).getStr();
 189     //printf("%s\n",sFullpathext.getStr());
 190     const char*    path         = sFullpathext.getStr();
 191
 192     // stat
 193     if( stat( path  , &statbuf ) < 0 ){   printf("warning: Can not stat %s" , path ); return; }// error }
 194
 195     if( S_ISDIR(statbuf.st_mode ) == 0 ) {  return; }// error }   return; // not dir
 196
 197     if( (dir = opendir( path ) ) == NULL  ) {printf("readerror 2 in %s \n",path); return; } // error } return; // error
 198
 199     sFullpathext += rtl::OString( "/" );
 200
 201     const rtl::OString sDot ( "." ) ;
 202     const rtl::OString sDDot( ".." );
 203
 204     chdir( path );
 205
 206     while(  ( dirp = readdir( dir ) ) != NULL )
 207     {
 208         rtl::OString sEntryName(  dirp->d_name );
 209
 210         if( sEntryName.equals( sDot )  || sEntryName.equals( sDDot ) )
 211             continue;
 212
 213         // add dir entry
 214         rtl::OString sEntity = sFullpathext;
 215         sEntity += sEntryName;
 216
 217         // stat new entry
 218         if( lstat( sEntity.getStr() , &statbuf2 ) < 0 )
 219         {
 220             printf("error on entry %s\n" , sEntity.getStr() ) ; // error
 221             continue;
 222         }
 223
 224         // add file / dir to vector
 225         switch( statbuf2.st_mode & S_IFMT )
 226         {
 227             case S_IFREG:
 228                         {
 229                             rtl::OString sFile = sFullpathext;
 230                             sFile += sEntryName ;
 231                             transex::File aFile( rtl::OStringToOUString( sEntity , RTL_TEXTENCODING_UTF8 , sEntity.getLength() ) ,
 232                                                  rtl::OStringToOUString( sEntryName , RTL_TEXTENCODING_UTF8 , sEntryName.getLength() )
 233                                                );
 234
 235                             aFileVec.push_back( aFile ) ;
 236                             break;
 237                          }
 238             case S_IFLNK:
 239                         {
 240                             if( bSkipLinks )    break;
 241                         }
 242             case S_IFDIR:
 243                         {
 244                             rtl::OString sDir = sFullpathext;
 245                             sDir += sEntryName ;
 246
 247                             transex::Directory aDir(
 248                                                      rtl::OStringToOUString( sEntity , RTL_TEXTENCODING_UTF8 , sEntity.getLength() ) ,
 249                                                      rtl::OStringToOUString( sEntryName , RTL_TEXTENCODING_UTF8 , sEntryName.getLength() )
 250                                                    ) ;
 251                             aDirVec.push_back( aDir ) ;
 252                             break;
 253                          }
 254         }
 255     }
 256     chdir( ".." );
 257     if( closedir( dir ) < 0 )   return ; // error
 258
 259     std::sort( aFileVec.begin() , aFileVec.end() , File::lessFile );
 260     std::sort( aDirVec.begin()  , aDirVec.end()  , Directory::lessDir  );
 261
 262 }
 263
 264 #endif
 265 }