<template>
  <div class="app-container">
    <el-row :gutter="20">
      <el-col :span="5" :xs="24">
        <div class="head-container">
          <el-input
            v-model="workUnitName"
            placeholder="请输入统计单元名称"
            clearable
            size="small"
            prefix-icon="el-icon-search"
            style="margin-bottom: 20px"
          />
        </div>
        <div class="head-container">
          <el-tree
            :data="workUnitOptions"
            :props="workUnitProps"
            :expand-on-click-node="false"
            :filter-node-method="filterNode"
            ref="treeRef"
            node-key="id"
            default-expand-all
            highlight-current
            @node-click="handleNodeClick"
          />
        </div>
      </el-col>

      <el-col :span="19" :xs="24">
        <el-form :model="queryParams" ref="queryFormRef" size="small" :inline="true" v-show="showSearch" label-width="100px">
          <el-form-item label="计量设备编号" prop="monitorCode">
            <el-input v-model="queryParams.monitorCode" placeholder="请输入计量设备编号" clearable @keyup.enter="handleQuery" />
          </el-form-item>

          <!--          <el-form-item label="统计单元编号" prop="workUnitCode">-->
          <!--            <el-input-->
          <!--              v-model="queryParams.workUnitCode"-->
          <!--              placeholder="请输入统计单元编号"-->
          <!--              clearable-->
          <!--              @keyup.enter="handleQuery"-->
          <!--            />-->
          <!--          </el-form-item>-->
          <!--          <el-form-item label="计量设备状态" prop="monitorStatus">-->
          <!--            <el-select v-model="queryParams.monitorStatus" placeholder="请选择计量设备状态" clearable>-->
          <!--              <el-option-->
          <!--                v-for="dict in dict.type.monitor_status"-->
          <!--                :key="dict.value"-->
          <!--                :label="dict.label"-->
          <!--                :value="dict.value"-->
          <!--              />-->
          <!--            </el-select>-->
          <!--          </el-form-item>-->
          <el-form-item label="设备计量类型" prop="monitorType">
            <el-select v-model="queryParams.monitorType" placeholder="请选择计算方式">
              <el-option v-for="dict in dict.type.monitor_type" :key="dict.value" :label="dict.label" :value="dict.value" />
            </el-select>
          </el-form-item>
          <el-form-item label="计算方式" prop="formulaMode">
            <el-select v-model="queryParams.formulaMode" placeholder="请选择计算方式" clearable>
              <el-option v-for="dict in dict.type.formula_mode" :key="dict.value" :label="dict.label" :value="dict.value" />
            </el-select>
          </el-form-item>
          <!--          <el-form-item label="启用标识" prop="isFlag">-->
          <!--            <el-select v-model="queryParams.isFlag" placeholder="请选择启用标识" clearable>-->
          <!--              <el-option-->
          <!--                v-for="dict in dict.type.is_flag"-->
          <!--                :key="dict.value"-->
          <!--                :label="dict.label"-->
          <!--                :value="dict.value"-->
          <!--              />-->
          <!--            </el-select>-->
          <!--          </el-form-item>-->
          <el-form-item>
            <el-button type="primary" icon="el-icon-search" size="mini" @click="handleQuery">搜索</el-button>
            <el-button icon="el-icon-refresh" size="mini" @click="resetQuery">重置</el-button>
          </el-form-item>
        </el-form>

        <el-row :gutter="10" class="mb8">
          <el-col :span="1.5">
            <el-button type="primary" plain icon="el-icon-plus" size="mini" @click="handleAdd" v-hasPermi="['ems/base:baseMonitorWorkUnit:add']"
              >新增
            </el-button>
          </el-col>
          <el-col :span="1.5">
            <el-button
              type="success"
              plain
              icon="el-icon-edit"
              size="mini"
              :disabled="single"
              @click="handleUpdate"
              v-hasPermi="['ems/base:baseMonitorWorkUnit:edit']"
              >修改
            </el-button>
          </el-col>
          <el-col :span="1.5">
            <el-button
              type="danger"
              plain
              icon="el-icon-delete"
              size="mini"
              :disabled="multiple"
              @click="handleDelete"
              v-hasPermi="['ems/base:baseMonitorWorkUnit:remove']"
              >删除
            </el-button>
          </el-col>
          <el-col :span="1.5">
            <el-button
              type="warning"
              plain
              icon="el-icon-download"
              size="mini"
              @click="handleExport"
              v-hasPermi="['ems/base:baseMonitorWorkUnit:export']"
              >导出
            </el-button>
          </el-col>
          <right-toolbar v-model:show-search="showSearch" @query-table="getList" :columns="columns"></right-toolbar>
        </el-row>

        <el-table v-loading="loading" :data="baseMonitorWorkUnitList" @selection-change="handleSelectionChange">
          <el-table-column type="selection" width="55" align="center" />
          <el-table-column label="主键标识" align="center" prop="objId" v-if="columns[0].visible" />
          <el-table-column label="计量设备编号" align="center" prop="monitorCode" v-if="columns[1].visible" />
          <el-table-column label="计量设备名称" align="center" prop="monitorName" v-if="columns[13].visible" />
          <el-table-column label="统计单元编号" align="center" prop="workUnitCode" v-if="columns[2].visible" />
          <el-table-column label="统计单元名称" align="center" prop="workUnitName" v-if="columns[14].visible" />
          <el-table-column label="计量设备状态" align="center" prop="monitorStatus" v-if="columns[3].visible">
            <template #default="scope">
              <dict-tag :options="dict.type.monitor_status" :value="scope.row.monitorStatus" />
            </template>
          </el-table-column>
          <el-table-column label="计量设备类型" align="center" prop="monitorType" v-if="columns[4].visible">
            <template #default="scope">
              <dict-tag :options="dict.type.monitor_type" :value="scope.row.monitorType" />
            </template>
          </el-table-column>
          <el-table-column label="计算方式" align="center" prop="formulaMode" v-if="columns[5].visible">
            <template #default="scope">
              <dict-tag :options="dict.type.formula_mode" :value="scope.row.formulaMode" />
            </template>
          </el-table-column>
          <el-table-column label="比率" align="center" prop="proportion" v-if="columns[6].visible" />
          <el-table-column label="备注" align="center" prop="remark" v-if="columns[7].visible" />
          <el-table-column label="启用标识" align="center" prop="isFlag" v-if="columns[8].visible">
            <template #default="scope">
              <dict-tag :options="dict.type.is_flag" :value="scope.row.isFlag" />
            </template>
          </el-table-column>
          <el-table-column label="创建人" align="center" prop="createdBy" v-if="columns[9].visible" />
          <el-table-column label="创建时间" align="center" prop="createdTime" width="180" v-if="columns[10].visible">
            <template #default="scope">
              <span>{{ parseTime(scope.row.createdTime, '{y}-{m}-{d} {h}:{i}:{s}') }}</span>
            </template>
          </el-table-column>
          <el-table-column label="修改人" align="center" prop="updatedBy" v-if="columns[11].visible" />
          <el-table-column label="修改时间" align="center" prop="updatedTime" width="180" v-if="columns[12].visible">
            <template #default="scope">
              <span>{{ parseTime(scope.row.updatedTime, '{y}-{m}-{d} {h}:{i}:{s}') }}</span>
            </template>
          </el-table-column>
          <el-table-column label="操作" align="center" class-name="small-padding fixed-width" width="100">
            <template #default="scope">
              <el-button
                size="mini"
                type="text"
                icon="el-icon-edit"
                @click="handleUpdate(scope.row)"
                v-hasPermi="['ems/base:baseMonitorWorkUnit:edit']"
                >修改
              </el-button>
              <el-button
                size="mini"
                type="text"
                icon="el-icon-delete"
                @click="handleDelete(scope.row)"
                v-hasPermi="['ems/base:baseMonitorWorkUnit:remove']"
                >删除
              </el-button>
            </template>
          </el-table-column>
        </el-table>

        <pagination v-show="total > 0" :total="total" v-model:page="queryParams.pageNum" v-model:limit="queryParams.pageSize" @pagination="getList" />
      </el-col>
    </el-row>
    <!-- 添加或修改统计计量信息对话框 -->
    <el-dialog :title="title" v-model="open" width="500px" append-to-body>
      <el-form ref="formRef" :model="form" :rules="rules" label-width="110px">
        <el-form-item label="统计单元" prop="workUnitCode">
          <EmsTreeSelect
            v-model="form.workUnitCode"
            :options="workUnitOptions"
            :normalizer="workUnitOptionsNormalizer"
            placeholder="请选择统计单元"
          />
        </el-form-item>
        <el-form-item label="计量设备名称" prop="monitorCode">
          <EmsTreeSelect v-model="form.monitorCode" :options="monitorInfoOptions" :normalizer="normalizer" placeholder="请选择计量设备" />
        </el-form-item>
        <el-form-item label="计量设备类型" prop="monitorType">
          <el-radio-group v-model="form.monitorType">
            <el-radio v-for="dict in dict.type.monitor_type" :key="dict.value" :label="parseInt(dict.value)">{{ dict.label }} </el-radio>
          </el-radio-group>
        </el-form-item>
        <el-form-item label="计算方式" prop="formulaMode">
          <el-radio-group v-model="form.formulaMode">
            <el-radio v-for="dict in dict.type.formula_mode" :key="dict.value" :label="parseInt(dict.value)">{{ dict.label }} </el-radio>
          </el-radio-group>
        </el-form-item>
        <el-form-item label="比率" prop="proportion">
          <el-input-number v-model="form.proportion" placeholder="请输入比率" />
        </el-form-item>
        <!--        <el-form-item label="备注" prop="remark">-->
        <!--          <el-input v-model="form.remark" placeholder="请输入备注"/>-->
        <!--        </el-form-item>-->
        <el-form-item label="启用标识" prop="isFlag">
          <el-radio-group v-model="form.isFlag">
            <el-radio v-for="dict in dict.type.is_flag" :key="dict.value" :label="dict.value">{{ dict.label }} </el-radio>
          </el-radio-group>
        </el-form-item>
      </el-form>
      <template #footer>
        <div class="dialog-footer">
          <el-button type="primary" @click="submitForm">确 定</el-button>
          <el-button @click="cancel">取 消</el-button>
        </div>
      </template>
    </el-dialog>
  </div>
