feat: admin set env

admin/service/route/system.js

@@ -1,4 +1,5 @@
 import jwt from 'jsonwebtoken';
+import { System } from '../schema.js';
 
 const adminAuth = {
   username: process.env.ADMIN_USER,
@@ -6,6 +7,14 @@ const adminAuth = {
 };
 const authSecret = process.env.ADMIN_SECRET;
 
+const postParent = () => {
+  fetch(`${process.env.PARENT_URL}/api/system/updateEnv`, {
+    headers: {
+      rootkey: process.env.PARENT_ROOT_KEY
+    }
+  });
+};
+
 export const useSystemRoute = (app) => {
   app.post('/api/login', (req, res) => {
     if (!adminAuth.username || !adminAuth.password) {
@@ -37,6 +46,63 @@ export const useSystemRoute = (app) => {
       res.status(401).end('username or password incorrect');
     }
   });
+  app.get('/system', auth(), async (req, res) => {
+    try {
+      const data = await System.find();
+      const totalCount = await System.countDocuments();
+
+      res.header('Access-Control-Expose-Headers', 'X-Total-Count');
+      res.header('X-Total-Count', totalCount);
+      res.json(
+        data.map((item) => {
+          const obj = item.toObject();
+          return {
+            ...obj,
+            id: obj._id
+          };
+        })
+      );
+    } catch (error) {
+      console.log(error);
+
+      res.status(500).json({ error: 'Error creating system env' });
+    }
+  });
+  app.post('/system', auth(), async (req, res) => {
+    try {
+      await System.create({
+        ...req.body,
+        sensitiveCheck: req.body.sensitiveCheck === 'true'
+      });
+      postParent();
+      res.json({});
+    } catch (error) {
+      res.status(500).json({ error: 'Error creating system env' });
+    }
+  });
+  app.put('/system/:id', auth(), async (req, res) => {
+    try {
+      const _id = req.params.id;
+      await System.findByIdAndUpdate(_id, {
+        ...req.body,
+        sensitiveCheck: req.body.sensitiveCheck === 'true'
+      });
+      postParent();
+      res.json({});
+    } catch (error) {
+      res.status(500).json({ error: 'Error updating system env' });
+    }
+  });
+  app.delete('/system/:id', auth(), async (req, res) => {
+    try {
+      const _id = req.params.id;
+      await System.findByIdAndDelete(_id);
+
+      res.json({});
+    } catch (error) {
+      res.status(500).json({ error: 'Error updating system env' });
+    }
+  });
 };
 
 export const auth = () => {

admin/service/schema.js

@@ -84,7 +84,35 @@ const modelSchema = new mongoose.Schema({
   updateTime: Date
 });
 
-export const Model = mongoose.model('Model', modelSchema);
-export const Kb = mongoose.model('Kb', kbSchema);
-export const User = mongoose.model('User', userSchema, 'users');
-export const Pay = mongoose.model('Pay', paySchema, 'pays');
+const SystemSchema = new mongoose.Schema({
+  openAIKeys: {
+    type: String,
+    default: ''
+  },
+  openAITrainingKeys: {
+    type: String,
+    default: ''
+  },
+  gpt4Key: {
+    type: String,
+    default: ''
+  },
+  vectorMaxProcess: {
+    type: Number,
+    default: 10
+  },
+  qaMaxProcess: {
+    type: Number,
+    default: 10
+  },
+  sensitiveCheck: {
+    type: Boolean,
+    default: false
+  }
+});
+
+export const Model = mongoose.models['model'] || mongoose.model('model', modelSchema);
+export const Kb = mongoose.models['kb'] || mongoose.model('kb', kbSchema);
+export const User = mongoose.models['user'] || mongoose.model('user', userSchema);
+export const Pay = mongoose.models['pay'] || mongoose.model('pay', paySchema);
+export const System = mongoose.models['system'] || mongoose.model('system', SystemSchema);