feat: output regex patterns for generated schemas

Rhys · Rhys · commit dcb1534763a5 · 2025-11-26T12:00:49.000-04:00
diff --git a/packages/rtk-query-codegen-openapi/src/generate.ts b/packages/rtk-query-codegen-openapi/src/generate.ts
@@ -2,13 +2,13 @@ import camelCase from 'lodash.camelcase';
 import path from 'node:path';
 import ApiGenerator, {
   getOperationName as _getOperationName,
-  getReferenceName,
-  isReference,
-  supportDeepObjects,
   createPropertyAssignment,
   createQuestionToken,
+  getReferenceName,
+  isReference,
   isValidIdentifier,
   keywordType,
+  supportDeepObjects,
 } from 'oazapfts/generate';
 import type { OpenAPIV3 } from 'openapi-types';
 import ts from 'typescript';
@@ -87,6 +87,56 @@ function withQueryComment<T extends ts.Node>(node: T, def: QueryArgDefinition, h
   return node;
 }
 
+function getPatternFromProperty(
+  property: OpenAPIV3.SchemaObject | OpenAPIV3.ReferenceObject,
+  apiGen: ApiGenerator
+): string | null {
+  const resolved = apiGen.resolve(property);
+  if (!resolved || typeof resolved !== 'object' || !('pattern' in resolved)) return null;
+  if (resolved.type !== 'string') return null;
+  const pattern = resolved.pattern;
+  return typeof pattern === 'string' && pattern.length > 0 ? pattern : null;
+}
+
+function generateRegexConstantsForType(
+  typeName: string,
+  schema: OpenAPIV3.SchemaObject | OpenAPIV3.ReferenceObject,
+  apiGen: ApiGenerator
+): ts.VariableStatement[] {
+  const resolvedSchema = apiGen.resolve(schema);
+  if (!resolvedSchema || !('properties' in resolvedSchema) || !resolvedSchema.properties) return [];
+
+  const constants: ts.VariableStatement[] = [];
+
+  for (const [propertyName, property] of Object.entries(resolvedSchema.properties)) {
+    const pattern = getPatternFromProperty(property, apiGen);
+    if (!pattern) continue;
+
+    const constantName = camelCase(`${typeName} ${propertyName} Pattern`);
+    const escapedPattern = pattern.replaceAll('/', String.raw`\/`);
+    const regexLiteral = factory.createRegularExpressionLiteral(`/${escapedPattern}/`);
+
+    constants.push(
+      factory.createVariableStatement(
+        [factory.createModifier(ts.SyntaxKind.ExportKeyword)],
+        factory.createVariableDeclarationList(
+          [
+            factory.createVariableDeclaration(
+              factory.createIdentifier(constantName),
+              undefined,
+              undefined,
+              regexLiteral
+            ),
+          ],
+          ts.NodeFlags.Const
+        )
+      )
+    );
+  }
+
+  return constants;
+}
+
 export function getOverrides(
   operation: OperationDefinition,
   endpointOverrides?: EndpointOverrides[]
@@ -119,6 +169,7 @@ export async function generateApi(
     httpResolverOptions,
     useUnknown = false,
     esmExtensions = false,
+    outputRegexConstants = false,
   }: GenerationOptions
 ) {
   const v3Doc = (v3DocCache[spec] ??= await getV3Doc(spec, httpResolverOptions));
@@ -206,7 +257,18 @@ export async function generateApi(
           undefined
         ),
         ...Object.values(interfaces),
-        ...apiGen.aliases,
+        ...(outputRegexConstants
+          ? apiGen.aliases.flatMap((alias) => {
+              if (!ts.isInterfaceDeclaration(alias) && !ts.isTypeAliasDeclaration(alias)) return [alias];
+
+              const typeName = alias.name.escapedText.toString();
+              const schema = v3Doc.components?.schemas?.[typeName];
+              if (!schema) return [alias];
+
+              const regexConstants = generateRegexConstantsForType(typeName, schema, apiGen);
+              return regexConstants.length > 0 ? [alias, ...regexConstants] : [alias];
+            })
+          : apiGen.aliases),
         ...apiGen.enumAliases,
         ...(hooks
           ? [
diff --git a/packages/rtk-query-codegen-openapi/src/types.ts b/packages/rtk-query-codegen-openapi/src/types.ts
@@ -126,6 +126,11 @@ export interface CommonOptions {
    * Will generate imports with file extension matching the expected compiled output of the api file
    */
   esmExtensions?: boolean;
+  /**
+   * @default false
+   * Will generate regex constants for pattern keywords in the schema
+   */
+  outputRegexConstants?: boolean;
 }
 
 export type TextMatcher = string | RegExp | (string | RegExp)[];
diff --git a/packages/rtk-query-codegen-openapi/test/fixtures/petstore.json b/packages/rtk-query-codegen-openapi/test/fixtures/petstore.json
@@ -1015,18 +1015,22 @@
           },
           "email": {
             "type": "string",
+            "pattern": "^[a-zA-Z0-9._%+-]+@[a-zA-Z0-9.-]+\\.[a-zA-Z]{2,}$",
             "example": "john@email.com"
           },
           "password": {
             "type": "string",
-            "example": "12345"
+            "example": "12345",
+            "pattern": ""
           },
           "phone": {
             "type": "string",
+            "pattern": "^\\+?[1-9]\\d{1,14}$",
             "example": "12345"
           },
           "userStatus": {
             "type": "integer",
+            "pattern": "^[1-9]\\d{0,2}$",
             "description": "User Status",
             "format": "int32",
             "example": 1
@@ -1044,7 +1048,8 @@
             "format": "int64"
           },
           "name": {
-            "type": "string"
+            "type": "string",
+            "pattern": "^\\S+$"
           }
         },
         "xml": {
diff --git a/packages/rtk-query-codegen-openapi/test/fixtures/petstore.yaml b/packages/rtk-query-codegen-openapi/test/fixtures/petstore.yaml
@@ -695,15 +695,19 @@ components:
           example: James
         email:
           type: string
+          pattern: '^[a-zA-Z0-9._%+-]+@[a-zA-Z0-9.-]+\.[a-zA-Z]{2,}$'
           example: john@email.com
         password:
           type: string
+          pattern: ''
           example: '12345'
         phone:
           type: string
+          pattern: '^\+?[1-9]\d{1,14}$'
           example: '12345'
         userStatus:
           type: integer
+          pattern: '^[1-9]\d{0,2}$'
           description: User Status
           format: int32
           example: 1
@@ -717,6 +721,7 @@ components:
           format: int64
         name:
           type: string
+          pattern: '^\S+$'
       xml:
         name: tag
     Pet:
diff --git a/packages/rtk-query-codegen-openapi/test/generateEndpoints.test.ts b/packages/rtk-query-codegen-openapi/test/generateEndpoints.test.ts
@@ -648,6 +648,53 @@ describe('query parameters', () => {
   });
 });
 
