feat(validator): 优化 when 的类型

fjc0k · Dec 30, 2021 · 5206139 · 5206139
1 parent 6db08f0
commit 5206139
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/src/validator/yupTypes/mixed.ts b/src/validator/yupTypes/mixed.ts
@@ -131,9 +131,19 @@ export interface MixedSchema<T = any> {
 
   notOneOf(arrayOfValues: T[], message?: MixedLocale['notOneOf']): this
 
-  /** @仅保留了类型友好的用法 */
   when(builder: (value: T, schema: this) => this): this
 
+  when<V>(
+    key: string,
+    builder: {
+      is: boolean | ((value: V) => boolean)
+      then: GetSchema<T>
+      otherwise: GetSchema<T>
+    },
+  ): this
+
+  when<V>(key: string, builder: (value: V, schema: this) => this): this
+
   test(
     test: SchemaTestOptions<this, T>['test'] | RegExp,
     message?: SchemaTestOptions<this, T>['message'],