img server

This commit is contained in:
2023-02-19 03:54:55 +08:00
parent 830710ae1d
commit 44afb1c797
3 changed files with 21 additions and 3 deletions

14
server/api/img/[id].ts Normal file
View File

@@ -0,0 +1,14 @@
import fs from 'fs'
export default defineEventHandler(async event => {
// 获取图片
if (event.node.req.method === 'GET') {
// 判断当前执行目录是否存在 outputs 文件夹, 否则在上一级文件夹
let path = `outputs/outputs/txt2img-samples/samples/${event.context.params.id}`
if (!fs.existsSync('outputs')) {
path = `../outputs/outputs/txt2img-samples/samples/${event.context.params.id}`
}
return await useStorage().getItem(`img:${event.context.params.id}`)
}
})