//
// Created by Hanno Gödecke on 15.07.22.
//

#pragma once

#include "box2d/b2_body.h"

#include "JSIBox2dVec2.h"
#include "JSIBox2dFixtureDef.h"

#include <jsi/jsi.h>
#include "jsi/JsiWrappingHostObjects.h"
#include "utils.h"

namespace Box2d {
    using namespace facebook;

    class JSIBox2dBody : public JsiWrappingHostObject<b2Body *> {
    public:
        /**
         * Constructor
         */
        JSIBox2dBody(b2Body *body)
                : JsiWrappingHostObject<b2Body *>(body) {}

        JSI_HOST_FUNCTION(GetPosition) {
            return JSIBox2dVec2::toValue(runtime, getObject()->GetPosition());
        }

        JSI_HOST_FUNCTION(GetAngle) {
            return jsi::Value((double) getObject()->GetAngle());
        }

        JSI_HOST_FUNCTION(CreateFixture) {
            if (arguments[0].isObject()) {
                const auto obj = arguments[0].asObject(runtime);
                if (obj.isHostObject<JSIBox2dFixtureDef>(runtime)) {
                    getObject()->CreateFixture(
                            JSIBox2dFixtureDef::fromValue(runtime, arguments[0]).get());
                    return jsi::Value::undefined();
                }
            }

            throw jsi::JSError(runtime, "Unsupported shape type");
        }

        JSI_HOST_FUNCTION(CreateFixture2) {
            b2Shape *shape = Utils::getShape(runtime, arguments[0]);
            getObject()->CreateFixture(shape, arguments[1].asNumber());
            return jsi::Value::undefined();
        }

        JSI_HOST_FUNCTION(SetLinearVelocity) {
            auto vector = JSIBox2dVec2::fromValue(runtime, arguments[0]).get();
            getObject()->SetLinearVelocity(*vector);
            return jsi::Value::undefined();
        }

        JSI_HOST_FUNCTION(GetLinearVelocity) {
            return JSIBox2dVec2::toValue(runtime, getObject()->GetLinearVelocity());
        }

        JSI_HOST_FUNCTION(SetLinearDamping) {
            auto linearDamping = arguments[0].asNumber();
            getObject()->SetLinearDamping(linearDamping);
            return jsi::Value::undefined();
        }

        JSI_HOST_FUNCTION(SetTransform) {
            auto position = JSIBox2dVec2::fromValue(runtime, arguments[0]).get();
            auto angle = arguments[1].asNumber();
            getObject()->SetTransform(*position, angle);
            return jsi::Value::undefined();
        }

//        JSI_HOST_FUNCTION(GetTransform) {
//            b2Transform transform;
//            getObject()->GetTransform(&transform);
//            return JSIBox2dTransform::toValue(runtime, transform);
//        }

        JSI_HOST_FUNCTION(ApplyForceToCenter) {
            auto vector = JSIBox2dVec2::fromValue(runtime, arguments[0]).get();
            getObject()->ApplyForceToCenter(*vector, arguments[1].getBool());
            return jsi::Value::undefined();
        }

        JSI_HOST_FUNCTION(ApplyLinearImpulseToCenter) {
            auto vector = JSIBox2dVec2::fromValue(runtime, arguments[0]).get();
            getObject()->ApplyLinearImpulseToCenter(*vector, arguments[1].getBool());
            return jsi::Value::undefined();
        }

        JSI_EXPORT_FUNCTIONS(JSI_EXPORT_FUNC(JSIBox2dBody, GetAngle),
                             JSI_EXPORT_FUNC(JSIBox2dBody, GetPosition),
                             JSI_EXPORT_FUNC(JSIBox2dBody, CreateFixture),
                             JSI_EXPORT_FUNC(JSIBox2dBody, CreateFixture2),
                             JSI_EXPORT_FUNC(JSIBox2dBody, SetLinearVelocity),
                             JSI_EXPORT_FUNC(JSIBox2dBody, GetLinearVelocity),
                             JSI_EXPORT_FUNC(JSIBox2dBody, SetLinearDamping),
                             JSI_EXPORT_FUNC(JSIBox2dBody, SetTransform),
                             JSI_EXPORT_FUNC(JSIBox2dBody, ApplyForceToCenter),
                             JSI_EXPORT_FUNC(JSIBox2dBody, ApplyLinearImpulseToCenter),
        );

        /**
        * Returns the underlying object from a host object of this type
        */
        static b2Body* fromValue(jsi::Runtime &runtime,
                                                 const jsi::Value &obj) {
            return obj.asObject(runtime)
                    .asHostObject<JSIBox2dBody>(runtime)
                    ->getObject();
        }

    };
}