gecko-dev/xpcom/ds/nsSimpleEnumerator.cpp
Kris Maglione 2dee0aae3c Bug 1484496: Part 4b - Add intrinsic type information to most nsSimpleEnumerators. r=froydnj
This allows JS callers to automatically get the correct types during
interation, without having to explicitly specify them.

Differential Revision: https://phabricator.services.mozilla.com/D3728

--HG--
extra : rebase_source : b708f382d8ea571d199c669bfed5b5a7ca9ffac4
extra : histedit_source : 7df6feb82088c8a5ca45dc28fe4d2b852c177fee
2018-08-18 21:06:32 -07:00

88 lines
2.1 KiB
C++

/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* vim: set ts=8 sts=2 et sw=2 tw=80: */
/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
#include "nsSimpleEnumerator.h"
#include "mozilla/dom/IteratorResultBinding.h"
#include "mozilla/dom/RootedDictionary.h"
#include "mozilla/dom/ToJSValue.h"
#include "mozilla/ResultExtensions.h"
#include "nsContentUtils.h"
using namespace mozilla;
using namespace mozilla::dom;
namespace {
class JSEnumerator final : public nsIJSEnumerator
{
NS_DECL_ISUPPORTS
NS_DECL_NSIJSENUMERATOR
explicit JSEnumerator(nsISimpleEnumerator* aEnumerator, const nsID& aIID)
: mEnumerator(aEnumerator)
, mIID(aIID)
{}
private:
~JSEnumerator() = default;
nsCOMPtr<nsISimpleEnumerator> mEnumerator;
const nsID mIID;
};
} // anonymous namespace
nsresult
JSEnumerator::Iterator(nsIJSEnumerator** aResult)
{
RefPtr<JSEnumerator> result(this);
result.forget(aResult);
return NS_OK;
}
nsresult
JSEnumerator::Next(JSContext* aCx, JS::MutableHandleValue aResult)
{
RootedDictionary<IteratorResult> result(aCx);
nsCOMPtr<nsISupports> elem;
if (NS_FAILED(mEnumerator->GetNext(getter_AddRefs(elem)))) {
result.mDone = true;
} else {
result.mDone = false;
JS::RootedValue value(aCx);
MOZ_TRY(nsContentUtils::WrapNative(aCx, elem, &mIID, &value));
result.mValue = value;
}
if (!ToJSValue(aCx, result, aResult)) {
return NS_ERROR_OUT_OF_MEMORY;
}
return NS_OK;
}
NS_IMPL_ISUPPORTS(JSEnumerator, nsIJSEnumerator)
nsresult
nsSimpleEnumerator::Iterator(nsIJSEnumerator **aResult)
{
auto result = MakeRefPtr<JSEnumerator>(this, DefaultInterface());
result.forget(aResult);
return NS_OK;
}
nsresult
nsSimpleEnumerator::Entries(const nsIID& aIface, nsIJSEnumerator **aResult)
{
auto result = MakeRefPtr<JSEnumerator>(this, aIface);
result.forget(aResult);
return NS_OK;
}
NS_IMPL_ISUPPORTS(nsSimpleEnumerator, nsISimpleEnumerator, nsISimpleEnumeratorBase)