SPT-Server-Build/Libraries/SPTarkov.Server.Core/Routers/ItemEvents/InventoryItemEventRouter.cs

using SPTarkov.DI.Annotations;
using SPTarkov.Server.Core.Callbacks;
using SPTarkov.Server.Core.DI;
using SPTarkov.Server.Core.Models.Eft.Common;
using SPTarkov.Server.Core.Models.Eft.Common.Request;
using SPTarkov.Server.Core.Models.Eft.Hideout;
using SPTarkov.Server.Core.Models.Eft.Inventory;
using SPTarkov.Server.Core.Models.Eft.ItemEvent;
using SPTarkov.Server.Core.Models.Eft.Quests;
using SPTarkov.Server.Core.Models.Enums;

namespace SPTarkov.Server.Core.Routers.ItemEvents;

[Injectable]
public class InventoryItemEventRouter : ItemEventRouterDefinition
{
    protected HideoutCallbacks _hideoutCallbacks;
    protected InventoryCallbacks _inventoryCallbacks;

    public InventoryItemEventRouter
    (
        InventoryCallbacks inventoryCallbacks,
        HideoutCallbacks hideoutCallbacks
    )
    {
        _inventoryCallbacks = inventoryCallbacks;
        _hideoutCallbacks = hideoutCallbacks;
    }

    protected override List<HandledRoute> GetHandledRoutes()
    {
        return new List<HandledRoute>
        {
            new(ItemEventActions.MOVE, false),
            new(ItemEventActions.REMOVE, false),
            new(ItemEventActions.SPLIT, false),
            new(ItemEventActions.MERGE, false),
            new(ItemEventActions.TRANSFER, false),
            new(ItemEventActions.SWAP, false),
            new(ItemEventActions.FOLD, false),
            new(ItemEventActions.TOGGLE, false),
            new(ItemEventActions.TAG, false),
            new(ItemEventActions.BIND, false),
            new(ItemEventActions.UNBIND, false),
            new(ItemEventActions.EXAMINE, false),
            new(ItemEventActions.READ_ENCYCLOPEDIA, false),
            new(ItemEventActions.APPLY_INVENTORY_CHANGES, false),
            new(ItemEventActions.CREATE_MAP_MARKER, false),
            new(ItemEventActions.DELETE_MAP_MARKER, false),
            new(ItemEventActions.EDIT_MAP_MARKER, false),
            new(ItemEventActions.OPEN_RANDOM_LOOT_CONTAINER, false),
            new(ItemEventActions.HIDEOUT_QTE_EVENT, false),
            new(ItemEventActions.REDEEM_PROFILE_REWARD, false),
            new(ItemEventActions.SET_FAVORITE_ITEMS, false),
            new(ItemEventActions.QUEST_FAIL, false),
            new(ItemEventActions.PIN_LOCK, false)
        };
    }

