/*
    This file is part of the WebKit open source project.
    This file has been generated by generate-bindings.pl. DO NOT MODIFY!

    This library is free software; you can redistribute it and/or
    modify it under the terms of the GNU Library General Public
    License as published by the Free Software Foundation; either
    version 2 of the License, or (at your option) any later version.

    This library is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
    Library General Public License for more details.

    You should have received a copy of the GNU Library General Public License
    along with this library; see the file COPYING.LIB.  If not, write to
    the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
    Boston, MA 02110-1301, USA.
*/

#include "config.h"
#include "JSGCObservation.h"

#include "JSDOMBinding.h"
#include <wtf/GetPtr.h>

using namespace JSC;

namespace WebCore {

// Attributes

JSC::EncodedJSValue jsGCObservationWasCollected(JSC::ExecState*, JSC::EncodedJSValue, JSC::PropertyName);
bool setJSGCObservationConstructor(JSC::ExecState*, JSC::EncodedJSValue, JSC::EncodedJSValue);

class JSGCObservationPrototype : public JSC::JSNonFinalObject {
public:
    typedef JSC::JSNonFinalObject Base;
    static JSGCObservationPrototype* create(JSC::VM& vm, JSC::JSGlobalObject* globalObject, JSC::Structure* structure)
    {
        JSGCObservationPrototype* ptr = new (NotNull, JSC::allocateCell<JSGCObservationPrototype>(vm.heap)) JSGCObservationPrototype(vm, globalObject, structure);
        ptr->finishCreation(vm);
        return ptr;
    }

    DECLARE_INFO;
    static JSC::Structure* createStructure(JSC::VM& vm, JSC::JSGlobalObject* globalObject, JSC::JSValue prototype)
    {
        return JSC::Structure::create(vm, globalObject, prototype, JSC::TypeInfo(JSC::ObjectType, StructureFlags), info());
    }

private:
    JSGCObservationPrototype(JSC::VM& vm, JSC::JSGlobalObject*, JSC::Structure* structure)
        : JSC::JSNonFinalObject(vm, structure)
    {
    }

    void finishCreation(JSC::VM&);
};

/* Hash table for prototype */

static const HashTableValue JSGCObservationPrototypeTableValues[] =
{
    { "wasCollected", ReadOnly | CustomAccessor, NoIntrinsic, { (intptr_t)static_cast<PropertySlot::GetValueFunc>(jsGCObservationWasCollected), (intptr_t) static_cast<PutPropertySlot::PutValueFunc>(0) } },
};

const ClassInfo JSGCObservationPrototype::s_info = { "GCObservationPrototype", &Base::s_info, 0, CREATE_METHOD_TABLE(JSGCObservationPrototype) };

void JSGCObservationPrototype::finishCreation(VM& vm)
{
    Base::finishCreation(vm);
    reifyStaticProperties(vm, JSGCObservationPrototypeTableValues, *this);
}

const ClassInfo JSGCObservation::s_info = { "GCObservation", &Base::s_info, 0, CREATE_METHOD_TABLE(JSGCObservation) };

JSGCObservation::JSGCObservation(Structure* structure, JSDOMGlobalObject& globalObject, Ref<GCObservation>&& impl)
    : JSDOMWrapper<GCObservation>(structure, globalObject, WTFMove(impl))
{
}

JSObject* JSGCObservation::createPrototype(VM& vm, JSGlobalObject* globalObject)
{
    return JSGCObservationPrototype::create(vm, globalObject, JSGCObservationPrototype::createStructure(vm, globalObject, globalObject->objectPrototype()));
}

JSObject* JSGCObservation::prototype(VM& vm, JSGlobalObject* globalObject)
{
    return getDOMPrototype<JSGCObservation>(vm, globalObject);
}

void JSGCObservation::destroy(JSC::JSCell* cell)
{
    JSGCObservation* thisObject = static_cast<JSGCObservation*>(cell);
    thisObject->JSGCObservation::~JSGCObservation();
}

EncodedJSValue jsGCObservationWasCollected(ExecState* state, EncodedJSValue thisValue, PropertyName)
{
    VM& vm = state->vm();
    auto throwScope = DECLARE_THROW_SCOPE(vm);
    UNUSED_PARAM(throwScope);
    UNUSED_PARAM(thisValue);
    JSValue decodedThisValue = JSValue::decode(thisValue);
    auto* castedThis = jsDynamicCast<JSGCObservation*>(decodedThisValue);
    if (UNLIKELY(!castedThis)) {
        return throwGetterTypeError(*state, throwScope, "GCObservation", "wasCollected");
    }
    auto& impl = castedThis->wrapped();
    JSValue result = jsBoolean(impl.wasCollected());
    return JSValue::encode(result);
}


bool setJSGCObservationConstructor(ExecState* state, EncodedJSValue thisValue, EncodedJSValue encodedValue)
{
    VM& vm = state->vm();
    auto throwScope = DECLARE_THROW_SCOPE(vm);
    JSValue value = JSValue::decode(encodedValue);
    JSGCObservationPrototype* domObject = jsDynamicCast<JSGCObservationPrototype*>(JSValue::decode(thisValue));
    if (UNLIKELY(!domObject)) {
        throwVMTypeError(state, throwScope);
        return false;
    }
    // Shadowing a built-in constructor
    return domObject->putDirect(state->vm(), state->propertyNames().constructor, value);
}

bool JSGCObservationOwner::isReachableFromOpaqueRoots(JSC::Handle<JSC::Unknown> handle, void*, SlotVisitor& visitor)
{
    UNUSED_PARAM(handle);
    UNUSED_PARAM(visitor);
    return false;
}

void JSGCObservationOwner::finalize(JSC::Handle<JSC::Unknown> handle, void* context)
{
    auto* jsGCObservation = jsCast<JSGCObservation*>(handle.slot()->asCell());
    auto& world = *static_cast<DOMWrapperWorld*>(context);
    uncacheWrapper(world, &jsGCObservation->wrapped(), jsGCObservation);
}

JSC::JSValue toJSNewlyCreated(JSC::ExecState*, JSDOMGlobalObject* globalObject, Ref<GCObservation>&& impl)
{
#if COMPILER(CLANG)
    // If you hit this failure the interface definition has the ImplementationLacksVTable
    // attribute. You should remove that attribute. If the class has subclasses
    // that may be passed through this toJS() function you should use the SkipVTableValidation
    // attribute to GCObservation.
    static_assert(!__is_polymorphic(GCObservation), "GCObservation is polymorphic but the IDL claims it is not");
#endif
    return createWrapper<GCObservation>(globalObject, WTFMove(impl));
}

JSC::JSValue toJS(JSC::ExecState* state, JSDOMGlobalObject* globalObject, GCObservation& impl)
{
    return wrap(state, globalObject, impl);
}

GCObservation* JSGCObservation::toWrapped(JSC::JSValue value)
{
    if (auto* wrapper = jsDynamicCast<JSGCObservation*>(value))
        return &wrapper->wrapped();
    return nullptr;
}

}
