diff options
Diffstat (limited to 'comm/mailnews/mime/src/nsSimpleMimeConverterStub.cpp')
| -rw-r--r-- | comm/mailnews/mime/src/nsSimpleMimeConverterStub.cpp | 191 |
1 files changed, 191 insertions, 0 deletions
diff --git a/comm/mailnews/mime/src/nsSimpleMimeConverterStub.cpp b/comm/mailnews/mime/src/nsSimpleMimeConverterStub.cpp new file mode 100644 index 0000000000..0ff3bd686a --- /dev/null +++ b/comm/mailnews/mime/src/nsSimpleMimeConverterStub.cpp @@ -0,0 +1,191 @@ +/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */ +/* 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/. */ + +#include "mimecth.h" +#include "mimeobj.h" +#include "mimetext.h" +#include "mimemoz2.h" +#include "mimecom.h" +#include "nsString.h" +#include "nsComponentManagerUtils.h" +#include "nsICategoryManager.h" +#include "nsCOMPtr.h" +#include "nsISimpleMimeConverter.h" +#include "nsServiceManagerUtils.h" +#include "nsSimpleMimeConverterStub.h" +#include "nsIMailChannel.h" + +typedef struct MimeSimpleStub MimeSimpleStub; +typedef struct MimeSimpleStubClass MimeSimpleStubClass; + +struct MimeSimpleStubClass { + MimeInlineTextClass text; +}; + +struct MimeSimpleStub { + MimeInlineText text; + nsCString* buffer; + nsCOMPtr<nsISimpleMimeConverter> innerScriptable; +}; + +#define MimeSimpleStubClassInitializer(ITYPE, CSUPER) \ + { MimeInlineTextClassInitializer(ITYPE, CSUPER) } + +MimeDefClass(MimeSimpleStub, MimeSimpleStubClass, mimeSimpleStubClass, NULL); + +static int BeginGather(MimeObject* obj) { + MimeSimpleStub* ssobj = (MimeSimpleStub*)obj; + int status = ((MimeObjectClass*)XPCOM_GetmimeLeafClass())->parse_begin(obj); + + if (status < 0) return status; + + if (!obj->output_p || !obj->options || !obj->options->write_html_p) { + return 0; + } + + ssobj->buffer->Truncate(); + return 0; +} + +static int GatherLine(const char* line, int32_t length, MimeObject* obj) { + MimeSimpleStub* ssobj = (MimeSimpleStub*)obj; + + if (!obj->output_p || !obj->options || !obj->options->output_fn) { + return 0; + } + + if (!obj->options->write_html_p) + return MimeObject_write(obj, line, length, true); + + ssobj->buffer->Append(line, length); + return 0; +} + +static int EndGather(MimeObject* obj, bool abort_p) { + MimeSimpleStub* ssobj = (MimeSimpleStub*)obj; + + if (obj->closed_p) return 0; + + int status = ((MimeObjectClass*)MIME_GetmimeInlineTextClass()) + ->parse_eof(obj, abort_p); + if (status < 0) return status; + + if (ssobj->buffer->IsEmpty()) return 0; + + mime_stream_data* msd = (mime_stream_data*)(obj->options->stream_closure); + nsIChannel* channel = msd->channel; // note the lack of ref counting... + if (channel) { + nsCOMPtr<nsIURI> uri; + channel->GetURI(getter_AddRefs(uri)); + ssobj->innerScriptable->SetUri(uri); + + nsCOMPtr<nsIMailChannel> mailChannel = do_QueryInterface(channel); + ssobj->innerScriptable->SetMailChannel(mailChannel); + } + // Remove possible embedded NULL bytes. + // Parsers can't handle this but e.g. calendar invitation might contain such + // as fillers. + ssobj->buffer->StripChar('\0'); + nsCString asHTML; + nsresult rv = ssobj->innerScriptable->ConvertToHTML( + nsDependentCString(obj->content_type), *ssobj->buffer, asHTML); + if (NS_FAILED(rv)) { + NS_ASSERTION(NS_SUCCEEDED(rv), "converter failure"); + return -1; + } + + // MimeObject_write wants a non-const string for some reason, but it doesn't + // mutate it. + status = MimeObject_write(obj, asHTML.get(), asHTML.Length(), true); + if (status < 0) return status; + return 0; +} + +static int Initialize(MimeObject* obj) { + MimeSimpleStub* ssobj = (MimeSimpleStub*)obj; + + nsresult rv; + nsCOMPtr<nsICategoryManager> catman = + do_GetService(NS_CATEGORYMANAGER_CONTRACTID, &rv); + if (NS_FAILED(rv)) return -1; + + nsAutoCString contentType; // lowercase + ToLowerCase(nsDependentCString(obj->content_type), contentType); + + nsCString value; + rv = catman->GetCategoryEntry(NS_SIMPLEMIMECONVERTERS_CATEGORY, contentType, + value); + if (NS_FAILED(rv) || value.IsEmpty()) return -1; + + ssobj->innerScriptable = do_CreateInstance(value.get(), &rv); + if (NS_FAILED(rv) || !ssobj->innerScriptable) return -1; + ssobj->buffer = new nsCString(); + ((MimeObjectClass*)XPCOM_GetmimeLeafClass())->initialize(obj); + + return 0; +} + +static void Finalize(MimeObject* obj) { + MimeSimpleStub* ssobj = (MimeSimpleStub*)obj; + ssobj->innerScriptable = nullptr; + delete ssobj->buffer; +} + +static int MimeSimpleStubClassInitialize(MimeSimpleStubClass* clazz) { + MimeObjectClass* oclass = (MimeObjectClass*)clazz; + oclass->parse_begin = BeginGather; + oclass->parse_line = GatherLine; + oclass->parse_eof = EndGather; + oclass->initialize = Initialize; + oclass->finalize = Finalize; + return 0; +} + +class nsSimpleMimeConverterStub : public nsIMimeContentTypeHandler { + public: + explicit nsSimpleMimeConverterStub(const char* aContentType) + : mContentType(aContentType) {} + + NS_DECL_ISUPPORTS + + NS_IMETHOD GetContentType(char** contentType) override { + *contentType = ToNewCString(mContentType); + return *contentType ? NS_OK : NS_ERROR_OUT_OF_MEMORY; + } + NS_IMETHOD CreateContentTypeHandlerClass( + const char* contentType, contentTypeHandlerInitStruct* initString, + MimeObjectClass** objClass) override; + + private: + virtual ~nsSimpleMimeConverterStub() {} + nsCString mContentType; +}; + +NS_IMPL_ISUPPORTS(nsSimpleMimeConverterStub, nsIMimeContentTypeHandler) + +NS_IMETHODIMP +nsSimpleMimeConverterStub::CreateContentTypeHandlerClass( + const char* contentType, contentTypeHandlerInitStruct* initStruct, + MimeObjectClass** objClass) { + NS_ENSURE_ARG_POINTER(objClass); + + *objClass = (MimeObjectClass*)&mimeSimpleStubClass; + (*objClass)->superclass = (MimeObjectClass*)XPCOM_GetmimeInlineTextClass(); + NS_ENSURE_TRUE((*objClass)->superclass, NS_ERROR_UNEXPECTED); + + initStruct->force_inline_display = true; + return NS_OK; + ; +} + +nsresult MIME_NewSimpleMimeConverterStub(const char* aContentType, + nsIMimeContentTypeHandler** aResult) { + RefPtr<nsSimpleMimeConverterStub> inst = + new nsSimpleMimeConverterStub(aContentType); + NS_ENSURE_TRUE(inst, NS_ERROR_OUT_OF_MEMORY); + + inst.forget(aResult); + return NS_OK; +} |