</template>

<script setup lang="ts">
import { useDict } from '@/utils/dict';
import {
  listBaseMonitorWorkUnit,
  getBaseMonitorWorkUnit,
  delBaseMonitorWorkUnit,
  addBaseMonitorWorkUnit,
  updateBaseMonitorWorkUnit
} from '@/api/ems/base/baseMonitorWorkUnit';
import EmsTreeSelect from '@/views/ems/components/EmsTreeSelect.vue';
import { listBaseMonitorInfo } from '@/api/ems/base/baseMonitorInfo';
import { getWorkUnitTrees } from '@/api/ems/base/baseWorkUnit';
import { handleTree } from '@/utils/ruoyi';

defineOptions({
  name: 'BaseMonitorWorkUnit',
  dicts: ['formula_mode', 'is_flag', 'monitor_status', 'monitor_type']
});

const { proxy } = getCurrentInstance() as ComponentInternalInstance;

const dict = reactive({
  type: useDict('formula_mode', 'is_flag', 'monitor_status', 'monitor_type')
});

const formRef = ref<ElFormInstance>();
const queryFormRef = ref<ElFormInstance>();
const treeRef = ref<ElTreeInstance>();

// 这里统一通过工厂函数重建表单，避免弹窗在新增/编辑切换时残留上一条业务数据。
const createFormData = () => ({
  objId: null,
  monitorCode: null,
  workUnitCode: null,
  monitorStatus: null,
  monitorType: '2',
  formulaMode: 0,
  proportion: 1,
  remark: null,
  isFlag: '0',
  createdBy: null,
  createdTime: null,
  updatedBy: null,
  updatedTime: null
});

