import type { VbenFormApi, VbenFormSchema } from '#/adapter/form';
import type { VxeTableGridOptions } from '#/adapter/vxe-table';
import { z } from '#/adapter/form';
import { getSimpleUserListByRoleCode } from '#/api/system/user';
import type { MesProTaskApi } from '#/api/mes/pro/task';
import type { MesProWorkOrderApi } from '#/api/mes/pro/workorder';

import type { Ref } from 'vue';

import { markRaw } from 'vue';

import { DICT_TYPE } from '@vben/constants';
import { getDictOptions } from '@vben/hooks';

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

import CrmCustomerSelect from '#/components/crm-customer-select.vue';
import { calculateTaskDuration } from '#/api/mes/pro/task';
import { getRangePickerDefaultProps } from '#/utils';
import { MdItemSelect } from '#/views/mes/md/item/components';
import { MdWorkstationSelect } from '#/views/mes/md/workstation/components';
import { ProProcessSelect } from '#/views/mes/process-design/process/components';
import { RouteColorPicker } from '#/views/mes/process-design/route/components';
import { ProWorkOrderSelect } from '#/views/mes/pro/workorder/components';

/** 待排产订单列表的搜索表单 */
export function useGridFormSchema(): VbenFormSchema[] {
  return [
    {
      fieldName: 'code',
      label: '订单编码',
      component: 'Input',
      componentProps: {
        allowClear: true,
        placeholder: '请输入订单编码',
      },
    },
    {
      fieldName: 'name',
      label: '订单名称',
      component: 'Input',
      componentProps: {
        allowClear: true,
        placeholder: '请输入订单名称',
      },
    },
    {
      fieldName: 'orderSourceCode',
      label: '来源单据',
      component: 'Input',
      componentProps: {
        allowClear: true,
        placeholder: '请输入来源单据编号',
      },
    },
    {
      fieldName: 'productId',
      label: '产品',
      component: markRaw(MdItemSelect),
      componentProps: {
        placeholder: '请选择产品',
      },
    },
    {
      fieldName: 'clientId',
      label: '客户',
      component: markRaw(CrmCustomerSelect),
      componentProps: {
        placeholder: '请选择客户',
      },
    },
    {
      fieldName: 'requestDate',
      label: '需求日期',
      component: 'RangePicker',
      componentProps: {
        ...getRangePickerDefaultProps(),
        allowClear: true,
      },
    },
  ];
}

/** 待排产订单列表的字段 */
export function useGridColumns(): VxeTableGridOptions<MesProWorkOrderApi.WorkOrder>['columns'] {
  return [
    {
      field: 'code',
      title: '订单编码',
      fixed: 'left',
      width: 200,
      treeNode: true,
      slots: { default: 'code' },
    },
    {
      field: 'name',
      title: '订单名称',
      minWidth: 150,
    },
    {
      field: 'orderSourceType',
      title: '订单来源',
      width: 100,
      cellRender: {
        name: 'CellDict',
        props: { type: DICT_TYPE.MES_PRO_WORK_ORDER_SOURCE_TYPE },
      },
    },
    {
      field: 'orderSourceCode',
      title: '来源单据编号',
      width: 140,
    },
    {
      field: 'productCode',
      title: '产品编码',
      width: 120,
    },
    {
      field: 'productName',
      title: '产品名称',
      minWidth: 120,
    },
    {
      field: 'productSpecification',
      title: '规格型号',
      width: 120,
    },
    {
      field: 'unitMeasureName',
      title: '单位',
      width: 80,
    },
    {
      field: 'status',
      title: '订单状态',
      width: 100,
      cellRender: {
        name: 'CellDict',
        props: { type: DICT_TYPE.MES_PRO_WORK_ORDER_STATUS },
      },
    },
    {
      field: 'productionStatus',
      title: '生产状态',
      width: 100,
      cellRender: {
        name: 'CellDict',
        props: { type: DICT_TYPE.ORDER_PRODUCTION_STATUS },
      },
    },
    {
      field: 'quantity',
      title: '订单数量',
      width: 100,
    },
    // {
    //   field: 'quantityChanged',
    //   title: '调整数量',
    //   width: 100,
    // },
    {
      field: 'quantityProduced',
      title: '已生产数量',
      width: 100,
    },
    {
      field: 'clientCode',
      title: '客户编码',
      width: 120,
    },
    {
      field: 'clientName',
      title: '客户名称',
      width: 120,
    },
    {
      field: 'requestDate',
      title: '需求日期',
      width: 120,
      formatter: 'formatDate',
    },

    {
      title: '操作',
      width: 160,
      fixed: 'right',
      slots: { default: 'actions' },
    },
  ];
}

