Northbound/Assets/Scripts/ServerResourceManager.cs

using Unity.Netcode;
using UnityEngine;
using System.Collections.Generic;

namespace Northbound
{
    public class ServerResourceManager : NetworkBehaviour
    {
        public static ServerResourceManager Instance { get; private set; }

        private Dictionary<ulong, int> _playerResources = new Dictionary<ulong, int>();
        private Dictionary<ulong, int> _playerMaxCapacity = new Dictionary<ulong, int>();
        private NetworkVariable<int> _resourcesData = new NetworkVariable<int>();

        private void Awake()
        {
            if (Instance != null && Instance != this)
            {
                Destroy(gameObject);
                return;
            }
            Instance = this;
        }

        public override void OnNetworkSpawn()
        {
            if (IsServer)
            {
                Instance = this;
                NetworkManager.Singleton.OnClientConnectedCallback += OnClientConnected;
                NetworkManager.Singleton.OnClientDisconnectCallback += OnClientDisconnected;
            }
        }

        public override void OnNetworkDespawn()
        {
            if (IsServer)
            {
                if (NetworkManager.Singleton != null)
                {
                    NetworkManager.Singleton.OnClientConnectedCallback -= OnClientConnected;
                    NetworkManager.Singleton.OnClientDisconnectCallback -= OnClientDisconnected;
                }
            }
        }

        private void OnClientConnected(ulong clientId)
        {
            _playerResources[clientId] = 0;
        }

        private void OnClientDisconnected(ulong clientId)
        {
            _playerResources.Remove(clientId);
            _playerMaxCapacity.Remove(clientId);
        }

        public int GetPlayerResourceAmount(ulong clientId)
        {
            if (_playerResources.TryGetValue(clientId, out var resource))
            {
                return resource;
            }
            return 0;
        }

        public bool CanAddResource(ulong clientId, int amount)
        {
            if (_playerResources.TryGetValue(clientId, out var resource) &&
                _playerMaxCapacity.TryGetValue(clientId, out var maxCapacity))
            {
                return resource + amount <= maxCapacity;
            }
            return false;
        }

        public int GetAvailableSpace(ulong clientId)
        {
            if (_playerResources.TryGetValue(clientId, out var resource) &&
                _playerMaxCapacity.TryGetValue(clientId, out var maxCapacity))
            {
                return maxCapacity - resource;
            }
            return 0;
        }

        public void AddResource(ulong clientId, int amount)
        {
            if (_playerResources.TryGetValue(clientId, out var resource) &&
                _playerMaxCapacity.TryGetValue(clientId, out var maxCapacity))
            {
                int actualAmount = Mathf.Min(amount, maxCapacity - resource);
                _playerResources[clientId] = resource + actualAmount;
            }
        }

        public void RemoveResource(ulong clientId, int amount)
        {
            if (_playerResources.TryGetValue(clientId, out var resource))
            {
                int actualAmount = Mathf.Min(amount, resource);
                _playerResources[clientId] = resource - actualAmount;
            }
        }

        public void SetPlayerMaxCapacity(ulong clientId, int maxCapacity)
        {
            _playerMaxCapacity[clientId] = maxCapacity;
        }

        public int GetPlayerMaxCapacity(ulong clientId)
        {
            if (_playerMaxCapacity.TryGetValue(clientId, out var maxCapacity))
            {
                return maxCapacity;
            }
            return 100;
        }
    }
}