    public override ValueTask<ItemEventRouterResponse> HandleItemEvent(string url, PmcData pmcData, BaseInteractionRequestData body, string sessionID,
        ItemEventRouterResponse output)
    {
        switch (url)
        {
            case ItemEventActions.MOVE:
                return new ValueTask<ItemEventRouterResponse>(_inventoryCallbacks.MoveItem(pmcData, body as InventoryMoveRequestData, sessionID, output));
            case ItemEventActions.REMOVE:
                return new ValueTask<ItemEventRouterResponse>(_inventoryCallbacks.RemoveItem(pmcData, body as InventoryRemoveRequestData, sessionID, output));
            case ItemEventActions.SPLIT:
                return new ValueTask<ItemEventRouterResponse>(_inventoryCallbacks.SplitItem(pmcData, body as InventorySplitRequestData, sessionID, output));
            case ItemEventActions.MERGE:
                return new ValueTask<ItemEventRouterResponse>(_inventoryCallbacks.MergeItem(pmcData, body as InventoryMergeRequestData, sessionID, output));
            case ItemEventActions.TRANSFER:
                return new ValueTask<ItemEventRouterResponse>(_inventoryCallbacks.TransferItem(pmcData, body as InventoryTransferRequestData, sessionID, output));
            case ItemEventActions.SWAP:
                return new ValueTask<ItemEventRouterResponse>(_inventoryCallbacks.SwapItem(pmcData, body as InventorySwapRequestData, sessionID));
            case ItemEventActions.FOLD:
                return new ValueTask<ItemEventRouterResponse>(_inventoryCallbacks.FoldItem(pmcData, body as InventoryFoldRequestData, sessionID));
            case ItemEventActions.TOGGLE:
                return new ValueTask<ItemEventRouterResponse>(_inventoryCallbacks.ToggleItem(pmcData, body as InventoryToggleRequestData, sessionID));
            case ItemEventActions.TAG:
                return new ValueTask<ItemEventRouterResponse>(_inventoryCallbacks.TagItem(pmcData, body as InventoryTagRequestData, sessionID));
            case ItemEventActions.BIND:
                return new ValueTask<ItemEventRouterResponse>(_inventoryCallbacks.BindItem(pmcData, body as InventoryBindRequestData, sessionID, output));
            case ItemEventActions.UNBIND:
                return new ValueTask<ItemEventRouterResponse>(_inventoryCallbacks.UnBindItem(pmcData, body as InventoryBindRequestData, sessionID, output));
            case ItemEventActions.EXAMINE:
                return new ValueTask<ItemEventRouterResponse>(_inventoryCallbacks.ExamineItem(pmcData, body as InventoryExamineRequestData, sessionID, output));
            case ItemEventActions.READ_ENCYCLOPEDIA:
                return new ValueTask<ItemEventRouterResponse>(_inventoryCallbacks.ReadEncyclopedia(pmcData, body as InventoryReadEncyclopediaRequestData, sessionID));
            case ItemEventActions.APPLY_INVENTORY_CHANGES:
                return new ValueTask<ItemEventRouterResponse>(_inventoryCallbacks.SortInventory(pmcData, body as InventorySortRequestData, sessionID, output));
            case ItemEventActions.CREATE_MAP_MARKER:
                return new ValueTask<ItemEventRouterResponse>(_inventoryCallbacks.CreateMapMarker(pmcData, body as InventoryCreateMarkerRequestData, sessionID, output));
            case ItemEventActions.DELETE_MAP_MARKER:
                return new ValueTask<ItemEventRouterResponse>(_inventoryCallbacks.DeleteMapMarker(pmcData, body as InventoryDeleteMarkerRequestData, sessionID, output));
            case ItemEventActions.EDIT_MAP_MARKER:
                return new ValueTask<ItemEventRouterResponse>(_inventoryCallbacks.EditMapMarker(pmcData, body as InventoryEditMarkerRequestData, sessionID, output));
            case ItemEventActions.OPEN_RANDOM_LOOT_CONTAINER:
                return new ValueTask<ItemEventRouterResponse>(_inventoryCallbacks.OpenRandomLootContainer(pmcData, body as OpenRandomLootContainerRequestData, sessionID, output));
            case ItemEventActions.HIDEOUT_QTE_EVENT:
                return new ValueTask<ItemEventRouterResponse>(_hideoutCallbacks.HandleQTEEvent(pmcData, body as HandleQTEEventRequestData, sessionID, output));
            case ItemEventActions.REDEEM_PROFILE_REWARD:
                return new ValueTask<ItemEventRouterResponse>(_inventoryCallbacks.RedeemProfileReward(pmcData, body as RedeemProfileRequestData, sessionID, output));
            case ItemEventActions.SET_FAVORITE_ITEMS:
                return new ValueTask<ItemEventRouterResponse>(_inventoryCallbacks.SetFavoriteItem(pmcData, body as SetFavoriteItems, sessionID, output));
            case ItemEventActions.QUEST_FAIL:
                return new ValueTask<ItemEventRouterResponse>(_inventoryCallbacks.FailQuest(pmcData, body as FailQuestRequestData, sessionID, output));
            case ItemEventActions.PIN_LOCK:
                return new ValueTask<ItemEventRouterResponse>(_inventoryCallbacks.PinOrLock(pmcData, body as PinOrLockItemRequest, sessionID, output));
            default:
                throw new Exception($"InventoryItemEventRouter being used when it cant handle route {url}");
        }
    }
}