参数补全

server/api/drawing.ts

@@ -22,9 +22,9 @@ export default defineEventHandler(async event => {
 
     // 修改任务状态(仅限bot访问)
     if (event.node.req.method === 'PATCH') {
-        let { ids } = await readBody(event)
+        const { ids } = await readBody(event)
         // 取出ids对应的任务
-        let tasks = await Promise.all(ids.map(async (id: any) => await useStorage().getItem(`task:${id}`)))
+        const tasks = await Promise.all(ids.map(async (id: any) => await useStorage().getItem(`task:${id}`)))
         // 修改任务状态
         tasks.forEach(async (task: any) => {
             task.status = 'finished'
@@ -35,17 +35,32 @@ export default defineEventHandler(async event => {
 
     // 添加任务到队列(生成一个32位随机字符串作为任务ID)
     if (event.node.req.method === 'POST') {
-        event.context.body = await readBody(event) // 读取body数据到
-        event.context.body.tid = Math.random().toString(36).substring(2, 34)
-        event.context.body.uid = 1234567890                 // 默认用户ID
-        event.context.body.status = 'waiting'               // 默认当前状态
-        event.context.body.createdAt = new Date().getTime() // 设定创建时间
-        await useStorage().setItem(`task:${event.context.body.tid}`, event.context.body)
+        const body = await readBody(event)                     // 读取body数据到
+        body.tid = Math.random().toString(36).substring(2, 34) // 生成随机任务ID
+        body.uid = 1234567890                                  // 用户ID
+        body.status = 'waiting'                                // 任务状态
+        body.createdAt = new Date().getTime()                  // 设定创建时间
+        // 验证需要输入的参数
+        body.model =  'SD2'                                    // 模型
+        body.ckpt = 'latest'                                   // 风格
+        body.prompt = '猫猫'                                    // 提示词
+        body.remove = '狗狗'                                    // 排除词
+        body.number = 1                                        // 生成数量
+        body.w = 512                                           // 图片宽度
+        body.h = 512                                           // 图片高度
+        body.seed = 0                                          // 随机种子
+        body.sampler = 'pndm'                                  // 扩散采样器
+        body.prompt_guidance = 0.5                             // 提示词权重
+        body.quality_details = 25                              // 质量和细节(步数)
+        body.image = ''                                        // 图片数据
+        body.data = null                                       // 生成结果
+        console.log('task:', body)
+        await useStorage().setItem(`task:${body.tid}`, body)   // 保存任务到内存
         // 筛选出所有的任务(并打印)
         let tasks = await useStorage().getKeys().then((keys: any[]) => keys.filter(key => key.startsWith('task:')))
-        console.log(tasks)
+        console.log('tasks:', tasks)
         // 如果无需排队，直接返回结果, 否则返回任务排队状态(预估等待时间)
-        return event.context.body
+        return body
     }
 
 })