src/common/textfile.cpp

   1 ///////////////////////////////////////////////////////////////////////////////
   2 // Name:        src/common/textfile.cpp
   3 // Purpose:     implementation of wxTextFile class
   4 // Author:      Vadim Zeitlin
   5 // Modified by:
   6 // Created:     03.04.98
   7 // RCS-ID:      $Id$
   8 // Copyright:   (c) 1998 Vadim Zeitlin <zeitlin@dptmaths.ens-cachan.fr>
   9 // Licence:     wxWindows licence
  10 ///////////////////////////////////////////////////////////////////////////////
  11
  12 // ============================================================================
  13 // headers
  14 // ============================================================================
  15
  16 #include  "wx/wxprec.h"
  17
  18 #ifdef    __BORLANDC__
  19     #pragma hdrstop
  20 #endif  //__BORLANDC__
  21
  22 #if !wxUSE_FILE || !wxUSE_TEXTBUFFER
  23     #undef wxUSE_TEXTFILE
  24     #define wxUSE_TEXTFILE 0
  25 #endif // wxUSE_FILE
  26
  27 #if wxUSE_TEXTFILE
  28
  29 #ifndef WX_PRECOMP
  30     #include "wx/string.h"
  31     #include "wx/intl.h"
  32     #include "wx/file.h"
  33     #include "wx/log.h"
  34 #endif
  35
  36 #include "wx/textfile.h"
  37 #include "wx/filename.h"
  38
  39 // ============================================================================
  40 // wxTextFile class implementation
  41 // ============================================================================
  42
  43 wxTextFile::wxTextFile(const wxString& strFileName)
  44           : wxTextBuffer(strFileName)
  45 {
  46 }
  47
  48
  49 // ----------------------------------------------------------------------------
  50 // file operations
  51 // ----------------------------------------------------------------------------
  52
  53 bool wxTextFile::OnExists() const
  54 {
  55     return wxFile::Exists(m_strBufferName);
  56 }
  57
  58
  59 bool wxTextFile::OnOpen(const wxString &strBufferName, wxTextBufferOpenMode OpenMode)
  60 {
  61     wxFile::OpenMode FileOpenMode;
  62
  63     switch ( OpenMode )
  64     {
  65         default:
  66             wxFAIL_MSG( _T("unknown open mode in wxTextFile::Open") );
  67             // fall through
  68
  69         case ReadAccess :
  70             FileOpenMode = wxFile::read;
  71             break;
  72
  73         case WriteAccess :
  74             FileOpenMode = wxFile::write;
  75             break;
  76     }
  77
  78     return m_file.Open(strBufferName.c_str(), FileOpenMode);
  79 }
  80
  81
  82 bool wxTextFile::OnClose()
  83 {
  84     return m_file.Close();
  85 }
  86
  87
  88 bool wxTextFile::OnRead(wxMBConv& conv)
  89 {
  90     // file should be opened and we must be in it's beginning
  91     wxASSERT( m_file.IsOpened() &&
  92                 (m_file.GetKind() != wxFILE_KIND_DISK || m_file.Tell() == 0) );
  93
  94     static const size_t BUF_SIZE = 1024;
  95 #if wxUSE_UNICODE
  96     static const size_t NUL_SIZE = 4;
  97 #else
  98     static const size_t NUL_SIZE = 1;
  99 #endif
 100
 101     char buf[BUF_SIZE + NUL_SIZE];
 102     wxChar chLast = '\0';
 103     wxString str;
 104
 105     for ( ;; )
 106     {
 107         // leave space for trailing NUL
 108         ssize_t nRead = m_file.Read(buf, BUF_SIZE);
 109
 110         if ( nRead == wxInvalidOffset )
 111         {
 112             // read error (error message already given in wxFile::Read)
 113             return false;
 114         }
 115
 116         if ( nRead == 0 )
 117             break;
 118
 119 #if wxUSE_UNICODE
 120         // we have to properly NUL-terminate the string for any encoding it may
 121         // use -- 4 NULs should be enough for everyone (this is why we add 4
 122         // extra bytes to the buffer)
 123         buf[nRead] =
 124         buf[nRead + 1] =
 125         buf[nRead + 2] =
 126         buf[nRead + 3] = '\0';
 127
 128         // append to the remains of the last block, don't overwrite
 129         wxString strbuf(buf, conv);
 130         if ( strbuf.empty() )
 131         {
 132             // conversion failed
 133             return false;
 134         }
 135
 136         str += strbuf;
 137 #else // ANSI
 138         buf[nRead] = '\0';
 139         str += buf;
 140 #endif // wxUSE_UNICODE/!wxUSE_UNICODE
 141
 142
 143         // the beginning of the current line, changes inside the loop
 144         const wxChar *lineStart = str.begin();
 145         const wxChar * const end = str.end();
 146         for ( const wxChar *p = lineStart; p != end; p++ )
 147         {
 148             const wxChar ch = *p;
 149             switch ( ch )
 150             {
 151                 case '\n':
 152                     // could be a DOS or Unix EOL
 153                     if ( chLast == '\r' )
 154                     {
 155                         AddLine(wxString(lineStart, p - 1), wxTextFileType_Dos);
 156                     }
 157                     else // bare '\n', Unix style
 158                     {
 159                         AddLine(wxString(lineStart, p), wxTextFileType_Unix);
 160                     }
 161
 162                     lineStart = p + 1;
 163                     break;
 164
 165                 case '\r':
 166                     if ( chLast == '\r' )
 167                     {
 168                         // Mac empty line
 169                         AddLine(wxEmptyString, wxTextFileType_Mac);
 170                         lineStart = p + 1;
 171                     }
 172                     //else: we don't what this is yet -- could be a Mac EOL or
 173                     //      start of DOS EOL so wait for next char
 174                     break;
 175
 176                 default:
 177                     if ( chLast == '\r' )
 178                     {
 179                         // Mac line termination
 180                         AddLine(wxString(lineStart, p - 1), wxTextFileType_Mac);
 181                         lineStart = p;
 182                     }
 183             }
 184
 185             chLast = ch;
 186         }
 187
 188         // remove the part we already processed
 189         str.erase(0, lineStart - str.begin());
 190     }
 191
 192     // anything in the last line?
 193     if ( !str.empty() )
 194     {
 195         AddLine(str, wxTextFileType_None); // no line terminator
 196     }
 197
 198     return true;
 199 }
 200
 201
 202 bool wxTextFile::OnWrite(wxTextFileType typeNew, wxMBConv& conv)
 203 {
 204     wxFileName fn = m_strBufferName;
 205
 206     // We do NOT want wxPATH_NORM_CASE here, or the case will not
 207     // be preserved.
 208     if ( !fn.IsAbsolute() )
 209         fn.Normalize(wxPATH_NORM_ENV_VARS | wxPATH_NORM_DOTS | wxPATH_NORM_TILDE |
 210                      wxPATH_NORM_ABSOLUTE | wxPATH_NORM_LONG);
 211
 212     wxTempFile fileTmp(fn.GetFullPath());
 213
 214     if ( !fileTmp.IsOpened() ) {
 215         wxLogError(_("can't write buffer '%s' to disk."), m_strBufferName.c_str());
 216         return false;
 217     }
 218
 219     size_t nCount = GetLineCount();
 220     for ( size_t n = 0; n < nCount; n++ ) {
 221         fileTmp.Write(GetLine(n) +
 222                       GetEOL(typeNew == wxTextFileType_None ? GetLineType(n)
 223                                                             : typeNew),
 224                       conv);
 225     }
 226
 227     // replace the old file with this one
 228     return fileTmp.Commit();
 229 }
 230
 231 #endif // wxUSE_TEXTFILE
 232