diff -r 000000000000 -r 6474c204b198 netwerk/streamconv/converters/nsTXTToHTMLConv.cpp --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/netwerk/streamconv/converters/nsTXTToHTMLConv.cpp Wed Dec 31 06:09:35 2014 +0100 @@ -0,0 +1,309 @@ +/* -*- 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 + +#define TOKEN_DELIMITERS MOZ_UTF16("\t\r\n ") + +// 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("\n"); + mBuffer.Append(mPageTitle); + mBuffer.AppendLiteral("\n\n"); + if (mPreFormatHTML) { // Use 
 tags
+        mBuffer.AppendLiteral("
\n");
+    }
+
+    // Push mBuffer to the listener now, so the initial HTML will not
+    // be parsed in OnDataAvailable().
+
+    nsCOMPtr 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 inputData;
+    rv = NS_NewStringInputStream(getter_AddRefs(inputData), mBuffer);
+    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("
\n");
+    }
+    mBuffer.AppendLiteral("\n");
+
+    nsCOMPtr inputData;
+
+    rv = NS_NewStringInputStream(getter_AddRefs(inputData), mBuffer);
+    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;
+    nsAutoArrayPtr buffer(new 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, aCount-amtRead, &read);
+        if (NS_FAILED(rv)) return rv;
+
+        buffer[read] = '\0';
+        // XXX charsets?? non-latin1 characters?? utf-16??
+        AppendASCIItoUTF16(buffer, 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 inputData;
+
+            rv = NS_NewStringInputStream(getter_AddRefs(inputData), pushBuffer);
+            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("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(""), cursor);
+        cursor += 4;
+    }
+    mToken = nullptr; // indicates completeness
+    return cursor;
+}