certd/packages/ui/certd-server/src/configuration.ts

import { App, Configuration } from "@midwayjs/core";
import * as koa from "@midwayjs/koa";
import { IMidwayKoaContext, NextFunction } from "@midwayjs/koa";
import * as orm from "@midwayjs/typeorm";
import * as cache from "@midwayjs/cache";
import * as validate from "@midwayjs/validate";
import * as info from "@midwayjs/info";
import * as staticFile from "@midwayjs/static-file";
import * as cron from "./modules/cron/index.js";
import * as flyway from "@certd/midway-flyway-js";
import cors from "@koa/cors";
import { GlobalExceptionMiddleware } from "./middleware/global-exception.js";
import { PreviewMiddleware } from "./middleware/preview.js";
import { AuthorityMiddleware } from "./middleware/authority.js";
import { logger } from "@certd/basic";
import { ResetPasswdMiddleware } from "./middleware/reset-passwd/middleware.js";
import DefaultConfig from "./config/config.default.js";
import * as libServer from "@certd/lib-server";
import * as commercial from "@certd/commercial-core";
import * as upload from "@midwayjs/upload";
import { setLogger } from "@certd/acme-client";
import { HiddenMiddleware } from "./middleware/hidden.js";
import { shouldSetDefaultNoCache } from "./configuration-cache.js";
// import * as swagger from '@midwayjs/swagger';
//@ts-ignore
// process.env.UV_THREADPOOL_SIZE = 2
process.on("uncaughtException", error => {
  console.error("未捕获的异常：", error);
  // 在这里可以添加日志记录、发送错误通知等操作
  if (error?.message?.includes("address family not supported")) {
    logger.error("您的服务器不支持监听IPV6格式的地址（::），请配置环境变量: certd_koa_hostname=0.0.0.0");
  }
});

// function startHeapLog() {
//   function format(bytes: any) {
//     return (bytes / 1024 / 1024).toFixed(2) + ' MB';
//   }
//   function log()  {
//     const mu = process.memoryUsage();
//     logger.info(`rss:${format(mu.rss)},heapUsed: ${format(mu.heapUsed)},heapTotal: ${format(mu.heapTotal)},external: ${format(mu.external)}`);
//   }
//   setInterval(log, 200);
//   log()
// }

// startHeapLog();

@Configuration({
  detectorOptions: {
    ignore: ["**/plugins/**"],
  },
  imports: [
    koa,
    orm,
    cache,
    flyway,
    cron,
    staticFile,
    validate,
    upload,
    libServer,
    commercial,
    // {
    //   component: swagger,
    //   enabledEnvironment: ['local']
    // },
    {
      component: info,
      enabledEnvironment: ["local"],
    },
  ],
  importConfigs: [
    {
      default: DefaultConfig,
    },
  ],
})
export class MainConfiguration {
  @App("koa")
  app: koa.Application;

  async onReady() {
    // 设置flyway logger

    // add middleware
    // this.app.useMiddleware([ReportMiddleware]);
    // add filter
    // this.app.useFilter([NotFoundFilter, DefaultErrorFilter]);
    //跨域
    this.app.use(
      cors({
        origin: "*",
      })
    );
    //
    // this.app.use(async (ctx, next) => {
    //   // 只在返回 'index.html' 的时候设置 maxAge
    //   if (ctx.path === '/') {
    //     // ctx.response.redirect('/index.html');
    //     ctx.send(file)
    //     return;
    //   }
    // });
    // bodyparser options see https://github.com/koajs/bodyparser
    //this.app.use(bodyParser());
    //请求日志打印
    this.app.useMiddleware([
      //统一异常处理
      GlobalExceptionMiddleware,
      //站点隐藏
      HiddenMiddleware,
      //预览模式限制修改id<1000的数据
      PreviewMiddleware,
      //授权处理
      AuthorityMiddleware,

      //resetPasswd,重置密码模式下不提供服务
      ResetPasswdMiddleware,
    ]);

    this.app.getMiddleware().insertFirst(async (ctx: IMidwayKoaContext, next: NextFunction) => {
      await next();
      const path = ctx.path;
      // 如果是首页则强制设置为不缓存
      if (shouldSetDefaultNoCache(path, ctx.response.get("Cache-Control"))) {
        ctx.response.set("Cache-Control", "public,max-age=0");
      }
    });

    //acme setlogger
    setLogger((text: string) => {
      logger.info(text);
    });

    logger.info("当前环境：", this.app.getEnv()); // prod

    
  }
}