local oklab = require("./oklab")
local types = require("../types")

type Animatable = types.Animatable

local function interpolate<T>(from: any, to: any, alpha: number): any
	local kind = typeof(from)

	if kind == "number" or kind == "vector" or kind == "Vector3" or kind == "Vector2" then
		return from + (to - from) * alpha
	elseif kind == "Color3" then
		local a = oklab.fromSRGB(vector.create(from.R, from.G, from.B))
		local b = oklab.fromSRGB(vector.create(to.R, to.G, to.B))
		local v = vector.max(oklab.toSRGB(a + (b - a) * alpha), vector.zero)

		return Color3.new(v.x, v.y, v.z)
	elseif kind == "UDim2" or kind == "CFrame" then
		return from:Lerp(to, alpha)
	elseif kind == "UDim" then
		return UDim.new(math.lerp(from.Scale, to.Scale, alpha), math.lerp(from.Offset, to.Offset, alpha))
	elseif kind == "Rect" then
		return Rect.new(
			math.lerp(from.Min.X, to.Min.X, alpha),
			math.lerp(from.Min.Y, to.Min.Y, alpha),
			math.lerp(from.Max.X, to.Max.X, alpha),
			math.lerp(from.Max.Y, to.Max.Y, alpha)
		)
	elseif kind == "table" then
		local result: { number | vector } = from

		for key, b in to do
			local a = from[key]

			if not a then
				continue
			end

			local value = a + (b - a) * alpha

			if result[key] ~= value then
				if result == from then
					result = table.clone(from)
				end
				result[key] = value
			end
		end

		return result
	else
		error(`Unsupported type for interpolation: {kind}`)
	end
end

return interpolate :: <T>(from: T, to: T, alpha: number) -> T