import type { VbenFormApi, VbenFormSchema } from "#/adapter/form";
import type { VxeTableGridOptions } from "#/adapter/vxe-table";

import { h } from 'vue';

import { CommonStatusEnum, DICT_TYPE } from "#/packages/constants/src";
import { getDictOptions } from "#/packages/effects/hooks/src";

import { z } from '#/adapter/form';
import { generateAutoCode } from '#/api/mes/md/autocode/record';
import { MesAutoCodeRuleCode } from '@vben/constants';
import { getItemTypeSimpleList } from '#/api/mes/md/item/type';
import { getUnitPage } from '#/api/mdm/unit';

import { Button } from 'ant-design-vue';

// 缓存单位列表
let unitListCache: any[] | null = null;
let unitListPromise: Promise<any[]> | null = null;

/** 将扁平分类列表转为 TreeSelect 所需的树形结构 */
function buildCategoryTree(list: any[]): any[] {
  const map = new Map<number, any>();
  const roots: any[] = [];
  for (const item of list) {
    map.set(item.id, { label: item.name, value: item.id, children: [] });
  }
  for (const item of list) {
    const node = map.get(item.id)!;
    if (item.parentId && map.has(item.parentId)) {
      map.get(item.parentId)!.children.push(node);
    } else {
      roots.push(node);
    }
  }
  return roots;
}

/** 获取单位列表（带缓存） */
async function getUnitListCached() {
  if (unitListCache) {
    return unitListCache;
  }
  if (unitListPromise) {
    return unitListPromise;
  }
  unitListPromise = getUnitPage({ pageNo: 1, pageSize: 100, status: 0 }).then((res) => {
    unitListCache = res.list || [];
    return unitListCache;
  }).finally(() => {
    unitListPromise = null;
  });
  return unitListPromise;
}

/** 清除单位列表缓存 */
export function clearUnitListCache() {
  unitListCache = null;
}

