61 lines
1.7 KiB
C#
61 lines
1.7 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using Chickensoft.AutoInject;
|
|
using Chickensoft.Introspection;
|
|
using Godot;
|
|
|
|
namespace Movementtests.managers;
|
|
|
|
[GlobalClass, Meta(typeof(IAutoNode))]
|
|
public partial class TokenManager : Node
|
|
{
|
|
public override void _Notification(int what) => this.Notify(what);
|
|
|
|
public record struct Token(ulong InstanceId, Action OnUse);
|
|
|
|
private Dictionary<ulong, Token> Tokens { get; set; } = [];
|
|
public int TokenCount => Tokens.Count;
|
|
|
|
private Queue<ulong> RequestQueue { get; set; } = [];
|
|
|
|
public int MaxTokens { get; set;}
|
|
|
|
public void Initialize(int maxTokens)
|
|
{
|
|
MaxTokens = maxTokens;
|
|
}
|
|
|
|
public Token? RequestToken(Node owner)
|
|
{
|
|
if (Tokens.ContainsKey(owner.GetInstanceId()))
|
|
{
|
|
GD.Print("Already has a token");
|
|
return null; // Already has a token
|
|
}
|
|
if (TokenCount >= MaxTokens)
|
|
{
|
|
GD.Print($"Max tokens reached: {TokenCount}");
|
|
return null; // Max tokens reached
|
|
}
|
|
if (!RequestQueue.Contains(owner.GetInstanceId()))
|
|
{
|
|
RequestQueue.Enqueue(owner.GetInstanceId());
|
|
}
|
|
if (RequestQueue.First() != owner.GetInstanceId())
|
|
{
|
|
GD.Print("Waiting its turn");
|
|
return null; // Next in line is not the requester
|
|
}
|
|
|
|
var ownerInstanceId = RequestQueue.Dequeue();
|
|
var token = new Token(ownerInstanceId, () => UseToken(ownerInstanceId));
|
|
Tokens.Add(token.InstanceId, token);
|
|
return token;
|
|
}
|
|
|
|
public void UseToken(ulong instanceId)
|
|
{
|
|
Tokens.Remove(instanceId);
|
|
}
|
|
} |