The Tor Browser / file revision

 /* -*- 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 );  //еее 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;

 }