feat: init pull version

Author: CrazyMrYan

.gitignore

@@ -42,6 +42,10 @@ build/Release
 node_modules/
 jspm_packages/
 
+# 
+public/files/*
+provisional/*
+
 # Snowpack dependency directory (https://snowpack.dev/)
 web_modules/
 

.user.ini

@@ -0,0 +1 @@
+open_basedir=/www/wwwroot/source.giao.club/:/tmp/

LICENSE

@@ -0,0 +1,178 @@
+// index.js
+const Koa = require('koa');
+const Router = require('koa-router');
+const { koaBody } = require('koa-body');
+const tinify = require('tinify');
+const path = require('path');
+const fs = require('fs');
+const mime = require('mime-types');
+const { checkAndCreateTable } = require('./utils/checkAndCreateTable');
+const pool = require('./utils/db');
+require('dotenv').config({ path: '.env.local' });
+
+const app = new Koa();
+const router = new Router();
+
+// TinyPNG API 密钥
+tinify.key = process.env.TINIFY_KEY;
+
+// Tinify 支持的文件格式
+const tinifySupportedMimeTypes = ['image/jpeg', 'image/png', 'image/webp'];
+
+// 所有图片格式
+const imageMimeTypes = [
+  'image/jpeg',
+  'image/png',
+  'image/webp',
+  'image/gif',
+  'image/bmp',
+  'image/tiff',
+  'image/x-icon',
+  'image/svg+xml'
+];
+
+// 设置静态文件服务目录
+app.use(require('koa-static')(path.join(__dirname, 'public')));
+
+// 创建必要的目录
+const createDirectories = () => {
+  const dirs = [
+    path.join(__dirname, 'provisional'),
+    path.join(__dirname, 'public', 'files')
+  ];
+  dirs.forEach((dir) => {
+    if (!fs.existsSync(dir)) {
+      fs.mkdirSync(dir, { recursive: true });
+    }
+  });
+};
+
+createDirectories();
+
+// 配置 koaBody 中间件处理文件上传
+app.use(
+  koaBody({
+    multipart: true,
+    formidable: {
+      uploadDir: path.join(__dirname, 'provisional'), // 临时上传目录
+      keepExtensions: true // 保留文件扩展名
+    }
+  })
+);
+
+// 文件上传和压缩的路由
+router.post('/upload', async (ctx) => {
+  const connection = await pool.getConnection();
+  try {
+    // 获取上传的文件
+    const files = ctx.request.files.file;
+    const fileList = Array.isArray(files) ? files : [files];
+    const responses = [];
+
+    // 检查是否需要压缩
+    const compress = ctx.query.compress !== 'false'; // 默认压缩
+    // 检查是否保留临时文件
+    const keepTemp = ctx.query.keepTemp === 'true'; // 默认不保留临时文件
+    // 获取请求的返回类型
+    const responseType = ctx.query.type;
+
+    for (const file of fileList) {
+      // 获取文件的 MIME 类型
+      const mimeType = mime.lookup(file.filepath);
+
+      // 生成输出文件路径
+      const outputFilePath = path.join(
+        __dirname,
+        'public',
+        'files',
+        path.basename(file.filepath)
+      );
+
+      if (compress && tinifySupportedMimeTypes.includes(mimeType)) {
+        // 如果文件类型受 Tinify 支持且要求压缩，使用 TinyPNG API 压缩文件
+        await tinify.fromFile(file.filepath).toFile(outputFilePath);
+      } else {
+        // 如果不支持压缩或者不要求压缩，保留临时文件则复制文件，否则移动文件
+        if (keepTemp) {
+          fs.copyFileSync(file.filepath, outputFilePath);
+        } else {
+          fs.renameSync(file.filepath, outputFilePath);
+        }
+      }
+
+      // 构建文件 URL
+      const fileUrl = `${process.env.PUBLIC_NETWORK_DOMAIN}/files/${path.basename(
+        outputFilePath
+      )}`;
+
+      // 插入文件信息到数据库
+      await connection.execute(
+        `INSERT INTO files (
+          filename, filesize, filelocation, created_by, is_public
+        ) VALUES (?, ?, ?, ?, ?)`,
+        [
+          path.basename(outputFilePath),
+          fs.statSync(outputFilePath).size,
+          fileUrl,
+          ctx.query.createdBy || 'anonymous',
+          ctx.query.isPublic === 'true'
+        ]
+      );
+
+      if (responseType === 'md' && imageMimeTypes.includes(mimeType)) {
+        // 如果文件是图片格式且请求类型是 'md'，返回 Markdown 格式
+        responses.push({
+          filepath: `![${path.basename(outputFilePath)}](${fileUrl})`
+        });
+      } else {
+        // 否则返回普通 URL
+        responses.push({ filepath: fileUrl });
+      }
+
+      if (!keepTemp && fs.existsSync(file.filepath)) {
+        // 删除临时上传文件
+        fs.unlinkSync(file.filepath);
+      }
+    }
+
+    // 返回响应
+    ctx.body = fileList.length > 1 ? responses : responses[0];
+  } catch (error) {
+    ctx.status = 500;
+    ctx.body = 'Error processing your request: ' + error.message;
+  } finally {
+    connection.release();
+  }
+});
+
+// 获取文件信息的路由
+router.get('/files', async (ctx) => {
+  const connection = await pool.getConnection();
+  try {
+    // 获取分页参数
+    const limit = parseInt(ctx.query.limit, 10) || 10; // 每页数量，默认为 10
+    const offset = parseInt(ctx.query.offset, 10) || 0; // 偏移量，默认为 0
+
+    // 执行查询
+    const [rows] = await connection.execute(
+      `SELECT * FROM files LIMIT ? OFFSET ?`,
+      [limit, offset]
+    );
+
+    ctx.body = rows;
+  } catch (error) {
+    ctx.status = 500;
+    ctx.body = 'Error retrieving files: ' + error.message;
+  } finally {
+    connection.release();
+  }
+});
+
+// 启动服务器
+app.use(router.routes()).use(router.allowedMethods());
+
+app.listen(process.env.SERVER_PORT, async () => {
+  console.log(`Server is running on ${process.env.INTERNAL_NETWORK_DOMAIN}`);
+  console.log(`Server is running on ${process.env.PUBLIC_NETWORK_DOMAIN}`);
+  await checkAndCreateTable();
+});

README.md

@@ -0,0 +1,27 @@
+{
+  "name": "upload-file",
+  "version": "1.0.0",
+  "main": "index.js",
+  "license": "MIT",
+  "engines": {
+    "node": ">=14.0.0"
+  },
+  "scripts": {
+    "dev": "pm2 start index.js --name upload-file-service --watch",
+    "stop": "pm2 delete upload-file-service"
+  },
+  "dependencies": {
+    "@koa/cors": "^5.0.0",
+    "dotenv": "^16.4.5",
+    "koa": "^2.15.3",
+    "koa-body": "^6.0.1",
+    "koa-router": "^12.0.1",
+    "koa-static": "^5.0.0",
+    "koa2-cors": "^2.0.6",
+    "mime-types": "^2.1.35",
+    "mysql2": "^3.10.1",
+    "pm2": "^5.4.0",
+    "tinify": "^1.7.1",
+    "uuid": "^10.0.0"
+  }
+}