michael@0: /* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
michael@0: /* This Source Code Form is subject to the terms of the Mozilla Public
michael@0:  * License, v. 2.0. If a copy of the MPL was not distributed with this
michael@0:  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
michael@0: 
michael@0: 
michael@0: #include <stddef.h>                     // for nullptr
michael@0: 
michael@0: #include "nsAString.h"
michael@0: #include "nsCOMPtr.h"                   // for nsCOMPtr, do_QueryInterface, etc
michael@0: #include "nsCSSStyleSheet.h"            // for nsCSSStyleSheet
michael@0: #include "nsDebug.h"                    // for NS_ENSURE_TRUE
michael@0: #include "nsError.h"                    // for NS_OK, etc
michael@0: #include "nsIDOMDocument.h"             // for nsIDOMDocument
michael@0: #include "nsIDocument.h"                // for nsIDocument
michael@0: #include "nsIDocumentObserver.h"        // for UPDATE_STYLE
michael@0: #include "nsIEditor.h"                  // for nsIEditor
michael@0: #include "nsStyleSheetTxns.h"
michael@0: 
michael@0: class nsIStyleSheet;
michael@0: 
michael@0: static void
michael@0: AddStyleSheet(nsIEditor* aEditor, nsIStyleSheet* aSheet)
michael@0: {
michael@0:   nsCOMPtr<nsIDOMDocument> domDoc;
michael@0:   aEditor->GetDocument(getter_AddRefs(domDoc));
michael@0:   nsCOMPtr<nsIDocument> doc = do_QueryInterface(domDoc);
michael@0:   if (doc) {
michael@0:     doc->BeginUpdate(UPDATE_STYLE);
michael@0:     doc->AddStyleSheet(aSheet);
michael@0:     doc->EndUpdate(UPDATE_STYLE);
michael@0:   }
michael@0: }
michael@0: 
michael@0: static void
michael@0: RemoveStyleSheet(nsIEditor *aEditor, nsIStyleSheet *aSheet)
michael@0: {
michael@0:   nsCOMPtr<nsIDOMDocument> domDoc;
michael@0:   aEditor->GetDocument(getter_AddRefs(domDoc));
michael@0:   nsCOMPtr<nsIDocument> doc = do_QueryInterface(domDoc);
michael@0:   if (doc) {
michael@0:     doc->BeginUpdate(UPDATE_STYLE);
michael@0:     doc->RemoveStyleSheet(aSheet);
michael@0:     doc->EndUpdate(UPDATE_STYLE);
michael@0:   }
michael@0: }
michael@0: 
michael@0: AddStyleSheetTxn::AddStyleSheetTxn()
michael@0: :  EditTxn()
michael@0: ,  mEditor(nullptr)
michael@0: {
michael@0: }
michael@0: 
michael@0: NS_IMPL_CYCLE_COLLECTION_INHERITED(AddStyleSheetTxn, EditTxn,
michael@0:                                    mSheet)
michael@0: 
michael@0: NS_INTERFACE_MAP_BEGIN_CYCLE_COLLECTION(AddStyleSheetTxn)
michael@0: NS_INTERFACE_MAP_END_INHERITING(EditTxn)
michael@0: 
michael@0: NS_IMETHODIMP
michael@0: AddStyleSheetTxn::Init(nsIEditor *aEditor, nsCSSStyleSheet *aSheet)
michael@0: {
michael@0:   NS_ENSURE_TRUE(aEditor && aSheet, NS_ERROR_INVALID_ARG);
michael@0: 
michael@0:   mEditor = aEditor;
michael@0:   mSheet = aSheet;
michael@0: 
michael@0:   return NS_OK;
michael@0: }
michael@0: 
michael@0: 
michael@0: NS_IMETHODIMP
michael@0: AddStyleSheetTxn::DoTransaction()
michael@0: {
michael@0:   NS_ENSURE_TRUE(mEditor && mSheet, NS_ERROR_NOT_INITIALIZED);
michael@0: 
michael@0:   AddStyleSheet(mEditor, mSheet);
michael@0:   return NS_OK;
michael@0: }
michael@0: 
michael@0: NS_IMETHODIMP
michael@0: AddStyleSheetTxn::UndoTransaction()
michael@0: {
michael@0:   NS_ENSURE_TRUE(mEditor && mSheet, NS_ERROR_NOT_INITIALIZED);
michael@0: 
michael@0:   RemoveStyleSheet(mEditor, mSheet);
michael@0:   return NS_OK;
michael@0: }
michael@0: 
michael@0: NS_IMETHODIMP
michael@0: AddStyleSheetTxn::GetTxnDescription(nsAString& aString)
michael@0: {
michael@0:   aString.AssignLiteral("AddStyleSheetTxn");
michael@0:   return NS_OK;
michael@0: }
michael@0: 
michael@0: 
michael@0: RemoveStyleSheetTxn::RemoveStyleSheetTxn()
michael@0: :  EditTxn()
michael@0: ,  mEditor(nullptr)
michael@0: {
michael@0: }
michael@0: 
michael@0: NS_IMPL_CYCLE_COLLECTION_INHERITED(RemoveStyleSheetTxn, EditTxn,
michael@0:                                    mSheet)
michael@0: 
michael@0: NS_INTERFACE_MAP_BEGIN_CYCLE_COLLECTION(RemoveStyleSheetTxn)
michael@0: NS_INTERFACE_MAP_END_INHERITING(EditTxn)
michael@0: 
michael@0: NS_IMETHODIMP
michael@0: RemoveStyleSheetTxn::Init(nsIEditor *aEditor, nsCSSStyleSheet *aSheet)
michael@0: {
michael@0:   NS_ENSURE_TRUE(aEditor && aSheet, NS_ERROR_INVALID_ARG);
michael@0: 
michael@0:   mEditor = aEditor;
michael@0:   mSheet = aSheet;
michael@0: 
michael@0:   return NS_OK;
michael@0: }
michael@0: 
michael@0: 
michael@0: NS_IMETHODIMP
michael@0: RemoveStyleSheetTxn::DoTransaction()
michael@0: {
michael@0:   NS_ENSURE_TRUE(mEditor && mSheet, NS_ERROR_NOT_INITIALIZED);
michael@0: 
michael@0:   RemoveStyleSheet(mEditor, mSheet);
michael@0:   return NS_OK;
michael@0: }
michael@0: 
michael@0: NS_IMETHODIMP
michael@0: RemoveStyleSheetTxn::UndoTransaction()
michael@0: {
michael@0:   NS_ENSURE_TRUE(mEditor && mSheet, NS_ERROR_NOT_INITIALIZED);
michael@0: 
michael@0:   AddStyleSheet(mEditor, mSheet);
michael@0:   return NS_OK;
michael@0: }
michael@0: 
michael@0: NS_IMETHODIMP
michael@0: RemoveStyleSheetTxn::GetTxnDescription(nsAString& aString)
michael@0: {
michael@0:   aString.AssignLiteral("RemoveStyleSheetTxn");
michael@0:   return NS_OK;
michael@0: }