diff options
Diffstat (limited to 'netwerk/streamconv/converters/nsTXTToHTMLConv.cpp')
| -rw-r--r-- | netwerk/streamconv/converters/nsTXTToHTMLConv.cpp | 314 |
1 files changed, 314 insertions, 0 deletions
diff --git a/netwerk/streamconv/converters/nsTXTToHTMLConv.cpp b/netwerk/streamconv/converters/nsTXTToHTMLConv.cpp new file mode 100644 index 000000000..ded3e9e6c --- /dev/null +++ b/netwerk/streamconv/converters/nsTXTToHTMLConv.cpp @@ -0,0 +1,314 @@ +/* -*- 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/. */ + +#include "nsTXTToHTMLConv.h" +#include "nsEscape.h" +#include "nsStringStream.h" +#include "nsAutoPtr.h" +#include "nsIChannel.h" +#include <algorithm> + +#include "mozilla/UniquePtrExtensions.h" + +#define TOKEN_DELIMITERS u"\t\r\n " + +using namespace mozilla; + +// nsISupports methods +NS_IMPL_ISUPPORTS(nsTXTToHTMLConv, + nsIStreamConverter, + nsITXTToHTMLConv, + nsIRequestObserver, + nsIStreamListener) + + +// nsIStreamConverter methods +NS_IMETHODIMP +nsTXTToHTMLConv::Convert(nsIInputStream *aFromStream, + const char *aFromType, const char *aToType, + nsISupports *aCtxt, nsIInputStream * *_retval) +{ + return NS_ERROR_NOT_IMPLEMENTED; +} + +NS_IMETHODIMP +nsTXTToHTMLConv::AsyncConvertData(const char *aFromType, + const char *aToType, + nsIStreamListener *aListener, + nsISupports *aCtxt) +{ + NS_ASSERTION(aListener, "null pointer"); + mListener = aListener; + return NS_OK; +} + + +// nsIRequestObserver methods +NS_IMETHODIMP +nsTXTToHTMLConv::OnStartRequest(nsIRequest* request, nsISupports *aContext) +{ + mBuffer.AssignLiteral("<html>\n<head><title>"); + mBuffer.Append(mPageTitle); + mBuffer.AppendLiteral("</title></head>\n<body>\n"); + if (mPreFormatHTML) { // Use <pre> tags + mBuffer.AppendLiteral("<pre>\n"); + } + + // Push mBuffer to the listener now, so the initial HTML will not + // be parsed in OnDataAvailable(). + + nsCOMPtr<nsIChannel> channel = do_QueryInterface(request); + if (channel) + channel->SetContentType(NS_LITERAL_CSTRING("text/html")); + // else, assume there is a channel somewhere that knows what it is doing! + + nsresult rv = mListener->OnStartRequest(request, aContext); + if (NS_FAILED(rv)) return rv; + + // The request may have been canceled, and if that happens, we want to + // suppress calls to OnDataAvailable. + request->GetStatus(&rv); + if (NS_FAILED(rv)) return rv; + + nsCOMPtr<nsIInputStream> inputData; + NS_LossyConvertUTF16toASCII asciiData(mBuffer); + rv = NS_NewCStringInputStream(getter_AddRefs(inputData), asciiData); + if (NS_FAILED(rv)) return rv; + + rv = mListener->OnDataAvailable(request, aContext, + inputData, 0, mBuffer.Length()); + if (NS_FAILED(rv)) return rv; + mBuffer.Truncate(); + return rv; +} + +NS_IMETHODIMP +nsTXTToHTMLConv::OnStopRequest(nsIRequest* request, nsISupports *aContext, + nsresult aStatus) +{ + nsresult rv = NS_OK; + if (mToken) { + // we still have an outstanding token + NS_ASSERTION(mToken->prepend, + "Non prepending tokens should be handled in " + "OnDataAvailable. There should only be a single " + "prepending token left to be processed."); + (void)CatHTML(0, mBuffer.Length()); + } + if (mPreFormatHTML) { + mBuffer.AppendLiteral("</pre>\n"); + } + mBuffer.AppendLiteral("\n</body></html>"); + + nsCOMPtr<nsIInputStream> inputData; + NS_LossyConvertUTF16toASCII asciiData(mBuffer); + rv = NS_NewCStringInputStream(getter_AddRefs(inputData), asciiData); + if (NS_FAILED(rv)) return rv; + + rv = mListener->OnDataAvailable(request, aContext, + inputData, 0, mBuffer.Length()); + if (NS_FAILED(rv)) return rv; + + return mListener->OnStopRequest(request, aContext, aStatus); +} + +// nsITXTToHTMLConv methods +NS_IMETHODIMP +nsTXTToHTMLConv::SetTitle(const char16_t *aTitle) +{ + mPageTitle.Assign(aTitle); + return NS_OK; +} + +NS_IMETHODIMP +nsTXTToHTMLConv::PreFormatHTML(bool value) +{ + mPreFormatHTML = value; + return NS_OK; +} + +// nsIStreamListener method +NS_IMETHODIMP +nsTXTToHTMLConv::OnDataAvailable(nsIRequest* request, nsISupports *aContext, + nsIInputStream *aInStream, + uint64_t aOffset, uint32_t aCount) +{ + nsresult rv = NS_OK; + nsString pushBuffer; + uint32_t amtRead = 0; + auto buffer = MakeUniqueFallible<char[]>(aCount+1); + if (!buffer) return NS_ERROR_OUT_OF_MEMORY; + + do { + uint32_t read = 0; + // XXX readSegments, to avoid the first copy? + rv = aInStream->Read(buffer.get(), aCount-amtRead, &read); + if (NS_FAILED(rv)) return rv; + + buffer[read] = '\0'; + // XXX charsets?? non-latin1 characters?? utf-16?? + AppendASCIItoUTF16(buffer.get(), mBuffer); + amtRead += read; + + int32_t front = -1, back = -1, tokenLoc = -1, cursor = 0; + + while ( (tokenLoc = FindToken(cursor, &mToken)) > -1) { + if (mToken->prepend) { + front = mBuffer.RFindCharInSet(TOKEN_DELIMITERS, tokenLoc); + front++; + back = mBuffer.FindCharInSet(TOKEN_DELIMITERS, tokenLoc); + } else { + front = tokenLoc; + back = front + mToken->token.Length(); + } + if (back == -1) { + // didn't find an ending, buffer up. + mBuffer.Left(pushBuffer, front); + cursor = front; + break; + } + // found the end of the token. + cursor = CatHTML(front, back); + } + + int32_t end = mBuffer.RFind(TOKEN_DELIMITERS, mBuffer.Length()); + mBuffer.Left(pushBuffer, std::max(cursor, end)); + mBuffer.Cut(0, std::max(cursor, end)); + cursor = 0; + + if (!pushBuffer.IsEmpty()) { + nsCOMPtr<nsIInputStream> inputData; + NS_LossyConvertUTF16toASCII asciiData(pushBuffer); + rv = NS_NewCStringInputStream(getter_AddRefs(inputData), asciiData); + if (NS_FAILED(rv)) + return rv; + + rv = mListener->OnDataAvailable(request, aContext, + inputData, 0, pushBuffer.Length()); + if (NS_FAILED(rv)) + return rv; + } + } while (amtRead < aCount); + + return rv; +} + +// nsTXTToHTMLConv methods +nsTXTToHTMLConv::nsTXTToHTMLConv() +{ + mToken = nullptr; + mPreFormatHTML = false; +} + +nsTXTToHTMLConv::~nsTXTToHTMLConv() +{ + mTokens.Clear(); +} + +nsresult +nsTXTToHTMLConv::Init() +{ + nsresult rv = NS_OK; + + // build up the list of tokens to handle + convToken *token = new convToken; + if (!token) return NS_ERROR_OUT_OF_MEMORY; + token->prepend = false; + token->token.Assign(char16_t('<')); + token->modText.AssignLiteral("<"); + mTokens.AppendElement(token); + + token = new convToken; + if (!token) return NS_ERROR_OUT_OF_MEMORY; + token->prepend = false; + token->token.Assign(char16_t('>')); + token->modText.AssignLiteral(">"); + mTokens.AppendElement(token); + + token = new convToken; + if (!token) return NS_ERROR_OUT_OF_MEMORY; + token->prepend = false; + token->token.Assign(char16_t('&')); + token->modText.AssignLiteral("&"); + mTokens.AppendElement(token); + + token = new convToken; + if (!token) return NS_ERROR_OUT_OF_MEMORY; + token->prepend = true; + token->token.AssignLiteral("http://"); // XXX need to iterate through all protos + mTokens.AppendElement(token); + + token = new convToken; + if (!token) return NS_ERROR_OUT_OF_MEMORY; + token->prepend = true; + token->token.Assign(char16_t('@')); + token->modText.AssignLiteral("mailto:"); + mTokens.AppendElement(token); + + return rv; +} + +int32_t +nsTXTToHTMLConv::FindToken(int32_t cursor, convToken* *_retval) +{ + int32_t loc = -1, firstToken = mBuffer.Length(); + int8_t token = -1; + for (uint8_t i=0; i < mTokens.Length(); i++) { + loc = mBuffer.Find(mTokens[i]->token, cursor); + if (loc != -1) + if (loc < firstToken) { + firstToken = loc; + token = i; + } + } + if (token == -1) + return -1; + + *_retval = mTokens[token]; + return firstToken; +} + +int32_t +nsTXTToHTMLConv::CatHTML(int32_t front, int32_t back) +{ + int32_t cursor = 0; + int32_t modLen = mToken->modText.Length(); + if (!mToken->prepend) { + // replace the entire token (from delimiter to delimiter) + mBuffer.Cut(front, back - front); + mBuffer.Insert(mToken->modText, front); + cursor = front+modLen; + } else { + nsString linkText; + // href is implied + mBuffer.Mid(linkText, front, back-front); + + mBuffer.Insert(NS_LITERAL_STRING("<a href=\""), front); + cursor += front+9; + if (modLen) { + mBuffer.Insert(mToken->modText, cursor); + cursor += modLen; + } + + NS_ConvertUTF16toUTF8 linkTextUTF8(linkText); + nsCString escaped; + if (NS_EscapeURL(linkTextUTF8.Data(), linkTextUTF8.Length(), esc_Minimal, escaped)) { + mBuffer.Cut(cursor, back - front); + CopyUTF8toUTF16(escaped, linkText); + mBuffer.Insert(linkText, cursor); + back = front + linkText.Length(); + } + + cursor += back-front; + mBuffer.Insert(NS_LITERAL_STRING("\">"), cursor); + cursor += 2; + mBuffer.Insert(linkText, cursor); + cursor += linkText.Length(); + mBuffer.Insert(NS_LITERAL_STRING("</a>"), cursor); + cursor += 4; + } + mToken = nullptr; // indicates completeness + return cursor; +} |