1.1 --- /dev/null Thu Jan 01 00:00:00 1970 +0000
1.2 +++ b/docshell/base/nsDocShellEnumerator.cpp Wed Dec 31 06:09:35 2014 +0100
1.3 @@ -0,0 +1,179 @@
1.4 +/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*-
1.5 + *
1.6 + * This Source Code Form is subject to the terms of the Mozilla Public
1.7 + * License, v. 2.0. If a copy of the MPL was not distributed with this
1.8 + * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
1.9 +
1.10 +
1.11 +#include "nsDocShellEnumerator.h"
1.12 +
1.13 +#include "nsIDocShellTreeItem.h"
1.14 +
1.15 +nsDocShellEnumerator::nsDocShellEnumerator(int32_t inEnumerationDirection)
1.16 +: mRootItem(nullptr)
1.17 +, mCurIndex(0)
1.18 +, mDocShellType(nsIDocShellTreeItem::typeAll)
1.19 +, mArrayValid(false)
1.20 +, mEnumerationDirection(inEnumerationDirection)
1.21 +{
1.22 +}
1.23 +
1.24 +nsDocShellEnumerator::~nsDocShellEnumerator()
1.25 +{
1.26 +}
1.27 +
1.28 +NS_IMPL_ISUPPORTS(nsDocShellEnumerator, nsISimpleEnumerator)
1.29 +
1.30 +
1.31 +/* nsISupports getNext (); */
1.32 +NS_IMETHODIMP nsDocShellEnumerator::GetNext(nsISupports **outCurItem)
1.33 +{
1.34 + NS_ENSURE_ARG_POINTER(outCurItem);
1.35 + *outCurItem = nullptr;
1.36 +
1.37 + nsresult rv = EnsureDocShellArray();
1.38 + if (NS_FAILED(rv)) return rv;
1.39 +
1.40 + if (mCurIndex >= mItemArray.Length()) {
1.41 + return NS_ERROR_FAILURE;
1.42 + }
1.43 +
1.44 + // post-increment is important here
1.45 + nsCOMPtr<nsISupports> item = do_QueryReferent(mItemArray[mCurIndex++], &rv);
1.46 + item.forget(outCurItem);
1.47 + return rv;
1.48 +}
1.49 +
1.50 +/* boolean hasMoreElements (); */
1.51 +NS_IMETHODIMP nsDocShellEnumerator::HasMoreElements(bool *outHasMore)
1.52 +{
1.53 + NS_ENSURE_ARG_POINTER(outHasMore);
1.54 + *outHasMore = false;
1.55 +
1.56 + nsresult rv = EnsureDocShellArray();
1.57 + if (NS_FAILED(rv)) return rv;
1.58 +
1.59 + *outHasMore = (mCurIndex < mItemArray.Length());
1.60 + return NS_OK;
1.61 +}
1.62 +
1.63 +nsresult nsDocShellEnumerator::GetEnumerationRootItem(nsIDocShellTreeItem * *aEnumerationRootItem)
1.64 +{
1.65 + NS_ENSURE_ARG_POINTER(aEnumerationRootItem);
1.66 + nsCOMPtr<nsIDocShellTreeItem> item = do_QueryReferent(mRootItem);
1.67 + item.forget(aEnumerationRootItem);
1.68 + return NS_OK;
1.69 +}
1.70 +
1.71 +nsresult nsDocShellEnumerator::SetEnumerationRootItem(nsIDocShellTreeItem * aEnumerationRootItem)
1.72 +{
1.73 + mRootItem = do_GetWeakReference(aEnumerationRootItem);
1.74 + ClearState();
1.75 + return NS_OK;
1.76 +}
1.77 +
1.78 +nsresult nsDocShellEnumerator::GetEnumDocShellType(int32_t *aEnumerationItemType)
1.79 +{
1.80 + NS_ENSURE_ARG_POINTER(aEnumerationItemType);
1.81 + *aEnumerationItemType = mDocShellType;
1.82 + return NS_OK;
1.83 +}
1.84 +
1.85 +nsresult nsDocShellEnumerator::SetEnumDocShellType(int32_t aEnumerationItemType)
1.86 +{
1.87 + mDocShellType = aEnumerationItemType;
1.88 + ClearState();
1.89 + return NS_OK;
1.90 +}
1.91 +
1.92 +nsresult nsDocShellEnumerator::First()
1.93 +{
1.94 + mCurIndex = 0;
1.95 + return EnsureDocShellArray();
1.96 +}
1.97 +
1.98 +nsresult nsDocShellEnumerator::EnsureDocShellArray()
1.99 +{
1.100 + if (!mArrayValid)
1.101 + {
1.102 + mArrayValid = true;
1.103 + return BuildDocShellArray(mItemArray);
1.104 + }
1.105 +
1.106 + return NS_OK;
1.107 +}
1.108 +
1.109 +nsresult nsDocShellEnumerator::ClearState()
1.110 +{
1.111 + mItemArray.Clear();
1.112 + mArrayValid = false;
1.113 + mCurIndex = 0;
1.114 + return NS_OK;
1.115 +}
1.116 +
1.117 +nsresult nsDocShellEnumerator::BuildDocShellArray(nsTArray<nsWeakPtr>& inItemArray)
1.118 +{
1.119 + NS_ENSURE_TRUE(mRootItem, NS_ERROR_NOT_INITIALIZED);
1.120 + inItemArray.Clear();
1.121 + nsCOMPtr<nsIDocShellTreeItem> item = do_QueryReferent(mRootItem);
1.122 + return BuildArrayRecursive(item, inItemArray);
1.123 +}
1.124 +
1.125 +nsresult nsDocShellForwardsEnumerator::BuildArrayRecursive(nsIDocShellTreeItem* inItem, nsTArray<nsWeakPtr>& inItemArray)
1.126 +{
1.127 + nsresult rv;
1.128 +
1.129 + // add this item to the array
1.130 + if (mDocShellType == nsIDocShellTreeItem::typeAll ||
1.131 + inItem->ItemType() == mDocShellType) {
1.132 + nsWeakPtr weakItem = do_GetWeakReference(inItem);
1.133 + if (!inItemArray.AppendElement(weakItem))
1.134 + return NS_ERROR_OUT_OF_MEMORY;
1.135 + }
1.136 +
1.137 + int32_t numChildren;
1.138 + rv = inItem->GetChildCount(&numChildren);
1.139 + if (NS_FAILED(rv)) return rv;
1.140 +
1.141 + for (int32_t i = 0; i < numChildren; ++i)
1.142 + {
1.143 + nsCOMPtr<nsIDocShellTreeItem> curChild;
1.144 + rv = inItem->GetChildAt(i, getter_AddRefs(curChild));
1.145 + if (NS_FAILED(rv)) return rv;
1.146 +
1.147 + rv = BuildArrayRecursive(curChild, inItemArray);
1.148 + if (NS_FAILED(rv)) return rv;
1.149 + }
1.150 +
1.151 + return NS_OK;
1.152 +}
1.153 +
1.154 +
1.155 +nsresult nsDocShellBackwardsEnumerator::BuildArrayRecursive(nsIDocShellTreeItem* inItem, nsTArray<nsWeakPtr>& inItemArray)
1.156 +{
1.157 + nsresult rv;
1.158 +
1.159 + int32_t numChildren;
1.160 + rv = inItem->GetChildCount(&numChildren);
1.161 + if (NS_FAILED(rv)) return rv;
1.162 +
1.163 + for (int32_t i = numChildren - 1; i >= 0; --i)
1.164 + {
1.165 + nsCOMPtr<nsIDocShellTreeItem> curChild;
1.166 + rv = inItem->GetChildAt(i, getter_AddRefs(curChild));
1.167 + if (NS_FAILED(rv)) return rv;
1.168 +
1.169 + rv = BuildArrayRecursive(curChild, inItemArray);
1.170 + if (NS_FAILED(rv)) return rv;
1.171 + }
1.172 +
1.173 + // add this item to the array
1.174 + if (mDocShellType == nsIDocShellTreeItem::typeAll ||
1.175 + inItem->ItemType() == mDocShellType) {
1.176 + nsWeakPtr weakItem = do_GetWeakReference(inItem);
1.177 + if (!inItemArray.AppendElement(weakItem))
1.178 + return NS_ERROR_OUT_OF_MEMORY;
1.179 + }
1.180 +
1.181 + return NS_OK;
1.182 +}
