<template>
  <div class="app-container">
    <el-form :model="queryParams" ref="queryRef" :inline="true" v-show="showSearch" label-width="68px">
      <el-form-item label="状态" prop="status">
        <el-select v-model="queryParams.status" placeholder="请选择状态" clearable>
          <el-option
            v-for="dict in sys_normal_disable"
            :key="dict.value"
            :label="dict.label"
            :value="dict.value"
          />
        </el-select>
      </el-form-item>
      <el-form-item label="用户ID" prop="userId">
        <el-input
          v-model="queryParams.userId"
          placeholder="请输入用户ID"
          clearable
          @keyup.enter="handleQuery"
        />
      </el-form-item>
      <el-form-item label="Key" prop="secretKey">
        <el-input
          v-model="queryParams.secretKey"
          placeholder="请输入Key"
          clearable
          @keyup.enter="handleQuery"
        />
      </el-form-item>
      <el-form-item label="模型" prop="model">
        <el-input
          v-model="queryParams.model"
          placeholder="请输入模型"
          clearable
          @keyup.enter="handleQuery"
        />
      </el-form-item>
      <el-form-item label="消息ID" prop="msgId">
        <el-input
          v-model="queryParams.msgId"
          placeholder="请输入消息ID"
          clearable
          @keyup.enter="handleQuery"
        />
      </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="['cert:usageRecord:add']"
        >新增</el-button>
      </el-col>
      <el-col :span="1.5">
        <el-button
          type="success"
          plain
          icon="Edit"
          :disabled="single"
          @click="handleUpdate"
          v-hasPermi="['cert:usageRecord:edit']"
        >修改</el-button>
      </el-col>
      <el-col :span="1.5">
        <el-button
          type="danger"
          plain
          icon="Delete"
          :disabled="multiple"
          @click="handleDelete"
          v-hasPermi="['cert:usageRecord:remove']"
        >删除</el-button>
      </el-col>
      <el-col :span="1.5">
        <el-button
          type="warning"
          plain
          icon="Download"
          @click="handleExport"
          v-hasPermi="['cert:usageRecord:export']"
        >导出</el-button>
      </el-col>
      <right-toolbar v-model:showSearch="showSearch" @queryTable="getList"></right-toolbar>
    </el-row>

    <el-table v-loading="loading" :data="usageRecordList" @selection-change="handleSelectionChange">
      <el-table-column type="selection" width="55" align="center" />
      <el-table-column label="时间" align="center" prop="createTime" />
      <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="用户ID" align="center" prop="userId" />
      <el-table-column label="Key" align="center" prop="secretKey" />
      <el-table-column label="模型" align="center" prop="model" />
      <el-table-column label="prompt" align="center" prop="promptTokens" />
      <el-table-column label="completion" align="center" prop="completionTokens" />
      <el-table-column label="total" align="center" prop="totalTokens" />
      <el-table-column label="消息ID" align="center" prop="msgId" />
      <el-table-column label="费用" align="center" prop="fee" />
      <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="['cert:usageRecord:edit']">修改</el-button>
          <el-button link type="primary" icon="Delete" @click="handleDelete(scope.row)" v-hasPermi="['cert:usageRecord: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-dialog :title="title" v-model="open" width="500px" append-to-body>
      <el-form ref="usageRecordRef" :model="form" :rules="rules" label-width="80px">
        <el-form-item label="状态" prop="status">
          <el-radio-group v-model="form.status">
            <el-radio
              v-for="dict in sys_normal_disable"
              :key="dict.value"
              :label="dict.value"
            >{{dict.label}}</el-radio>
          </el-radio-group>
        </el-form-item>
        <el-form-item label="用户ID" prop="userId">
          <el-input v-model="form.userId" placeholder="请输入用户ID" />
        </el-form-item>
        <el-form-item label="Key" prop="secretKey">
          <el-input v-model="form.secretKey" placeholder="请输入Key" />
        </el-form-item>
        <el-form-item label="模型" prop="model">
          <el-input v-model="form.model" placeholder="请输入模型" />
        </el-form-item>
        <el-form-item label="问题tokens" prop="promptTokens">
          <el-input v-model="form.promptTokens" placeholder="请输入问题tokens" />
        </el-form-item>
        <el-form-item label="回复tokens" prop="completionTokens">
          <el-input v-model="form.completionTokens" placeholder="请输入回复tokens" />
        </el-form-item>
        <el-form-item label="总tokens" prop="totalTokens">
          <el-input v-model="form.totalTokens" placeholder="请输入总tokens" />
        </el-form-item>
        <el-form-item label="消息ID" prop="msgId">
          <el-input v-model="form.msgId" placeholder="请输入消息ID" />
        </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 name="UsageRecord">
import { listUsageRecord, getUsageRecord, delUsageRecord, addUsageRecord, updateUsageRecord } from "@/api/cert/usageRecord";

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

const usageRecordList = 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 data = reactive({
  form: {},
  queryParams: {
    pageNum: 1,
    pageSize: 10,
    status: null,
    userId: null,
    secretKey: null,
    model: null,
    msgId: null
  },
  rules: {
    status: [
      { required: true, message: "状态不能为空", trigger: "change" }
    ],
    createBy: [
      { required: true, message: "不能为空", trigger: "blur" }
    ],
    updateBy: [
      { required: true, message: "不能为空", trigger: "blur" }
    ],
    userId: [
      { required: true, message: "用户ID不能为空", trigger: "blur" }
    ],
    secretKey: [
      { required: true, message: "Key不能为空", trigger: "blur" }
    ],
    model: [
      { required: true, message: "模型不能为空", trigger: "blur" }
    ],
    promptTokens: [
      { required: true, message: "问题tokens不能为空", trigger: "blur" }
    ],
    completionTokens: [
      { required: true, message: "回复tokens不能为空", trigger: "blur" }
    ],
    totalTokens: [
      { required: true, message: "总tokens不能为空", trigger: "blur" }
    ],
    msgId: [
      { required: true, message: "消息ID不能为空", trigger: "blur" }
    ]
  }
});

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

/** 查询使用记录列表 */
function getList() {
  loading.value = true;
  listUsageRecord(queryParams.value).then(response => {
    usageRecordList.value = response.rows;
    total.value = response.total;
    loading.value = false;
  });
}

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

// 表单重置
function reset() {
  form.value = {
    id: null,
    status: null,
    delFlag: null,
    createBy: null,
    createTime: null,
    updateBy: null,
    updateTime: null,
    userId: null,
    secretKey: null,
    model: null,
    promptTokens: null,
    completionTokens: null,
    totalTokens: null,
    msgId: null
  };
  proxy.resetForm("usageRecordRef");
}

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

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

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

/** 新增按钮操作 */
function handleAdd() {
  reset();
  open.value = true;
  title.value = "添加使用记录";
}

/** 修改按钮操作 */
function handleUpdate(row) {
  reset();
  const _id = row.id || ids.value
  getUsageRecord(_id).then(response => {
    form.value = response.data;
    open.value = true;
    title.value = "修改使用记录";
  });
}

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

/** 删除按钮操作 */
function handleDelete(row) {
  const _ids = row.id || ids.value;
  proxy.$modal.confirm('是否确认删除使用记录编号为"' + _ids + '"的数据项？').then(function() {
    return delUsageRecord(_ids);
  }).then(() => {
    getList();
    proxy.$modal.msgSuccess("删除成功");
  }).catch(() => {});
}

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

getList();
</script>