/** 排产对话框只读订单信息的表单 */
export function useScheduleFormSchema(): VbenFormSchema[] {
  return [
    {
      fieldName: 'code',
      label: '订单编码',
      component: 'Input',
      componentProps: {
        disabled: true,
      },
    },
    {
      fieldName: 'name',
      label: '订单名称',
      component: 'Input',
      componentProps: {
        disabled: true,
      },
    },
    {
      fieldName: 'orderSourceType',
      label: '订单来源',
      component: 'Select',
      componentProps: {
        disabled: true,
        options: getDictOptions(
          DICT_TYPE.MES_PRO_WORK_ORDER_SOURCE_TYPE,
          'number',
        ),
      },
    },
    {
      fieldName: 'orderSourceCode',
      label: '来源单据编号',
      component: 'Input',
      componentProps: {
        disabled: true,
      },
    },
    {
      fieldName: 'type',
      label: '订单类型',
      component: 'Select',
      componentProps: {
        disabled: true,
        options: getDictOptions(DICT_TYPE.MES_PRO_WORK_ORDER_TYPE, 'number'),
      },
    },
    {
      fieldName: 'productId',
      label: '产品',
      component: markRaw(MdItemSelect),
      componentProps: {
        disabled: true,
      },
    },
    {
      fieldName: 'productSpecification',
      label: '规格型号',
      component: 'Input',
      componentProps: {
        disabled: true,
      },
    },
    {
      fieldName: 'unitMeasureName',
      label: '单位',
      component: 'Input',
      componentProps: {
        disabled: true,
      },
    },
    {
      fieldName: 'quantity',
      label: '订单数量',
      component: 'InputNumber',
      componentProps: {
        class: '!w-full',
        disabled: true,
        precision: 2,
      },
    },
    {
      fieldName: 'clientId',
      label: '客户',
      component: markRaw(CrmCustomerSelect),
      componentProps: {
        disabled: true,
      },
    },
    {
      fieldName: 'batchCode',
      label: '批次号',
      component: 'Input',
      componentProps: {
        disabled: true,
      },
    },
    {
      fieldName: 'requestDate',
      label: '需求日期',
      component: 'DatePicker',
      componentProps: {
        class: '!w-full',
        disabled: true,
        format: 'YYYY-MM-DD',
        valueFormat: 'YYYY-MM-DD',
      },
    },
    {
      fieldName: 'status',
      label: '订单状态',
      component: 'Select',
      componentProps: {
        disabled: true,
        options: getDictOptions(DICT_TYPE.MES_PRO_WORK_ORDER_STATUS, 'number'),
      },
    },
    {
      fieldName: 'remark',
      label: '备注',
      component: 'Textarea',
      formItemClass: 'col-span-3',
      componentProps: {
        disabled: true,
        rows: 2,
      },
    },
  ];
}

