// Copyright 2024 The Lynx Authors. All rights reserved.
// Licensed under the Apache License Version 2.0 that can be found in the
// LICENSE file in the root directory of this source tree.

import {
  createRunOnMainThread,
  takeDelayedRunOnMainThreadData,
} from '../../../core/thread-function-call/main-thread.js';
import { resetFunctionCallReturnListener } from '../../../core/thread-function-call/return-value.js';
import { isRendering } from '../../lifecycle/isRendering.js';
import { __globalSnapshotPatch } from '../../lifecycle/patch/snapshotPatch.js';

const runOnMainThreadImpl = createRunOnMainThread({
  shouldDispatchRunOnMainThreadDirectly() {
    return __globalSnapshotPatch !== undefined && !isRendering.value;
  },
});

/**
 * `runOnMainThread` allows triggering main thread functions on the main thread asynchronously.
 * @param fn - The main thread functions to be called.
 * @returns A function. Calling which with the arguments to be passed to the main thread function to trigger it on the main thread. This function returns a promise that resolves to the return value of the main thread function.
 * @example
 * ```ts
 * import { runOnMainThread } from '@lynx-js/react';
 *
 * async function someFunction() {
 *   const fn = runOnMainThread(() => {
 *     'main thread';
 *     return 'hello';
 *   });
 *   const result = await fn();
 * }
 * ```
 * @public
 *
 * @group Functions
 */
export function runOnMainThread<R, Fn extends (...args: any[]) => R>(
  fn: Fn,
): (...args: Parameters<Fn>) => Promise<R> {
  return runOnMainThreadImpl(fn);
}

export function resetSnapshotMainThreadFunctionRuntime(): void {
  takeDelayedRunOnMainThreadData();
  resetFunctionCallReturnListener();
}