using System;
using System.Threading;
using System.Threading.Tasks;
using Windows.Win32;

namespace Flow.Launcher.Infrastructure
{
    /// <summary>
    /// Provides static helper for Win32.
    /// Codes are edited from: https://github.com/files-community/Files.
    /// </summary>
    public static class Win32Helper
    {
        public static Task StartSTATaskAsync(Action action)
        {
            var taskCompletionSource = new TaskCompletionSource();
            Thread thread = new(() =>
            {
                PInvoke.OleInitialize();

                try
                {
                    action();
                    taskCompletionSource.SetResult();
                }
                catch (System.Exception)
                {
                    taskCompletionSource.SetResult();
                }
                finally
                {
                    PInvoke.OleUninitialize();
                }
            })
            {
                IsBackground = true,
                Priority = ThreadPriority.Normal
            };

            thread.SetApartmentState(ApartmentState.STA);
            thread.Start();

            return taskCompletionSource.Task;
        }

        public static Task StartSTATaskAsync(Func<Task> func)
        {
            var taskCompletionSource = new TaskCompletionSource();
            Thread thread = new(async () =>
            {
                PInvoke.OleInitialize();

                try
                {
                    await func();
                    taskCompletionSource.SetResult();
                }
                catch (System.Exception)
                {
                    taskCompletionSource.SetResult();
                }
                finally
                {
                    PInvoke.OleUninitialize();
                }
            })
            {
                IsBackground = true,
                Priority = ThreadPriority.Normal
            };

            thread.SetApartmentState(ApartmentState.STA);
            thread.Start();

            return taskCompletionSource.Task;
        }

        public static Task<T?> StartSTATaskAsync<T>(Func<T> func)
        {
            var taskCompletionSource = new TaskCompletionSource<T?>();

            Thread thread = new(() =>
            {
                PInvoke.OleInitialize();

                try
                {
                    taskCompletionSource.SetResult(func());
                }
                catch (System.Exception)
                {
                    taskCompletionSource.SetResult(default);
                }
                finally
                {
                    PInvoke.OleUninitialize();
                }
            })
            {
                IsBackground = true,
                Priority = ThreadPriority.Normal
            };

            thread.SetApartmentState(ApartmentState.STA);
            thread.Start();

            return taskCompletionSource.Task;
        }

        public static Task<T?> StartSTATaskAsync<T>(Func<Task<T>> func)
        {
            var taskCompletionSource = new TaskCompletionSource<T?>();

            Thread thread = new(async () =>
            {
                PInvoke.OleInitialize();
                try
                {
                    taskCompletionSource.SetResult(await func());
                }
                catch (System.Exception)
                {
                    taskCompletionSource.SetResult(default);
                }
                finally
                {
                    PInvoke.OleUninitialize();
                }
            })
            {
                IsBackground = true,
                Priority = ThreadPriority.Normal
            };

            thread.SetApartmentState(ApartmentState.STA);
            thread.Start();

            return taskCompletionSource.Task;
        }
    }
}