using AngleSharp.Common;
using System.Collections.Concurrent;

namespace EnvelopeGenerator.ServiceHost.Jobs;

public class JobStateManager(Dictionary<string, State>? initialState = null)
{
    private readonly ConcurrentDictionary<Type, State> _states = new();

    public State GetState<TJob>() => _states.GetOrAdd(typeof(TJob), type => initialState?.GetOrDefault(type.Name, State.Stopped) ?? State.Stopped);

    public State SetState<TJob>(State state) => _states[typeof(TJob)] = state;
}

public enum State
{
    Running,
    Stopped
}