local a=require(script.Parent.TestEnum)local c={}c.__index=c function c.new()local a={_stack={}}return setmetatable(a,c)end function c.getBeforeEachHooks(c)local G=a.NodeType.BeforeEach local y={}for a,c in ipairs(c._stack)do for a,c in ipairs(c[G])do table.insert(y,c)end end return y end function c.getAfterEachHooks(c)local G=a.NodeType.AfterEach local y={}for a,c in ipairs(c._stack)do for a,c in ipairs(c[G])do table.insert(y,1,c)end end return y end function c.popHooks(a)table.remove(a._stack,#a._stack)end function c.pushHooksFrom(G,c)assert(c~=nil)table.insert(G._stack,{[a.NodeType.BeforeAll]=G:_getHooksOfType(c.children,a.NodeType.BeforeAll),[a.NodeType.AfterAll]=G:_getHooksOfType(c.children,a.NodeType.AfterAll);[a.NodeType.BeforeEach]=G:_getHooksOfType(c.children,a.NodeType.BeforeEach),[a.NodeType.AfterEach]=G:_getHooksOfType(c.children,a.NodeType.AfterEach)})end function c.getBeforeAllHooks(c)return c._stack[#c._stack][a.NodeType.BeforeAll]end function c.getAfterAllHooks(c)return c._stack[#c._stack][a.NodeType.AfterAll]end function c._getHooksOfType(G,a,c)local y={}for a,G in ipairs(a)do if G.type==c then table.insert(y,G.callback)end end return y end return c
local TestEnum = require(script.Parent.TestEnum)

local LifecycleHooks = {}
LifecycleHooks.__index = LifecycleHooks

function LifecycleHooks.new()
	local self = {
		_stack = {},
	}
	return setmetatable(self, LifecycleHooks)
end

--[[
	Returns an array of `beforeEach` hooks in FIFO order
]]
function LifecycleHooks:getBeforeEachHooks()
	local key = TestEnum.NodeType.BeforeEach
	local hooks = {}

	for _, level in ipairs(self._stack) do
		for _, hook in ipairs(level[key]) do
			table.insert(hooks, hook)
		end
	end

	return hooks
end

--[[
	Returns an array of `afterEach` hooks in FILO order
]]
function LifecycleHooks:getAfterEachHooks()
	local key = TestEnum.NodeType.AfterEach
	local hooks = {}

	for _, level in ipairs(self._stack) do
		for _, hook in ipairs(level[key]) do
			table.insert(hooks, 1, hook)
		end
	end

	return hooks
end

--[[
	Pushes uncalled beforeAll and afterAll hooks back up the stack
]]
function LifecycleHooks:popHooks()
	table.remove(self._stack, #self._stack)
end

function LifecycleHooks:pushHooksFrom(planNode)
	assert(planNode ~= nil)

	table.insert(self._stack, {
		[TestEnum.NodeType.BeforeAll] = self:_getHooksOfType(planNode.children, TestEnum.NodeType.BeforeAll),
		[TestEnum.NodeType.AfterAll] = self:_getHooksOfType(planNode.children, TestEnum.NodeType.AfterAll),
		[TestEnum.NodeType.BeforeEach] = self:_getHooksOfType(planNode.children, TestEnum.NodeType.BeforeEach),
		[TestEnum.NodeType.AfterEach] = self:_getHooksOfType(planNode.children, TestEnum.NodeType.AfterEach),
	})
end

--[[
	Get the beforeAll hooks from the current level.
]]
function LifecycleHooks:getBeforeAllHooks()
	return self._stack[#self._stack][TestEnum.NodeType.BeforeAll]
end

--[[
	Get the afterAll hooks from the current level.
]]
function LifecycleHooks:getAfterAllHooks()
	return self._stack[#self._stack][TestEnum.NodeType.AfterAll]
end

function LifecycleHooks:_getHooksOfType(nodes, key)
	local hooks = {}

	for _, node in ipairs(nodes) do
		if node.type == key then
			table.insert(hooks, node.callback)
		end
	end

	return hooks
end

return LifecycleHooks
