feat: add ip versions support

ItMaga · Jul 27, 2023 · b9052de · b9052de
1 parent 75db05c
commit b9052de
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 5 deletions.
diff --git a/lib/generators/StringGenerator.ts b/lib/generators/StringGenerator.ts
@@ -24,7 +24,7 @@ export default class StringGenerator<T extends z.ZodString> implements BaseGener
             string = '😀';
             break;
           case 'ip':
-            string = this.getRandomIP();
+            string = this.getRandomIP(check.version!);
             break;
           case 'includes':
           case 'startsWith':
@@ -62,7 +62,10 @@ export default class StringGenerator<T extends z.ZodString> implements BaseGener
     });
   }
 
-  public getRandomIP(): string {
-    return Array.from({ length: 4 }, () => Math.floor(Math.random() * 256)).join('.');
+  public getRandomIP(version: 'v4' | 'v6'): string {
+    if (version === 'v4') {
+      return [0, 0, 0, 0].map(() => Math.floor(Math.random() * 256)).join('.');
+    }
+    return [0, 0, 0, 0, 0, 0, 0, 0].map(() => Math.floor(Math.random() * 65536).toString(16)).join(':');
   }
 }
diff --git a/tests/string.test.ts b/tests/string.test.ts
@@ -54,9 +54,11 @@ describe('String', () => {
     expect(schema);
   });
   test('ip', () => {
-    const schema = z.string().ip();
+    const schemaV4 = z.string().ip({ version: 'v4' });
+    const schemaV6 = z.string().ip({ version: 'v6' });
 
-    expect(schema);
+    expect(schemaV4);
+    expect(schemaV6);
   });
   test('startsWith', () => {
     const schema = z.string().startsWith('__MARK__');