<template>
  <ContentWrap>
    <!-- 搜索工作栏 -->
    <el-form
      class="-mb-15px"
      :model="queryParams"
      ref="queryFormRef"
      :inline="true"
      label-width="68px"
    >
      <el-form-item label="名称" prop="name">
        <el-input
          v-model="queryParams.name"
          placeholder="请输入名称"
          clearable
          @keyup.enter="handleQuery"
          class="!w-240px"
        />
      </el-form-item>
      
      <el-form-item label="创建时间" prop="createTime">
        <el-date-picker
          v-model="queryParams.createTime"
          value-format="YYYY-MM-DD HH:mm:ss"
          type="daterange"
          start-placeholder="开始日期"
          end-placeholder="结束日期"
          :default-time="[new Date('1 00:00:00'), new Date('1 23:59:59')]"
          class="!w-240px"
        />
      </el-form-item>
      <el-form-item>
        <el-button @click="handleQuery"><Icon icon="ep:search" class="mr-5px" /> 搜索</el-button>
        <el-button @click="resetQuery"><Icon icon="ep:refresh" class="mr-5px" /> 重置</el-button>
        <el-button
          type="primary"
          plain
          @click="openForm('create')"
          v-hasPermi="['annotation:dataset:create']"
        >
          <Icon icon="ep:plus" class="mr-5px" /> 新增
        </el-button>
        <el-button
          type="success"
          plain
          @click="handleExport"
          :loading="exportLoading"
          v-hasPermi="['annotation:dataset:export']"
        >
          <Icon icon="ep:download" class="mr-5px" /> 导出
        </el-button>
      </el-form-item>
    </el-form>
  </ContentWrap>

  <!-- 列表 -->
  <ContentWrap>
    <el-table v-loading="loading" :data="list" :stripe="true" :show-overflow-tooltip="true">
      <el-table-column label="id" align="center" prop="id" />
      <el-table-column label="名称" align="center" prop="name" />
      <el-table-column label="路径" align="center" prop="path" />
      <el-table-column
        label="创建时间"
        align="center"
        prop="createTime"
        :formatter="dateFormatter"
        width="180px"
      />
      <el-table-column label="操作" align="center">
        <template #default="scope">
          <el-button
            link
            type="primary"
            @click="openForm('update', scope.row.id)"
            v-hasPermi="['annotation:dataset:update']"
          >
            编辑
          </el-button>
          <el-button
            link
            type="success"
            @click="handleAddImages(scope.row.id)"
            v-hasPermi="['annotation:dataset:update']"
          >
            添加图片
          </el-button>
          
          <el-button
            link
            type="danger"
            @click="handleDelete(scope.row.id)"
            v-hasPermi="['annotation:dataset:delete']"
          >
            删除
          </el-button>
        </template>
      </el-table-column>
    </el-table>
    <!-- 分页 -->
    <Pagination
      :total="total"
      v-model:page="queryParams.pageNo"
      v-model:limit="queryParams.pageSize"
      @pagination="getList"
    />
  </ContentWrap>

  <!-- 表单弹窗：添加/修改 -->
  <DatasetForm ref="formRef" @success="getList" />

  <!-- 添加图片弹窗 -->
  <el-dialog v-model="addImagesDialogVisible" title="添加图片" width="500px">
    <el-upload
      v-model:file-list="addImagesFileList"
      :auto-upload="false"
      accept=".zip,image/*"
      drag
      multiple
    >
      <Icon icon="ep:upload" class="el-icon--upload" />
      <div class="el-upload__text">
        拖拽文件到此处或 <em>点击上传</em>
      </div>
      <template #tip>
        <div class="el-upload__tip">
          支持上传 zip 压缩包或多张图片
        </div>
      </template>
    </el-upload>
    <template #footer>
      <el-button @click="addImagesDialogVisible = false">取 消</el-button>
      <el-button type="primary" @click="submitAddImages" :loading="addImagesLoading">
        确 定
      </el-button>
    </template>
  </el-dialog>
</template>

<script setup lang="ts">
import { dateFormatter } from '@/utils/formatTime'
import download from '@/utils/download'
import { DatasetApi, DatasetVO } from '@/api/annotation/dataset'
import DatasetForm from './DatasetForm.vue'
import { UploadUserFile } from 'element-plus'

/** 识别数据集 列表 */
defineOptions({ name: 'Dataset' })

const message = useMessage() // 消息弹窗
const { t } = useI18n() // 国际化

const loading = ref(true) // 列表的加载中
const list = ref<DatasetVO[]>([]) // 列表的数据
const total = ref(0) // 列表的总页数
const queryParams = reactive({
  pageNo: 1,
  pageSize: 10,
  name: undefined as string | undefined,
  path: undefined as string | undefined,
  createTime: [] as string[],
})
const queryFormRef = ref() // 搜索的表单
const exportLoading = ref(false) // 导出的加载中

// 添加图片相关
const addImagesDialogVisible = ref(false)
const addImagesFileList = ref<UploadUserFile[]>([])
const addImagesLoading = ref(false)
const currentDatasetId = ref<number>()

/** 查询列表 */
const getList = async () => {
  loading.value = true
  try {
    const data = await DatasetApi.getDatasetPage(queryParams)
    list.value = data.list
    total.value = data.total
  } finally {
    loading.value = false
  }
}

/** 搜索按钮操作 */
const handleQuery = () => {
  queryParams.pageNo = 1
  getList()
}

/** 重置按钮操作 */
const resetQuery = () => {
  queryFormRef.value.resetFields()
  handleQuery()
}

/** 添加/修改操作 */
const formRef = ref()
const openForm = (type: string, id?: number) => {
  formRef.value.open(type, id)
}

/** 添加图片操作 */
const handleAddImages = (id: number) => {
  currentDatasetId.value = id
  addImagesFileList.value = []
  addImagesDialogVisible.value = true
}

/** 刷新按钮操作 */
const handleRefresh = async (row: any) => {
  try {
    const data = await DatasetApi.getDataset(row.id)
    // 更新列表中对应行的数据
    const index = list.value.findIndex((item) => item.id === row.id)
    if (index !== -1) {
      list.value[index] = data
    }
    message.success('刷新成功')
  } catch {
    message.error('刷新失败')
  }
}

/** 提交添加图片 */
const submitAddImages = async () => {
  if (addImagesFileList.value.length === 0) {
    message.error('请选择要上传的文件')
    return
  }
  if (!currentDatasetId.value) {
    return
  }

  addImagesLoading.value = true
  try {
    const files = addImagesFileList.value.map((file: any) => file.raw)
    await DatasetApi.addImages({ id: currentDatasetId.value, files })
    message.success('添加成功')
    addImagesDialogVisible.value = false
    // 刷新列表
    await getList()
  } finally {
    addImagesLoading.value = false
  }
}

/** 删除按钮操作 */
const handleDelete = async (id: number) => {
  try {
    // 删除的二次确认
    await message.delConfirm()
    // 发起删除
    await DatasetApi.deleteDataset(id)
    message.success(t('common.delSuccess'))
    // 刷新列表
    await getList()
  } catch {}
}

/** 导出按钮操作 */
const handleExport = async () => {
  try {
    // 导出的二次确认
    await message.exportConfirm()
    // 发起导出
    exportLoading.value = true
    const data = await DatasetApi.exportDataset(queryParams)
    download.excel(data, '识别数据集.xls')
  } catch {
  } finally {
    exportLoading.value = false
  }
}

/** 初始化 **/
onMounted(() => {
  getList()
})
</script>