import { updateKey, removeIndicator, GlobalStateContext } from 'piral-core';

import type { FormDataState } from './types';

function getNewFormState(newState: FormDataState, patch: Partial<FormDataState>) {
  if (patch.active === false && !newState.submitting) {
    return removeIndicator;
  }

  return newState.active || newState.submitting || newState.changed ? newState : removeIndicator;
}

export function updateFormState(
  ctx: GlobalStateContext,
  id: string,
  initial: FormDataState,
  patch: Partial<FormDataState>,
) {
  ctx.dispatch((state) => {
    const newState = {
      ...initial,
      ...(state.forms[id] || {}),
      ...patch,
    };
    const updatedState = getNewFormState(newState, patch);
    return {
      ...state,
      forms: updateKey(state.forms, id, updatedState),
    };
  });
}