/** 新增/修改的表单 */
export function useFormSchema(formApi?: VbenFormApi): VbenFormSchema[] {
  return [
    {
      component: "Input",
      fieldName: "id",
      dependencies: {
        triggerFields: [""],
        show: () => false,
      },
    },
    {
      component: "Input",
      fieldName: "code",
      label: "物料编码",
      componentProps: {
        maxLength: 64,
        placeholder: "请输入物料编码",
      },
      rules: z.string().min(1, '物料编码不能为空').max(64),
      suffix: () =>
        h(
          Button,
          {
            type: 'default',
            onClick: async () => {
              const code = await generateAutoCode(
                MesAutoCodeRuleCode.MD_ITEM_CODE,
              );
              await formApi?.setFieldValue('code', code);
            },
          },
          { default: () => '生成' },
        ),
    },
    {
      component: "Input",
      fieldName: "name",
      label: "物料名称",
      componentProps: {
        placeholder: "请输入物料名称",
      },
      rules: "required",
    },
    {
      component: "ApiTreeSelect",
      fieldName: "categoryId",
      label: "物料分类",
      rules: "selectRequired",
      componentProps: {
        placeholder: "请选择物料分类",
        allowClear: true,
        treeDefaultExpandAll: true,
        api: async () => {
          const res = await getItemTypeSimpleList();
          return buildCategoryTree(res || []);
        },
      },
    },
    {
      component: "Input",
      fieldName: "specification",
      label: "规格型号",
      componentProps: {
        placeholder: "请输入规格型号",
      },
    },
    {
      component: "ApiSelect",
      fieldName: "unitMeasureId",
      label: "计量单位",
      rules: "required",
      formItemClass: "col-span-1",
      componentProps: {
        placeholder: "请选择计量单位",
        allowClear: true,
        api: getUnitListCached,
        labelField: "name",
        valueField: "id",
      },
    },
    {
      component: "ApiSelect",
      fieldName: "unitMeasureId2",
      label: "辅单位1",
      formItemClass: "col-span-1",
      componentProps: {
        placeholder: "请选择辅单位1",
        allowClear: true,
        api: getUnitListCached,
        labelField: "name",
        valueField: "id",
      },
    },
    {
      component: "InputNumber",
      fieldName: "unitMeasureRate1",
      label: "辅单位1换算比率",
      formItemClass: "col-span-1",
      componentProps: {
        class: "!w-full",
        min: 0,
        precision: 6,
        placeholder: "请输入换算比率",
      },
    },
    {
      component: "ApiSelect",
      fieldName: "unitMeasureId3",
      label: "辅单位2",
      formItemClass: "col-span-1",
      componentProps: {
        placeholder: "请选择辅单位2",
        allowClear: true,
        api: getUnitListCached,
        labelField: "name",
        valueField: "id",
      },
    },
    {
      component: "InputNumber",
      fieldName: "unitMeasureRate2",
      label: "辅单位2换算比率",
      formItemClass: "col-span-1",
      componentProps: {
        class: "!w-full",
        min: 0,
        precision: 6,
        placeholder: "请输入换算比率",
      },
    },
    {
      component: 'Input',
      fieldName: 'warehouseId',
      label: '默认仓库',
      componentProps: {
        placeholder: '请输入默认仓库ID',
      },
    },
    {
      component: "Input",
      fieldName: "barCode",
      label: "条码",
      componentProps: {
        placeholder: "请输入条码",
      },
    },
    {
      component: "InputNumber",
      fieldName: "purchasePrice",
      label: "采购价",
      componentProps: {
        class: "!w-full",
        min: 0,
        precision: 2,
        placeholder: "请输入采购价",
      },
    },
    {
      component: "InputNumber",
      fieldName: "salesPrice",
      label: "销售价",
      componentProps: {
        class: "!w-full",
        min: 0,
        precision: 2,
        placeholder: "请输入销售价",
      },
    },
    {
      component: "InputNumber",
      fieldName: "costPrice",
      label: "成本价",
      componentProps: {
        class: "!w-full",
        min: 0,
        precision: 2,
        placeholder: "请输入成本价",
      },
    },
    {
      component: "InputNumber",
      fieldName: "safetyStock",
      label: "安全库存",
      componentProps: {
        class: "!w-full",
        min: 0,
        precision: 2,
        placeholder: "请输入安全库存",
      },
    },
    {
      component: "InputNumber",
      fieldName: "minStock",
      label: "最低库存",
      componentProps: {
        class: "!w-full",
        min: 0,
        precision: 2,
        placeholder: "请输入最低库存",
      },
    },
    {
      component: "InputNumber",
      fieldName: "maxStock",
      label: "最高库存",
      componentProps: {
        class: "!w-full",
        min: 0,
        precision: 2,
        placeholder: "请输入最高库存",
      },
    },
    {
      component: 'Switch',
      fieldName: 'isBatchManaged',
      label: '批次管理',
      defaultValue: false,
      componentProps: {
        class: "w-auto",
        checkedChildren: "是",
        unCheckedChildren: "否",
      },
      rules: z.boolean().default(false),
    },
    {
      component: "Switch",
      fieldName: "syncMes",
      label: "同步到 MES",
      componentProps: {
        class: "w-auto",
        checkedChildren: "是",
        unCheckedChildren: "否",
      },
      rules: z.boolean().default(false),
    },
    {
      component: "InputNumber",
      fieldName: "expiryDay",
      label: "有效期(天)",
      componentProps: {
        class: "!w-full",
        min: 0,
        placeholder: "请输入有效期",
      },
    },
    {
      component: "RadioGroup",
      fieldName: "status",
      label: "状态",
      componentProps: {
        options: getDictOptions(DICT_TYPE.COMMON_STATUS, "number"),
        buttonStyle: "solid",
        optionType: "button",
      },
      rules: z.number().default(CommonStatusEnum.ENABLE),
    },
    {
      component: "Textarea",
      fieldName: "remark",
      label: "备注",
      formItemClass: "col-span-2",
      componentProps: {
        placeholder: "请输入备注",
        rows: 3,
      },
    },
    // ===== 批次属性配置（isBatchManaged = true 时显示） =====
    {
      component: "Divider",
      fieldName: "batchConfigDivider",
      label: "批次属性配置",
      formItemClass: "col-span-2",
      componentProps: {
        orientation: "left",
        plain: true,
      },
      dependencies: {
        triggerFields: ["isBatchManaged"],
        show: (values) => values.isBatchManaged === true,
      },
    },
    {
      component: "Switch",
      fieldName: "produceDateFlag",
      label: "生产日期必填",
      defaultValue: false,
      componentProps: {
        class: "w-auto",
        checkedChildren: "是",
        unCheckedChildren: "否",
      },
      dependencies: {
        triggerFields: ["isBatchManaged"],
        show: (values) => values.isBatchManaged === true,
      },
    },
    {
      component: "Switch",
      fieldName: "expireDateFlag",
      label: "有效期必填",
      defaultValue: false,
      componentProps: {
        class: "w-auto",
        checkedChildren: "是",
        unCheckedChildren: "否",
      },
      dependencies: {
        triggerFields: ["isBatchManaged"],
        show: (values) => values.isBatchManaged === true,
      },
    },
    {
      component: "Switch",
      fieldName: "receiptDateFlag",
      label: "入库日期必填",
      defaultValue: false,
      componentProps: {
        class: "w-auto",
        checkedChildren: "是",
        unCheckedChildren: "否",
      },
      dependencies: {
        triggerFields: ["isBatchManaged"],
        show: (values) => values.isBatchManaged === true,
      },
    },
    {
      component: "Switch",
      fieldName: "vendorFlag",
      label: "供应商必填",
      defaultValue: false,
      componentProps: {
        class: "w-auto",
        checkedChildren: "是",
        unCheckedChildren: "否",
      },
      dependencies: {
        triggerFields: ["isBatchManaged"],
        show: (values) => values.isBatchManaged === true,
      },
    },
    {
      component: "Switch",
      fieldName: "purchaseOrderCodeFlag",
      label: "采购订单号必填",
      defaultValue: false,
      componentProps: {
        class: "w-auto",
        checkedChildren: "是",
        unCheckedChildren: "否",
      },
      dependencies: {
        triggerFields: ["isBatchManaged"],
        show: (values) => values.isBatchManaged === true,
      },
    },
    {
      component: "Switch",
      fieldName: "lotNumberFlag",
      label: "生产批号必填",
      defaultValue: false,
      componentProps: {
        class: "w-auto",
        checkedChildren: "是",
        unCheckedChildren: "否",
      },
      dependencies: {
        triggerFields: ["isBatchManaged"],
        show: (values) => values.isBatchManaged === true,
      },
    },
    {
      component: "Switch",
      fieldName: "qualityStatusFlag",
      label: "质量状态必填",
      defaultValue: false,
      componentProps: {
        class: "w-auto",
        checkedChildren: "是",
        unCheckedChildren: "否",
      },
      dependencies: {
        triggerFields: ["isBatchManaged"],
        show: (values) => values.isBatchManaged === true,
      },
    },
  ];
}

