上传

.gitignore

@@ -30,3 +30,4 @@ coverage
 
 # Environment variables files
 /service/.env
+/service/uploads

service/.env.example

@@ -46,5 +46,10 @@ HTTPS_PROXY=
 MJ_SERVER=
 
 # MJ_API_SECRET 
-MJ_API_SECRET=
+MJ_API_SECRET= 
 
+#API_UPLOADER 是否可以上传 1 可以其他都不可以
+API_UPLOADER=
+
+#HIDE_SERVER 隐藏服务端 1
+HIDE_SERVER=

service/package.json

@@ -35,6 +35,7 @@
     "http-proxy-middleware": "^2.0.6",
     "https-proxy-agent": "^5.0.1",
     "isomorphic-fetch": "^3.0.0",
+    "multer": "1.4.5-lts.1",
     "node-fetch": "^3.3.0",
     "socks-proxy-agent": "^7.0.0"
   },

service/src/index.ts

@@ -4,7 +4,10 @@ import type { ChatMessage } from './chatgpt'
 import { chatConfig, chatReplyProcess, currentModel } from './chatgpt'
 import { auth } from './middleware/auth'
 import { limiter } from './middleware/limiter'
-import { isNotEmptyString } from './utils/is'
+import { isNotEmptyString,formattedDate } from './utils/is'
+import multer from "multer"
+import path from "path"
+import fs from "fs" 
 // const { createProxyMiddleware } = require('http-proxy-middleware');
 //import {createProxyMiddleware} from "http-proxy-middleware"
 import  proxy from "express-http-proxy"
@@ -18,6 +21,8 @@ app.use(express.static('public'))
 //app.use(express.json())
 app.use(bodyParser.json({ limit: '10mb' })); //大文件传输
 
+
+
 app.all('*', (_, res, next) => {
   res.header('Access-Control-Allow-Origin', '*')
   res.header('Access-Control-Allow-Headers', 'authorization, Content-Type')
@@ -65,7 +70,9 @@ router.post('/session', async (req, res) => {
   try {
     const AUTH_SECRET_KEY = process.env.AUTH_SECRET_KEY
     const hasAuth = isNotEmptyString(AUTH_SECRET_KEY)
-    res.send({ status: 'Success', message: '', data: { auth: hasAuth, model: currentModel() } })
+    const isUpload= isNotEmptyString(  process.env.API_UPLOADER )
+    const isHideServer= isNotEmptyString(  process.env.HIDE_SERVER )
+    res.send({ status: 'Success', message: '', data: {isHideServer,isUpload, auth: hasAuth, model: currentModel() } })
   }
   catch (error) {
     res.send({ status: 'Fail', message: error.message, data: null })
@@ -106,6 +113,48 @@ app.use('/mjapi', proxy(process.env.MJ_SERVER?process.env.MJ_SERVER:'https://api
 
 }));
 
+
+
+// 设置存储引擎和文件保存路径
+const storage = multer.diskStorage({
+  destination: function (req, file, cb) {
+    let uploadFolderPath=`./uploads/${formattedDate()}/`;//`
+
+    console.log('dir', __dirname   ) ;
+
+    if(!fs.existsSync('./uploads/')) {
+      fs.mkdirSync('./uploads/');
+    }
+    if(!fs.existsSync(uploadFolderPath)) {
+      fs.mkdirSync(uploadFolderPath);
+    }
+    cb(null, `uploads/${formattedDate()}/`); 
+  },
+  filename: function (req, file, cb) {
+    let filename=  Date.now() + path.extname(file.originalname);
+    console.log( 'file',  filename );
+    cb(null, filename);
+  }
+});
+const upload = multer({ storage: storage });
+// 处理文件上传的路由
+const isUpload= isNotEmptyString(  process.env.API_UPLOADER )
+if(isUpload){
+  app.use('/openapi/v1/upload', upload.single('file'), (req, res) => {
+    //res.send('文件上传成功！');
+    res.setHeader('Content-type', 'application/json' ); 
+    if(req.file.filename) res.json({ url:`/uploads/${formattedDate()}/${ req.file.filename  }`,created:Date.now() })
+    else res.json({ error:`uploader fail`,created:Date.now() })
+  });
+}else {
+  app.use('/openapi/v1/upload',  (req, res) => {
+    //res.send('文件上传成功！');
+     res.json({ error:`server is no open uploader `,created:Date.now() })
+  });
+}
+app.use('/uploads', express.static('uploads'));
+
+
 //代理openai 接口
 app.use('/openapi', proxy(API_BASE_URL, {
   https: false, limit: '10mb',
@@ -117,8 +166,7 @@ app.use('/openapi', proxy(API_BASE_URL, {
     proxyReqOpts.headers['Content-Type'] = 'application/json';
     return proxyReqOpts;
   },
-  //limit: '10mb'
-
+  //limit: '10mb' 
 }));
 
 

service/src/utils/is.ts

@@ -17,3 +17,12 @@ export function isBoolean<T extends boolean>(value: T | unknown): value is boole
 export function isFunction<T extends (...args: any[]) => any | void | never>(value: T | unknown): value is T {
   return Object.prototype.toString.call(value) === '[object Function]'
 }
+
+export const formattedDate=()=>{
+  const currentDate = new Date(); 
+  const year = currentDate.getFullYear();
+  const month = (currentDate.getMonth() + 1).toString().padStart(2, '0');
+  const day = currentDate.getDate().toString().padStart(2, '0'); 
+  return  `${year}${month}${day}`;
+
+}