-- Compiled with roblox-ts v2.3.0-dev-30dae68
local TS = _G[script]
--[[
	*
	 * @license
	 * Copyright 2024 Daymon Littrell-Reyes
	 *
	 * Licensed under the Apache License, Version 2.0 (the "License");
	 * you may not use this file except in compliance with the License.
	 * You may obtain a copy of the License at
	 *
	 *   http://www.apache.org/licenses/LICENSE-2.0
	 *
	 * Unless required by applicable law or agreed to in writing, software
	 * distributed under the License is distributed on an "AS IS" BASIS,
	 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
	 * See the License for the specific language governing permissions and
	 * limitations under the License.
	 
]]
local LogContext = TS.import(script, script.Parent, "log-context").LogContext
--[[
	*
	 * A callback that take in a {@link LogContext} and optionally returns a value.
	 *
	 * @see {@link withLogContext}, {@link withLogContextAsync}
	 *
	 * @typeParam R - The type of value returned by the function, or void.
	 * @public
	 
]]
local function isContextCallback(value)
	local _value = value
	return type(_value) == "function"
end
--[[
	*
	 * Wraps around a callback, automatically creating and managing the lifecycle
	 * for a {@link LogContext}.
	 *
	 * @remarks
	 *
	 * The callback will be invoked immediately, and within the same thread.
	 *
	 * Any errors thrown within the callback will be re-thrown after calling
	 * {@link LogContext.stop | stop} on the created context to avoid memory leaks.
	 *
	 * Any value returned from the callback will also be propagated appropriately.
	 *
	 * @param config - Config to create the context with.
	 * @param callback - {@link ContextCallback} scope to run and provide the context for.
	 *
	 * @see {@link withLogContextAsync}
	 *
	 * @example
	 * ```ts
	 * remotes.buyPet.connect((player: Player, pet: PetId) => {
	 *   // automatically starts and stops the context
	 *   withLogContext({ minLogLevel: LogLevel.DEBUG }, (context) => {
	 *     buyPet(context, player.UserId, pet);
	 *   });
	 * });
	 * ```
	 *
	 * @typeParam R - The type of value returned by the function, or void.
	 * @public
	 
]]
--[[
	*
	 * Wraps around a callback, automatically creating and managing the lifecycle
	 * for a {@link LogContext}.
	 *
	 * @remarks
	 *
	 * The callback will be invoked immediately, and within the same thread.
	 *
	 * Any errors thrown within the callback will be re-thrown after calling
	 * {@link LogContext.stop | stop} on the created context to avoid memory leaks.
	 *
	 * Any value returned from the callback will also be propagated appropriately.
	 *
	 * @param callback - {@link ContextCallback} scope to run and provide the context for.
	 *
	 * @see {@link withLogContextAsync}
	 *
	 * @example
	 * ```ts
	 * remotes.buyPet.connect((player: Player, pet: PetId) => {
	 *   // automatically starts and stops the context
	 *   withLogContext((context) => {
	 *     buyPet(context, player.UserId, pet);
	 *   });
	 * });
	 * ```
	 *
	 * @typeParam R - The type of value returned by the function, or void.
	 * @public
	 
]]
local function withLogContext(arg1, arg2)
	if isContextCallback(arg1) then
		local createdContext = LogContext:start()
		local _exitType, _returns = TS.try(function()
			return TS.TRY_RETURN, { arg1(createdContext) }
		end, nil, function()
			createdContext:stop()
		end)
		if _exitType then
			return unpack(_returns)
		end
	elseif isContextCallback(arg2) then
		local createdContext = LogContext:start(arg1)
		local _exitType, _returns = TS.try(function()
			return TS.TRY_RETURN, { arg2(createdContext) }
		end, nil, function()
			createdContext:stop()
		end)
		if _exitType then
			return unpack(_returns)
		end
	else
		error(`withLogContext called with invalid arguments:\narg1:"{arg1}"\narg2:"{arg2}"`)
	end
end
--[[
	*
	 * Wraps around an async callback, automatically creating and managing the lifecycle
	 * for a {@link LogContext}.
	 *
	 * @remarks
	 *
	 * Will call {@link LogContext.stop | stop} on the created context when the executed
	 * scope is finished- regardless if the promise was cancelled or threw an error.
	 *
	 * Any value returned from the callback will also be propagated appropriately.
	 *
	 * @param config - Config to create the context with.
	 * @param callback - {@link ContextCallback} scope to run and provide the context for.
	 *
	 * @see {@link withLogContext}
	 *
	 * @example
	 * ```ts
	 * remotes.buyPet.onRequest((player: Player, pet: PetId) =>
	 *   // automatically starts and stops the context
	 *   withLogContextAsync({ minLogLevel: LogLevel.DEBUG }, async (context) => {
	 *     return buyPet(context, player.UserId, pet);
	 *   }),
	 * );
	 * ```
	 *
	 * @typeParam R - The type of value returned by the function, or void.
	 * @public
	 
]]
--[[
	*
	 * Wraps around an async callback, automatically creating and managing the lifecycle
	 * for a {@link LogContext}.
	 *
	 * @remarks
	 *
	 * Will call {@link LogContext.stop | stop} on the created context when the executed
	 * scope is finished- regardless if the promise was cancelled or threw an error.
	 *
	 * Any value returned from the callback will also be propagated appropriately.
	 *
	 * @param callback - {@link ContextCallback} scope to run and provide the context for.
	 *
	 * @see {@link withLogContext}
	 *
	 * @example
	 * ```ts
	 * remotes.buyPet.onRequest((player: Player, pet: PetId) =>
	 *   // automatically starts and stops the context
	 *   withLogContextAsync(async (context) => {
	 *     return buyPet(context, player.UserId, pet);
	 *   }),
	 * );
	 * ```
	 *
	 * @typeParam R - The type of value returned by the function, or void.
	 * @public
	 
]]
local withLogContextAsync = TS.async(function(arg1, arg2)
	if isContextCallback(arg1) then
		local createdContext = LogContext:start()
		return arg1(createdContext):finally(function()
			createdContext:stop()
		end)
	elseif isContextCallback(arg2) then
		local createdContext = LogContext:start(arg1)
		return arg2(createdContext):finally(function()
			createdContext:stop()
		end)
	else
		return TS.Promise.reject(`withLogContext called with invalid arguments:\narg1:"{arg1}"\narg2:"{arg2}"`)
	end
end)
return {
	withLogContext = withLogContext,
	withLogContextAsync = withLogContextAsync,
}