/** 生产任务子表的字段 */
export function useTaskGridColumns(
  editable: boolean,
): VxeTableGridOptions<MesProTaskApi.Task>['columns'] {
  return [
    {
      field: 'code',
      title: '任务编码',
      width: 140,
    },
    {
      field: 'name',
      title: '任务名称',
      minWidth: 150,
    },
    {
      field: 'workstationCode',
      title: '工作站编号',
      width: 120,
    },
    {
      field: 'workstationName',
      title: '工作站名称',
      width: 120,
    },
    {
      field: 'quantity',
      title: '排产数量',
      width: 100,
    },
    {
      field: 'producedQuantity',
      title: '已生产数量',
      width: 100,
    },
    {
      field: 'startTime',
      title: '开始生产时间',
      width: 170,
      formatter: 'formatDateTime',
    },
    {
      field: 'duration',
      title: '生产时长(h)',
      width: 120,
    },
    {
      field: 'endTime',
      title: '预计完成时间',
      width: 170,
      formatter: 'formatDateTime',
    },
    {
      field: 'colorCode',
      title: '显示颜色',
      width: 100,
      slots: { default: 'colorCode' },
    },
    ...(editable
      ? [
          {
            title: '操作',
            width: 160,
            fixed: 'right',
            slots: { default: 'actions' },
          } as const,
        ]
      : []),
  ];
}

/** 生产任务新增/修改的表单 */
export function useTaskFormSchema(formApi?: VbenFormApi, workOrderQuantity?: Ref<number | undefined>): VbenFormSchema[] {
  return [
    {
      fieldName: 'workstationId',
      label: '工作站',
      component: markRaw(MdWorkstationSelect),
      componentProps: {
        placeholder: '请选择工作站',
        onChange: () => recalcDuration(formApi),
      },
      rules: 'selectRequired',
    },
    {
      fieldName: 'quantity',
      label: '排产数量',
      component: 'InputNumber',
      description: () => workOrderQuantity?.value != null ? `订单数量：${workOrderQuantity.value}` : '',
      componentProps: {
        class: '!w-full',
        min: 0.01,
        placeholder: '请输入排产数量',
        precision: 2,
        onChange: () => recalcDuration(formApi),
      },
      rules: 'required',
    },
    {
      fieldName: 'colorCode',
      label: '甘特颜色',
      component: markRaw(RouteColorPicker),
    },
    {
      fieldName: 'startTime',
      label: '开始时间',
      component: 'DatePicker',
      componentProps: {
        class: '!w-full',
        placeholder: '请先选择工作站',
        showTime: true,
        valueFormat: 'YYYY-MM-DD HH:mm:ss',
        onChange: () => recalcEndTime(formApi),
      },
      dependencies: {
        triggerFields: ['workstationId'],
        componentProps: (values) => ({
          disabled: !values.workstationId,
          placeholder: values.workstationId ? '请选择开始时间' : '请先选择工作站',
        }),
      },
      rules: 'required',
    },
    {
      fieldName: 'duration',
      label: '生产时长(h)',
      component: 'InputNumber',
      componentProps: {
        class: '!w-full',
        min: 1,
        placeholder: '请输入生产时长',
        precision: 0,
        // 生产时长变更：重新计算结束时间
        onChange: () => recalcEndTime(formApi),
      },
      rules: 'required',
    },
    {
      fieldName: 'endTime',
      label: '结束时间',
      component: 'DatePicker',
      componentProps: {
        class: '!w-full',
        disabled: true,
        showTime: true,
        valueFormat: 'YYYY-MM-DD HH:mm:ss',
      },
    },
    {
      fieldName: 'assigneeUserIds',
      label: '报工人',
      component: 'ApiSelect',
      componentProps: {
        placeholder: '请选择报工人',
        api: getSimpleUserListByRoleCode,
        params: { code: 'production_submit_work' },
        labelField: 'nickname',
        valueField: 'id',
        mode: 'multiple',
      },
      rules: z.array(z.number()).min(1, '请选择报工人').default([]),
    },
    {
      fieldName: 'remark',
      label: '备注',
      component: 'Textarea',
      formItemClass: 'col-span-3',
      componentProps: {
        placeholder: '请输入备注',
        rows: 2,
      },
    },
  ];
}

