configuration.ts

import {
  Configuration,
  Inject,
  MidwayDecoratorService,
  WEB_ROUTER_PARAM_KEY,
  IMidwayContainer,
  Init,
} from '@midwayjs/core';
import * as DefaultConfig from './config/config.default';
import { ValidateService } from './service';
import * as i18n from '@midwayjs/i18n';
import {
  DecoratorValidPipe,
  ParseBoolPipe,
  ParseFloatPipe,
  ParseIntPipe,
  ValidationPipe,
} from './pipe';
import { VALID_KEY } from './constants';

@Configuration({
  namespace: 'validate',
  imports: [i18n],
  importConfigs: [
    {
      default: DefaultConfig,
    },
  ],
})
export class ValidateConfiguration {
  @Inject()
  decoratorService: MidwayDecoratorService;

  @Inject()
  validateService: ValidateService;

  @Init()
  async init() {
    this.decoratorService.registerParameterHandler(
      VALID_KEY,
      ({ parameterIndex, originParamType, originArgs, metadata }) => {
        if (!metadata.schema) {
          metadata.schema = this.validateService.getSchema(originParamType);
        }
        return originArgs[parameterIndex];
      }
    );
  }

  async onReady(container: IMidwayContainer) {
    await container.getAsync(ValidationPipe);
    await container.getAsync(ParseIntPipe);
    await container.getAsync(ParseBoolPipe);
    await container.getAsync(ParseFloatPipe);
    await container.getAsync(DecoratorValidPipe);

    this.decoratorService.registerParameterHandler(
      VALID_KEY,
      ({ parameterIndex, originParamType, originArgs, metadata }) => {
        if (!metadata.schema) {
          metadata.schema = this.validateService.getSchema(originParamType);
        }
        return originArgs[parameterIndex];
      }
    );

    // register web param default pipe
    this.decoratorService.registerParameterPipes(WEB_ROUTER_PARAM_KEY, [
      ValidationPipe,
    ]);
  }
}