diff options
Diffstat (limited to 'comm/mailnews/import/src/ImportOutFile.h')
-rw-r--r-- | comm/mailnews/import/src/ImportOutFile.h | 94 |
1 files changed, 94 insertions, 0 deletions
diff --git a/comm/mailnews/import/src/ImportOutFile.h b/comm/mailnews/import/src/ImportOutFile.h new file mode 100644 index 0000000000..f682c4dc4f --- /dev/null +++ b/comm/mailnews/import/src/ImportOutFile.h @@ -0,0 +1,94 @@ +/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */ +/* This Source Code Form is subject to the terms of the Mozilla Public + * License, v. 2.0. If a copy of the MPL was not distributed with this + * file, You can obtain one at http://mozilla.org/MPL/2.0/. */ + +#ifndef ImportOutFile_h___ +#define ImportOutFile_h___ + +#include "nsImportTranslator.h" +#include "nsIOutputStream.h" +#include "nsIFile.h" + +#define kMaxMarkers 10 + +class ImportOutFile; + +class ImportOutFile { + public: + ImportOutFile(); + ImportOutFile(nsIFile* pFile, uint8_t* pBuf, uint32_t sz); + ~ImportOutFile(); + + bool InitOutFile(nsIFile* pFile, uint32_t bufSz = 4096); + void InitOutFile(nsIFile* pFile, uint8_t* pBuf, uint32_t sz); + inline bool WriteData(const uint8_t* pSrc, uint32_t len); + inline bool WriteByte(uint8_t byte); + bool WriteStr(const char* pStr) { + return WriteU8NullTerm((const uint8_t*)pStr, false); + } + bool WriteU8NullTerm(const uint8_t* pSrc, bool includeNull); + bool WriteEol(void) { return WriteStr("\x0D\x0A"); } + bool Done(void) { return Flush(); } + + // Marker support + bool SetMarker(int markerID); + void ClearMarker(int markerID); + bool WriteStrAtMarker(int markerID, const char* pStr); + + // 8-bit to 7-bit translation + bool Set8bitTranslator(nsImportTranslator* pTrans); + bool End8bitTranslation(bool* pEngaged, nsCString& useCharset, + nsCString& encoding); + + protected: + bool Flush(void); + + protected: + nsCOMPtr<nsIFile> m_pFile; + nsCOMPtr<nsIOutputStream> m_outputStream; + uint8_t* m_pBuf; + uint32_t m_bufSz; + uint32_t m_pos; + bool m_ownsFileAndBuffer; + + // markers + uint32_t m_markers[kMaxMarkers]; + + // 8 bit to 7 bit translations + nsImportTranslator* m_pTrans; + bool m_engaged; + bool m_supports8to7; + ImportOutFile* m_pTransOut; + uint8_t* m_pTransBuf; +}; + +inline bool ImportOutFile::WriteData(const uint8_t* pSrc, uint32_t len) { + while ((len + m_pos) > m_bufSz) { + if ((m_bufSz - m_pos)) { + memcpy(m_pBuf + m_pos, pSrc, m_bufSz - m_pos); + len -= (m_bufSz - m_pos); + pSrc += (m_bufSz - m_pos); + m_pos = m_bufSz; + } + if (!Flush()) return false; + } + + if (len) { + memcpy(m_pBuf + m_pos, pSrc, len); + m_pos += len; + } + + return true; +} + +inline bool ImportOutFile::WriteByte(uint8_t byte) { + if (m_pos == m_bufSz) { + if (!Flush()) return false; + } + *(m_pBuf + m_pos) = byte; + m_pos++; + return true; +} + +#endif /* ImportOutFile_h__ */ |