init.lua

--!strict

-- Copyright (C) 2023 Tria
-- This Source Code Form is subject to the terms of the Mozilla Public License, v. 2.0.
-- If a copy of the MPL was not distributed with this file, You can obtain one at http://mozilla.org/MPL/2.0/.

local Players = game:GetService("Players")
local ReplicatedStorage = game:GetService("ReplicatedStorage")
local RunService = game:GetService("RunService")

local PlayerStates = require(ReplicatedStorage.shared.PlayerStates)

--[=[
	@class Cleanup

	This is a MapLib Feature. It can be accessed by `MapLib:GetFeature("Cleanup")`.
]=]

--[=[
	@within Cleanup
	@since 0.11
	@function Janitor.new
	@param name string?
	@return Janitor
]=]

--< Main >--
local Cleanup = {}

Cleanup.Janitors = {}
Cleanup.Janitor = require(script.Janitor) :: any

--[=[
	@since 0.11
	@return Janitor
	This method returns a Janitor class with the given name
]=]
function Cleanup:GetJanitor(janitorName: string | number)
	if self.Janitors[janitorName] then
		return self.Janitors[janitorName]
	else
		return self.Janitors[#self.Janitors]
	end
end

--[=[
	@since 0.11
	This method returns all the active Janitor classes.
	@return {Janitor}
]=]
function Cleanup:GetJanitors()
	return self.Janitors
end

local function cleanup()
	for _, v in pairs(Cleanup:GetJanitors()) do
		if v.ClassName == "Janitor" then
			v:Destroy()
		end
	end
end

if RunService:IsClient() then
	local character = Players.LocalPlayer.Character or Players.LocalPlayer.CharacterAdded:Wait()
	character:WaitForChild("Humanoid").Died:Connect(cleanup)

	PlayerStates.LocalStateChanged:Connect(function(newState)
		if newState == PlayerStates.SURVIVED then
			cleanup()
		end
	end)
end

return Cleanup