export default defineEventHandler(async event => {

    // 获取任务列表的状态进度(普通用户只能看到自己的)
    if (event.node.req.method === 'GET') {
        let {status, model, uid, bot} = getQuery(event)
        let tasks = await useStorage().getKeys().then((keys: any[]) => keys.filter(key => key.startsWith('task:')))
        tasks = await Promise.all(tasks.map(async (key: any) => await useStorage().getItem(key))) // 读取所有任务的数据
        tasks.sort((a: any, b: any) => a.createdAt - b.createdAt)                    // 按照创建时间排序
        if (tasks.length === 0) return []                                            // 如果没有任务，直接返回空数组
        if (status) tasks = tasks.filter((task: any) => task.status === status)      // 筛选出指定状态的任务
        if (model) tasks = tasks.filter((task: any) => task.model === model)         // 筛选出指定model的任务
        if (uid) tasks = tasks.filter((task: any) => task.uid === uid)               // 筛选出指定用户的任务
        if (bot) {
            tasks = tasks.filter((task: any) => task.model === tasks[0].model)  // 筛选出首位model的任务
            tasks.forEach(async (task: any) => {
                task.status = 'processing'
                await useStorage().setItem(`task:${task.id}`, task)
            })
        }
        return tasks
    }

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

})