chore: src/modules/base/middleware/translate.ts

2026-07-01 17:56:10 +08:00
parent 24ea83bc33
commit 9e284d636b
1 changed files with 119 additions and 0 deletions
@@ -0,0 +1,119 @@
+import { Config, ILogger, Middleware } from '@midwayjs/core';
+import { NextFunction, Context } from '@midwayjs/koa';
+import { IMiddleware, Inject } from '@midwayjs/core';
+import { BaseTranslateService } from '../service/translate';
+import * as _ from 'lodash';
+import { RESCODE } from '@cool-midway/core';
+/**
+ * 翻译中间件
+ */
+@Middleware()
+export class BaseTranslateMiddleware
+  implements IMiddleware<Context, NextFunction>
+{
+  @Inject()
+  baseTranslateService: BaseTranslateService;
+
+  @Inject()
+  logger: ILogger;
+
+  @Config('cool.i18n')
+  config: {
+    /** 是否开启 */
+    enable: boolean;
+    /** 语言 */
+    languages: string[];
+    /** 翻译服务 */
+    serviceUrl?: string;
+  };
+
+  resolve() {
+    return async (ctx, next: NextFunction) => {
+      const url = ctx.url;
+      const language = ctx.get('language');
+      let data;
+      try {
+        data = await next();
+      } catch (error) {
+        this.logger.error(error);
+        // 处理翻译消息
+        if (error.name == 'CoolCommException') {
+          if (error.message && error.message !== 'success') {
+            ctx.status = error.statusCode || 200;
+            ctx.body = {
+              code: RESCODE.COMMFAIL,
+              message: await this.baseTranslateService.translate(
+                'msg',
+                language,
+                error.message
+              ),
+            };
+            return;
+          }
+        }
+        ctx.status = 200;
+        ctx.body = {
+          code: RESCODE.COMMFAIL,
+          message: error.message,
+        };
+        return;
+      }
+      if (!this.config.enable) {
+        return;
+      }
+      // 处理菜单翻译
+      if (url == '/admin/base/comm/permmenu') {
+        for (const menu of data.data.menus) {
+          if (menu.name) {
+            menu.name = await this.baseTranslateService.translate(
+              'menu',
+              language,
+              menu.name
+            );
+          }
+        }
+      }
+      if (url == '/admin/base/sys/menu/list') {
+        for (const menu of data.data) {
+          if (menu.name) {
+            menu.name = await this.baseTranslateService.translate(
+              'menu',
+              language,
+              menu.name
+            );
+          }
+        }
+      }
+      // 处理字典翻译
+      if (url == '/admin/dict/info/list') {
+        for (const dict of data.data) {
+          dict.name = await this.baseTranslateService.translate(
+            'dict:info',
+            language,
+            dict.name
+          );
+        }
+      }
+      if (url == '/admin/dict/type/page') {
+        for (const dict of data.data.list) {
+          dict.name = await this.baseTranslateService.translate(
+            'dict:type',
+            language,
+            dict.name
+          );
+        }
+      }
+      if (url == '/admin/dict/info/data' || url == '/app/dict/info/data') {
+        for (const key in data.data) {
+          for (const item of data.data[key]) {
+            item.name = await this.baseTranslateService.translate(
+              'dict:info',
+              language,
+              item.name
+            );
+          }
+        }
+      }
+    };
+  }
+}