添加系统更新操作和设置删除日志频率

back/services/schedule.ts

@@ -0,0 +1,61 @@
+import { Service, Inject } from 'typedi';
+import winston from 'winston';
+import nodeSchedule from 'node-schedule';
+import { Crontab } from '../data/cron';
+import { exec } from 'child_process';
+
+@Service()
+export default class ScheduleService {
+  private scheduleStacks = new Map<string, nodeSchedule.Job>();
+
+  constructor(@Inject('logger') private logger: winston.Logger) {}
+
+  async generateSchedule({ _id = '', command, name, schedule }: Crontab) {
+    this.logger.info(
+      '[创建定时任务]，任务ID: %s，cron: %s，任务名: %s，任务方法: %s',
+      _id,
+      schedule,
+      name,
+    );
+
+    this.scheduleStacks.set(
+      _id,
+      nodeSchedule.scheduleJob(_id, schedule, async () => {
+        try {
+          exec(command, async (error, stdout, stderr) => {
+            if (error) {
+              await this.logger.info(
+                '执行任务`%s`失败，时间：%s, 错误信息：%j',
+                name,
+                new Date().toLocaleString(),
+                error,
+              );
+            }
+
+            if (stderr) {
+              await this.logger.info(
+                '执行任务`%s`失败，时间：%s, 错误信息：%j',
+                name,
+                new Date().toLocaleString(),
+                stderr,
+              );
+            }
+          });
+        } catch (error) {
+          await this.logger.info(
+            '执行任务`%s`失败，时间：%s, 错误信息：%j',
+            name,
+            new Date().toLocaleString(),
+            error,
+          );
+        } finally {
+        }
+      }),
+    );
+  }
+
+  async cancelSchedule(id: string, jobName: string) {
+    this.logger.info('[取消定时任务]，任务名：%s', jobName);
+    this.scheduleStacks.has(id) && this.scheduleStacks.get(id)?.cancel();
+  }
+}

back/services/sock.ts

@@ -0,0 +1,33 @@
+import { Service, Inject } from 'typedi';
+import winston from 'winston';
+import { Connection } from 'sockjs';
+
+@Service()
+export default class SockService {
+  private clients: Connection[] = [];
+
+  constructor(@Inject('logger') private logger: winston.Logger) {}
+
+  public getClients() {
+    return this.clients;
+  }
+
+  public addClient(conn: Connection) {
+    if (this.clients.indexOf(conn) === -1) {
+      this.clients.push(conn);
+    }
+  }
+
+  public removeClient(conn: Connection) {
+    const index = this.clients.indexOf(conn);
+    if (index !== -1) {
+      this.clients.splice(index, 1);
+    }
+  }
+
+  public sendMessage(msg: string) {
+    this.clients.forEach((x) => {
+      x.write(msg);
+    });
+  }
+}

back/services/user.ts

@@ -11,6 +11,10 @@ import { AuthDataType, AuthInfo, LoginStatus } from '../data/auth';
 import { NotificationInfo } from '../data/notify';
 import NotificationService from './notify';
 import { Request } from 'express';
+import ScheduleService from './schedule';
+import { spawn } from 'child_process';
+import SockService from './sock';
+import got from 'got';
 
 @Service()
 export default class UserService {
@@ -18,7 +22,11 @@ export default class UserService {
   private notificationService!: NotificationService;
   private authDb = new DataStore({ filename: config.authDbFile });
 
-  constructor(@Inject('logger') private logger: winston.Logger) {
+  constructor(
+    @Inject('logger') private logger: winston.Logger,
+    private scheduleService: ScheduleService,
+    private sockService: SockService,
+  ) {
     this.authDb.loadDatabase((err) => {
       if (err) throw err;
     });
@@ -330,7 +338,7 @@ export default class UserService {
           if (err) {
             resolve({} as NotificationInfo);
           } else {
-            resolve(doc.info);
+            resolve({ ...doc.info, _id: doc._id });
           }
         },
       );
@@ -354,4 +362,62 @@ export default class UserService {
       return { code: 400, data: '通知发送失败，请检查参数' };
     }
   }
+
+  public async updateLogRemoveFrequency(frequency: number) {
+    const result = await this.updateNotificationDb({
+      type: AuthDataType.removeLogFrequency,
+      info: { frequency },
+    });
+    const cron = {
+      _id: result._id,
+      name: '删除日志',
+      command: `ql rmlog ${frequency}`,
+      schedule: `5 23 */${frequency} * *`,
+    };
+    await this.scheduleService.generateSchedule(cron);
+    return { code: 200, data: { ...cron } };
+  }
+
+  public async checkUpdate() {
+    try {
+      const { version } = await import(config.versionFile);
+      const lastVersionFileContent = await got.get(config.lastVersionFile);
+      const filePath = `${config.rootPath}/.version.ts`;
+      fs.writeFileSync(filePath, lastVersionFileContent.body, {
+        encoding: 'utf-8',
+      });
+      const result = await import(config.versionFile);
+
+      return {
+        code: 200,
+        data: {
+          hasNewVersion: version !== result.version,
+          ...result,
+        },
+      };
+    } catch (error) {
+      return {
+        code: 400,
+        data: '获取版本文件失败',
+      };
+    }
+  }
+
+  public async updateSystem() {
+    const cp = spawn('ql update', { shell: '/bin/bash' });
+
+    cp.stdout.on('data', (data) => {
+      this.sockService.sendMessage(data.toString());
+    });
+
+    cp.stderr.on('data', (data) => {
+      this.sockService.sendMessage(data.toString());
+    });
+
+    cp.on('error', (err) => {
+      this.sockService.sendMessage(JSON.stringify(err));
+    });
+
+    return { code: 200 };
+  }
 }