270 lines
8.8 KiB
C++
270 lines
8.8 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 "nsXMLPrettyPrinter.h"
|
|
#include "nsContentUtils.h"
|
|
#include "nsICSSDeclaration.h"
|
|
#include "nsIDOMDocumentXBL.h"
|
|
#include "nsIObserver.h"
|
|
#include "nsIXSLTProcessor.h"
|
|
#include "nsSyncLoadService.h"
|
|
#include "nsPIDOMWindow.h"
|
|
#include "nsIDOMDocument.h"
|
|
#include "nsIDOMElement.h"
|
|
#include "nsIServiceManager.h"
|
|
#include "nsNetUtil.h"
|
|
#include "mozilla/dom/Element.h"
|
|
#include "nsIDOMDocumentFragment.h"
|
|
#include "nsBindingManager.h"
|
|
#include "nsXBLService.h"
|
|
#include "nsIScriptSecurityManager.h"
|
|
#include "mozilla/Preferences.h"
|
|
#include "nsIDocument.h"
|
|
#include "nsVariant.h"
|
|
#include "mozilla/dom/CustomEvent.h"
|
|
|
|
using namespace mozilla;
|
|
using namespace mozilla::dom;
|
|
|
|
NS_IMPL_ISUPPORTS(nsXMLPrettyPrinter,
|
|
nsIDocumentObserver,
|
|
nsIMutationObserver)
|
|
|
|
nsXMLPrettyPrinter::nsXMLPrettyPrinter() : mDocument(nullptr),
|
|
mUnhookPending(false)
|
|
{
|
|
}
|
|
|
|
nsXMLPrettyPrinter::~nsXMLPrettyPrinter()
|
|
{
|
|
NS_ASSERTION(!mDocument, "we shouldn't be referencing the document still");
|
|
}
|
|
|
|
nsresult
|
|
nsXMLPrettyPrinter::PrettyPrint(nsIDocument* aDocument,
|
|
bool* aDidPrettyPrint)
|
|
{
|
|
*aDidPrettyPrint = false;
|
|
|
|
// Check for iframe with display:none. Such iframes don't have presshells
|
|
if (!aDocument->GetShell()) {
|
|
return NS_OK;
|
|
}
|
|
|
|
// check if we're in an invisible iframe
|
|
nsPIDOMWindowOuter *internalWin = aDocument->GetWindow();
|
|
nsCOMPtr<Element> frameElem;
|
|
if (internalWin) {
|
|
frameElem = internalWin->GetFrameElementInternal();
|
|
}
|
|
|
|
if (frameElem) {
|
|
nsCOMPtr<nsICSSDeclaration> computedStyle;
|
|
if (nsIDocument* frameOwnerDoc = frameElem->OwnerDoc()) {
|
|
nsPIDOMWindowOuter* window = frameOwnerDoc->GetDefaultView();
|
|
if (window) {
|
|
nsCOMPtr<nsPIDOMWindowInner> innerWindow =
|
|
window->GetCurrentInnerWindow();
|
|
|
|
ErrorResult dummy;
|
|
computedStyle = innerWindow->GetComputedStyle(*frameElem,
|
|
EmptyString(),
|
|
dummy);
|
|
dummy.SuppressException();
|
|
}
|
|
}
|
|
|
|
if (computedStyle) {
|
|
nsAutoString visibility;
|
|
computedStyle->GetPropertyValue(NS_LITERAL_STRING("visibility"),
|
|
visibility);
|
|
if (!visibility.EqualsLiteral("visible")) {
|
|
|
|
return NS_OK;
|
|
}
|
|
}
|
|
}
|
|
|
|
// check the pref
|
|
if (!Preferences::GetBool("layout.xml.prettyprint", true)) {
|
|
return NS_OK;
|
|
}
|
|
|
|
// Ok, we should prettyprint. Let's do it!
|
|
*aDidPrettyPrint = true;
|
|
nsresult rv = NS_OK;
|
|
|
|
// Load the XSLT
|
|
nsCOMPtr<nsIURI> xslUri;
|
|
rv = NS_NewURI(getter_AddRefs(xslUri),
|
|
NS_LITERAL_CSTRING("chrome://global/content/xml/XMLPrettyPrint.xsl"));
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
nsCOMPtr<nsIDOMDocument> xslDocument;
|
|
rv = nsSyncLoadService::LoadDocument(xslUri, nsIContentPolicy::TYPE_XSLT,
|
|
nsContentUtils::GetSystemPrincipal(),
|
|
nsILoadInfo::SEC_ALLOW_CROSS_ORIGIN_DATA_IS_NULL,
|
|
nullptr, true, mozilla::net::RP_Default,
|
|
getter_AddRefs(xslDocument));
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
// Transform the document
|
|
nsCOMPtr<nsIXSLTProcessor> transformer =
|
|
do_CreateInstance("@mozilla.org/document-transformer;1?type=xslt", &rv);
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
rv = transformer->ImportStylesheet(xslDocument);
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
nsCOMPtr<nsIDOMDocumentFragment> resultFragment;
|
|
nsCOMPtr<nsIDOMDocument> sourceDocument = do_QueryInterface(aDocument);
|
|
rv = transformer->TransformToFragment(sourceDocument, sourceDocument,
|
|
getter_AddRefs(resultFragment));
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
//
|
|
// Apply the prettprint XBL binding.
|
|
//
|
|
// We take some shortcuts here. In particular, we don't bother invoking the
|
|
// contstructor (since the binding has no constructor), and we don't bother
|
|
// calling LoadBindingDocument because it's a chrome:// URI and thus will get
|
|
// sync loaded no matter what.
|
|
//
|
|
|
|
// Grab the XBL service.
|
|
nsXBLService* xblService = nsXBLService::GetInstance();
|
|
NS_ENSURE_TRUE(xblService, NS_ERROR_NOT_AVAILABLE);
|
|
|
|
// Compute the binding URI.
|
|
nsCOMPtr<nsIURI> bindingUri;
|
|
rv = NS_NewURI(getter_AddRefs(bindingUri),
|
|
NS_LITERAL_STRING("chrome://global/content/xml/XMLPrettyPrint.xml#prettyprint"));
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
// Compute the bound element.
|
|
nsCOMPtr<nsIContent> rootCont = aDocument->GetRootElement();
|
|
NS_ENSURE_TRUE(rootCont, NS_ERROR_UNEXPECTED);
|
|
|
|
// Grab the system principal.
|
|
nsCOMPtr<nsIPrincipal> sysPrincipal;
|
|
nsContentUtils::GetSecurityManager()->
|
|
GetSystemPrincipal(getter_AddRefs(sysPrincipal));
|
|
|
|
// Load the bindings.
|
|
RefPtr<nsXBLBinding> unused;
|
|
bool ignored;
|
|
rv = xblService->LoadBindings(rootCont, bindingUri, sysPrincipal,
|
|
getter_AddRefs(unused), &ignored);
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
// Fire an event at the bound element to pass it |resultFragment|.
|
|
RefPtr<CustomEvent> event =
|
|
NS_NewDOMCustomEvent(rootCont, nullptr, nullptr);
|
|
MOZ_ASSERT(event);
|
|
nsCOMPtr<nsIWritableVariant> resultFragmentVariant = new nsVariant();
|
|
rv = resultFragmentVariant->SetAsISupports(resultFragment);
|
|
MOZ_ASSERT(NS_SUCCEEDED(rv));
|
|
rv = event->InitCustomEvent(NS_LITERAL_STRING("prettyprint-dom-created"),
|
|
/* bubbles = */ false, /* cancelable = */ false,
|
|
/* detail = */ resultFragmentVariant);
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
event->SetTrusted(true);
|
|
bool dummy;
|
|
rv = rootCont->DispatchEvent(event, &dummy);
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
// Observe the document so we know when to switch to "normal" view
|
|
aDocument->AddObserver(this);
|
|
mDocument = aDocument;
|
|
|
|
NS_ADDREF_THIS();
|
|
|
|
return NS_OK;
|
|
}
|
|
|
|
void
|
|
nsXMLPrettyPrinter::MaybeUnhook(nsIContent* aContent)
|
|
{
|
|
// If there either aContent is null (the document-node was modified) or
|
|
// there isn't a binding parent we know it's non-anonymous content.
|
|
if ((!aContent || !aContent->GetBindingParent()) && !mUnhookPending) {
|
|
// Can't blindly to mUnhookPending after AddScriptRunner,
|
|
// since AddScriptRunner _could_ in theory run us
|
|
// synchronously
|
|
mUnhookPending = true;
|
|
nsContentUtils::AddScriptRunner(
|
|
NewRunnableMethod(this, &nsXMLPrettyPrinter::Unhook));
|
|
}
|
|
}
|
|
|
|
void
|
|
nsXMLPrettyPrinter::Unhook()
|
|
{
|
|
mDocument->RemoveObserver(this);
|
|
nsCOMPtr<Element> element = mDocument->GetDocumentElement();
|
|
|
|
if (element) {
|
|
mDocument->BindingManager()->ClearBinding(element);
|
|
}
|
|
|
|
mDocument = nullptr;
|
|
|
|
NS_RELEASE_THIS();
|
|
}
|
|
|
|
void
|
|
nsXMLPrettyPrinter::AttributeChanged(nsIDocument* aDocument,
|
|
Element* aElement,
|
|
int32_t aNameSpaceID,
|
|
nsIAtom* aAttribute,
|
|
int32_t aModType,
|
|
const nsAttrValue* aOldValue)
|
|
{
|
|
MaybeUnhook(aElement);
|
|
}
|
|
|
|
void
|
|
nsXMLPrettyPrinter::ContentAppended(nsIDocument* aDocument,
|
|
nsIContent* aContainer,
|
|
nsIContent* aFirstNewContent,
|
|
int32_t aNewIndexInContainer)
|
|
{
|
|
MaybeUnhook(aContainer);
|
|
}
|
|
|
|
void
|
|
nsXMLPrettyPrinter::ContentInserted(nsIDocument* aDocument,
|
|
nsIContent* aContainer,
|
|
nsIContent* aChild,
|
|
int32_t aIndexInContainer)
|
|
{
|
|
MaybeUnhook(aContainer);
|
|
}
|
|
|
|
void
|
|
nsXMLPrettyPrinter::ContentRemoved(nsIDocument* aDocument,
|
|
nsIContent* aContainer,
|
|
nsIContent* aChild,
|
|
int32_t aIndexInContainer,
|
|
nsIContent* aPreviousSibling)
|
|
{
|
|
MaybeUnhook(aContainer);
|
|
}
|
|
|
|
void
|
|
nsXMLPrettyPrinter::NodeWillBeDestroyed(const nsINode* aNode)
|
|
{
|
|
mDocument = nullptr;
|
|
NS_RELEASE_THIS();
|
|
}
|
|
|
|
|
|
nsresult NS_NewXMLPrettyPrinter(nsXMLPrettyPrinter** aPrinter)
|
|
{
|
|
*aPrinter = new nsXMLPrettyPrinter;
|
|
NS_ADDREF(*aPrinter);
|
|
return NS_OK;
|
|
}
|