<template>
  <div class="app-container">
    <el-form :model="queryParams" ref="queryRef" :inline="true" v-show="showSearch" label-width="68px">
      <el-form-item label="库区编码" prop="areaCode">
        <el-input v-model="queryParams.areaCode" placeholder="请输入库区编码" clearable @keyup.enter="handleQuery" />
      </el-form-item>
      <el-form-item label="库区名称" prop="areaName">
        <el-input v-model="queryParams.areaName" placeholder="请输入库区名称" clearable @keyup.enter="handleQuery" />
      </el-form-item>
      <el-form-item label="所属仓库" prop="warehouseId">
        <el-input v-model="queryParams.warehouseId" placeholder="请选择仓库" readonly clearable style="width: 280px;">
          <template #append>
            <el-button type="primary" @click="openQueryWarehouseSelect">选择仓库</el-button>
          </template>
        </el-input>
      </el-form-item>
      <el-form-item>
        <el-button type="primary" icon="Search" @click="handleQuery">搜索</el-button>
        <el-button icon="Refresh" @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="Plus" @click="handleAdd" v-hasPermi="['ware:wmsArea:add']">新增</el-button>
      </el-col>
      <el-col :span="1.5">
        <el-button type="success" plain icon="Edit" :disabled="single" @click="handleUpdate"
          v-hasPermi="['ware:wmsArea:edit']">修改</el-button>
      </el-col>
      <el-col :span="1.5">
        <el-button type="danger" plain icon="Delete" :disabled="multiple" @click="handleDelete"
          v-hasPermi="['ware:wmsArea:remove']">删除</el-button>
      </el-col>
      <el-col :span="1.5">
        <el-button type="warning" plain icon="Download" @click="handleExport"
          v-hasPermi="['ware:wmsArea:export']">导出</el-button>
      </el-col>
      <right-toolbar v-model:showSearch="showSearch" @queryTable="getList"></right-toolbar>
    </el-row>

    <el-table v-loading="loading" :data="wmsAreaList" default-expand-all @selection-change="handleSelectionChange"
      row-key="areaId" :tree-props="{ children: 'children' }">
      <el-table-column type="selection" width="55" align="center" />
      <el-table-column label="库区编码" align="center" prop="areaCode" />
      <el-table-column label="库区名称" align="center" prop="areaName" />
      <el-table-column label="所属仓库ID" align="center" prop="warehouseId" />
      <el-table-column label="库区类型" align="center" prop="areaType" />
      <el-table-column label="温度范围" align="center" prop="temperatureRange" />
      <el-table-column label="湿度范围" align="center" prop="humidityRange" />
      <el-table-column label="库区容量" align="center" prop="capacity" />
      <el-table-column label="排序" align="center" prop="sortOrder" />
      <el-table-column label="状态" align="center" prop="status">
        <template #default="scope">
          <dict-tag :options="sys_normal_disable" :value="scope.row.status" />
        </template>
      </el-table-column>
      <el-table-column label="操作" align="center" class-name="small-padding fixed-width">
        <template #default="scope">
          <el-button link type="primary" icon="Edit" @click="handleUpdate(scope.row)"
            v-hasPermi="['ware:wmsArea:edit']">修改</el-button>
          <el-button link type="primary" icon="Delete" @click="handleDelete(scope.row)"
            v-hasPermi="['ware:wmsArea:remove']">删除</el-button>
          <el-button link type="primary" icon="Plus" @click="handleAddSub(scope.row)"
            v-hasPermi="['ware:wmsArea:add']">添加下级</el-button>
        </template>
      </el-table-column>
    </el-table>

    <!-- 添加或修改库区信息对话框 -->
    <el-dialog :title="title" v-model="open" width="500px" append-to-body>
      <el-form ref="wmsAreaRef" :model="form" :rules="rules" label-width="80px">
        <el-form-item label="库区编码" prop="areaCode">
          <el-input v-model="form.areaCode" placeholder="请输入库区编码" />
        </el-form-item>
        <el-form-item label="库区名称" prop="areaName">
          <el-input v-model="form.areaName" placeholder="请输入库区名称" />
        </el-form-item>
        <el-form-item label="所属仓库" prop="warehouseId">
          <el-input v-model="form.warehouseId" placeholder="请选择仓库" readonly clearable>
            <template #append>
              <el-button type="primary" @click="openWarehouseSelect">选择仓库</el-button>
            </template>
          </el-input>
        </el-form-item>
        <el-form-item label="父库区" prop="parentId">
          <el-cascader v-model="form.parentId" :options="wmsAreaOptions"
            :props="{ value: 'areaId', label: 'areaName', checkStrictly: true }" placeholder="请选择父库区" clearable />
        </el-form-item>
        <el-form-item label="温度范围" prop="temperatureRange">
          <el-input v-model="form.temperatureRange" placeholder="请输入温度范围" />
        </el-form-item>
        <el-form-item label="湿度范围" prop="humidityRange">
          <el-input v-model="form.humidityRange" placeholder="请输入湿度范围" />
        </el-form-item>
        <el-form-item label="库区容量" prop="capacity">
          <el-input v-model="form.capacity" placeholder="请输入库区容量" />
        </el-form-item>
        <el-form-item label="排序" prop="sortOrder">
          <el-input v-model="form.sortOrder" placeholder="请输入显示顺序" />
        </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>

    <!-- 仓库选择对话框组件 -->
    <WarehouseSelectDialog ref="warehouseSelectDialogRef" v-model:visible="warehouseDialogVisible" title="选择仓库"
      @confirm="handleWarehouseConfirm" />
  </div>
</template>

