speckle-sharp-connectors/Sdk/Speckle.Connectors.Common/Threading/ThreadContext.cs

using Speckle.InterfaceGenerator;

namespace Speckle.Connectors.Common.Threading;

[GenerateAutoInterface]
public abstract class ThreadContext : IThreadContext
{
  private static readonly Task<object?> s_empty = Task.FromResult<object?>(null);
  public virtual bool IsMainThread => Environment.CurrentManagedThreadId == 1;

  public async Task RunOnThread(Action action, bool useMain)
  {
    if (useMain)
    {
      if (IsMainThread)
      {
        await RunMain(action);
      }
      else
      {
        await WorkerToMain(() =>
        {
          action();
          return s_empty;
        });
      }
    }
    else
    {
      if (IsMainThread)
      {
        await MainToWorker(() =>
        {
          action();
          return s_empty;
        });
      }
      else
      {
        await RunWorker(action);
      }
    }
  }

  public Task<T> RunOnThread<T>(Func<T> action, bool useMain)
  {
    if (useMain)
    {
      if (IsMainThread)
      {
        return RunMainAsync(action);
      }

      return WorkerToMain(action);
    }
    if (IsMainThread)
    {
      return MainToWorker(action);
    }

    return RunWorkerAsync(action);
  }

  public async Task RunOnThreadAsync(Func<Task> action, bool useMain)
  {
    if (useMain)
    {
      if (IsMainThread)
      {
        await RunMainAsync(action);
      }
      else
      {
        await WorkerToMainAsync<object?>(async () =>
        {
          await action();
          return s_empty;
        });
      }
    }
    else
    {
      if (IsMainThread)
      {
        await MainToWorkerAsync<object?>(async () =>
        {
          await action();
          return s_empty;
        });
      }
      else
      {
        if (useMain)
        {
          await RunMainAsync(action);
        }
        else
        {
          await RunWorkerAsync(action);
        }
      }
    }
  }

  public Task<T> RunOnThreadAsync<T>(Func<Task<T>> action, bool useMain)
  {
    if (useMain)
    {
      if (IsMainThread)
      {
        return RunMainAsync(action);
      }

      return WorkerToMainAsync(action);
    }
    if (IsMainThread)
    {
      return MainToWorkerAsync(action);
    }
    return RunWorkerAsync(action);
  }

  protected abstract Task<T> WorkerToMainAsync<T>(Func<Task<T>> action);

  protected abstract Task<T> MainToWorkerAsync<T>(Func<Task<T>> action);
  protected abstract Task<T> WorkerToMain<T>(Func<T> action);

  protected abstract Task<T> MainToWorker<T>(Func<T> action);

  protected virtual Task RunMain(Action action)
  {
    action();
    return Task.CompletedTask;
  }

  protected virtual Task RunWorker(Action action)
  {
    action();
    return Task.CompletedTask;
  }

  protected virtual Task<T> RunMainAsync<T>(Func<T> action) => Task.FromResult(action());

  protected virtual Task<T> RunWorkerAsync<T>(Func<T> action) => Task.FromResult(action());

  protected virtual Task RunMainAsync(Func<Task> action) => Task.FromResult(action());

  protected virtual Task RunWorkerAsync(Func<Task> action) => Task.FromResult(action());

  protected virtual Task<T> RunMainAsync<T>(Func<Task<T>> action) => action();

  protected virtual Task<T> RunWorkerAsync<T>(Func<Task<T>> action) => action();
}