1 files changed, 105 insertions, 0 deletions
diff --git a/src/core/basetypes/MCDataStreamDecoder.cpp b/src/core/basetypes/MCDataStreamDecoder.cpp
new file mode 100644
index 00000000..120cfb65
--- /dev/null
+++ b/src/core/basetypes/MCDataStreamDecoder.cpp
@@ -0,0 +1,105 @@
+#include "MCDataStreamDecoder.h"
+
+#include "MCString.h"
+#include "MCUtils.h"
+#include "MCDataDecoderUtils.h"
+
+using namespace mailcore;
+
+DataStreamDecoder::DataStreamDecoder()
+{
+    mFilename = NULL;
+    mEncoding = Encoding7Bit;
+    mRemainingData = NULL;
+    mFile = NULL;
+}
+
+DataStreamDecoder::~DataStreamDecoder()
+{
+    MC_SAFE_RELEASE(mRemainingData);
+    MC_SAFE_RELEASE(mFilename);
+    if (mFile != NULL) {
+        fclose(mFile);
+        mFile = NULL;
+    }
+}
+
+void DataStreamDecoder::setEncoding(Encoding encoding)
+{
+    mEncoding = encoding;
+}
+
+void DataStreamDecoder::setFilename(String * filename)
+{
+    MC_SAFE_REPLACE_COPY(String, mFilename, filename);
+}
+
+ErrorCode DataStreamDecoder::appendData(Data * data)
+{
+    Data * dataForDecode;
+    if (mRemainingData && mRemainingData->length()) {
+        // the data remains from previous append
+        dataForDecode = (Data *) MC_SAFE_COPY(mRemainingData);
+        dataForDecode->appendData(data);
+    } else {
+        dataForDecode = (Data *) MC_SAFE_RETAIN(data);
+    }
+
+    Data * remainingData = NULL;
+    Data * decodedData = MCDecodeData(dataForDecode, mEncoding, true, &remainingData);
+
+    ErrorCode errorCode = appendDecodedData(decodedData);
+
+    if (errorCode == ErrorNone) {
+        MC_SAFE_REPLACE_RETAIN(Data, mRemainingData, remainingData);
+    }
+
+    MC_SAFE_RELEASE(dataForDecode);
+    return errorCode;
+}
+
+ErrorCode DataStreamDecoder::flushData()
+{
+    if (mRemainingData == NULL || mRemainingData->length() == 0) {
+        return ErrorNone;
+    }
+
+    Data * unused = NULL;
+    Data * decodedData = MCDecodeData(mRemainingData, mEncoding, false, &unused);
+
+    ErrorCode errorCode = appendDecodedData(decodedData);
+
+    if (errorCode == ErrorNone) {
+        if (mFile != NULL) {
+            if (fclose(mFile) != 0) {
+                return ErrorFile;
+            }
+        }
+
+        MC_SAFE_RELEASE(mRemainingData);
+    }
+
+    return errorCode;
+}
+
+ErrorCode DataStreamDecoder::appendDecodedData(Data * decodedData)
+{
+    if (mFilename == NULL) {
+        return ErrorFile;
+    }
+
+    if (mFile == NULL) {
+        mFile = fopen(mFilename->fileSystemRepresentation(), "wb");
+
+        if (mFile == NULL) {
+            return ErrorFile;
+        }
+    }
+
+    size_t result = fwrite(decodedData->bytes(), decodedData->length(), 1, mFile);
+    if (result == 0) {
+        return ErrorFile;
+    }
+
+    return ErrorNone;
+}