<script setup name="WmsArea">
import { listWmsArea, getWmsArea, delWmsArea, addWmsArea, updateWmsArea } from "@/api/ware/wmsArea"
import WarehouseSelectDialog from "@/components/WarehouseSelectDialog.vue"
import { handleTree } from "@/utils/ruoyi"

const { proxy } = getCurrentInstance()

const { sys_normal_disable } = proxy.useDict('sys_normal_disable')

const wmsAreaList = ref([])
const open = ref(false)
const loading = ref(true)
const showSearch = ref(true)
const ids = ref([])
const single = ref(true)
const multiple = ref(true)
const total = ref(0)
const title = ref("")

// 仓库选择对话框状态
const warehouseDialogVisible = ref(false)
// 仓库选择对话框实例引用
const warehouseSelectDialogRef = ref(null)
// 标记当前是否为查询表单的仓库选择
const isQueryWarehouseSelect = ref(false)

// 库区下拉树选项
const wmsAreaOptions = ref([])

const data = reactive({
  form: {},
  queryParams: {
    pageNum: 1,
    pageSize: 100,
    areaCode: null,
    areaName: null,
    warehouseId: null,
    parentId: null,
    areaType: null,
    temperatureRange: null,
    humidityRange: null,
    capacity: null,
    sortOrder: null,
    status: null,
  },
  rules: {
    areaCode: [
      { required: true, message: "库区编码不能为空", trigger: "blur" }
    ],
    areaName: [
      { required: true, message: "库区名称不能为空", trigger: "blur" }
    ],
    warehouseId: [
      { required: true, message: "所属仓库ID不能为空", trigger: "change" }
    ],
  }
})

const { queryParams, form, rules } = toRefs(data)

/** 查询库区信息列表 */
function getList() {
  loading.value = true
  listWmsArea(queryParams.value).then(response => {
    // 将扁平数据转换为树形结构
    wmsAreaList.value = handleTree(response.rows, 'areaId', 'parentId')
    wmsAreaOptions.value = [
      { areaId: 0, areaName: '顶级节点', children: wmsAreaList.value }
    ]
    total.value = response.total
    loading.value = false
  })
}

// 取消按钮
function cancel() {
  open.value = false
  reset()
}

// 表单重置
function reset() {
  form.value = {
    areaId: null,
    areaCode: null,
    areaName: null,
    warehouseId: null,
    parentId: null,
    areaType: null,
    temperatureRange: null,
    humidityRange: null,
    capacity: null,
    sortOrder: null,
    status: null,
    createBy: null,
    createTime: null,
    updateBy: null,
    updateTime: null
  }
  proxy.resetForm("wmsAreaRef")
}

/** 搜索按钮操作 */
function handleQuery() {
  queryParams.value.pageNum = 1
  getList()
}

/** 重置按钮操作 */
function resetQuery() {
  proxy.resetForm("queryRef")
  handleQuery()
}

// 多选框选中数据
function handleSelectionChange(selection) {
  ids.value = selection.map(item => item.areaId)
  single.value = selection.length != 1
  multiple.value = !selection.length
}

/** 新增按钮操作 */
function handleAdd() {
  reset()
  open.value = true
  title.value = "添加库区信息"
}

/** 添加下级按钮操作 */
function handleAddSub(row) {
  reset()
  // 设置父库区ID
  form.value.parentId = row.areaId
  open.value = true
  title.value = "添加下级库区"
}

/** 修改按钮操作 */
async function handleUpdate(row) {
  reset()
  const _areaId = row.areaId || ids.value
  getWmsArea(_areaId).then(response => {
    form.value = response.data
    open.value = true
    title.value = "修改库区信息"
  })
}

/** 提交按钮 */
function submitForm() {
  proxy.$refs["wmsAreaRef"].validate(valid => {
    if (valid) {
      if (form.value.areaId != null) {
        updateWmsArea(form.value).then(response => {
          proxy.$modal.msgSuccess("修改成功")
          open.value = false
          getList()
        })
      } else {
        addWmsArea(form.value).then(response => {
          proxy.$modal.msgSuccess("新增成功")
          open.value = false
          getList()
        })
      }
    }
  })
}

/** 删除按钮操作 */
function handleDelete(row) {
  const _areaIds = row.areaId || ids.value
  proxy.$modal.confirm('是否确认删除库区信息编号为"' + _areaIds + '"的数据项？').then(function () {
    return delWmsArea(_areaIds)
  }).then(() => {
    getList()
    proxy.$modal.msgSuccess("删除成功")
  }).catch(() => { })
}

/** 导出按钮操作 */
function handleExport() {
  proxy.download('ware/wmsArea/export', {
    ...queryParams.value
  }, `wmsArea_${new Date().getTime()}.xlsx`)
}

/** 打开搜索表单仓库选择对话框 */
function openQueryWarehouseSelect() {
  isQueryWarehouseSelect.value = true
  warehouseDialogVisible.value = true
  if (warehouseSelectDialogRef.value) {
    warehouseSelectDialogRef.value.handleQuery()
  }
}

/** 打开编辑表单仓库选择对话框 */
function openWarehouseSelect() {
  isQueryWarehouseSelect.value = false
  warehouseDialogVisible.value = true
  if (warehouseSelectDialogRef.value) {
    warehouseSelectDialogRef.value.handleQuery()
  }
}

/** 处理仓库选择结果 */
function handleWarehouseConfirm(selectedWarehouses) {
  if (selectedWarehouses.length > 0) {
    const warehouse = selectedWarehouses[0]
    if (isQueryWarehouseSelect.value) {
      queryParams.value.warehouseId = warehouse.warehouseId
    } else {
      form.value.warehouseId = warehouse.warehouseId
    }
    // 可以根据需要添加更多仓库信息字段
  }
}

getList()
</script>