xinyin025/ChatGPT-Next-Web
1
0
Fork 0
mirror of https://github.com/ChatGPTNextWeb/ChatGPT-Next-Web.git synced 2025-11-13 12:43:42 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

feat: support other type file upload

Browse Source
This commit is contained in:
Hk-Gosuto
2024-03-31 23:07:17 +08:00
parent aea5bedb68
commit 80a077a3db
9 changed files with 223 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
app/api/file/upload/route.ts
View File

@@ -4,6 +4,7 @@ import { auth } from "@/app/api/auth";
import LocalFileStorage from "@/app/utils/local_file_storage";
import { getServerSideConfig } from "@/app/config/server";
import S3FileStorage from "@/app/utils/s3_file_storage";
import path from "path";
async function handle(req: NextRequest) {
if (req.method === "OPTIONS") {
@@ -19,20 +20,21 @@ async function handle(req: NextRequest) {
try {
const formData = await req.formData();
const image = formData.get("file") as File;
const file = formData.get("file") as File;
const originalFileName = file?.name;
const imageReader = image.stream().getReader();
const imageData: number[] = [];
const fileReader = file.stream().getReader();
const fileData: number[] = [];
while (true) {
const { done, value } = await imageReader.read();
const { done, value } = await fileReader.read();
if (done) break;
imageData.push(...value);
fileData.push(...value);
}
const buffer = Buffer.from(imageData);
var fileName = `${Date.now()}.png`;
const buffer = Buffer.from(fileData);
const fileType = path.extname(originalFileName).slice(1);
var fileName = `${Date.now()}.${fileType}`;
var filePath = "";
const serverConfig = getServerSideConfig();
if (serverConfig.isStoreFileToLocal) {