256 lines
7.5 KiB
C++
256 lines
7.5 KiB
C++
/* -*- 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 "nsHTMLFormatConverter.h"
|
||
|
||
#include "nsCRT.h"
|
||
#include "nsISupportsArray.h"
|
||
#include "nsIComponentManager.h"
|
||
#include "nsCOMPtr.h"
|
||
#include "nsXPCOM.h"
|
||
#include "nsISupportsPrimitives.h"
|
||
|
||
#include "nsITransferable.h" // for mime defs, this is BAD
|
||
|
||
// HTML convertor stuff
|
||
#include "nsPrimitiveHelpers.h"
|
||
#include "nsIDocumentEncoder.h"
|
||
#include "nsContentUtils.h"
|
||
|
||
nsHTMLFormatConverter::nsHTMLFormatConverter()
|
||
{
|
||
}
|
||
|
||
nsHTMLFormatConverter::~nsHTMLFormatConverter()
|
||
{
|
||
}
|
||
|
||
NS_IMPL_ISUPPORTS(nsHTMLFormatConverter, nsIFormatConverter)
|
||
|
||
//
|
||
// GetInputDataFlavors
|
||
//
|
||
// Creates a new list and returns the list of all the flavors this converter
|
||
// knows how to import. In this case, it's just HTML.
|
||
//
|
||
// Flavors (strings) are wrapped in a primitive object so that JavaScript can
|
||
// access them easily via XPConnect.
|
||
//
|
||
NS_IMETHODIMP
|
||
nsHTMLFormatConverter::GetInputDataFlavors(nsISupportsArray **_retval)
|
||
{
|
||
if ( !_retval )
|
||
return NS_ERROR_INVALID_ARG;
|
||
|
||
nsresult rv = NS_NewISupportsArray ( _retval ); // addrefs for us
|
||
if ( NS_SUCCEEDED(rv) )
|
||
rv = AddFlavorToList ( *_retval, kHTMLMime );
|
||
|
||
return rv;
|
||
|
||
} // GetInputDataFlavors
|
||
|
||
|
||
//
|
||
// GetOutputDataFlavors
|
||
//
|
||
// Creates a new list and returns the list of all the flavors this converter
|
||
// knows how to export (convert). In this case, it's all sorts of things that HTML can be
|
||
// converted to.
|
||
//
|
||
// Flavors (strings) are wrapped in a primitive object so that JavaScript can
|
||
// access them easily via XPConnect.
|
||
//
|
||
NS_IMETHODIMP
|
||
nsHTMLFormatConverter::GetOutputDataFlavors(nsISupportsArray **_retval)
|
||
{
|
||
if ( !_retval )
|
||
return NS_ERROR_INVALID_ARG;
|
||
|
||
nsresult rv = NS_NewISupportsArray ( _retval ); // addrefs for us
|
||
if ( NS_SUCCEEDED(rv) ) {
|
||
rv = AddFlavorToList ( *_retval, kHTMLMime );
|
||
if ( NS_FAILED(rv) )
|
||
return rv;
|
||
#if NOT_NOW
|
||
// pinkerton
|
||
// no one uses this flavor right now, so it's just slowing things down. If anyone cares I
|
||
// can put it back in.
|
||
rv = AddFlavorToList ( *_retval, kAOLMailMime );
|
||
if ( NS_FAILED(rv) )
|
||
return rv;
|
||
#endif
|
||
rv = AddFlavorToList ( *_retval, kUnicodeMime );
|
||
if ( NS_FAILED(rv) )
|
||
return rv;
|
||
}
|
||
return rv;
|
||
|
||
} // GetOutputDataFlavors
|
||
|
||
|
||
//
|
||
// AddFlavorToList
|
||
//
|
||
// Convenience routine for adding a flavor wrapped in an nsISupportsCString object
|
||
// to a list
|
||
//
|
||
nsresult
|
||
nsHTMLFormatConverter :: AddFlavorToList ( nsISupportsArray* inList, const char* inFlavor )
|
||
{
|
||
nsresult rv;
|
||
|
||
nsCOMPtr<nsISupportsCString> dataFlavor =
|
||
do_CreateInstance(NS_SUPPORTS_CSTRING_CONTRACTID, &rv);
|
||
if ( dataFlavor ) {
|
||
dataFlavor->SetData ( nsDependentCString(inFlavor) );
|
||
// add to list as an nsISupports so the correct interface gets the addref
|
||
// in AppendElement()
|
||
nsCOMPtr<nsISupports> genericFlavor ( do_QueryInterface(dataFlavor) );
|
||
inList->AppendElement ( genericFlavor);
|
||
}
|
||
return rv;
|
||
|
||
} // AddFlavorToList
|
||
|
||
|
||
//
|
||
// CanConvert
|
||
//
|
||
// Determines if we support the given conversion. Currently, this method only
|
||
// converts from HTML to others.
|
||
//
|
||
NS_IMETHODIMP
|
||
nsHTMLFormatConverter::CanConvert(const char *aFromDataFlavor, const char *aToDataFlavor, bool *_retval)
|
||
{
|
||
if ( !_retval )
|
||
return NS_ERROR_INVALID_ARG;
|
||
|
||
*_retval = false;
|
||
if ( !nsCRT::strcmp(aFromDataFlavor, kHTMLMime) ) {
|
||
if ( !nsCRT::strcmp(aToDataFlavor, kHTMLMime) )
|
||
*_retval = true;
|
||
else if ( !nsCRT::strcmp(aToDataFlavor, kUnicodeMime) )
|
||
*_retval = true;
|
||
#if NOT_NOW
|
||
// pinkerton
|
||
// no one uses this flavor right now, so it's just slowing things down. If anyone cares I
|
||
// can put it back in.
|
||
else if ( toFlavor.Equals(kAOLMailMime) )
|
||
*_retval = true;
|
||
#endif
|
||
}
|
||
return NS_OK;
|
||
|
||
} // CanConvert
|
||
|
||
|
||
|
||
//
|
||
// Convert
|
||
//
|
||
// Convert data from one flavor to another. The data is wrapped in primitive objects so that it is
|
||
// accessible from JS. Currently, this only accepts HTML input, so anything else is invalid.
|
||
//
|
||
//XXX This method copies the data WAAAAY too many time for my liking. Grrrrrr. Mostly it's because
|
||
//XXX we _must_ put things into nsStrings so that the parser will accept it. Lame lame lame lame. We
|
||
//XXX also can't just get raw unicode out of the nsString, so we have to allocate heap to get
|
||
//XXX unicode out of the string. Lame lame lame.
|
||
//
|
||
NS_IMETHODIMP
|
||
nsHTMLFormatConverter::Convert(const char *aFromDataFlavor, nsISupports *aFromData, uint32_t aDataLen,
|
||
const char *aToDataFlavor, nsISupports **aToData, uint32_t *aDataToLen)
|
||
{
|
||
if ( !aToData || !aDataToLen )
|
||
return NS_ERROR_INVALID_ARG;
|
||
|
||
nsresult rv = NS_OK;
|
||
*aToData = nullptr;
|
||
*aDataToLen = 0;
|
||
|
||
if ( !nsCRT::strcmp(aFromDataFlavor, kHTMLMime) ) {
|
||
nsAutoCString toFlavor ( aToDataFlavor );
|
||
|
||
// HTML on clipboard is going to always be double byte so it will be in a primitive
|
||
// class of nsISupportsString. Also, since the data is in two byte chunks the
|
||
// length represents the length in 1-byte chars, so we need to divide by two.
|
||
nsCOMPtr<nsISupportsString> dataWrapper0 ( do_QueryInterface(aFromData) );
|
||
if (!dataWrapper0) {
|
||
return NS_ERROR_INVALID_ARG;
|
||
}
|
||
|
||
nsAutoString dataStr;
|
||
dataWrapper0->GetData ( dataStr ); //<2F><><EFBFBD> COPY #1
|
||
// note: conversion to text/plain is done inside the clipboard. we do not need to worry
|
||
// about it here.
|
||
if ( toFlavor.Equals(kHTMLMime) || toFlavor.Equals(kUnicodeMime) ) {
|
||
nsresult res;
|
||
if (toFlavor.Equals(kHTMLMime)) {
|
||
int32_t dataLen = dataStr.Length() * 2;
|
||
nsPrimitiveHelpers::CreatePrimitiveForData ( toFlavor.get(), dataStr.get(), dataLen, aToData );
|
||
if ( *aToData )
|
||
*aDataToLen = dataLen;
|
||
} else {
|
||
nsAutoString outStr;
|
||
res = ConvertFromHTMLToUnicode(dataStr, outStr);
|
||
if (NS_SUCCEEDED(res)) {
|
||
int32_t dataLen = outStr.Length() * 2;
|
||
nsPrimitiveHelpers::CreatePrimitiveForData ( toFlavor.get(), outStr.get(), dataLen, aToData );
|
||
if ( *aToData )
|
||
*aDataToLen = dataLen;
|
||
}
|
||
}
|
||
} // else if HTML or Unicode
|
||
else if ( toFlavor.Equals(kAOLMailMime) ) {
|
||
nsAutoString outStr;
|
||
if ( NS_SUCCEEDED(ConvertFromHTMLToAOLMail(dataStr, outStr)) ) {
|
||
int32_t dataLen = outStr.Length() * 2;
|
||
nsPrimitiveHelpers::CreatePrimitiveForData ( toFlavor.get(), outStr.get(), dataLen, aToData );
|
||
if ( *aToData )
|
||
*aDataToLen = dataLen;
|
||
}
|
||
} // else if AOL mail
|
||
else {
|
||
rv = NS_ERROR_FAILURE;
|
||
}
|
||
} // if we got html mime
|
||
else
|
||
rv = NS_ERROR_FAILURE;
|
||
|
||
return rv;
|
||
|
||
} // Convert
|
||
|
||
|
||
//
|
||
// ConvertFromHTMLToUnicode
|
||
//
|
||
// Takes HTML and converts it to plain text but in unicode.
|
||
//
|
||
NS_IMETHODIMP
|
||
nsHTMLFormatConverter::ConvertFromHTMLToUnicode(const nsAutoString & aFromStr, nsAutoString & aToStr)
|
||
{
|
||
return nsContentUtils::ConvertToPlainText(aFromStr,
|
||
aToStr,
|
||
nsIDocumentEncoder::OutputSelectionOnly |
|
||
nsIDocumentEncoder::OutputAbsoluteLinks |
|
||
nsIDocumentEncoder::OutputNoScriptContent |
|
||
nsIDocumentEncoder::OutputNoFramesContent,
|
||
0);
|
||
} // ConvertFromHTMLToUnicode
|
||
|
||
|
||
NS_IMETHODIMP
|
||
nsHTMLFormatConverter::ConvertFromHTMLToAOLMail(const nsAutoString & aFromStr,
|
||
nsAutoString & aToStr)
|
||
{
|
||
aToStr.AssignLiteral("<HTML>");
|
||
aToStr.Append(aFromStr);
|
||
aToStr.AppendLiteral("</HTML>");
|
||
|
||
return NS_OK;
|
||
}
|
||
|