Burst limit in lua

burst_limit.lua

-- Burst Limit for Garrysmod/Wiremod
local Burst = setmetatable({}, {
	__call = function(self, max, regen_time)
		return setmetatable({
			max = max, -- Will start full.

			regen = regen_time,
			tracker = WireLib.RegisterPlayerTable()
		}, self)
	end
})
Burst.__index = Burst

function Burst:get(ply)
	local data = self.tracker[ply]
	if data then
		return data.stock
	else
		return self.max
	end
end

function Burst:check(ply)
	local data = self.tracker[ply]
	if data then
		local last = data.last
		local stock = data.stock
		if stock == 0 then
			stock = math.min( math.floor(last / self.regen), self.max)
			if stock == 0 then
				return false
			end
		else
			return true
		end
	else
		return true
	end
end

function Burst:use(ply)
	local data = self.tracker[ply]
	if data then
		local last = data.last
		local stock = math.min( data.stock + math.floor(last/self.regen), self.max )

		if stock > 0 then
			data.stock = stock - 1
			data.last = CurTime()
			return true
		else
			return false
		end
	else
		self.tracker[ply] = {
			stock = self.max - 1,
			last = CurTime()
		}
		return true
	end
end