using System;
using System.IO;
using System.Threading;
using System.Threading.Tasks;

namespace Wabbajack.Paths.IO;

public class TemporaryFileManager : IDisposable, IAsyncDisposable
{
    private readonly AbsolutePath _basePath;
    private readonly bool _deleteOnDispose;

    public TemporaryFileManager() : this(KnownFolders.EntryPoint.Combine("temp"))
    {
    }

    public TemporaryFileManager(AbsolutePath basePath, bool deleteOnDispose = true)
    {
        _deleteOnDispose = deleteOnDispose;
        _basePath = basePath;
        _basePath.CreateDirectory();
    }

    public void Dispose()
    {
        if (!_deleteOnDispose) return;
        for (var retries = 0; retries < 10; retries++)
        {
            try
            {
                if (!_basePath.DirectoryExists())
                    return;
                _basePath.DeleteDirectory();
                return;
            }
            catch (IOException)
            {
                Thread.Sleep(1000);
            }
        }
    }
    
    
    public async ValueTask DisposeAsync()
    {
        if (!_deleteOnDispose) return;
        for (var retries = 0; retries < 10; retries++)
        {
            try
            {
                if (!_basePath.DirectoryExists())
                    return;
                _basePath.DeleteDirectory();
                return;
            }
            catch (IOException)
            {
                await Task.Delay(1000);
            }
        }
    }

    public TemporaryPath CreateFile(Extension? ext = default, bool deleteOnDispose = true)
    {
        var path = _basePath.Combine(Guid.NewGuid().ToString());
        if (path.Extension != default)
            path = path.WithExtension(ext);
        return new TemporaryPath(path);
    }

    public TemporaryPath CreateFolder()
    {
        var path = _basePath.Combine(Guid.NewGuid().ToString());
        path.CreateDirectory();
        return new TemporaryPath(path);
    }

}