michael@0: /* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*-
michael@0:  *
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: #include "nsWebBrowserFind.h"
michael@0: 
michael@0: // Only need this for NS_FIND_CONTRACTID,
michael@0: // else we could use nsIDOMRange.h and nsIFind.h.
michael@0: #include "nsFind.h"
michael@0: 
michael@0: #include "nsIComponentManager.h"
michael@0: #include "nsIScriptSecurityManager.h"
michael@0: #include "nsIInterfaceRequestor.h"
michael@0: #include "nsIInterfaceRequestorUtils.h"
michael@0: #include "nsPIDOMWindow.h"
michael@0: #include "nsIURI.h"
michael@0: #include "nsIDocShell.h"
michael@0: #include "nsIPresShell.h"
michael@0: #include "nsPresContext.h"
michael@0: #include "nsIDocument.h"
michael@0: #include "nsIDOMDocument.h"
michael@0: #include "nsISelectionController.h"
michael@0: #include "nsISelection.h"
michael@0: #include "nsIFrame.h"
michael@0: #include "nsITextControlFrame.h"
michael@0: #include "nsReadableUtils.h"
michael@0: #include "nsIDOMHTMLElement.h"
michael@0: #include "nsIDOMHTMLDocument.h"
michael@0: #include "nsIContent.h"
michael@0: #include "nsContentCID.h"
michael@0: #include "nsIServiceManager.h"
michael@0: #include "nsIObserverService.h"
michael@0: #include "nsISupportsPrimitives.h"
michael@0: #include "nsFind.h"
michael@0: #include "nsError.h"
michael@0: #include "nsFocusManager.h"
michael@0: #include "mozilla/Services.h"
michael@0: #include "mozilla/dom/Element.h"
michael@0: #include "nsISimpleEnumerator.h"
michael@0: 
michael@0: #if DEBUG
michael@0: #include "nsIWebNavigation.h"
michael@0: #include "nsXPIDLString.h"
michael@0: #endif
michael@0: 
michael@0: //*****************************************************************************
michael@0: // nsWebBrowserFind
michael@0: //*****************************************************************************
michael@0: 
michael@0: nsWebBrowserFind::nsWebBrowserFind() :
michael@0:     mFindBackwards(false),
michael@0:     mWrapFind(false),
michael@0:     mEntireWord(false),
michael@0:     mMatchCase(false),
michael@0:     mSearchSubFrames(true),
michael@0:     mSearchParentFrames(true)
michael@0: {
michael@0: }
michael@0: 
michael@0: nsWebBrowserFind::~nsWebBrowserFind()
michael@0: {
michael@0: }
michael@0: 
michael@0: NS_IMPL_ISUPPORTS(nsWebBrowserFind, nsIWebBrowserFind, nsIWebBrowserFindInFrames)
michael@0: 
michael@0: 
michael@0: /* boolean findNext (); */
michael@0: NS_IMETHODIMP nsWebBrowserFind::FindNext(bool *outDidFind)
michael@0: {
michael@0:     NS_ENSURE_ARG_POINTER(outDidFind);
michael@0:     *outDidFind = false;
michael@0: 
michael@0:     NS_ENSURE_TRUE(CanFindNext(), NS_ERROR_NOT_INITIALIZED);
michael@0: 
michael@0:     nsresult rv = NS_OK;
michael@0:     nsCOMPtr<nsIDOMWindow> searchFrame = do_QueryReferent(mCurrentSearchFrame);
michael@0:     NS_ENSURE_TRUE(searchFrame, NS_ERROR_NOT_INITIALIZED);
michael@0: 
michael@0:     nsCOMPtr<nsIDOMWindow> rootFrame = do_QueryReferent(mRootSearchFrame);
michael@0:     NS_ENSURE_TRUE(rootFrame, NS_ERROR_NOT_INITIALIZED);
michael@0:     
michael@0:     // first, if there's a "cmd_findagain" observer around, check to see if it
michael@0:     // wants to perform the find again command . If it performs the find again
michael@0:     // it will return true, in which case we exit ::FindNext() early.
michael@0:     // Otherwise, nsWebBrowserFind needs to perform the find again command itself
michael@0:     // this is used by nsTypeAheadFind, which controls find again when it was
michael@0:     // the last executed find in the current window.
michael@0:     nsCOMPtr<nsIObserverService> observerSvc =
michael@0:       mozilla::services::GetObserverService();
michael@0:     if (observerSvc) {
michael@0:         nsCOMPtr<nsISupportsInterfacePointer> windowSupportsData = 
michael@0:           do_CreateInstance(NS_SUPPORTS_INTERFACE_POINTER_CONTRACTID, &rv);
michael@0:         NS_ENSURE_SUCCESS(rv, rv);
michael@0:         nsCOMPtr<nsISupports> searchWindowSupports =
michael@0:           do_QueryInterface(rootFrame);
michael@0:         windowSupportsData->SetData(searchWindowSupports);
michael@0:         NS_NAMED_LITERAL_STRING(dnStr, "down");
michael@0:         NS_NAMED_LITERAL_STRING(upStr, "up");
michael@0:         observerSvc->NotifyObservers(windowSupportsData, 
michael@0:                                      "nsWebBrowserFind_FindAgain", 
michael@0:                                      mFindBackwards? upStr.get(): dnStr.get());
michael@0:         windowSupportsData->GetData(getter_AddRefs(searchWindowSupports));
michael@0:         // findnext performed if search window data cleared out
michael@0:         *outDidFind = searchWindowSupports == nullptr;
michael@0:         if (*outDidFind)
michael@0:             return NS_OK;
michael@0:     }
michael@0: 
michael@0:     // next, look in the current frame. If found, return.
michael@0: 
michael@0:     // Beware! This may flush notifications via synchronous
michael@0:     // ScrollSelectionIntoView.
michael@0:     rv = SearchInFrame(searchFrame, false, outDidFind);
michael@0:     if (NS_FAILED(rv)) return rv;
michael@0:     if (*outDidFind)
michael@0:         return OnFind(searchFrame);     // we are done
michael@0: 
michael@0:     // if we are not searching other frames, return
michael@0:     if (!mSearchSubFrames && !mSearchParentFrames)
michael@0:         return NS_OK;
michael@0: 
michael@0:     nsIDocShell *rootDocShell = GetDocShellFromWindow(rootFrame);
michael@0:     if (!rootDocShell) return NS_ERROR_FAILURE;
michael@0:     
michael@0:     int32_t enumDirection;
michael@0:     if (mFindBackwards)
michael@0:         enumDirection = nsIDocShell::ENUMERATE_BACKWARDS;
michael@0:     else
michael@0:         enumDirection = nsIDocShell::ENUMERATE_FORWARDS;
michael@0:         
michael@0:     nsCOMPtr<nsISimpleEnumerator> docShellEnumerator;
michael@0:     rv = rootDocShell->GetDocShellEnumerator(nsIDocShellTreeItem::typeAll,
michael@0:             enumDirection, getter_AddRefs(docShellEnumerator));    
michael@0:     if (NS_FAILED(rv)) return rv;
michael@0:         
michael@0:     // remember where we started
michael@0:     nsCOMPtr<nsIDocShellTreeItem> startingItem =
michael@0:         do_QueryInterface(GetDocShellFromWindow(searchFrame), &rv);
michael@0:     if (NS_FAILED(rv)) return rv;
michael@0: 
michael@0:     nsCOMPtr<nsIDocShellTreeItem> curItem;
michael@0: 
michael@0:     // XXX We should avoid searching in frameset documents here.
michael@0:     // We also need to honour mSearchSubFrames and mSearchParentFrames.
michael@0:     bool hasMore, doFind = false;
michael@0:     while (NS_SUCCEEDED(docShellEnumerator->HasMoreElements(&hasMore)) && hasMore)
michael@0:     {
michael@0:         nsCOMPtr<nsISupports> curSupports;
michael@0:         rv = docShellEnumerator->GetNext(getter_AddRefs(curSupports));
michael@0:         if (NS_FAILED(rv)) break;
michael@0:         curItem = do_QueryInterface(curSupports, &rv);
michael@0:         if (NS_FAILED(rv)) break;
michael@0: 
michael@0:         if (doFind)
michael@0:         {
michael@0:             searchFrame = do_GetInterface(curItem, &rv);
michael@0:             if (NS_FAILED(rv)) break;
michael@0: 
michael@0:             OnStartSearchFrame(searchFrame);
michael@0: 
michael@0:             // Beware! This may flush notifications via synchronous
michael@0:             // ScrollSelectionIntoView.
michael@0:             rv = SearchInFrame(searchFrame, false, outDidFind);
michael@0:             if (NS_FAILED(rv)) return rv;
michael@0:             if (*outDidFind)
michael@0:                 return OnFind(searchFrame);     // we are done
michael@0: 
michael@0:             OnEndSearchFrame(searchFrame);
michael@0:         }
michael@0: 
michael@0:         if (curItem.get() == startingItem.get())
michael@0:             doFind = true;       // start looking in frames after this one
michael@0:     };
michael@0: 
michael@0:     if (!mWrapFind)
michael@0:     {
michael@0:         // remember where we left off
michael@0:         SetCurrentSearchFrame(searchFrame);
michael@0:         return NS_OK;
michael@0:     }
michael@0: 
michael@0:     // From here on, we're wrapping, first through the other frames,
michael@0:     // then finally from the beginning of the starting frame back to
michael@0:     // the starting point.
michael@0: 
michael@0:     // because nsISimpleEnumerator is totally lame and isn't resettable, I
michael@0:     // have to make a new one
michael@0:     docShellEnumerator = nullptr;
michael@0:     rv = rootDocShell->GetDocShellEnumerator(nsIDocShellTreeItem::typeAll,
michael@0:             enumDirection, getter_AddRefs(docShellEnumerator));    
michael@0:     if (NS_FAILED(rv)) return rv;
michael@0:     
michael@0:     while (NS_SUCCEEDED(docShellEnumerator->HasMoreElements(&hasMore)) && hasMore)
michael@0:     {
michael@0:         nsCOMPtr<nsISupports> curSupports;
michael@0:         rv = docShellEnumerator->GetNext(getter_AddRefs(curSupports));
michael@0:         if (NS_FAILED(rv)) break;
michael@0:         curItem = do_QueryInterface(curSupports, &rv);
michael@0:         if (NS_FAILED(rv)) break;
michael@0: 
michael@0:         searchFrame = do_GetInterface(curItem, &rv);
michael@0:         if (NS_FAILED(rv)) break;
michael@0: 
michael@0:         if (curItem.get() == startingItem.get())
michael@0:         {
michael@0:             // Beware! This may flush notifications via synchronous
michael@0:             // ScrollSelectionIntoView.
michael@0:             rv = SearchInFrame(searchFrame, true, outDidFind);
michael@0:             if (NS_FAILED(rv)) return rv;
michael@0:             if (*outDidFind)
michael@0:                 return OnFind(searchFrame);        // we are done
michael@0:             break;
michael@0:         }
michael@0: 
michael@0:         OnStartSearchFrame(searchFrame);
michael@0: 
michael@0:         // Beware! This may flush notifications via synchronous
michael@0:         // ScrollSelectionIntoView.
michael@0:         rv = SearchInFrame(searchFrame, false, outDidFind);
michael@0:         if (NS_FAILED(rv)) return rv;
michael@0:         if (*outDidFind)
michael@0:             return OnFind(searchFrame);        // we are done
michael@0:         
michael@0:         OnEndSearchFrame(searchFrame);
michael@0:     }
michael@0: 
michael@0:     // remember where we left off
michael@0:     SetCurrentSearchFrame(searchFrame);
michael@0:     
michael@0:     NS_ASSERTION(NS_SUCCEEDED(rv), "Something failed");
michael@0:     return rv;
michael@0: }
michael@0: 
michael@0: 
michael@0: /* attribute wstring searchString; */
michael@0: NS_IMETHODIMP nsWebBrowserFind::GetSearchString(char16_t * *aSearchString)
michael@0: {
michael@0:     NS_ENSURE_ARG_POINTER(aSearchString);
michael@0:     *aSearchString = ToNewUnicode(mSearchString);
michael@0:     return NS_OK;
michael@0: }
michael@0: 
michael@0: NS_IMETHODIMP nsWebBrowserFind::SetSearchString(const char16_t * aSearchString)
michael@0: {
michael@0:     mSearchString.Assign(aSearchString);
michael@0:     return NS_OK;
michael@0: }
michael@0: 
michael@0: /* attribute boolean findBackwards; */
michael@0: NS_IMETHODIMP nsWebBrowserFind::GetFindBackwards(bool *aFindBackwards)
michael@0: {
michael@0:     NS_ENSURE_ARG_POINTER(aFindBackwards);
michael@0:     *aFindBackwards = mFindBackwards;
michael@0:     return NS_OK;
michael@0: }
michael@0: 
michael@0: NS_IMETHODIMP nsWebBrowserFind::SetFindBackwards(bool aFindBackwards)
michael@0: {
michael@0:     mFindBackwards = aFindBackwards;
michael@0:     return NS_OK;
michael@0: }
michael@0: 
michael@0: /* attribute boolean wrapFind; */
michael@0: NS_IMETHODIMP nsWebBrowserFind::GetWrapFind(bool *aWrapFind)
michael@0: {
michael@0:     NS_ENSURE_ARG_POINTER(aWrapFind);
michael@0:     *aWrapFind = mWrapFind;
michael@0:     return NS_OK;
michael@0: }
michael@0: NS_IMETHODIMP nsWebBrowserFind::SetWrapFind(bool aWrapFind)
michael@0: {
michael@0:     mWrapFind = aWrapFind;
michael@0:     return NS_OK;
michael@0: }
michael@0: 
michael@0: /* attribute boolean entireWord; */
michael@0: NS_IMETHODIMP nsWebBrowserFind::GetEntireWord(bool *aEntireWord)
michael@0: {
michael@0:     NS_ENSURE_ARG_POINTER(aEntireWord);
michael@0:     *aEntireWord = mEntireWord;
michael@0:     return NS_OK;
michael@0: }
michael@0: NS_IMETHODIMP nsWebBrowserFind::SetEntireWord(bool aEntireWord)
michael@0: {
michael@0:     mEntireWord = aEntireWord;
michael@0:     return NS_OK;
michael@0: }
michael@0: 
michael@0: /* attribute boolean matchCase; */
michael@0: NS_IMETHODIMP nsWebBrowserFind::GetMatchCase(bool *aMatchCase)
michael@0: {
michael@0:     NS_ENSURE_ARG_POINTER(aMatchCase);
michael@0:     *aMatchCase = mMatchCase;
michael@0:     return NS_OK;
michael@0: }
michael@0: NS_IMETHODIMP nsWebBrowserFind::SetMatchCase(bool aMatchCase)
michael@0: {
michael@0:     mMatchCase = aMatchCase;
michael@0:     return NS_OK;
michael@0: }
michael@0: 
michael@0: static bool
michael@0: IsInNativeAnonymousSubtree(nsIContent* aContent)
michael@0: {
michael@0:     while (aContent) {
michael@0:         nsIContent* bindingParent = aContent->GetBindingParent();
michael@0:         if (bindingParent == aContent) {
michael@0:             return true;
michael@0:         }
michael@0: 
michael@0:         aContent = bindingParent;
michael@0:     }
michael@0: 
michael@0:     return false;
michael@0: }
michael@0: 
michael@0: void nsWebBrowserFind::SetSelectionAndScroll(nsIDOMWindow* aWindow,
michael@0:                                              nsIDOMRange*  aRange)
michael@0: {
michael@0:   nsCOMPtr<nsIDOMDocument> domDoc;    
michael@0:   aWindow->GetDocument(getter_AddRefs(domDoc));
michael@0:   if (!domDoc) return;
michael@0: 
michael@0:   nsCOMPtr<nsIDocument> doc(do_QueryInterface(domDoc));
michael@0:   nsIPresShell* presShell = doc->GetShell();
michael@0:   if (!presShell) return;
michael@0: 
michael@0:   nsCOMPtr<nsIDOMNode> node;
michael@0:   aRange->GetStartContainer(getter_AddRefs(node));
michael@0:   nsCOMPtr<nsIContent> content(do_QueryInterface(node));
michael@0:   nsIFrame* frame = content->GetPrimaryFrame();
michael@0:   if (!frame)
michael@0:       return;
michael@0:   nsCOMPtr<nsISelectionController> selCon;
michael@0:   frame->GetSelectionController(presShell->GetPresContext(),
michael@0:                                 getter_AddRefs(selCon));
michael@0:   
michael@0:   // since the match could be an anonymous textnode inside a
michael@0:   // <textarea> or text <input>, we need to get the outer frame
michael@0:   nsITextControlFrame *tcFrame = nullptr;
michael@0:   for ( ; content; content = content->GetParent()) {
michael@0:     if (!IsInNativeAnonymousSubtree(content)) {
michael@0:       nsIFrame* f = content->GetPrimaryFrame();
michael@0:       if (!f)
michael@0:         return;
michael@0:       tcFrame = do_QueryFrame(f);
michael@0:       break;
michael@0:     }
michael@0:   }
michael@0: 
michael@0:   nsCOMPtr<nsISelection> selection;
michael@0: 
michael@0:   selCon->SetDisplaySelection(nsISelectionController::SELECTION_ON);
michael@0:   selCon->GetSelection(nsISelectionController::SELECTION_NORMAL,
michael@0:     getter_AddRefs(selection));
michael@0:   if (selection) {
michael@0:     selection->RemoveAllRanges();
michael@0:     selection->AddRange(aRange);
michael@0: 
michael@0:     nsCOMPtr<nsIFocusManager> fm = do_GetService(FOCUSMANAGER_CONTRACTID);
michael@0:     if (fm) {
michael@0:       if (tcFrame) {
michael@0:         nsCOMPtr<nsIDOMElement> newFocusedElement(do_QueryInterface(content));
michael@0:         fm->SetFocus(newFocusedElement, nsIFocusManager::FLAG_NOSCROLL);
michael@0:       }
michael@0:       else  {
michael@0:         nsCOMPtr<nsIDOMElement> result;
michael@0:         fm->MoveFocus(aWindow, nullptr, nsIFocusManager::MOVEFOCUS_CARET,
michael@0:                       nsIFocusManager::FLAG_NOSCROLL,
michael@0:                       getter_AddRefs(result));
michael@0:       }
michael@0:     }
michael@0: 
michael@0:     // Scroll if necessary to make the selection visible:
michael@0:     // Must be the last thing to do - bug 242056
michael@0: 
michael@0:     // After ScrollSelectionIntoView(), the pending notifications might be
michael@0:     // flushed and PresShell/PresContext/Frames may be dead. See bug 418470.
michael@0:     selCon->ScrollSelectionIntoView
michael@0:       (nsISelectionController::SELECTION_NORMAL,
michael@0:        nsISelectionController::SELECTION_WHOLE_SELECTION,
michael@0:        nsISelectionController::SCROLL_CENTER_VERTICALLY |
michael@0:        nsISelectionController::SCROLL_SYNCHRONOUS);
michael@0:   }
michael@0: }
michael@0: 
michael@0: // Adapted from nsTextServicesDocument::GetDocumentContentRootNode
michael@0: nsresult nsWebBrowserFind::GetRootNode(nsIDOMDocument* aDomDoc,
michael@0:                                        nsIDOMNode **aNode)
michael@0: {
michael@0:   nsresult rv;
michael@0: 
michael@0:   NS_ENSURE_ARG_POINTER(aNode);
michael@0:   *aNode = 0;
michael@0: 
michael@0:   nsCOMPtr<nsIDOMHTMLDocument> htmlDoc = do_QueryInterface(aDomDoc);
michael@0:   if (htmlDoc)
michael@0:   {
michael@0:     // For HTML documents, the content root node is the body.
michael@0:     nsCOMPtr<nsIDOMHTMLElement> bodyElement;
michael@0:     rv = htmlDoc->GetBody(getter_AddRefs(bodyElement));
michael@0:     NS_ENSURE_SUCCESS(rv, rv);
michael@0:     NS_ENSURE_ARG_POINTER(bodyElement);
michael@0:     return bodyElement->QueryInterface(NS_GET_IID(nsIDOMNode),
michael@0:                                        (void **)aNode);
michael@0:   }
michael@0: 
michael@0:   // For non-HTML documents, the content root node will be the doc element.
michael@0:   nsCOMPtr<nsIDOMElement> docElement;
michael@0:   rv = aDomDoc->GetDocumentElement(getter_AddRefs(docElement));
michael@0:   NS_ENSURE_SUCCESS(rv, rv);
michael@0:   NS_ENSURE_ARG_POINTER(docElement);
michael@0:   return docElement->QueryInterface(NS_GET_IID(nsIDOMNode), (void **)aNode);
michael@0: }
michael@0: 
michael@0: nsresult nsWebBrowserFind::SetRangeAroundDocument(nsIDOMRange* aSearchRange,
michael@0:                                                   nsIDOMRange* aStartPt,
michael@0:                                                   nsIDOMRange* aEndPt,
michael@0:                                                   nsIDOMDocument* aDoc)
michael@0: {
michael@0:     nsCOMPtr<nsIDOMNode> bodyNode;
michael@0:     nsresult rv = GetRootNode(aDoc, getter_AddRefs(bodyNode));
michael@0:     nsCOMPtr<nsIContent> bodyContent (do_QueryInterface(bodyNode));
michael@0:     NS_ENSURE_SUCCESS(rv, rv);
michael@0:     NS_ENSURE_ARG_POINTER(bodyContent);
michael@0: 
michael@0:     uint32_t childCount = bodyContent->GetChildCount();
michael@0: 
michael@0:     aSearchRange->SetStart(bodyNode, 0);
michael@0:     aSearchRange->SetEnd(bodyNode, childCount);
michael@0: 
michael@0:     if (mFindBackwards)
michael@0:     {
michael@0:         aStartPt->SetStart(bodyNode, childCount);
michael@0:         aStartPt->SetEnd(bodyNode, childCount);
michael@0:         aEndPt->SetStart(bodyNode, 0);
michael@0:         aEndPt->SetEnd(bodyNode, 0);
michael@0:     }
michael@0:     else
michael@0:     {
michael@0:         aStartPt->SetStart(bodyNode, 0);
michael@0:         aStartPt->SetEnd(bodyNode, 0);
michael@0:         aEndPt->SetStart(bodyNode, childCount);
michael@0:         aEndPt->SetEnd(bodyNode, childCount);
michael@0:     }
michael@0: 
michael@0:     return NS_OK;
michael@0: }
michael@0: 
michael@0: // Set the range to go from the end of the current selection to the
michael@0: // end of the document (forward), or beginning to beginning (reverse).
michael@0: // or around the whole document if there's no selection.
michael@0: nsresult
michael@0: nsWebBrowserFind::GetSearchLimits(nsIDOMRange* aSearchRange,
michael@0:                                   nsIDOMRange* aStartPt,
michael@0:                                   nsIDOMRange* aEndPt,
michael@0:                                   nsIDOMDocument* aDoc,
michael@0:                                   nsISelection* aSel,
michael@0:                                   bool aWrap)
michael@0: {
michael@0:     NS_ENSURE_ARG_POINTER(aSel);
michael@0: 
michael@0:     // There is a selection.
michael@0:     int32_t count = -1;
michael@0:     nsresult rv = aSel->GetRangeCount(&count);
michael@0:     NS_ENSURE_SUCCESS(rv, rv);
michael@0:     if (count < 1)
michael@0:         return SetRangeAroundDocument(aSearchRange, aStartPt, aEndPt, aDoc);
michael@0: 
michael@0:     // Need bodyNode, for the start/end of the document
michael@0:     nsCOMPtr<nsIDOMNode> bodyNode;
michael@0:     rv = GetRootNode(aDoc, getter_AddRefs(bodyNode));
michael@0:     NS_ENSURE_SUCCESS(rv, rv);
michael@0: 
michael@0:     nsCOMPtr<nsIContent> bodyContent (do_QueryInterface(bodyNode));
michael@0:     NS_ENSURE_ARG_POINTER(bodyContent);
michael@0: 
michael@0:     uint32_t childCount = bodyContent->GetChildCount();
michael@0: 
michael@0:     // There are four possible range endpoints we might use:
michael@0:     // DocumentStart, SelectionStart, SelectionEnd, DocumentEnd.
michael@0: 
michael@0:     nsCOMPtr<nsIDOMRange> range;
michael@0:     nsCOMPtr<nsIDOMNode> node;
michael@0:     int32_t offset;
michael@0: 
michael@0:     // Forward, not wrapping: SelEnd to DocEnd
michael@0:     if (!mFindBackwards && !aWrap)
michael@0:     {
michael@0:         // This isn't quite right, since the selection's ranges aren't
michael@0:         // necessarily in order; but they usually will be.
michael@0:         aSel->GetRangeAt(count-1, getter_AddRefs(range));
michael@0:         if (!range) return NS_ERROR_UNEXPECTED;
michael@0:         range->GetEndContainer(getter_AddRefs(node));
michael@0:         if (!node) return NS_ERROR_UNEXPECTED;
michael@0:         range->GetEndOffset(&offset);
michael@0: 
michael@0:         aSearchRange->SetStart(node, offset);
michael@0:         aSearchRange->SetEnd(bodyNode, childCount);
michael@0:         aStartPt->SetStart(node, offset);
michael@0:         aStartPt->SetEnd(node, offset);
michael@0:         aEndPt->SetStart(bodyNode, childCount);
michael@0:         aEndPt->SetEnd(bodyNode, childCount);
michael@0:     }
michael@0:     // Backward, not wrapping: DocStart to SelStart
michael@0:     else if (mFindBackwards && !aWrap)
michael@0:     {
michael@0:         aSel->GetRangeAt(0, getter_AddRefs(range));
michael@0:         if (!range) return NS_ERROR_UNEXPECTED;
michael@0:         range->GetStartContainer(getter_AddRefs(node));
michael@0:         if (!node) return NS_ERROR_UNEXPECTED;
michael@0:         range->GetStartOffset(&offset);
michael@0: 
michael@0:         aSearchRange->SetStart(bodyNode, 0);
michael@0:         aSearchRange->SetEnd(bodyNode, childCount);
michael@0:         aStartPt->SetStart(node, offset);
michael@0:         aStartPt->SetEnd(node, offset);
michael@0:         aEndPt->SetStart(bodyNode, 0);
michael@0:         aEndPt->SetEnd(bodyNode, 0);
michael@0:     }
michael@0:     // Forward, wrapping: DocStart to SelEnd
michael@0:     else if (!mFindBackwards && aWrap)
michael@0:     {
michael@0:         aSel->GetRangeAt(count-1, getter_AddRefs(range));
michael@0:         if (!range) return NS_ERROR_UNEXPECTED;
michael@0:         range->GetEndContainer(getter_AddRefs(node));
michael@0:         if (!node) return NS_ERROR_UNEXPECTED;
michael@0:         range->GetEndOffset(&offset);
michael@0: 
michael@0:         aSearchRange->SetStart(bodyNode, 0);
michael@0:         aSearchRange->SetEnd(bodyNode, childCount);
michael@0:         aStartPt->SetStart(bodyNode, 0);
michael@0:         aStartPt->SetEnd(bodyNode, 0);
michael@0:         aEndPt->SetStart(node, offset);
michael@0:         aEndPt->SetEnd(node, offset);
michael@0:     }
michael@0:     // Backward, wrapping: SelStart to DocEnd
michael@0:     else if (mFindBackwards && aWrap)
michael@0:     {
michael@0:         aSel->GetRangeAt(0, getter_AddRefs(range));
michael@0:         if (!range) return NS_ERROR_UNEXPECTED;
michael@0:         range->GetStartContainer(getter_AddRefs(node));
michael@0:         if (!node) return NS_ERROR_UNEXPECTED;
michael@0:         range->GetStartOffset(&offset);
michael@0: 
michael@0:         aSearchRange->SetStart(bodyNode, 0);
michael@0:         aSearchRange->SetEnd(bodyNode, childCount);
michael@0:         aStartPt->SetStart(bodyNode, childCount);
michael@0:         aStartPt->SetEnd(bodyNode, childCount);
michael@0:         aEndPt->SetStart(node, offset);
michael@0:         aEndPt->SetEnd(node, offset);
michael@0:     }
michael@0:     return NS_OK;
michael@0: }
michael@0: 
michael@0: /* attribute boolean searchFrames; */
michael@0: NS_IMETHODIMP nsWebBrowserFind::GetSearchFrames(bool *aSearchFrames)
michael@0: {
michael@0:     NS_ENSURE_ARG_POINTER(aSearchFrames);
michael@0:     // this only returns true if we are searching both sub and parent
michael@0:     // frames. There is ambiguity if the caller has previously set
michael@0:     // one, but not both of these.
michael@0:     *aSearchFrames = mSearchSubFrames && mSearchParentFrames;
michael@0:     return NS_OK;
michael@0: }
michael@0: 
michael@0: NS_IMETHODIMP nsWebBrowserFind::SetSearchFrames(bool aSearchFrames)
michael@0: {
michael@0:     mSearchSubFrames = aSearchFrames;
michael@0:     mSearchParentFrames = aSearchFrames;
michael@0:     return NS_OK;
michael@0: }
michael@0: 
michael@0: /* attribute nsIDOMWindow currentSearchFrame; */
michael@0: NS_IMETHODIMP nsWebBrowserFind::GetCurrentSearchFrame(nsIDOMWindow * *aCurrentSearchFrame)
michael@0: {
michael@0:     NS_ENSURE_ARG_POINTER(aCurrentSearchFrame);
michael@0:     nsCOMPtr<nsIDOMWindow> searchFrame = do_QueryReferent(mCurrentSearchFrame);
michael@0:     NS_IF_ADDREF(*aCurrentSearchFrame = searchFrame);
michael@0:     return (*aCurrentSearchFrame) ? NS_OK : NS_ERROR_NOT_INITIALIZED;
michael@0: }
michael@0: 
michael@0: NS_IMETHODIMP nsWebBrowserFind::SetCurrentSearchFrame(nsIDOMWindow * aCurrentSearchFrame)
michael@0: {
michael@0:     // is it ever valid to set this to null?
michael@0:     NS_ENSURE_ARG(aCurrentSearchFrame);
michael@0:     mCurrentSearchFrame = do_GetWeakReference(aCurrentSearchFrame);
michael@0:     return NS_OK;
michael@0: }
michael@0: 
michael@0: /* attribute nsIDOMWindow rootSearchFrame; */
michael@0: NS_IMETHODIMP nsWebBrowserFind::GetRootSearchFrame(nsIDOMWindow * *aRootSearchFrame)
michael@0: {
michael@0:     NS_ENSURE_ARG_POINTER(aRootSearchFrame);
michael@0:     nsCOMPtr<nsIDOMWindow> searchFrame = do_QueryReferent(mRootSearchFrame);
michael@0:     NS_IF_ADDREF(*aRootSearchFrame = searchFrame);
michael@0:     return (*aRootSearchFrame) ? NS_OK : NS_ERROR_NOT_INITIALIZED;
michael@0: }
michael@0: 
michael@0: NS_IMETHODIMP nsWebBrowserFind::SetRootSearchFrame(nsIDOMWindow * aRootSearchFrame)
michael@0: {
michael@0:     // is it ever valid to set this to null?
michael@0:     NS_ENSURE_ARG(aRootSearchFrame);
michael@0:     mRootSearchFrame = do_GetWeakReference(aRootSearchFrame);
michael@0:     return NS_OK;
michael@0: }
michael@0: 
michael@0: /* attribute boolean searchSubframes; */
michael@0: NS_IMETHODIMP nsWebBrowserFind::GetSearchSubframes(bool *aSearchSubframes)
michael@0: {
michael@0:     NS_ENSURE_ARG_POINTER(aSearchSubframes);
michael@0:     *aSearchSubframes = mSearchSubFrames;
michael@0:     return NS_OK;
michael@0: }
michael@0: 
michael@0: NS_IMETHODIMP nsWebBrowserFind::SetSearchSubframes(bool aSearchSubframes)
michael@0: {
michael@0:     mSearchSubFrames = aSearchSubframes;
michael@0:     return NS_OK;
michael@0: }
michael@0: 
michael@0: /* attribute boolean searchParentFrames; */
michael@0: NS_IMETHODIMP nsWebBrowserFind::GetSearchParentFrames(bool *aSearchParentFrames)
michael@0: {
michael@0:     NS_ENSURE_ARG_POINTER(aSearchParentFrames);
michael@0:     *aSearchParentFrames = mSearchParentFrames;
michael@0:     return NS_OK;
michael@0: }
michael@0: 
michael@0: NS_IMETHODIMP nsWebBrowserFind::SetSearchParentFrames(bool aSearchParentFrames)
michael@0: {
michael@0:     mSearchParentFrames = aSearchParentFrames;
michael@0:     return NS_OK;
michael@0: }
michael@0: 
michael@0: /*
michael@0:     This method handles finding in a single window (aka frame).
michael@0: 
michael@0: */
michael@0: nsresult nsWebBrowserFind::SearchInFrame(nsIDOMWindow* aWindow,
michael@0:                                          bool aWrapping,
michael@0:                                          bool* aDidFind)
michael@0: {
michael@0:     NS_ENSURE_ARG(aWindow);
michael@0:     NS_ENSURE_ARG_POINTER(aDidFind);
michael@0: 
michael@0:     *aDidFind = false;
michael@0: 
michael@0:     nsCOMPtr<nsIDOMDocument> domDoc;    
michael@0:     nsresult rv = aWindow->GetDocument(getter_AddRefs(domDoc));
michael@0:     NS_ENSURE_SUCCESS(rv, rv);
michael@0:     if (!domDoc) return NS_ERROR_FAILURE;
michael@0: 
michael@0:     // Do security check, to ensure that the frame we're searching is
michael@0:     // acccessible from the frame where the Find is being run.
michael@0: 
michael@0:     // get a uri for the window
michael@0:     nsCOMPtr<nsIDocument> theDoc = do_QueryInterface(domDoc);
michael@0:     if (!theDoc) return NS_ERROR_FAILURE;
michael@0: 
michael@0:     nsCOMPtr<nsIScriptSecurityManager> secMan =
michael@0:       do_GetService(NS_SCRIPTSECURITYMANAGER_CONTRACTID, &rv);
michael@0:     NS_ENSURE_SUCCESS(rv, rv);
michael@0:   
michael@0:     nsCOMPtr<nsIPrincipal> subject;
michael@0:     rv = secMan->GetSubjectPrincipal(getter_AddRefs(subject));
michael@0:     NS_ENSURE_SUCCESS(rv, rv);
michael@0: 
michael@0:     if (subject) {
michael@0:         bool subsumes;
michael@0:         rv = subject->Subsumes(theDoc->NodePrincipal(), &subsumes);
michael@0:         NS_ENSURE_SUCCESS(rv, rv);
michael@0:         if (!subsumes) {
michael@0:             return NS_ERROR_DOM_PROP_ACCESS_DENIED;
michael@0:         }
michael@0:     }
michael@0: 
michael@0:     nsCOMPtr<nsIFind> find = do_CreateInstance(NS_FIND_CONTRACTID, &rv);
michael@0:     NS_ENSURE_SUCCESS(rv, rv);
michael@0: 
michael@0:     (void) find->SetCaseSensitive(mMatchCase);
michael@0:     (void) find->SetFindBackwards(mFindBackwards);
michael@0: 
michael@0:     // XXX Make and set a line breaker here, once that's implemented.
michael@0:     (void) find->SetWordBreaker(0);
michael@0: 
michael@0:     // Now make sure the content (for actual finding) and frame (for
michael@0:     // selection) models are up to date.
michael@0:     theDoc->FlushPendingNotifications(Flush_Frames);
michael@0: 
michael@0:     nsCOMPtr<nsISelection> sel;
michael@0:     GetFrameSelection(aWindow, getter_AddRefs(sel));
michael@0:     NS_ENSURE_ARG_POINTER(sel);
michael@0: 
michael@0:     nsCOMPtr<nsIDOMRange> searchRange = nsFind::CreateRange(theDoc);
michael@0:     NS_ENSURE_ARG_POINTER(searchRange);
michael@0:     nsCOMPtr<nsIDOMRange> startPt  = nsFind::CreateRange(theDoc);
michael@0:     NS_ENSURE_ARG_POINTER(startPt);
michael@0:     nsCOMPtr<nsIDOMRange> endPt  = nsFind::CreateRange(theDoc);
michael@0:     NS_ENSURE_ARG_POINTER(endPt);
michael@0: 
michael@0:     nsCOMPtr<nsIDOMRange> foundRange;
michael@0: 
michael@0:     // If !aWrapping, search from selection to end
michael@0:     if (!aWrapping)
michael@0:         rv = GetSearchLimits(searchRange, startPt, endPt, domDoc, sel,
michael@0:                              false);
michael@0: 
michael@0:     // If aWrapping, search the part of the starting frame
michael@0:     // up to the point where we left off.
michael@0:     else
michael@0:         rv = GetSearchLimits(searchRange, startPt, endPt, domDoc, sel,
michael@0:                              true);
michael@0: 
michael@0:     NS_ENSURE_SUCCESS(rv, rv);
michael@0: 
michael@0:     rv =  find->Find(mSearchString.get(), searchRange, startPt, endPt,
michael@0:                      getter_AddRefs(foundRange));
michael@0: 
michael@0:     if (NS_SUCCEEDED(rv) && foundRange)
michael@0:     {
michael@0:         *aDidFind = true;
michael@0:         sel->RemoveAllRanges();
michael@0:         // Beware! This may flush notifications via synchronous
michael@0:         // ScrollSelectionIntoView.
michael@0:         SetSelectionAndScroll(aWindow, foundRange);
michael@0:     }
michael@0: 
michael@0:     return rv;
michael@0: }
michael@0: 
michael@0: 
michael@0: // called when we start searching a frame that is not the initial
michael@0: // focussed frame. Prepare the frame to be searched.
michael@0: // we clear the selection, so that the search starts from the top
michael@0: // of the frame.
michael@0: nsresult nsWebBrowserFind::OnStartSearchFrame(nsIDOMWindow *aWindow)
michael@0: {
michael@0:     return ClearFrameSelection(aWindow);
michael@0: }
michael@0: 
michael@0: // called when we are done searching a frame and didn't find anything,
michael@0: // and about about to start searching the next frame.
michael@0: nsresult nsWebBrowserFind::OnEndSearchFrame(nsIDOMWindow *aWindow)
michael@0: {
michael@0:     return NS_OK;
michael@0: }
michael@0: 
michael@0: void
michael@0: nsWebBrowserFind::GetFrameSelection(nsIDOMWindow* aWindow, 
michael@0:                                     nsISelection** aSel)
michael@0: {
michael@0:     *aSel = nullptr;
michael@0: 
michael@0:     nsCOMPtr<nsIDOMDocument> domDoc;    
michael@0:     aWindow->GetDocument(getter_AddRefs(domDoc));
michael@0:     if (!domDoc) return;
michael@0: 
michael@0:     nsCOMPtr<nsIDocument> doc(do_QueryInterface(domDoc));
michael@0:     nsIPresShell* presShell = doc->GetShell();
michael@0:     if (!presShell) return;
michael@0: 
michael@0:     // text input controls have their independent selection controllers
michael@0:     // that we must use when they have focus.
michael@0:     nsPresContext *presContext = presShell->GetPresContext();
michael@0: 
michael@0:     nsCOMPtr<nsPIDOMWindow> window(do_QueryInterface(aWindow));
michael@0: 
michael@0:     nsCOMPtr<nsPIDOMWindow> focusedWindow;
michael@0:     nsCOMPtr<nsIContent> focusedContent =
michael@0:       nsFocusManager::GetFocusedDescendant(window, false, getter_AddRefs(focusedWindow));
michael@0: 
michael@0:     nsIFrame *frame = focusedContent ? focusedContent->GetPrimaryFrame() : nullptr;
michael@0: 
michael@0:     nsCOMPtr<nsISelectionController> selCon;
michael@0:     if (frame) {
michael@0:         frame->GetSelectionController(presContext, getter_AddRefs(selCon));
michael@0:         selCon->GetSelection(nsISelectionController::SELECTION_NORMAL, aSel);
michael@0:         if (*aSel) {
michael@0:             int32_t count = -1;
michael@0:             (*aSel)->GetRangeCount(&count);
michael@0:             if (count > 0) {
michael@0:                 return;
michael@0:             }
michael@0:             NS_RELEASE(*aSel);
michael@0:         }
michael@0:     }
michael@0: 
michael@0:     selCon = do_QueryInterface(presShell);
michael@0:     selCon->GetSelection(nsISelectionController::SELECTION_NORMAL, aSel);
michael@0: }
michael@0: 
michael@0: nsresult nsWebBrowserFind::ClearFrameSelection(nsIDOMWindow *aWindow)
michael@0: {
michael@0:     NS_ENSURE_ARG(aWindow);
michael@0:     nsCOMPtr<nsISelection> selection;
michael@0:     GetFrameSelection(aWindow, getter_AddRefs(selection));
michael@0:     if (selection)
michael@0:         selection->RemoveAllRanges();
michael@0:     
michael@0:     return NS_OK;
michael@0: }
michael@0: 
michael@0: nsresult nsWebBrowserFind::OnFind(nsIDOMWindow *aFoundWindow)
michael@0: {
michael@0:     SetCurrentSearchFrame(aFoundWindow);
michael@0: 
michael@0:     // We don't want a selection to appear in two frames simultaneously
michael@0:     nsCOMPtr<nsIDOMWindow> lastFocusedWindow = do_QueryReferent(mLastFocusedWindow);
michael@0:     if (lastFocusedWindow && lastFocusedWindow != aFoundWindow)
michael@0:         ClearFrameSelection(lastFocusedWindow);
michael@0: 
michael@0:     nsCOMPtr<nsIFocusManager> fm = do_GetService(FOCUSMANAGER_CONTRACTID);
michael@0:     if (fm) {
michael@0:       // get the containing frame and focus it. For top-level windows,
michael@0:       // the right window should already be focused.
michael@0:       nsCOMPtr<nsPIDOMWindow> window(do_QueryInterface(aFoundWindow));
michael@0:       NS_ENSURE_TRUE(window, NS_ERROR_FAILURE);
michael@0: 
michael@0:       nsCOMPtr<nsIDOMElement> frameElement =
michael@0:         do_QueryInterface(window->GetFrameElementInternal());
michael@0:       if (frameElement)
michael@0:         fm->SetFocus(frameElement, 0);
michael@0: 
michael@0:       mLastFocusedWindow = do_GetWeakReference(aFoundWindow);
michael@0:     }
michael@0: 
michael@0:     return NS_OK;
michael@0: }
michael@0: 
michael@0: /*---------------------------------------------------------------------------
michael@0: 
michael@0:   GetDocShellFromWindow
michael@0: 
michael@0:   Utility method. This will always return nullptr if no docShell
michael@0:   is returned. Oh why isn't there a better way to do this?
michael@0: ----------------------------------------------------------------------------*/
michael@0: nsIDocShell *
michael@0: nsWebBrowserFind::GetDocShellFromWindow(nsIDOMWindow *inWindow)
michael@0: {
michael@0:     nsCOMPtr<nsPIDOMWindow> window(do_QueryInterface(inWindow));
michael@0:     if (!window) return nullptr;
michael@0: 
michael@0:     return window->GetDocShell();
michael@0: }
michael@0: