Add rules option to AutoIncrementPlugin in identities.schema.ts

Author: tacxou

src/_common/plugins/mongoose/auto-increment.interface.ts

@@ -33,6 +33,8 @@ export interface AutoIncrementPluginOptions {
    * Defaults to `document.constructor.modelName`
    */
   overwriteModelName?: string | OverwriteModelNameFunction;
+
+  rules?: Function;
 }
 
 /**

src/_common/plugins/mongoose/auto-increment.plugin.ts

@@ -1,6 +1,7 @@
 import { Connection, Model, Schema } from 'mongoose';
 import { AutoIncrementPluginOptions, AutoIncrementPluginTrackerSpec } from './auto-increment.interface';
 import { logger } from './auto-increment.logger';
+import { set } from 'radash';
 
 const DEFAULT_INCREMENT = 1;
 
@@ -95,6 +96,15 @@ export function AutoIncrementPlugin(schema: Schema<any>, options: any): void {
       return;
     }
 
+
+    if (typeof opt.rules === 'function') {
+      const result = opt.rules(this);
+      if (!result) {
+        logger.info('Rule function returned false, not incrementing');
+        return next();
+      }
+    }
+
     const leandoc = await model
       .findOneAndUpdate(
         {
@@ -119,7 +129,9 @@ export function AutoIncrementPlugin(schema: Schema<any>, options: any): void {
     }
 
     logger.info('Setting "%s" to "%d"', opt.field, leandoc.count);
-    this[opt.field] = leandoc.count;
+
+    set(this, opt.field, leandoc.count);
+    // this[opt.field] = leandoc.count;
 
     return next();
   });

src/management/identities/_dto/_parts/inetOrgPerson.dto.ts

@@ -12,6 +12,7 @@ export class inetOrgPersonCreateDto {
 
   @IsString()
   @ApiProperty()
+  @IsOptional()
   public employeeNumber: string;
 
   @IsString()

src/management/identities/_schemas/_parts/inetOrgPerson.part.ts

@@ -1,5 +1,7 @@
 import { Prop, Schema, SchemaFactory } from '@nestjs/mongoose';
 import { Document } from 'mongoose';
+import { AutoIncrementPluginOptions } from '~/_common/plugins/mongoose/auto-increment.interface';
+import { AutoIncrementPlugin } from '~/_common/plugins/mongoose/auto-increment.plugin';
 
 @Schema({ _id: false })
 export class inetOrgPerson extends Document {
@@ -42,7 +44,10 @@ export class inetOrgPerson extends Document {
   @Prop({ type: String, required: true })
   public uid: string;
 
-  @Prop({ type: String, required: true })
+  @Prop({
+    type: String,
+    validate: [(employeeNumber: string) => /[A-Za-z0-9_-]+/.test(employeeNumber), 'EmployeeNumber invalide.'],
+  })
   public employeeNumber: string;
 
   @Prop({ type: String, required: true })

src/management/identities/_schemas/identities.schema.ts

@@ -8,6 +8,8 @@ import { AdditionalFieldsPart, AdditionalFieldsPartSchema } from './_parts/addit
 import { InitStatesEnum } from '~/management/identities/_enums/init-state.enum';
 import { InitInfoPart, InitInfoPartSchema } from '~/management/identities/_schemas/_parts/init-info.part.schema';
 import { MixedValue } from '~/_common/types/mixed-value.type';
+import { AutoIncrementPlugin } from '~/_common/plugins/mongoose/auto-increment.plugin';
+import { AutoIncrementPluginOptions } from '~/_common/plugins/mongoose/auto-increment.interface';
 
 export type IdentitiesDocument = Identities & Document;
 
@@ -43,4 +45,11 @@ export class Identities extends AbstractSchema {
   public customFields?: { [key: string]: MixedValue }
 }
 
-export const IdentitiesSchema = SchemaFactory.createForClass(Identities);
+export const IdentitiesSchema = SchemaFactory.createForClass(Identities).plugin(AutoIncrementPlugin, <AutoIncrementPluginOptions>{
+  incrementBy: 1,
+  field: 'inetOrgPerson.employeeNumber',
+  startAt: 1,
+  rules: (ctx) => {
+    return ctx.inetOrgPerson.employeeType === 'LOCAL';
+  },
+});