Peacock/components/inventory.ts

/*
 *     The Peacock Project - a HITMAN server replacement.
 *     Copyright (C) 2021-2022 The Peacock Project Team
 *
 *     This program is free software: you can redistribute it and/or modify
 *     it under the terms of the GNU Affero General Public License as published by
 *     the Free Software Foundation, either version 3 of the License, or
 *     (at your option) any later version.
 *
 *     This program is distributed in the hope that it will be useful,
 *     but WITHOUT ANY WARRANTY; without even the implied warranty of
 *     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *     GNU Affero General Public License for more details.
 *
 *     You should have received a copy of the GNU Affero General Public License
 *     along with this program.  If not, see <https://www.gnu.org/licenses/>.
 */

import { getVersionedConfig } from "./configSwizzleManager"
import type { GameVersion, Unlockable } from "./types/types"
import {
    brokenItems,
    DELUXE_UNLOCKABLES,
    EXECUTIVE_UNLOCKABLES,
    H1_GOTY_UNLOCKABLES,
    H1_REQUIEM_UNLOCKABLES,
    H2_RACCOON_STINGRAY_UNLOCKABLES,
    SIN_ENVY_UNLOCKABLES,
    SIN_GLUTTONY_UNLOCKABLES,
    SIN_GREED_UNLOCKABLES,
    SIN_LUST_UNLOCKABLES,
    SIN_PRIDE_UNLOCKABLES,
    SIN_SLOTH_UNLOCKABLES,
    SIN_WRATH_UNLOCKABLES,
    TRINITY_UNLOCKABLES,
    WINTERSPORTS_UNLOCKABLES,
} from "./ownership"
import { EPIC_NAMESPACE_2016 } from "./platformEntitlements"

/**
 * An inventory item.
 */
export interface InventoryItem {
    InstanceId: string
    ProfileId: string
    Unlockable: Unlockable
    Properties: Record<string, string>
}

const inventoryUserCache: Map<string, InventoryItem[]> = new Map()

/**
 * Clears a user's inventory.
 *
 * @param userId The user's ID.
 */
export function clearInventoryFor(userId: string): void {
    inventoryUserCache.delete(userId)
}

/**
 * Clears the entire inventory cache.
 */
export function clearInventoryCache(): void {
    inventoryUserCache.clear()
}

