import { assign } from 'vest-utils';

import {
  DoneCallback,
  useDoneCallbacks,
  useFieldCallbacks,
} from '../../core/Runtime';
import { TFieldName } from '../../suiteResult/SuiteResultTypes';

export function useDeferDoneCallback(
  doneCallback: DoneCallback,
  fieldName?: TFieldName,
): void {
  const [, setFieldCallbacks] = useFieldCallbacks();
  const [, setDoneCallbacks] = useDoneCallbacks();

  if (fieldName) {
    setFieldCallbacks(fieldCallbacks =>
      assign(fieldCallbacks, {
        [fieldName]: (fieldCallbacks[fieldName] || []).concat(doneCallback),
      }),
    );

    return;
  }

  setDoneCallbacks(doneCallbacks => doneCallbacks.concat(doneCallback));
}