Cleanuparr/code/backend/Cleanuparr.Infrastructure/Features/Context/ContextProvider.cs

using System.Collections.Immutable;

namespace Cleanuparr.Infrastructure.Features.Context;

public static class ContextProvider
{
    private static readonly AsyncLocal<ImmutableDictionary<string, object>> _asyncLocalDict = new();

    public static void Set(string key, object value)
    {
        ImmutableDictionary<string, object> currentDict = _asyncLocalDict.Value ?? ImmutableDictionary<string, object>.Empty;
        _asyncLocalDict.Value = currentDict.SetItem(key, value);
    }
    
    public static void Set<T>(T value) where T : class
    {
        string key = typeof(T).Name ?? throw new Exception("Type name is null");
        Set(key, value);
    }

    public static object? Get(string key)
    {
        return _asyncLocalDict.Value?.TryGetValue(key, out object? value) is true ? value : null;
    }
    
    public static T Get<T>(string key) where T : class
    {
        return Get(key) as T ?? throw new Exception($"failed to get \"{key}\" from context");
    }
    
    public static T Get<T>() where T : class
    {
        string key = typeof(T).Name ?? throw new Exception("Type name is null");
        return Get<T>(key);
    }

    public const string JobRunIdKey = "JobRunId";

    public static Guid GetJobRunId() =>
        Get(JobRunIdKey) as Guid? ?? throw new InvalidOperationException("JobRunId not set in context");

    public static Guid? TryGetJobRunId() => Get(JobRunIdKey) as Guid?;

    public static void SetJobRunId(Guid id) => Set(JobRunIdKey, id);

    public static class Keys
    {
        public const string Version = "version";
        public const string ItemName = "itemName";
        public const string Hash = "hash";
        public const string DownloadClientUrl = "downloadClientUrl";
        public const string DownloadClientType = "downloadClientType";
        public const string DownloadClientName = "downloadClientName";
        public const string ArrInstanceUrl = "arrInstanceUrl";
    }
}