DieInstance.ttslua

local EventManager = require('ge_tts.EventManager')
local Instance = require('ge_tts.Instance')
local Object = require('ge_tts.Object')
local ObjectUtils = require('ge_tts.ObjectUtils')
local Vector3 = require('ge_tts.Vector3')

---@class ge_tts__DieInstance : ge_tts__Instance

---@class ge_tts__static_DieInstance
---@overload fun(savedState: ge_tts__Instance_SavedState): ge_tts__Instance
---@overload fun(object: tts__Object): ge_tts__Instance
---@overload fun(guid: string, container: tts__Container): ge_tts__Instance
---@overload fun(objectOrSavedState: tts__Object | ge_tts__Instance_SavedState): ge_tts__Instance
---@overload fun(objectOrGuidOrSavedState: tts__Object | string | ge_tts__Instance_SavedState, nilOrContainer: nil | tts__Container): ge_tts__Instance
---@field TYPE string
local DieInstance = {}

DieInstance.TYPE = 'Die'

setmetatable(DieInstance, {
    ---@param objectOrGuidOrSavedState tts__Object | string | ge_tts__Instance_SavedState
    ---@param nilOrContainer nil | tts__Container
    __call = function(_, objectOrGuidOrSavedState, nilOrContainer)
        local self = --[[---@type ge_tts__DieInstance]] Instance(objectOrGuidOrSavedState, nilOrContainer)

        ObjectUtils.getSpawnedObject(self.getObject(), function(object)
            object.registerCollisions()
        end)

        ---@return string
        function self.getType()
            return DieInstance.TYPE
        end

        ---@param value number | string
        ---@param playerColor nil | tts__PlayerColor
        function self.onRolled(value, playerColor)
        end

        ---@param value number | string
        ---@param playerColor nil | tts__PlayerColor
        function self.onRotationValueUpdated(value, playerColor)
        end

        return self
    end,
    __index = Instance,
})

local MIN_ROLL_ANGULAR_VELOCITY_SQUARED = 2 * math.pi * math.pi

---@type table<tts__Object, true>
local monitoredDice = {}

---@param object tts__Object
---@param playerColor nil | tts__PlayerColor
local onObjectUpdating = function(object, playerColor)
    if monitoredDice[object] or object.tag ~= Object.Tag.Die then
        return
    end

    local instance = Instance.getOneInstance(object)

    if instance and (--[[---@type ge_tts__DieInstance]] instance).onRolled ~= nil then
        local dieInstance = (--[[---@type ge_tts__DieInstance]] instance)
        local initialRotationValue = object.getRotationValue()
        local isRolling = false

        monitoredDice[object] = true

        Wait.condition(function()
            if isRolling then
                Wait.condition(function()
                    monitoredDice[object] = nil

                    if object ~= nil then
                        local value = object.getRotationValue()
                        dieInstance.onRolled(value, playerColor)
                        dieInstance.onRotationValueUpdated(value, playerColor)
                    end
                end, function()
                    return object == nil or object.resting
                end)
            else
                if object ~= nil then
                    local value = object.getRotationValue()

                    if value ~= initialRotationValue then
                        dieInstance.onRotationValueUpdated(value, playerColor)
                    end
                end
                monitoredDice[object] = nil
            end
        end, function()
            if object == nil then
                return true
            end

            isRolling = object.getRotationValue() ~= initialRotationValue and Vector3.lengthSquared(object.getAngularVelocity()) > MIN_ROLL_ANGULAR_VELOCITY_SQUARED
            return isRolling or object.resting
        end, 20)
    end
end

EventManager.addHandler('onObjectRandomize', onObjectUpdating)
EventManager.addHandler('onObjectDrop', function(playerColor, object) onObjectUpdating(object, playerColor) end)
EventManager.addHandler('onObjectCollisionExit', function(registeredObject) onObjectUpdating(registeredObject, nil) end)

return DieInstance