export function createInventory(
    profileId: string,
    gameVersion: GameVersion,
    entP: string[],
): InventoryItem[] {
    if (inventoryUserCache.has(profileId)) {
        return inventoryUserCache.get(profileId)!
    }

    // add all unlockables to player's inventory
    const allunlockables = getVersionedConfig<Unlockable[]>(
        "allunlockables",
        gameVersion,
        true,
    ).filter((u) => u.Type !== "location") // locations not in inventory

    // ts-expect-error It cannot be undefined.
    const filtered: InventoryItem[] = allunlockables
        .map((unlockable) => {
            if (brokenItems.includes(unlockable.Guid)) {
                return undefined
            }

            if (unlockable.Guid === "1efe1010-4fff-4ee2-833e-7c58b6518e3e") {
                unlockable.Properties.Name =
                    "char_reward_hero_halloweenoutfit_m_pro140008_name_ebf1e362-671f-47e8-8c88-dd490d8ad866"
                unlockable.Properties.Description =
                    "char_reward_hero_halloweenoutfit_m_pro140008_description_ebf1e362-671f-47e8-8c88-dd490d8ad866"
            }

            unlockable.GameAsset = null
            unlockable.DisplayNameLocKey = `UI_${unlockable.Id}_NAME`
            return {
                InstanceId: unlockable.Guid,
                ProfileId: profileId,
                Unlockable: unlockable,
                Properties: {},
            }
        })
        // filter again, this time removing legacy unlockables
        .filter((unlockContainer) => {
            if (!unlockContainer) {
                return false
            }

            if (gameVersion === "h1") {
                return true
            }

            const e = entP
            const { Id: id } = unlockContainer!.Unlockable

            if (!e) {
                return false
            }

            if (gameVersion === "h3") {
                if (WINTERSPORTS_UNLOCKABLES.includes(id)) {
                    return (
                        e.includes("afa4b921503f43339c360d4b53910791") ||
                        e.includes("1829590")
                    )
                }

                if (EXECUTIVE_UNLOCKABLES.includes(id)) {
                    return (
                        e.includes("6408de14f7dc46b9a33adcf6cbc4d159") ||
                        e.includes("afa4b921503f43339c360d4b53910791") ||
                        e.includes("1829590")
                    )
                }

                if (H1_REQUIEM_UNLOCKABLES.includes(id)) {
                    return (
                        e.includes("e698e1a4b63947b0bc9349a5ae2dc015") ||
                        e.includes("1843460")
                    )
                }

                if (H1_GOTY_UNLOCKABLES.includes(id)) {
                    return (
                        e.includes("894d1e6771044f48a8fdde934b8e443a") ||
                        e.includes("1843460") ||
                        e.includes("1829595")
                    )
                }

                if (H2_RACCOON_STINGRAY_UNLOCKABLES.includes(id)) {
                    return (
                        e.includes("afa4b921503f43339c360d4b53910791") ||
                        e.includes("1829590")
                    )
                }
            } else if (gameVersion === "h2") {
                if (WINTERSPORTS_UNLOCKABLES.includes(id)) {
                    return e.includes("957693")
                }
            } else if (
                // @ts-expect-error The types do actually overlap, but there is no way to show that.
                gameVersion === "h1" &&
                (e.includes("0a73eaedcac84bd28b567dbec764c5cb") ||
                    e.includes(EPIC_NAMESPACE_2016))
            ) {
                // h1 EGS
                if (
                    H1_REQUIEM_UNLOCKABLES.includes(id) ||
                    H1_GOTY_UNLOCKABLES.includes(id)
                ) {
                    return e.includes("81aecb49a60b47478e61e1cbd68d63c5")
                }
            }

            if (DELUXE_UNLOCKABLES.includes(id)) {
                return (
                    e.includes("bc610b36c75442299edcbe99f6f0fb60") ||
                    e.includes("1829591")
                )
            }

            /*
            TODO: Fix this entitlement check (confirmed its broken with Blazer)
            if (LEGACY_UNLOCKABLES.includes(id)) {
                return (
                    e.includes("0b59243cb8aa420691b66be1ecbe68c0") ||
                    e.includes("1829593")
                )
            }
             */

            if (SIN_GREED_UNLOCKABLES.includes(id)) {
                return (
                    e.includes("0e8632b4cdfb415e94291d97d727b98d") ||
                    e.includes("1829580")
                )
            }

            if (SIN_PRIDE_UNLOCKABLES.includes(id)) {
                return (
                    e.includes("3f9adc216dde44dda5e829f11740a0a2") ||
                    e.includes("1829581")
                )
            }

            if (SIN_SLOTH_UNLOCKABLES.includes(id)) {
                return (
                    e.includes("aece009ff59441c0b526f8aa69e24cfb") ||
                    e.includes("1829582")
                )
            }

            if (SIN_LUST_UNLOCKABLES.includes(id)) {
                return (
                    e.includes("dfe5aeb89976450ba1e0e2c208b63d33") ||
                    e.includes("1829583")
                )
            }

            if (SIN_GLUTTONY_UNLOCKABLES.includes(id)) {
                return (
                    e.includes("30107bff80024d1ab291f9cd3bac9fac") ||
                    e.includes("1829584")
                )
            }

            if (SIN_ENVY_UNLOCKABLES.includes(id)) {
                return (
                    e.includes("0403062df0d347619c8dcf043c65c02e") ||
                    e.includes("1829585")
                )
            }

            if (SIN_WRATH_UNLOCKABLES.includes(id)) {
                return (
                    e.includes("9e936ed2507a473db6f53ad24d2da587") ||
                    e.includes("1829586")
                )
            }

            if (TRINITY_UNLOCKABLES.includes(id)) {
                return (
                    e.includes("5d06a6c6af9b4875b3530d5328f61287") ||
                    e.includes("1829596")
                )
            }

            return true
        })

    for (const unlockable of filtered) {
        unlockable!.ProfileId = profileId
    }

    inventoryUserCache.set(profileId, filtered)
    return filtered
}