dom/base/nsDOMWindowList.cpp

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