// 这里保留原页面字段结构，优先保证 EMS 业务行为与接口入参不发生漂移。
const state = reactive({
  monitorInfoOptions: [],
  workUnitOptions: [],
  workUnitName: undefined,
  // 遮罩层
  loading: true,
  // 选中数组
  ids: [],
  // 非单个禁用
  single: true,
  // 非多个禁用
  multiple: true,
  // 显示搜索条件
  showSearch: true,
  // 总条数
  total: 0,
  // 统计计量信息表格数据
  baseMonitorWorkUnitList: [],
  // 弹出层标题
  title: '',
  // 是否显示弹出层
  open: false,
  // 查询参数
  queryParams: {
    pageNum: 1,
    pageSize: 10,
    monitorCode: null,
    workUnitCode: null,
    monitorStatus: null,
    monitorType: '2',
    formulaMode: null,
    proportion: null,
    isFlag: null,
    createdBy: null,
    createdTime: null,
    updatedBy: null,
    updatedTime: null
  },
  // 左侧树点击时添加默认赋值
  addWorkUnitCode: null,
  // 表单参数
  form: createFormData(),
  workUnitProps: {
    children: 'children',
    label: 'label'
  },
  // 表单校验
  rules: {
    objId: [
      {
        required: true,
        message: '主键标识不能为空',
        trigger: 'blur'
      }
    ],
    monitorCode: [
      {
        required: true,
        message: '计量设备编号不能为空',
        trigger: 'blur'
      }
    ]
  },
  columns: [
    {
      key: 0,
      label: `主键标识`,
      visible: false
    },
    {
      key: 1,
      label: `计量设备编号`,
      visible: true
    },
    {
      key: 2,
      label: `统计单元编号`,
      visible: true
    },
    {
      key: 3,
      label: `计量设备状态`,
      visible: false
    },
    {
      key: 4,
      label: `计量设备类型`,
      visible: true
    },
    {
      key: 5,
      label: `计算方式`,
      visible: true
    },
    {
      key: 6,
      label: `比率`,
      visible: true
    },
    {
      key: 7,
      label: `备注`,
      visible: false
    },
    {
      key: 8,
      label: `启用标识`,
      visible: true
    },
    {
      key: 9,
      label: `创建人`,
      visible: false
    },
    {
      key: 10,
      label: `创建时间`,
      visible: false
    },
    {
      key: 11,
      label: `修改人`,
      visible: false
    },
    {
      key: 12,
      label: `修改时间`,
      visible: false
    },
    {
      key: 13,
      label: `计量设备名称`,
      visible: true
    },
    {
      key: 14,
      label: `统计单元名称`,
      visible: true
    }
  ]
} as any);

