import { FunctionCodeEnum } from './config';
import { ModbusCommandValueType } from './type';
import { GenModbusCommandType } from '/@/api/task/model';

const registerInfo = (record: ModbusCommandValueType): Partial<GenModbusCommandType> => {
  const {
    method,
    registerNumber,
    registerValue,
    registerValues,
    coilNumber,
    coilValue,
    coilValues,
  } = record;
  switch (method) {
    case FunctionCodeEnum.READ_COIL_STATE_01 || FunctionCodeEnum.READ_INPUT_STATE_02:
      return { registerNum: coilNumber };
    case FunctionCodeEnum.READ_KEEP_REGISTER_03 || FunctionCodeEnum.READ_INPUT_REGISTER_04:
      return { registerNum: registerNumber };
    case FunctionCodeEnum.WRITE_SINGLE_COIL_REGISTER_05:
      return { registerValues: [coilValue] };
    case FunctionCodeEnum.WRITE_SINGLE_KEEP_COIL_REGISTER_06:
      return { registerValues: [registerValue] };
    case FunctionCodeEnum.WRITE_MULTIPLE_COIL_STATE_15:
      return {
        registerNum: coilNumber,
        registerValues: coilValues.map((item) => (item.value ? 1 : 0)),
      };
    case FunctionCodeEnum.WRITE_MULTIPLE_KEEP_REGISTER_16:
      return {
        registerNum: registerNumber,
        registerValues: registerValues.map((item) => item.value),
      };
    default:
      return {};
  }
};

export const composeModbusModalData = (record: ModbusCommandValueType): GenModbusCommandType => {
  const { crc, deviceCode, method, registerAddr } = record;
  return {
    crc,
    deviceCode,
    method,
    registerAddr,
    ...registerInfo(record),
  };
};