using System;
using System.Collections;
using System.Collections.Concurrent;
using System.Collections.Generic;
using System.Threading;
using System.Threading.Tasks;

namespace Wabbajack.Common
{
    public class AsyncBlockingCollection<T> : IDisposable
    {
        private readonly ConcurrentStack<T> _collection = new ConcurrentStack<T>();
        private bool isDisposed = false;

        public int Count => _collection.Count;
        
        public void Add(T val)
        {
            _collection.Push(val);
        }

        public async ValueTask<(bool found, T val)> TryTake(TimeSpan timeout, CancellationToken token)
        {
            var startTime = DateTime.Now;
            while (true)
            {
                if (_collection.TryPop(out var result))
                {
                    return (true, result);
                }

                if (DateTime.Now - startTime > timeout || token.IsCancellationRequested || isDisposed)
                    return (false, default)!;
                await Task.Delay(100);
            }
        }

        public void Dispose()
        {
            isDisposed = true;
        }
    }
}