ChatGPT/FastGPT
1
0
Fork 0
mirror of https://github.com/labring/FastGPT.git synced 2025-10-20 18:54:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

perf: upload img

Browse Source
This commit is contained in:
archer
2023-06-14 22:45:47 +08:00
parent 7c52cec0ea
commit cf3eb3b7b5
1 changed files with 17 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
client/src/pages/api/system/uploadImage.ts
View File

@@ -3,20 +3,20 @@ import { jsonRes } from '@/service/response';
import { connectToDatabase, Image } from '@/service/mongo'; import { connectToDatabase, Image } from '@/service/mongo';
import { authUser } from '@/service/utils/auth'; import { authUser } from '@/service/utils/auth';
// get the models available to the system type Props = { base64Img: string };
export default async function handler(req: NextApiRequest, res: NextApiResponse) { export default async function handler(req: NextApiRequest, res: NextApiResponse) {
try { try {
await connectToDatabase(); await connectToDatabase();
const { userId } = await authUser({ req, authToken: true }); const { userId } = await authUser({ req, authToken: true });
const { base64Img } = req.body; const { base64Img } = req.body as Props;
const base64Data = base64Img.split(',')[1];
const { _id } = await Image.create({ const data = await uploadImg({
userId, userId,
binary: Buffer.from(base64Data, 'base64') base64Img
}); });
jsonRes(res, { data: `/api/system/img/${_id}` }); jsonRes(res, { data });
} catch (error) { } catch (error) {
jsonRes(res, { jsonRes(res, {
code: 500, code: 500,
@@ -24,3 +24,14 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
}); });
} }
} }
export async function uploadImg({ base64Img, userId }: Props & { userId: string }) {
const base64Data = base64Img.split(',')[1];
const { _id } = await Image.create({
userId,
binary: Buffer.from(base64Data, 'base64')
});
return `/api/system/img/${_id}`;
}