HexUnlockedWeb/store/useInventoryStore.ts

import { create } from 'zustand';
import { persist } from 'zustand/middleware';

export interface InventoryState {
	unlockedCharacters: string[];
	unlockedCustomizations: string[];
	unlockedDLCs: string[];
	unlockedPerks: string[];
	items: Record<string, number>;
	offerings: Record<string, number>;
	addons: Record<string, number>;

	spoofItems: boolean;
	spoofPerks: boolean;
	spoofCatalog: boolean;
	spoofDLCs: boolean;

	toggleItem: (
		id: string,
		category:
			| 'characters'
			| 'customizations'
			| 'items'
			| 'offerings'
			| 'addons'
			| 'dlcs'
			| 'perks'
	) => void;
	setItemQuantity: (id: string, qty: number) => void;
	setOfferingQuantity: (id: string, qty: number) => void;
	setAddonQuantity: (id: string, qty: number) => void;
	setToggle: (
		key: 'spoofItems' | 'spoofPerks' | 'spoofCatalog' | 'spoofDLCs',
		val: boolean
	) => void;
	importToggles: (toggles: any) => void;

	unlockAllInCategory: (
		category:
			| 'characters'
			| 'customizations'
			| 'items'
			| 'offerings'
			| 'addons'
			| 'dlcs'
			| 'perks',
		allIds: string[],
		qty?: number
	) => void;
	clearCategory: (
		category:
			| 'characters'
			| 'customizations'
			| 'items'
			| 'offerings'
			| 'addons'
			| 'dlcs'
			| 'perks'
	) => void;
	clearAll: () => void;
	importProfile: (profile: any) => void;
}