+describe('regex constants', () => {
+  it('should export regex constants for patterns', async () => {
+    const api = await generateEndpoints({
+      unionUndefined: true,
+      apiFile: './fixtures/emptyApi.ts',
+      schemaFile: resolve(__dirname, 'fixtures/petstore.json'),
+      outputRegexConstants: true,
+    });
+
+    expect(api).toContain(String.raw`export const tagNamePattern = /^\S+$/`);
+    expect(api).toContain(String.raw`export const userEmailPattern`);
+    expect(api).toContain(String.raw`/^[a-zA-Z0-9._%+-]+@[a-zA-Z0-9.-]+\.[a-zA-Z]{2,}$/`);
+    expect(api).toContain(String.raw`export const userPhonePattern = /^\+?[1-9]\d{1,14}$/`);
+  });
+
+  it('should not export constants for invalid patterns', async () => {
+    const api = await generateEndpoints({
+      unionUndefined: true,
+      apiFile: './fixtures/emptyApi.ts',
+      schemaFile: resolve(__dirname, 'fixtures/petstore.json'),
+      outputRegexConstants: true,
+    });
+
+    // Empty pattern should not generate a constant
+    expect(api).not.toContain('userPasswordPattern');
+    expect(api).not.toContain('passwordPattern');
+
+    // Pattern on non-string property (integer) should not generate a constant
+    expect(api).not.toContain('userUserStatusPattern');
+    expect(api).not.toContain('userStatusPattern');
+  });
+
+  it('should export regex constants for patterns from YAML file', async () => {
+    const api = await generateEndpoints({
+      unionUndefined: true,
+      apiFile: './fixtures/emptyApi.ts',
+      schemaFile: resolve(__dirname, 'fixtures/petstore.yaml'),
+      outputRegexConstants: true,
+    });
+
+    expect(api).toContain(String.raw`export const tagNamePattern = /^\S+$/`);
+    expect(api).toContain(String.raw`export const userEmailPattern`);
+    expect(api).toContain(String.raw`/^[a-zA-Z0-9._%+-]+@[a-zA-Z0-9.-]+\.[a-zA-Z]{2,}$/`);
+    expect(api).toContain(String.raw`export const userPhonePattern = /^\+?[1-9]\d{1,14}$/`);
+  });
+});
+
 describe('esmExtensions option', () => {
   beforeAll(async () => {
     if (!(await isDir(tmpDir))) {
