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         wxUnusedVar(conv);
 139         buf[nRead] = '\0';
 140         str += buf;
 141 #endif // wxUSE_UNICODE/!wxUSE_UNICODE
 142
 143
 144         // the beginning of the current line, changes inside the loop
 145         wxString::const_iterator lineStart = str.begin();
 146         const wxString::const_iterator end = str.end();
 147         for ( wxString::const_iterator p = lineStart; p != end; p++ )
 148         {
 149             const wxChar ch = *p;
 150             switch ( ch )
 151             {
 152                 case '\n':
 153                     // could be a DOS or Unix EOL
 154                     if ( chLast == '\r' )
 155                     {
 156                         AddLine(wxString(lineStart, p - 1), wxTextFileType_Dos);
 157                     }
 158                     else // bare '\n', Unix style
 159                     {
 160                         AddLine(wxString(lineStart, p), wxTextFileType_Unix);
 161                     }
 162
 163                     lineStart = p + 1;
 164                     break;
 165
 166                 case '\r':
 167                     if ( chLast == '\r' )
 168                     {
 169                         // Mac empty line
 170                         AddLine(wxEmptyString, wxTextFileType_Mac);
 171                         lineStart = p + 1;
 172                     }
 173                     //else: we don't what this is yet -- could be a Mac EOL or
 174                     //      start of DOS EOL so wait for next char
 175                     break;
 176
 177                 default:
 178                     if ( chLast == '\r' )
 179                     {
 180                         // Mac line termination
 181                         AddLine(wxString(lineStart, p - 1), wxTextFileType_Mac);
 182                         lineStart = p;
 183                     }
 184             }
 185
 186             chLast = ch;
 187         }
 188
 189         // remove the part we already processed
 190         str.erase(0, lineStart - str.begin());
 191     }
 192
 193     // anything in the last line?
 194     if ( !str.empty() )
 195     {
 196         AddLine(str, wxTextFileType_None); // no line terminator
 197     }
 198
 199     return true;
 200 }
 201
 202
 203 bool wxTextFile::OnWrite(wxTextFileType typeNew, wxMBConv& conv)
 204 {
 205     wxFileName fn = m_strBufferName;
 206
 207     // We do NOT want wxPATH_NORM_CASE here, or the case will not
 208     // be preserved.
 209     if ( !fn.IsAbsolute() )
 210         fn.Normalize(wxPATH_NORM_ENV_VARS | wxPATH_NORM_DOTS | wxPATH_NORM_TILDE |
 211                      wxPATH_NORM_ABSOLUTE | wxPATH_NORM_LONG);
 212
 213     wxTempFile fileTmp(fn.GetFullPath());
 214
 215     if ( !fileTmp.IsOpened() ) {
 216         wxLogError(_("can't write buffer '%s' to disk."), m_strBufferName.c_str());
 217         return false;
 218     }
 219
 220     size_t nCount = GetLineCount();
 221     for ( size_t n = 0; n < nCount; n++ ) {
 222         fileTmp.Write(GetLine(n) +
 223                       GetEOL(typeNew == wxTextFileType_None ? GetLineType(n)
 224                                                             : typeNew),
 225                       conv);
 226     }
 227
 228     // replace the old file with this one
 229     return fileTmp.Commit();
 230 }
 231
 232 #endif // wxUSE_TEXTFILE
 233