export const useInventoryStore = create<InventoryState>()(
	persist(
		(set) => ({
			unlockedCharacters: [],
			unlockedCustomizations: [],
			unlockedDLCs: [],
			unlockedPerks: [],
			items: {},
			offerings: {},
			addons: {},

			spoofItems: false,
			spoofPerks: false,
			spoofCatalog: false,
			spoofDLCs: false,

			setToggle: (key, val) => set({ [key]: val }),

			importToggles: (toggles) =>
				set((state) => {
					if (!toggles) return {};
					return {
						spoofItems:
							typeof toggles.spoofItems === 'boolean'
								? toggles.spoofItems
								: state.spoofItems,
						spoofPerks:
							typeof toggles.spoofPerks === 'boolean'
								? toggles.spoofPerks
								: state.spoofPerks,
						spoofCatalog:
							typeof toggles.spoofCatalog === 'boolean'
								? toggles.spoofCatalog
								: state.spoofCatalog,
						spoofDLCs:
							typeof toggles.spoofDLCs === 'boolean'
								? toggles.spoofDLCs
								: state.spoofDLCs
					};
				}),

			toggleItem: (id, category) =>
				set((state) => {
					if (category === 'characters') {
						const arr = state.unlockedCharacters;
						return {
							unlockedCharacters: arr.includes(id)
								? arr.filter((i) => i !== id)
								: [...arr, id]
						};
					} else if (category === 'customizations') {
						const arr = state.unlockedCustomizations;
						return {
							unlockedCustomizations: arr.includes(id)
								? arr.filter((i) => i !== id)
								: [...arr, id]
						};
					} else if (category === 'dlcs') {
						const arr = state.unlockedDLCs;
						return {
							unlockedDLCs: arr.includes(id)
								? arr.filter((i) => i !== id)
								: [...arr, id]
						};
					} else if (category === 'perks') {
						const arr = state.unlockedPerks;
						return {
							unlockedPerks: arr.includes(id)
								? arr.filter((i) => i !== id)
								: [...arr, id]
						};
					} else if (category === 'items') {
						const newItems = { ...state.items };
						if ((newItems[id] || 0) > 0) delete newItems[id];
						else newItems[id] = 100;
						return { items: newItems };
					} else if (category === 'addons') {
						const newAddons = { ...state.addons };
						if ((newAddons[id] || 0) > 0) delete newAddons[id];
						else newAddons[id] = 100;
						return { addons: newAddons };
					} else {
						const newOfferings = { ...state.offerings };
						if ((newOfferings[id] || 0) > 0) delete newOfferings[id];
						else newOfferings[id] = 100;
						return { offerings: newOfferings };
					}
				}),

			setItemQuantity: (id, qty) =>
				set((state) => {
					const newItems = { ...state.items };
					if (qty <= 0) delete newItems[id];
					else newItems[id] = Math.min(32767, Math.max(0, qty));
					return { items: newItems };
				}),

			setOfferingQuantity: (id, qty) =>
				set((state) => {
					const newOfferings = { ...state.offerings };
					if (qty <= 0) delete newOfferings[id];
					else newOfferings[id] = Math.min(32767, Math.max(0, qty));
					return { offerings: newOfferings };
				}),

			setAddonQuantity: (id, qty) =>
				set((state) => {
					const newAddons = { ...state.addons };
					if (qty <= 0) delete newAddons[id];
					else newAddons[id] = Math.min(32767, Math.max(0, qty));
					return { addons: newAddons };
				}),

			unlockAllInCategory: (category, allIds, qty = 100) =>
				set((state) => {
					if (category === 'characters') return { unlockedCharacters: allIds };
					if (category === 'customizations')
						return { unlockedCustomizations: allIds };
					if (category === 'dlcs') return { unlockedDLCs: allIds };
					if (category === 'perks') return { unlockedPerks: allIds };
					if (category === 'items') {
						const newItems = { ...state.items };
						allIds.forEach((id) => {
							newItems[id] = qty;
						});
						return { items: newItems };
					} else if (category === 'addons') {
						const newAddons = { ...state.addons };
						allIds.forEach((id) => {
							newAddons[id] = qty;
						});
						return { addons: newAddons };
					} else {
						const newOfferings = { ...state.offerings };
						allIds.forEach((id) => {
							newOfferings[id] = qty;
						});
						return { offerings: newOfferings };
					}
				}),

			clearCategory: (category) =>
				set(() => {
					if (category === 'characters') return { unlockedCharacters: [] };
					if (category === 'customizations')
						return { unlockedCustomizations: [] };
					if (category === 'dlcs') return { unlockedDLCs: [] };
					if (category === 'perks') return { unlockedPerks: [] };
					if (category === 'items') return { items: {} };
					if (category === 'addons') return { addons: {} };
					return { offerings: {} };
				}),

			clearAll: () =>
				set({
					unlockedCharacters: [],
					unlockedCustomizations: [],
					unlockedDLCs: [],
					unlockedPerks: [],
					items: {},
					offerings: {},
					addons: {}
				}),

			importProfile: (profile) =>
				set((state) => {
					if (!profile) return {};
					return {
						unlockedCharacters: Array.isArray(profile.unlockedCharacters)
							? profile.unlockedCharacters
							: state.unlockedCharacters,
						unlockedCustomizations: Array.isArray(
							profile.unlockedCustomizations
						)
							? profile.unlockedCustomizations
							: state.unlockedCustomizations,
						unlockedDLCs: Array.isArray(profile.unlockedDLCs)
							? profile.unlockedDLCs
							: state.unlockedDLCs,
						unlockedPerks: Array.isArray(profile.unlockedPerks)
							? profile.unlockedPerks
							: state.unlockedPerks,
						items:
							profile.items && typeof profile.items === 'object'
								? profile.items
								: state.items,
						offerings:
							profile.offerings && typeof profile.offerings === 'object'
								? profile.offerings
								: state.offerings,
						addons:
							profile.addons && typeof profile.addons === 'object'
								? profile.addons
								: state.addons
					};
				})
		}),
		{
			name: 'hex-unlocked-inventory'
		}
	)
);