feat: 添加头文件信息，添加图床工具。

server/api/tools/imgBed/download/[hash].ts

@@ -0,0 +1,39 @@
+// @ts-ignore
+import DB from 'better-sqlite3';
+import { defineEventHandler, getRouterParam } from 'h3';
+import fs from 'fs';
+import { sendStream } from 'h3';
+
+export default defineEventHandler(async (event) => {
+    try {
+        const hash = getRouterParam(event, 'hash');
+        if (!hash) {
+            return { code: 400, msg: 'Invalid hash parameter' };
+        }
+        const db = new DB('./userData/db/imgBed.db');
+        // 从数据库中查询文件信息
+        const fileInfo = db.prepare(`
+            SELECT name, expire FROM imgBed WHERE hash =?
+        `).get(hash) as { name: string, expire: number };
+        db.close();
+
+        if (!fileInfo) {
+            return { code: 404, msg: 'File not found' };
+        }
+
+        const fileName = fileInfo.name;
+        const filePath = `./userData/imgBed/${hash}`;
+        // 检查文件是否存在
+        if (!fs.existsSync(filePath)) {
+            return { code: 404, msg: 'File not found on disk' };
+        }
+        const fileStream = fs.createReadStream(filePath);
+        
+        setResponseHeader(event, 'Content-Disposition', `attachment; filename="${fileName}"`);
+        // 发送文件流
+        return sendStream(event, fileStream);
+    } catch (error) {
+        console.error(error);
+        return { code: 500, msg: 'Internal Server Error' };
+    }
+});

server/api/tools/imgBed/upload.ts

@@ -0,0 +1,39 @@
+// @ts-ignore
+import DB from 'better-sqlite3';
+import { defineEventHandler, readBody } from 'h3';
+import crypto from 'crypto';
+import fs from 'fs';
+
+export default defineEventHandler(async (event) => {
+    try {
+        const data = await readFormData(event)
+        const file = data.get('file') as File
+        const expire = data.get('expire') as string
+        console.log({
+            file: file.name,
+            expire,
+        })
+        const fileBuffer = Buffer.from(await file.arrayBuffer());
+        // 计算文件的 MD5 哈希值
+        const hash = crypto.createHash('md5').update(fileBuffer).digest('hex');
+        const fileName = file.name;
+        const filePath = `./userData/imgBed/${hash}`;
+        // 将文件写入磁盘
+        await fs.promises.writeFile(filePath, fileBuffer);
+        // 获取文件大小
+        const fileSize = fileBuffer.length;
+
+        const db = new DB('./userData/db/imgBed.db');
+        // 插入数据到数据库
+        db.prepare(`
+            INSERT INTO imgBed (name, hash, size, expire)
+            VALUES (?,?,?,?)
+        `).run([fileName, hash, fileSize, expire]);
+        db.close();
+
+        return { code: 200, hash, fileName, fileSize, expire: Number(expire) + Date.now() };
+    } catch (error) {
+        console.error(error);
+        return { code: 500, msg: 'Internal Server Error' };
+    }
+});