/** 选择工作站或修改数量后，自动计算生产时长 */
async function recalcDuration(formApi?: VbenFormApi) {
  if (!formApi) return;
  const values = await formApi.getValues();
  if (!values.workstationId || !values.quantity) return;
  try {
    const res = await calculateTaskDuration(values.workstationId, values.quantity);
    await formApi.setFieldValue('duration', res.duration);
    if (res.duration === 0) {
      message.warning('该工作站未绑定设备，或绑定设备未配置生产效率');
    }
    // 重新计算结束时间
    await recalcEndTime(formApi);
  } catch {
    // 计算失败不阻塞用户操作
  }
}

/** 计算结束时间：开始时间 + 生产时长（小时） */
async function recalcEndTime(formApi?: VbenFormApi) {
  if (!formApi) {
    return;
  }
  const values = await formApi.getValues();
  if (values.startTime && values.duration) {
    const start = new Date(values.startTime).getTime();
    const endTime = start + values.duration * 3600 * 1000;
    const pad = (n: number) => String(n).padStart(2, '0');
    const d = new Date(endTime);
    const formatted = `${d.getFullYear()}-${pad(d.getMonth() + 1)}-${pad(d.getDate())} ${pad(d.getHours())}:${pad(d.getMinutes())}:${pad(d.getSeconds())}`;
    await formApi.setFieldValue('endTime', formatted);
  }
}

/** 任务选择弹窗的搜索表单 */
export function useTaskSelectGridFormSchema(): VbenFormSchema[] {
  return [
    {
      fieldName: 'workOrderId',
      label: '生产订单',
      component: markRaw(ProWorkOrderSelect),
      componentProps: {
        allowClear: true,
        placeholder: '请选择生产订单',
      },
    },
    {
      fieldName: 'processId',
      label: '所属工序',
      component: markRaw(ProProcessSelect),
      componentProps: {
        allowClear: true,
        placeholder: '请选择工序',
      },
    },
    {
      fieldName: 'workstationId',
      label: '工作站',
      component: markRaw(MdWorkstationSelect),
      componentProps: {
        allowClear: true,
        placeholder: '请选择工作站',
      },
    },
    {
      fieldName: 'code',
      label: '任务编号',
      component: 'Input',
      componentProps: {
        allowClear: true,
        placeholder: '请输入任务编号',
      },
    },
    {
      fieldName: 'name',
      label: '任务名称',
      component: 'Input',
      componentProps: {
        allowClear: true,
        placeholder: '请输入任务名称',
      },
    },
  ];
}

/** 任务选择弹窗的字段 */
export function useTaskSelectGridColumns(
  multiple = false,
): VxeTableGridOptions<MesProTaskApi.Task>['columns'] {
  return [
    { type: multiple ? 'checkbox' : 'radio', width: 50 },
    { field: 'code', title: '任务编号', width: 180 },
    { field: 'name', title: '任务名称', minWidth: 140 },
    { field: 'workstationCode', title: '工作站编码', width: 140 },
    { field: 'workstationName', title: '工作站名称', width: 140 },
    { field: 'processName', title: '工序', width: 120 },
    {
      field: 'checkFlag',
      title: '是否质检',
      width: 100,
      cellRender: {
        name: 'CellDict',
        props: { type: DICT_TYPE.INFRA_BOOLEAN_STRING },
      },
    },
    { field: 'itemCode', title: '物料编码', width: 140 },
    { field: 'itemName', title: '物料名称', width: 140 },
    { field: 'itemSpecification', title: '规格型号', width: 120 },
    { field: 'quantity', title: '排产数量', width: 100 },
    { field: 'producedQuantity', title: '已生产数量', width: 110 },
    {
      field: 'startTime',
      title: '开始生产时间',
      width: 170,
      formatter: 'formatDateTime',
    },
    { field: 'duration', title: '生产时长(h)', width: 120 },
    {
      field: 'endTime',
      title: '预计完成时间',
      width: 170,
      formatter: 'formatDateTime',
    },
    {
      field: 'status',
      title: '任务状态',
      width: 100,
      cellRender: {
        name: 'CellDict',
        props: { type: DICT_TYPE.MES_PRO_TASK_STATUS },
      },
    },
  ];
}