summaryrefslogtreecommitdiffstats
path: root/netwerk/streamconv/converters/nsTXTToHTMLConv.cpp
diff options
context:
space:
mode:
authorMatt A. Tobin <mattatobin@localhost.localdomain>2018-02-02 04:16:08 -0500
committerMatt A. Tobin <mattatobin@localhost.localdomain>2018-02-02 04:16:08 -0500
commit5f8de423f190bbb79a62f804151bc24824fa32d8 (patch)
tree10027f336435511475e392454359edea8e25895d /netwerk/streamconv/converters/nsTXTToHTMLConv.cpp
parent49ee0794b5d912db1f95dce6eb52d781dc210db5 (diff)
downloadUXP-5f8de423f190bbb79a62f804151bc24824fa32d8.tar
UXP-5f8de423f190bbb79a62f804151bc24824fa32d8.tar.gz
UXP-5f8de423f190bbb79a62f804151bc24824fa32d8.tar.lz
UXP-5f8de423f190bbb79a62f804151bc24824fa32d8.tar.xz
UXP-5f8de423f190bbb79a62f804151bc24824fa32d8.zip
Add m-esr52 at 52.6.0
Diffstat (limited to 'netwerk/streamconv/converters/nsTXTToHTMLConv.cpp')
-rw-r--r--netwerk/streamconv/converters/nsTXTToHTMLConv.cpp314
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("&lt;");
+ 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("&gt;");
+ 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("&amp;");
+ 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;
+}