const {
  addWorkUnitCode,
  baseMonitorWorkUnitList,
  columns,
  form,
  ids,
  loading,
  monitorInfoOptions,
  multiple,
  open,
  queryParams,
  rules,
  showSearch,
  single,
  title,
  total,
  workUnitName,
  workUnitOptions,
  workUnitProps
} = toRefs(state);

const getList = () => {
  loading.value = true;
  listBaseMonitorWorkUnit(queryParams.value).then((response) => {
    baseMonitorWorkUnitList.value = response.rows;
    total.value = response.total;
    loading.value = false;
  });
};

const cancel = () => {
  open.value = false;
  reset();
};

const reset = () => {
  form.value = {
    objId: null,
    monitorCode: null,
    workUnitCode: null,
    monitorStatus: null,
    monitorType: queryParams.value.monitorType,
    formulaMode: 0,
    proportion: 1,
    remark: null,
    isFlag: '0',
    createdBy: null,
    createdTime: null,
    updatedBy: null,
    updatedTime: null
  };
  formRef.value?.resetFields();
};

const handleQuery = () => {
  queryParams.value.pageNum = 1;
  getList();
};

const resetQuery = () => {
  queryParams.value.workUnitCode = null;
  queryFormRef.value?.resetFields();
  handleQuery();
};

