[patch] feat: add tests for TypeScript schema loading and route discovery patterns

Co-authored-by: Copilot <copilot@github.com>

Author: Michael-Obele

src/lib/base-spec.spec.ts

@@ -105,6 +105,48 @@ describe('base-spec', () => {
 			});
 		});
 
+		it('should load and merge schemas from a TypeScript baseSchemasPath', () => {
+			const libDir = join(tempDir, 'src', 'lib');
+			mkdirSync(libDir, { recursive: true });
+
+			writeFileSync(
+				join(libDir, 'schemas.ts'),
+				`
+import type { SchemaObject } from 'openapi-types';
+
+/**
+ * @swagger
+ * components:
+ *   schemas:
+ *     Session:
+ *       type: object
+ *       properties:
+ *         token:
+ *           type: string
+ */
+export const SessionSchema: SchemaObject = {};
+`,
+				'utf-8'
+			);
+
+			const spec = createBaseSpec({
+				rootDir: tempDir,
+				baseSchemasPath: 'src/lib/schemas.ts',
+				info: {
+					title: 'Test API',
+					version: '1.0.0'
+				}
+			});
+
+			expect(spec.components?.schemas).toHaveProperty('Session');
+			expect(spec.components?.schemas?.Session).toMatchObject({
+				type: 'object',
+				properties: {
+					token: { type: 'string' }
+				}
+			});
+		});
+
 		it('should merge security schemes from baseSchemasPath', () => {
 			// Create schemas file with security schemes
 			const libDir = join(tempDir, 'src', 'lib');

src/lib/base-spec.ts

@@ -1,8 +1,11 @@
 import swaggerJsdoc from 'swagger-jsdoc';
 import type { OpenAPIV3 } from 'openapi-types';
 import { resolve } from 'path';
-import { readFileSync } from 'fs';
+import { writeFileSync, mkdirSync, readFileSync, rmSync } from 'fs';
+import { dirname, join } from 'path';
 import yaml from 'js-yaml';
+import ts from 'typescript';
+import { tmpdir } from 'os';
 import { createLogger } from './logger.js';
 
 export interface BaseSpecOptions {
@@ -52,6 +55,19 @@ export function createBaseSpec(options: BaseSpecOptions): OpenAPIV3.Document {
 
 	const logger = createLogger(silent);
 
+	const stripTypeScript = (code: string): string => {
+		const result = ts.transpileModule(code, {
+			compilerOptions: {
+				target: ts.ScriptTarget.ESNext,
+				module: ts.ModuleKind.ESNext,
+				removeComments: false,
+				isolatedModules: true
+			}
+		});
+
+		return result.outputText;
+	};
+
 	// Start with minimal base spec
 	const baseSpec: OpenAPIV3.Document = {
 		openapi: '3.0.0',
@@ -66,10 +82,24 @@ export function createBaseSpec(options: BaseSpecOptions): OpenAPIV3.Document {
 
 	// If baseSchemasPath is provided, parse it for shared schemas
 	if (baseSchemasPath) {
+		const tempDir = join(
+			tmpdir(),
+			`openapi-base-spec-${Date.now()}-${Math.random().toString(36).slice(2)}`
+		);
+		mkdirSync(tempDir, { recursive: true });
+
 		try {
 			const schemaFilePath = resolve(rootDir, baseSchemasPath);
 			const apis = [schemaFilePath];
 
+			if (schemaFilePath.endsWith('.ts')) {
+				const sourceCode = readFileSync(schemaFilePath, 'utf-8');
+				const tempSchemaPath = join(tempDir, 'base-schemas.js');
+				mkdirSync(dirname(tempSchemaPath), { recursive: true });
+				writeFileSync(tempSchemaPath, stripTypeScript(sourceCode), 'utf-8');
+				apis[0] = tempSchemaPath;
+			}
+
 			const schemaSpec = swaggerJsdoc({
 				definition: {
 					openapi: '3.0.0',
@@ -125,6 +155,8 @@ export function createBaseSpec(options: BaseSpecOptions): OpenAPIV3.Document {
 				`[openapi] Failed to parse base schemas from ${baseSchemasPath}:`,
 				error instanceof Error ? error.message : error
 			);
+		} finally {
+			rmSync(tempDir, { recursive: true, force: true });
 		}
 	}
 

src/lib/generator.spec.ts

@@ -150,6 +150,58 @@ export const GET: RequestHandler = async () => {
 			expect(spec.paths?.['/api/upload/status/{uploadId}']?.get?.summary).toBe('Get upload status');
 		});
 
+		it('should honor include and exclude patterns when discovering routes', () => {
+			const includedDir = join(tempDir, 'src', 'routes', 'api', 'included');
+			const excludedDir = join(tempDir, 'src', 'routes', 'api', 'excluded');
+			mkdirSync(includedDir, { recursive: true });
+			mkdirSync(excludedDir, { recursive: true });
+
+			writeFileSync(
+				join(includedDir, '+server.js'),
+				`
+/**
+ * @swagger
+ * /api/included:
+ *   get:
+ *     summary: Included route
+ */
+export async function GET() {
+	return new Response('included');
+}
+`,
+				'utf-8'
+			);
+
+			writeFileSync(
+				join(excludedDir, '+server.js'),
+				`
+/**
+ * @swagger
+ * /api/excluded:
+ *   get:
+ *     summary: Excluded route
+ */
+export async function GET() {
+	return new Response('excluded');
+}
+`,
+				'utf-8'
+			);
+
+			const spec = generateSpec({
+				rootDir: tempDir,
+				include: ['src/routes/api/**/+server.js'],
+				exclude: ['**/excluded/**'],
+				info: {
+					title: 'Test API',
+					version: '1.0.0'
+				}
+			});
+
+			expect(Object.keys(spec.paths || {})).toContain('/api/included');
+			expect(Object.keys(spec.paths || {})).not.toContain('/api/excluded');
+		});
+
 		it('should merge shared schemas from baseSchemasPath', () => {
 			// Create shared schemas file
 			const libDir = join(tempDir, 'src', 'lib');