certd/packages/ui/certd-server/src/middleware/preview.ts

import { Config, Provide } from "@midwayjs/core";
import { IMidwayKoaContext, NextFunction, IWebMiddleware } from "@midwayjs/koa";
import { PreviewException } from "@certd/lib-server";

/**
 * 预览模式
 */
@Provide()
export class PreviewMiddleware implements IWebMiddleware {
  @Config("preview.enabled")
  private preview: boolean;

  resolve() {
    return async (ctx: IMidwayKoaContext, next: NextFunction) => {
      if (!this.preview) {
        await next();
        return;
      }
      // eslint-disable-next-line prefer-const
      let { url, request } = ctx;
      const body: any = request.body;
      let id = body.id || request.query.id;
      const roleId = body.roleId;
      if (id == null && roleId != null) {
        id = roleId;
      }
      if (id != null && typeof id === "string") {
        id = parseInt(id);
      }
      if (url.indexOf("?") !== -1) {
        url = url.substring(0, url.indexOf("?"));
      }
      const isModify = url.endsWith("update") || url.endsWith("delete") || url.endsWith("authz");
      const isPreviewId = id < 1000;
      if (this.preview && isModify && isPreviewId) {
        throw new PreviewException("对不起，预览环境不允许修改此数据，如需体验请添加新数据");
      }
      await next();
      return;
    };
  }
}