/** 列表的搜索表单 */
export function useGridFormSchema(): VbenFormSchema[] {
  return [
    {
      fieldName: "code",
      label: "物料编码",
      component: "Input",
      componentProps: {
        placeholder: "请输入物料编码",
        allowClear: true,
      },
    },
    {
      fieldName: "name",
      label: "物料名称",
      component: "Input",
      componentProps: {
        placeholder: "请输入物料名称",
        allowClear: true,
      },
    },
    {
      fieldName: "status",
      label: "状态",
      component: "Select",
      componentProps: {
        placeholder: "请选择状态",
        allowClear: true,
        options: getDictOptions(DICT_TYPE.COMMON_STATUS, "number"),
      },
    },
  ];
}

/** 列表的字段 */
export function useGridColumns(): VxeTableGridOptions["columns"] {
  return [
    { type: "checkbox", width: 40 },
    {
      field: "code",
      title: "物料编码",
      minWidth: 120,
    },
    {
      field: "name",
      title: "物料名称",
      minWidth: 180,
    },
    {
      field: "specification",
      title: "规格型号",
      minWidth: 150,
    },
    {
      field: "categoryName",
      title: "物料分类",
      minWidth: 120,
      slots: { default: "categoryName" },
    },
    {
      field: "unitMeasureName",
      title: "单位",
      minWidth: 80,
    },
    {
      field: "unitMeasureName2",
      title: "辅单位1",
      minWidth: 80,
    },
    {
      field: "unitMeasureRate1",
      title: "辅单位1换算比率",
      minWidth: 130,
    },
    {
      field: "unitMeasureName3",
      title: "辅单位2",
      minWidth: 80,
    },
    {
      field: "unitMeasureRate2",
      title: "辅单位2换算比率",
      minWidth: 130,
    },
    {
      field: "warehouseName",
      title: "默认仓库",
      minWidth: 100,
    },
    {
      field: "purchasePrice",
      title: "采购价",
      minWidth: 100,
      formatter: ({ cellValue }) => {
        return cellValue != null ? `¥${Number(cellValue).toFixed(2)}` : "";
      },
    },
    {
      field: "salesPrice",
      title: "销售价",
      minWidth: 100,
      formatter: ({ cellValue }) => {
        return cellValue != null ? `¥${Number(cellValue).toFixed(2)}` : "";
      },
    },
    {
      field: "isBatchManaged",
      title: "批次管理",
      minWidth: 80,
      slots: { default: "isBatchManaged" },
    },
    {
      field: "status",
      title: "状态",
      minWidth: 80,
      cellRender: {
        name: "CellDict",
        props: { type: DICT_TYPE.COMMON_STATUS },
      },
    },
    {
      field: "createTime",
      title: "创建时间",
      minWidth: 180,
      formatter: "formatDateTime",
    },
    {
      title: "操作",
      width: 180,
      fixed: "right",
      slots: { default: "actions" },
    },
  ];
}