Line data    Source code

       1             : /* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
       2             : /* vim: set ts=8 sts=2 et sw=2 tw=80: */
       3             : /* This Source Code Form is subject to the terms of the Mozilla Public
       4             :  * License, v. 2.0. If a copy of the MPL was not distributed with this
       5             :  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
       6             : 
       7             : // Local Includes
       8             : #include "nsDOMWindowList.h"
       9             : 
      10             : // Helper classes
      11             : #include "nsCOMPtr.h"
      12             : 
      13             : // Interfaces needed
      14             : #include "nsIDocument.h"
      15             : #include "nsIDOMDocument.h"
      16             : #include "nsIDOMWindow.h"
      17             : #include "nsIDocShell.h"
      18             : #include "nsIInterfaceRequestorUtils.h"
      19             : #include "nsIScriptGlobalObject.h"
      20             : #include "nsIWebNavigation.h"
      21             : 
      22           2 : nsDOMWindowList::nsDOMWindowList(nsIDocShell *aDocShell)
      23             : {
      24           2 :   SetDocShell(aDocShell);
      25           2 : }
      26             : 
      27           0 : nsDOMWindowList::~nsDOMWindowList()
      28             : {
      29           0 : }
      30             : 
      31           2 : NS_IMPL_ADDREF(nsDOMWindowList)
      32           0 : NS_IMPL_RELEASE(nsDOMWindowList)
      33             : 
      34           0 : NS_INTERFACE_MAP_BEGIN(nsDOMWindowList)
      35           0 :    NS_INTERFACE_MAP_ENTRY(nsIDOMWindowCollection)
      36           0 :    NS_INTERFACE_MAP_ENTRY(nsISupports)
      37           0 : NS_INTERFACE_MAP_END
      38             : 
      39             : NS_IMETHODIMP
      40           2 : nsDOMWindowList::SetDocShell(nsIDocShell* aDocShell)
      41             : {
      42           2 :   mDocShellNode = aDocShell; // Weak Reference
      43             : 
      44           2 :   return NS_OK;
      45             : }
      46             : 
      47             : void
      48         384 : nsDOMWindowList::EnsureFresh()
      49             : {
      50         768 :   nsCOMPtr<nsIWebNavigation> shellAsNav = do_QueryInterface(mDocShellNode);
      51             : 
      52         384 :   if (shellAsNav) {
      53         768 :     nsCOMPtr<nsIDOMDocument> domdoc;
      54         384 :     shellAsNav->GetDocument(getter_AddRefs(domdoc));
      55             : 
      56         768 :     nsCOMPtr<nsIDocument> doc = do_QueryInterface(domdoc);
      57             : 
      58         384 :     if (doc) {
      59         384 :       doc->FlushPendingNotifications(FlushType::ContentAndNotify);
      60             :     }
      61             :   }
      62         384 : }
      63             : 
      64             : uint32_t
      65         384 : nsDOMWindowList::GetLength()
      66             : {
      67         384 :   EnsureFresh();
      68             : 
      69         384 :   NS_ENSURE_TRUE(mDocShellNode, 0);
      70             : 
      71             :   int32_t length;
      72         384 :   nsresult rv = mDocShellNode->GetChildCount(&length);
      73         384 :   NS_ENSURE_SUCCESS(rv, 0);
      74             : 
      75         384 :   return uint32_t(length);
      76             : }
      77             : 
      78             : NS_IMETHODIMP
      79           0 : nsDOMWindowList::GetLength(uint32_t* aLength)
      80             : {
      81           0 :   *aLength = GetLength();
      82           0 :   return NS_OK;
      83             : }
      84             : 
      85             : already_AddRefed<nsPIDOMWindowOuter>
      86           0 : nsDOMWindowList::IndexedGetter(uint32_t aIndex)
      87             : {
      88           0 :   nsCOMPtr<nsIDocShellTreeItem> item = GetDocShellTreeItemAt(aIndex);
      89           0 :   if (!item) {
      90           0 :     return nullptr;
      91             :   }
      92             : 
      93           0 :   nsCOMPtr<nsPIDOMWindowOuter> window = item->GetWindow();
      94           0 :   MOZ_ASSERT(window);
      95             : 
      96           0 :   return window.forget();
      97             : }
      98             : 
      99             : NS_IMETHODIMP
     100           0 : nsDOMWindowList::Item(uint32_t aIndex, mozIDOMWindowProxy** aReturn)
     101             : {
     102           0 :   nsCOMPtr<nsPIDOMWindowOuter> window = IndexedGetter(aIndex);
     103           0 :   window.forget(aReturn);
     104           0 :   return NS_OK;
     105             : }
     106             : 
     107             : NS_IMETHODIMP
     108           0 : nsDOMWindowList::NamedItem(const nsAString& aName, mozIDOMWindowProxy** aReturn)
     109             : {
     110           0 :   nsCOMPtr<nsIDocShellTreeItem> item;
     111             : 
     112           0 :   *aReturn = nullptr;
     113             : 
     114           0 :   EnsureFresh();
     115             : 
     116           0 :   if (mDocShellNode) {
     117           0 :     mDocShellNode->FindChildWithName(aName, false, false, nullptr,
     118           0 :                                      nullptr, getter_AddRefs(item));
     119             : 
     120           0 :     nsCOMPtr<nsIScriptGlobalObject> globalObject(do_GetInterface(item));
     121           0 :     if (globalObject) {
     122           0 :       CallQueryInterface(globalObject.get(), aReturn);
     123             :     }
     124             :   }
     125             : 
     126           0 :   return NS_OK;
     127             : }