Thread System

local RunService = game:GetService("RunService")

local Threads = {}
Threads._tasks = {}

local function handler(err)
	warn(debug.traceback(err, 2))
	return err
end

local function stopTask(taskData)
	if not taskData then return end

	taskData.running = false

	if taskData.connection then
		taskData.connection:Disconnect()
		taskData.connection = nil
	end

	taskData.thread = nil
end

function Threads:IsRunning(name)
	local t = self._tasks[name]
	return t and t.running == true
end

function Threads:StartLoop(name, callback, waitTime)
	if self:IsRunning(name) then return false end

	local taskData = { running = true }
	self._tasks[name] = taskData

	taskData.thread = task.spawn(function()
		while taskData.running do
			local ok, err = xpcall(callback, handler)
			if not ok then
				error(err)
			end

			if not taskData.running then
				break
			end

			if waitTime then
				task.wait(waitTime)
			else
				task.wait()
			end
		end
	end)

	return true
end

function Threads:StartHeartbeat(name, callback)
	if self:IsRunning(name) then return false end

	local taskData = { running = true }
	self._tasks[name] = taskData

	taskData.connection = RunService.Heartbeat:Connect(function(dt)
		if not taskData.running then return end
		pcall(callback, dt)
	end)

	return true
end

function Threads:RunOnce(name, callback)
	local existing = self._tasks[name]
	if existing then
		stopTask(existing)
		self._tasks[name] = nil
	end

	local taskData = { running = true }
	self._tasks[name] = taskData

	taskData.thread = task.spawn(function()
		pcall(callback)
		taskData.running = false
		Threads._tasks[name] = nil
	end)
end

function Threads:Stop(name)
	local taskData = self._tasks[name]
	if not taskData then return end

	stopTask(taskData)
	self._tasks[name] = nil
end

function Threads:StopAll()
	for name, taskData in pairs(self._tasks) do
		stopTask(taskData)
		self._tasks[name] = nil
	end
end

return Threads