import { FormItem, FormModalUseOptions, useFormModalProps, AnFormModalProps } from '@/components/AnForm';
import { merge } from 'lodash-es';
import { ExtendFormItem } from './useSearchForm';
import { TableUseOptions } from './useTable';

export type ModifyForm = Omit<FormModalUseOptions, 'items' | 'trigger'> & {
  /**
   * 是否继承新建弹窗
   * @default
   * ```ts
   * false
   * ```
   */
  extend?: boolean;
  /**
   * 表单项
   * ```tsx
   * [{
   *   extend: 'name', // 从 create.items 中继承
   * }]
   * ```
   */
  items?: ExtendFormItem[];
};

export function useModifyForm(options: TableUseOptions): AnFormModalProps | undefined {
  const { create, modify } = options;

  if (!modify) {
    return undefined;
  }

  let result: FormModalUseOptions = { items: [] };
  if (modify.extend && create) {
    result = merge({}, create);
  }
  result = merge(result, modify);

  if (modify.items) {
    const items: FormItem[] = [];
    const createItemMap: Record<string, FormItem> = {};
    for (const item of create?.items ?? []) {
      createItemMap[item.field] = item;
    }
    for (let item of modify.items ?? []) {
      if (item.extend) {
        item = merge({}, createItemMap[item.field!] ?? {}, item);
      }
      items.push(item as any);
    }
    if (items.length) {
      result.items = items;
    }
  }

  if (modify.width || create?.width) {
    if (!result.modalProps) {
      (result as any).modalProps = {};
    }
    (result.modalProps as any).width = modify.width || create?.width;
  }

  if (modify.formClass || create?.formClass) {
    if (!result.formProps) {
      (result as any).formProps = {};
    }
    result.formProps!.class = modify.formClass || create?.formClass;
  }

  return useFormModalProps(result);
}