local UserInputService = game:GetService("UserInputService")
local Environment = {}

Environment.EnvGlobalInjectionKey = "__hotreload_env_global_injection__" -- the longer the better

function SearchInEnv(key: string, def: any?)
	local env = Environment.GetEnvGlobalInjection()

	return (env and env[key]) or def
end

function Environment.GetEnvGlobalInjection()
	local env = getfenv()

	return env[Environment.EnvGlobalInjectionKey]
end
function Environment.IsStory()
	local env = Environment.GetEnvGlobalInjection()

	return env ~= nil
end
function UserInputFallback(userInput: any?)
	if not userInput then
		return UserInputService
	end
	if userInput == UserInputService then
		return UserInputService
	end

	local clone = table.clone(userInput)
	return setmetatable(clone, {
		__index = function(_, key)
			local value = UserInputService[key]
			if typeof(value) == "function" then
				-- Wrap the function to replace self
				return function(_, ...)
					return value(UserInputService, ...)
				end
			end
			return value
		end,
	})
end

Environment.Unmount = SearchInEnv("Unmount", function() end) :: () -> ()

Environment.Reload = SearchInEnv("Reload", function() end) :: () -> ()
Environment.CreateSnapshot = SearchInEnv("CreateSnapshot", function() end) :: (name: string?) -> ()
Environment.SetStoryHolder = SearchInEnv("SetStoryHolder", function() end) :: (holder: Instance?) -> ()
Environment.GetJanitor = function()
	local janitor = SearchInEnv("StoryJanitor")
	return janitor
end

type InputSignals = {
	InputBegan: RBXScriptSignal,
	InputEnded: RBXScriptSignal,
	InputChanged: RBXScriptSignal,
	MouseMoved: RBXScriptSignal,
}

Environment.InputListener = SearchInEnv("InputListener", nil) :: InputSignals
Environment.UserInput = UserInputFallback(SearchInEnv("InputListener", UserInputService)) :: UserInputService

Environment.EnvironmentUID = SearchInEnv("EnvironmentUID", "") :: string
Environment.PreviewUID = SearchInEnv("PreviewUID", "") :: string
Environment.OriginalG = SearchInEnv("OriginalG", _G) :: any
Environment.PluginWidget = SearchInEnv("PluginWidget", nil) :: DockWidgetPluginGui
Environment.Plugin = SearchInEnv("Plugin", plugin) :: Plugin

return Environment