STA thread as an awaitable task

StaTask.cs

// Author: Atif Aziz
// License: This code is released by "Author" into the public domain.

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

static partial class StaTask
{
    public static Task RunAsync(Action action) =>
        RunAsync(_ => action(), CancellationToken.None);

    public static Task RunAsync(Action<CancellationToken> action, CancellationToken cancellationToken)
    {
        var completionSource = new TaskCompletionSource();
        var thread = new Thread(() =>
        {
            try
            {
                action(cancellationToken);
                completionSource.SetResult();
            }
            catch (OperationCanceledException ex) when (ex.CancellationToken == cancellationToken)
            {
                completionSource.SetCanceled(cancellationToken);
            }
            catch (Exception ex)
            {
                completionSource.SetException(ex);
            }
        });
        thread.SetApartmentState(ApartmentState.STA);
        thread.Start();
        return completionSource.Task;
    }
}