/* 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 "mozilla/dom/TestInterfaceAsyncIterableSingleWithArgs.h"

#include "ScriptSettings.h"
#include "js/Value.h"
#include "mozilla/dom/BindingUtils.h"
#include "mozilla/dom/IterableIterator.h"
#include "mozilla/dom/Promise-inl.h"
#include "mozilla/dom/TestInterfaceJSMaplikeSetlikeIterableBinding.h"
#include "nsPIDOMWindow.h"

namespace mozilla::dom {

NS_IMPL_CYCLE_COLLECTION_CLASS(TestInterfaceAsyncIterableSingleWithArgs)

NS_IMPL_CYCLE_COLLECTION_UNLINK_BEGIN_INHERITED(
    TestInterfaceAsyncIterableSingleWithArgs, TestInterfaceAsyncIterableSingle)
NS_IMPL_CYCLE_COLLECTION_UNLINK_END

NS_IMPL_CYCLE_COLLECTION_TRAVERSE_BEGIN_INHERITED(
    TestInterfaceAsyncIterableSingleWithArgs, TestInterfaceAsyncIterableSingle)
NS_IMPL_CYCLE_COLLECTION_TRAVERSE_END

NS_IMPL_CYCLE_COLLECTION_TRACE_BEGIN_INHERITED(
    TestInterfaceAsyncIterableSingleWithArgs, TestInterfaceAsyncIterableSingle)
  NS_IMPL_CYCLE_COLLECTION_TRACE_JS_MEMBER_CALLBACK(mReturnLastCalledWith)
NS_IMPL_CYCLE_COLLECTION_TRACE_END

NS_IMPL_ADDREF_INHERITED(TestInterfaceAsyncIterableSingleWithArgs,
                         TestInterfaceAsyncIterableSingle)
NS_IMPL_RELEASE_INHERITED(TestInterfaceAsyncIterableSingleWithArgs,
                          TestInterfaceAsyncIterableSingle)

NS_INTERFACE_MAP_BEGIN_CYCLE_COLLECTION(
    TestInterfaceAsyncIterableSingleWithArgs)
NS_INTERFACE_MAP_END_INHERITING(TestInterfaceAsyncIterableSingle)

// static
already_AddRefed<TestInterfaceAsyncIterableSingleWithArgs>
TestInterfaceAsyncIterableSingleWithArgs::Constructor(
    const GlobalObject& aGlobal, ErrorResult& aRv) {
  nsCOMPtr<nsPIDOMWindowInner> window =
      do_QueryInterface(aGlobal.GetAsSupports());
  if (!window) {
    aRv.Throw(NS_ERROR_FAILURE);
    return nullptr;
  }

  RefPtr<TestInterfaceAsyncIterableSingleWithArgs> r =
      new TestInterfaceAsyncIterableSingleWithArgs(window);
  return r.forget();
}

JSObject* TestInterfaceAsyncIterableSingleWithArgs::WrapObject(
    JSContext* aCx, JS::Handle<JSObject*> aGivenProto) {
  return TestInterfaceAsyncIterableSingleWithArgs_Binding::Wrap(aCx, this,
                                                                aGivenProto);
}

void TestInterfaceAsyncIterableSingleWithArgs::InitAsyncIteratorData(
    IteratorData& aData, Iterator::IteratorType aType,
    const TestInterfaceAsyncIteratorOptions& aOptions, ErrorResult& aError) {
  aData.mMultiplier = aOptions.mMultiplier;
  aData.mBlockingPromises = aOptions.mBlockingPromises;
  aData.mFailNextAfter = aOptions.mFailNextAfter;
  aData.mThrowFromNext = aOptions.mThrowFromNext;
  if (aOptions.mThrowFromReturn.WasPassed()) {
    aData.mThrowFromReturn = &aOptions.mThrowFromReturn.Value();
  }
}

already_AddRefed<Promise>
TestInterfaceAsyncIterableSingleWithArgs::GetNextIterationResult(
    Iterator* aIterator, ErrorResult& aRv) {
  if (aIterator->Data().mThrowFromNext) {
    AutoJSAPI jsapi;
    MOZ_RELEASE_ASSERT(jsapi.Init(GetParentObject()));
    JS_ReportErrorASCII(jsapi.cx(), "Throwing from next().");
    aRv.MightThrowJSException();
    aRv.StealExceptionFromJSContext(jsapi.cx());
    return nullptr;
  }

  if (aIterator->Data().mIndex == aIterator->Data().mFailNextAfter) {
    aRv.ThrowTypeError("Failed because we of 'failNextAfter'.");
    return nullptr;
  }
  return TestInterfaceAsyncIterableSingle::GetNextIterationResult(
      aIterator, aIterator->Data(), aRv);
}

already_AddRefed<Promise>
TestInterfaceAsyncIterableSingleWithArgs::IteratorReturn(
    JSContext* aCx, Iterator* aIterator, JS::Handle<JS::Value> aValue,
    ErrorResult& aRv) {
  ++mReturnCallCount;

  if (RefPtr<TestThrowingCallback> throwFromReturn =
          aIterator->Data().mThrowFromReturn) {
    throwFromReturn->Call(aRv, nullptr, CallbackFunction::eRethrowExceptions);
    return nullptr;
  }

  RefPtr<Promise> promise = Promise::Create(GetParentObject()->AsGlobal(), aRv);
  if (NS_WARN_IF(aRv.Failed())) {
    return nullptr;
  }

  mReturnLastCalledWith = aValue;
  promise->MaybeResolve(JS::UndefinedHandleValue);
  return promise.forget();
}

}  // namespace mozilla::dom