【代码优化】AI：image 作品中心的代码

src/views/ai/image/index/components/ImageList.vue

@@ -1,6 +1,10 @@
 <template>
   <el-card class="dr-task" body-class="task-card" shadow="never">
-    <template #header>绘画任务</template>
+    <template #header>
+      绘画任务
+      <!-- TODO @fan：看看，怎么优化下这个样子哈。 -->
+      <el-button @click="handleViewPublic">绘画作品</el-button>
+    </template>
     <!-- 图片列表 -->
     <div class="task-image-list" ref="imageListRef">
       <ImageCard
@@ -42,6 +46,7 @@ import { AiImageStatusEnum } from '@/views/ai/utils/constants'
 import download from '@/utils/download'
 
 const message = useMessage() // 消息弹窗
+const router = useRouter() // 路由
 
 // 图片分页相关的参数
 const queryParams = reactive({
@@ -59,6 +64,13 @@ const inProgressTimer = ref<any>() // 生成中的 image 定时器，轮询生
 const isShowImageDetail = ref<boolean>(false) // 图片详情是否展示
 const showImageDetailId = ref<number>(0) // 图片详情的图片编号
 
+/** 处理查看绘图作品 */
+const handleViewPublic = () => {
+  router.push({
+    name: 'AiImageSquare'
+  })
+}
+
 /** 查看图片的详情  */
 const handleDetailOpen = async () => {
   isShowImageDetail.value = true

src/views/ai/image/square/index.vue

@@ -1,48 +1,67 @@
 <template>
   <div class="square-container">
+    <!-- TODO @fan：style 建议换成 unocss -->
+    <!-- TODO @fan：Search 可以换成 Icon 组件么？ -->
     <el-input
-      v-model="searchText"
+      v-model="queryParams.prompt"
       style="width: 100%; margin-bottom: 20px"
       size="large"
       placeholder="请输入要搜索的内容"
       :suffix-icon="Search"
-      @keyup.enter="handleSearch"
+      @keyup.enter="handleQuery"
     />
     <div class="gallery">
-      <div v-for="item in publicList" :key="item.id" class="gallery-item">
+      <!-- TODO @fan：这个图片的风格，要不和 ImageCard.vue 界面一致？（只有卡片，没有操作）；因为看着更有相框的感觉~~~ -->
+      <div v-for="item in list" :key="item.id" class="gallery-item">
         <img :src="item.picUrl" class="img" />
       </div>
     </div>
+    <!-- TODO @fan：缺少翻页 -->
+    <!-- 分页 -->
+    <Pagination
+      :total="total"
+      v-model:page="queryParams.pageNo"
+      v-model:limit="queryParams.pageSize"
+      @pagination="getList"
+    />
   </div>
 </template>
 <script setup lang="ts">
 import { ImageApi, ImageVO } from '@/api/ai/image'
 import { Search } from '@element-plus/icons-vue'
 
-/** 属性 */
-// TODO @fan：queryParams 里面搞分页哈。
-const pageNo = ref<number>(1)
-const pageSize = ref<number>(20)
-const publicList = ref<ImageVO[]>([])
-const searchText = ref<string>('')
+// TODO @fan：加个 loading 加载中的状态
+const loading = ref(true) // 列表的加载中
+const list = ref<ImageVO[]>([]) // 列表的数据
+const total = ref(0) // 列表的总页数
+const queryParams = reactive({
+  pageNo: 1,
+  pageSize: 10,
+  publicStatus: true,
+  prompt: undefined
+})
 
-/** 获取数据 */
-const getListData = async () => {
-  const res = await ImageApi.getImagePagePublic({
-    pageNo: pageNo.value,
-    pageSize: pageSize.value,
-    prompt: searchText.value
-  })
-  publicList.value = res.list as ImageVO[]
+/** 查询列表 */
+const getList = async () => {
+  loading.value = true
+  try {
+    const data = await ImageApi.getImagePageMy(queryParams)
+    list.value = data.list
+    total.value = data.total
+  } finally {
+    loading.value = false
+  }
 }
 
-/** 搜索 */
-const handleSearch = async () => {
-  await getListData()
+/** 搜索按钮操作 */
+const handleQuery = () => {
+  queryParams.pageNo = 1
+  getList()
 }
 
+/** 初始化 */
 onMounted(async () => {
-  await getListData()
+  await getList()
 })
 </script>
 <style scoped lang="scss">