const handleSelectionChange = (selection) => {
  ids.value = selection.map((item) => item.objId);
  single.value = selection.length !== 1;
  multiple.value = !selection.length;
};

const handleAdd = () => {
  reset();
  form.value.workUnitCode = addWorkUnitCode.value;
  open.value = true;
  title.value = '添加统计计量信息';
};

const handleUpdate = (row) => {
  reset();
  const objId = row.objId || ids.value;
  getBaseMonitorWorkUnit(objId).then((response) => {
    form.value = response.data;
    open.value = true;
    title.value = '修改统计计量信息';
  });
};

const submitForm = () => {
  formRef.value.validate((valid) => {
    if (valid) {
      if (form.value.objId != null) {
        updateBaseMonitorWorkUnit(form.value).then((response) => {
          proxy?.$modal.msgSuccess('修改成功');
          open.value = false;
          getList();
        });
      } else {
        addBaseMonitorWorkUnit(form.value).then((response) => {
          proxy?.$modal.msgSuccess('新增成功');
          open.value = false;
          getList();
        });
      }
    }
  });
};

const handleDelete = (row) => {
  const objIds = row.objId || ids.value;
  proxy?.$modal
    .confirm('是否确认删除统计计量信息编号为"' + objIds + '"的数据项？')
    .then(function () {
      return delBaseMonitorWorkUnit(objIds);
    })
    .then(() => {
      getList();
      proxy?.$modal.msgSuccess('删除成功');
    })
    .catch(() => {});
};

const handleExport = () => {
  proxy?.download(
    'ems/base/baseMonitorWorkUnit/export',
    {
      ...queryParams.value
    },
    `baseMonitorWorkUnit_${new Date().getTime()}.xlsx`
  );
};

const getTreeselect = () => {
  listBaseMonitorInfo({
    monitorType: queryParams.value.monitorType
  }).then((response) => {
    monitorInfoOptions.value = [];
    monitorInfoOptions.value = handleTree(response.data, 'objId', 'parentId');
  });
};

const normalizer = (node) => {
  if (node.children && !node.children.length) {
    delete node.children;
  }
  return {
    id: node.monitorCode,
    label: node.monitorName,
    children: node.children
  };
};

const workUnitOptionsNormalizer = (node) => {
  if (node.children && !node.children.length) {
    delete node.children;
  }
  return {
    id2: node.workUnitCode,
    label2: node.workUnitName,
    children2: node.children
  };
};

const getDeptTrees = () => {
  getWorkUnitTrees().then((response) => {
    workUnitOptions.value = [];
    workUnitOptions.value = JSON.parse(JSON.stringify(response.data).replaceAll('id', 'id2').replaceAll('code', 'id'));
  });
};

const filterNode = (value, data) => {
  if (!value) return true;
  return data.label.indexOf(value) !== -1;
};

const handleNodeClick = (data) => {
  queryParams.value.workUnitCode = data.id;
  addWorkUnitCode.value = data.id;
  handleQuery();
};

watch(workUnitName, (val) => {
  treeRef.value.filter(val);
});
watch(
  () => queryParams.monitorType,
  (newVal, oldVal) => {
    // 执行方法，可以在这里调用你的方法
    getTreeselect();
  }
);

onMounted(() => {
  getTreeselect();
  getDeptTrees();
  getList();
});
</script>
