mirror of
https://github.com/classilla/tenfourfox.git
synced 2024-06-07 13:53:33 +00:00
312 lines
9.3 KiB
C++
312 lines
9.3 KiB
C++
/* -*- 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 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("<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;
|
|
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("</pre>\n");
|
|
}
|
|
mBuffer.AppendLiteral("\n</body></html>");
|
|
|
|
nsCOMPtr<nsIInputStream> 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;
|
|
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;
|
|
|
|
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("<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;
|
|
}
|