ts-class-to-openapi 1.3.4 → 1.4.1

package/README.md

@@ -27,7 +27,26 @@ class User {
 }
 
 const schema = transform(User)
-// Returns complete OpenAPI schema ready for Swagger/API documentation
+
+console.log(JSON.stringify(shema), null, 2)
+```
+
+**Generated output:**
+
+```json
+{
+  "name": "User",
+  "schema": {
+    "type": "object",
+    "properties": {
+      "id": { "type": "number" },
+      "name": { "type": "string" },
+      "email": { "type": "string" },
+      "age": { "type": "number" }
+    },
+    "required": ["id", "name", "email"]
+  }
+}
 ```
 
 ## 📦 Installation
@@ -55,57 +74,7 @@ npm install ts-class-to-openapi class-validator
 
 ## 🎨 Class Transformation Examples
 
-### 1. Basic TypeScript Class
-
-Fundamental method: transform any TypeScript class without decorators:
-
-```typescript
-import { transform } from 'ts-class-to-openapi'
-
-// Basic TypeScript class - no decorators required
-class User {
-  id: number
-  name: string
-  email: string
-  age: number
-  active: boolean
-  tags: string[]
-  createdAt: Date
-}
-
-// Transform class to OpenAPI schema
-const result = transform(User)
-console.log(JSON.stringify(result, null, 2))
-```
-
-**Generated output:**
-
-```json
-{
-  "name": "User",
-  "schema": {
-    "type": "object",
-    "properties": {
-      "id": { "type": "number" },
-      "name": { "type": "string" },
-      "email": { "type": "string" },
-      "age": { "type": "number" },
-      "active": { "type": "boolean" },
-      "tags": {
-        "type": "array",
-        "items": { "type": "string" }
-      },
-      "createdAt": {
-        "type": "string",
-        "format": "date-time"
-      }
-    },
-    "required": ["id", "name", "email", "age", "active", "tags", "createdAt"]
-  }
-}
-```
-
-### 2. Class with Advanced Validations
+### 1. Class with Advanced Validations
 
 For more detailed schemas, class-validator decorators can be incorporated:
 
@@ -156,7 +125,7 @@ const result = transform(User)
 }
 ```
 
-### 3. Nested Objects and Arrays
+### 2. Nested Objects and Arrays
 
 Automatic processing of complex relationships:
 
@@ -229,9 +198,9 @@ const schema = transform(User)
 }
 ```
 
-### 4. Enumerations and Special Types
+### 3. Enumerations and Special Types
 
-Full compatibility with TypeScript enumerations (both decorated and pure):
+Full compatibility with TypeScript enumerations (both decorated and pure), and literal object as enums:
 
 ```typescript
 import { transform } from 'ts-class-to-openapi'
@@ -299,59 +268,6 @@ const schema = transform(Task)
 }
 ```
 
-### 5. File Upload
-
-Integrated support for binary file handling:
-
-```typescript
-import { transform } from 'ts-class-to-openapi'
-import { IsNotEmpty, IsOptional } from 'class-validator'
-
-// Custom file type definition
-class UploadFile {}
-
-class UserProfile {
-  @IsNotEmpty()
-  profilePicture: UploadFile
-
-  @IsOptional()
-  resume: UploadFile
-
-  documents: UploadFile[] // Multiple files
-}
-
-const schema = transform(UserProfile)
-```
-
-**Generated output:**
-
-```json
-{
-  "name": "UserProfile",
-  "schema": {
-    "type": "object",
-    "properties": {
-      "profilePicture": {
-        "type": "string",
-        "format": "binary"
-      },
-      "resume": {
-        "type": "string",
-        "format": "binary"
-      },
-      "documents": {
-        "type": "array",
-        "items": {
-          "type": "string",
-          "format": "binary"
-        }
-      }
-    },
-    "required": ["profilePicture", "documents"]
-  }
-}
-```
-
 ## 🌐 REST API Integration
 
 ### Implementation with Express.js and Swagger UI

package/dist/index.cjs

@@ -189,6 +189,7 @@ var SchemaTransformer = class SchemaTransformer {
 	maxCacheSize;
 	autoCleanup;
 	classFileIndex = /* @__PURE__ */ new Map();
+	transformCallIndex = /* @__PURE__ */ new Map();
 	constructor(tsConfigPath = constants.TS_CONFIG_DEFAULT_PATH, options = {}) {
 		this.maxCacheSize = options.maxCacheSize ?? 100;
 		this.autoCleanup = options.autoCleanup ?? true;
@@ -200,19 +201,7 @@ var SchemaTransformer = class SchemaTransformer {
 		const { options: tsOptions, fileNames } = typescript.default.parseJsonConfigFileContent(config, typescript.default.sys, "./");
 		this.program = typescript.default.createProgram(fileNames, tsOptions);
 		this.checker = this.program.getTypeChecker();
-		this.program.getSourceFiles().forEach((sf) => {
-			sf.statements.forEach((stmt) => {
-				if (typescript.default.isClassDeclaration(stmt) && stmt.name) {
-					const name = stmt.name.text;
-					const entry = this.classFileIndex.get(name) || [];
-					entry.push({
-						sourceFile: sf,
-						node: stmt
-					});
-					this.classFileIndex.set(name, entry);
-				}
-			});
-		});
+		this.buildTransformCallIndex();
 	}
 	getPropertiesByClassDeclaration(classNode, visitedDeclarations = /* @__PURE__ */ new Set(), genericTypeMap = /* @__PURE__ */ new Map()) {
 		if (visitedDeclarations.has(classNode)) return [];
@@ -260,6 +249,18 @@ var SchemaTransformer = class SchemaTransformer {
 				const isOptional = !!member.questionToken;
 				const isGeneric = this.isPropertyTypeGeneric(member);
 				const isEnum = this.isEnum(member);
+				const isPrimitive = this.isPrimitiveType(type) || isEnum;
+				const isClassType = this.isClassType(member);
+				const isArray = this.isArrayProperty(member);
+				const isTypeLiteral = this.isTypeLiteral(member);
+				let genericClassReference = void 0;
+				if (isGeneric && !isPrimitive) {
+					const baseTypeName = type.replace(/\[\]$/, "").trim();
+					if (!this.isPrimitiveType(baseTypeName)) {
+						const matches = this.classFileIndex.get(baseTypeName);
+						if (matches && matches.length > 0 && matches[0]) genericClassReference = matches[0].node;
+					}
+				}
 				const property = {
 					name: propertyName,
 					type,
@@ -267,12 +268,13 @@ var SchemaTransformer = class SchemaTransformer {
 					isOptional,
 					isGeneric,
 					originalProperty: member,
-					isPrimitive: this.isPrimitiveType(type) || isEnum,
-					isClassType: this.isClassType(member),
-					isArray: this.isArrayProperty(member),
+					isPrimitive,
+					isClassType,
+					isArray,
 					isEnum,
 					isRef: false,
-					isTypeLiteral: this.isTypeLiteral(member)
+					isTypeLiteral,
+					genericClassReference
 				};
 				if (property.isClassType) {
 					const declaration = this.getDeclarationProperty(property);
@@ -310,6 +312,7 @@ var SchemaTransformer = class SchemaTransformer {
 			if (typeName.toLowerCase() === "uploadfile") return "UploadFile";
 			if (typeName.toLowerCase() === "uploadfiledto") return "UploadFileDto";
 			if (typeNode.typeArguments && typeNode.typeArguments.length > 0) {
+				if (typeName === "Array") return `${this.getTypeNodeToString(typeNode.typeArguments[0], genericTypeMap)}[]`;
 				const firstTypeArg = typeNode.typeArguments[0];
 				if (firstTypeArg && typescript.default.isTypeReferenceNode(firstTypeArg) && typescript.default.isIdentifier(firstTypeArg.typeName)) {
 					if (firstTypeArg.typeName.text.toLowerCase() === "uploadfile") return "UploadFile";
@@ -332,6 +335,11 @@ var SchemaTransformer = class SchemaTransformer {
 				if (types.length > 0 && types[0]) return types[0];
 				return "object";
 			default:
+				if (typescript.default.isIndexedAccessTypeNode(typeNode)) {
+					const resolvedType = this.checker.getTypeAtLocation(typeNode);
+					const resolved = this.checker.typeToString(resolvedType);
+					if (this.isPrimitiveType(resolved)) return resolved;
+				}
 				const typeText = typeNode.getText();
 				if (genericTypeMap && genericTypeMap.has(typeText)) return genericTypeMap.get(typeText);
 				if (typeText.startsWith("Date")) return constants.jsPrimitives.Date.type;
@@ -406,7 +414,7 @@ var SchemaTransformer = class SchemaTransformer {
 	isGenericTypeFromSymbol(type) {
 		if (this.isSimpleArrayType(type)) return false;
 		if (type.aliasTypeArguments && type.aliasTypeArguments.length > 0) return true;
-		if (type.typeArguments && type.typeArguments.length > 0) {
+		if (type.typeArguments && type.typeArguments.length > 0 && type.typeArguments[0].symbol.getName() === "Array") {
 			const symbol = type.getSymbol();
 			if (symbol && symbol.getName() === "Array") {
 				const elementType = type.typeArguments[0];
@@ -451,6 +459,7 @@ var SchemaTransformer = class SchemaTransformer {
 			if (!elementType) return false;
 			if (this.isUtilityTypeFromType(elementType)) return false;
 			if (elementType.typeArguments && elementType.typeArguments.length > 0) return false;
+			if (type.typeArguments && type.typeArguments[0].symbol && type.typeArguments[0].symbol.getName() !== "Array") return false;
 			return true;
 		}
 		return false;
@@ -508,22 +517,6 @@ var SchemaTransformer = class SchemaTransformer {
 		}
 		return matches[0];
 	}
-	checkTypeMatch(value, typeNode) {
-		const runtimeType = typeof value;
-		if (runtimeType === "string" && typeNode.kind === typescript.default.SyntaxKind.StringKeyword) return true;
-		if (runtimeType === "number" && typeNode.kind === typescript.default.SyntaxKind.NumberKeyword) return true;
-		if (runtimeType === "boolean" && typeNode.kind === typescript.default.SyntaxKind.BooleanKeyword) return true;
-		if (Array.isArray(value) && typescript.default.isArrayTypeNode(typeNode)) {
-			if (value.length === 0) return true;
-			const firstItem = value[0];
-			const elementType = typeNode.elementType;
-			return this.checkTypeMatch(firstItem, elementType);
-		}
-		if (runtimeType === "object" && value !== null && !Array.isArray(value)) {
-			if (typescript.default.isTypeReferenceNode(typeNode) || typeNode.kind === typescript.default.SyntaxKind.ObjectKeyword) return true;
-		}
-		return false;
-	}
 	findBestMatch(cls, matches) {
 		let instance = {};
 		try {
@@ -565,7 +558,11 @@ var SchemaTransformer = class SchemaTransformer {
 	isClassType(propertyDeclaration) {
 		if (!propertyDeclaration.type) return false;
 		if (this.isArrayProperty(propertyDeclaration)) {
-			const elementType = propertyDeclaration.type.elementType;
+			var _propertyDeclaration$;
+			let elementType;
+			if (typescript.default.isArrayTypeNode(propertyDeclaration.type)) elementType = propertyDeclaration.type.elementType;
+			else if (typescript.default.isTypeReferenceNode(propertyDeclaration.type) && ((_propertyDeclaration$ = propertyDeclaration.type.typeArguments) === null || _propertyDeclaration$ === void 0 ? void 0 : _propertyDeclaration$[0])) elementType = propertyDeclaration.type.typeArguments[0];
+			if (!elementType) return false;
 			if (typescript.default.isTypeReferenceNode(elementType) && elementType.typeArguments && elementType.typeArguments.length > 0) {
 				const firstTypeArg = elementType.typeArguments[0];
 				if (firstTypeArg) {
@@ -626,7 +623,9 @@ var SchemaTransformer = class SchemaTransformer {
 	}
 	isArrayProperty(propertyDeclaration) {
 		if (!propertyDeclaration.type) return false;
-		return typescript.default.isArrayTypeNode(propertyDeclaration.type);
+		if (typescript.default.isArrayTypeNode(propertyDeclaration.type)) return true;
+		if (typescript.default.isTypeReferenceNode(propertyDeclaration.type) && typescript.default.isIdentifier(propertyDeclaration.type.typeName) && propertyDeclaration.type.typeName.text === "Array") return true;
+		return false;
 	}
 	getSchemaFromProperties({ properties, visitedClass, transformedSchema, classDeclaration }) {
 		let schema = {};
@@ -661,6 +660,17 @@ var SchemaTransformer = class SchemaTransformer {
 			visitedClass,
 			transformedSchema
 		});
+		else if (property.isGeneric) if (property.genericClassReference) schema = this.getSchemaFromClass({
+			isArray: property.isArray,
+			visitedClass,
+			transformedSchema,
+			declaration: property.genericClassReference
+		});
+		else schema = {
+			type: "object",
+			properties: {},
+			additionalProperties: true
+		};
 		else schema = {
 			type: "object",
 			properties: {},
@@ -838,6 +848,11 @@ var SchemaTransformer = class SchemaTransformer {
 		if (decorator.arguments.length === 0) return;
 		const arg = decorator.arguments[0];
 		if (arg && typeof arg === "object" && "kind" in arg) {
+			if (typescript.default.isArrayLiteralExpression(arg)) {
+				const values = this.extractValuesFromArrayLiteral(arg);
+				if (values.length > 0) this.applyEnumValues(values, schema);
+				return;
+			}
 			const type = this.checker.getTypeAtLocation(arg);
 			if (type.symbol && type.symbol.exports) {
 				const values = [];
@@ -849,15 +864,46 @@ var SchemaTransformer = class SchemaTransformer {
 					}
 				});
 				if (values.length > 0) {
-					schema.enum = values;
-					const isString = values.every((v) => typeof v === "string");
-					const isNumber = values.every((v) => typeof v === "number");
-					if (isString) schema.type = "string";
-					else if (isNumber) schema.type = "number";
-					else schema.type = "string";
+					this.applyEnumValues(values, schema);
+					return;
 				}
 			}
+			const values = this.extractValuesFromObjectLiteral(type);
+			if (values.length > 0) this.applyEnumValues(values, schema);
+		}
+	}
+	extractValuesFromArrayLiteral(arrayLiteral) {
+		const values = [];
+		for (const element of arrayLiteral.elements) if (typescript.default.isStringLiteral(element)) values.push(element.text);
+		else if (typescript.default.isNumericLiteral(element)) values.push(Number(element.text));
+		else if (typescript.default.isPrefixUnaryExpression(element) && element.operator === typescript.default.SyntaxKind.MinusToken && typescript.default.isNumericLiteral(element.operand)) values.push(-Number(element.operand.text));
+		return values;
+	}
+	extractValuesFromObjectLiteral(type) {
+		const values = [];
+		const properties = type.getProperties();
+		if (!properties || properties.length === 0) return values;
+		for (const prop of properties) {
+			const propType = this.checker.getTypeOfSymbolAtLocation(prop, prop.valueDeclaration);
+			if (propType.isStringLiteral()) values.push(propType.value);
+			else if (propType.isNumberLiteral()) values.push(propType.value);
+			else if (prop.valueDeclaration && typescript.default.isPropertyAssignment(prop.valueDeclaration)) {
+				const initializer = prop.valueDeclaration.initializer;
+				if (typescript.default.isStringLiteral(initializer)) values.push(initializer.text);
+				else if (typescript.default.isNumericLiteral(initializer)) values.push(Number(initializer.text));
+			}
 		}
+		return values;
+	}
+	applyEnumValues(values, schema) {
+		schema.enum = values;
+		const isString = values.every((v) => typeof v === "string");
+		const isNumber = values.every((v) => typeof v === "number");
+		if (isString) schema.type = "string";
+		else if (isNumber) schema.type = "number";
+		else schema.type = "string";
+		delete schema.properties;
+		delete schema.additionalProperties;
 	}
 	applyDecorators(property, schema) {
 		for (const decorator of property.decorators) switch (decorator.name) {
@@ -935,16 +981,65 @@ var SchemaTransformer = class SchemaTransformer {
 				break;
 			case constants.validatorDecorators.IsEnum.name:
 				if (!property.isArray) this.applyEnumDecorator(decorator, schema);
-				else if (schema.items) this.applyEnumDecorator(decorator, schema.items);
+				else {
+					if (!schema.items) {
+						schema.type = "array";
+						schema.items = {};
+					}
+					this.applyEnumDecorator(decorator, schema.items);
+				}
 				break;
 		}
 	}
+	/**
+	* Scans all non-declaration source files in the program once and records
+	* every call of the form `transform(Foo<Bar, Baz>)`, keyed by class name.
+	* This means generic resolution in transform() is a pure O(1) Map lookup
+	* with no runtime stack inspection.
+	*/
+	buildTransformCallIndex() {
+		this.program.getSourceFiles().forEach((sf) => {
+			if (sf.isDeclarationFile) return;
+			sf.statements.forEach((stmt) => {
+				if (typescript.default.isClassDeclaration(stmt) && stmt.name) {
+					const name = stmt.name.text;
+					const entry = this.classFileIndex.get(name) || [];
+					entry.push({
+						sourceFile: sf,
+						node: stmt
+					});
+					this.classFileIndex.set(name, entry);
+				}
+			});
+			const visit = (node) => {
+				if (typescript.default.isCallExpression(node) && node.arguments.length > 0) {
+					const callee = node.expression;
+					if (typescript.default.isIdentifier(callee) && callee.text === "transform" || typescript.default.isPropertyAccessExpression(callee) && callee.name.text === "transform") {
+						const firstArg = node.arguments[0];
+						const typeArgs = firstArg.typeArguments;
+						if (typeArgs && typeArgs.length > 0) {
+							const baseExpr = firstArg.expression ?? firstArg;
+							if (typescript.default.isIdentifier(baseExpr)) {
+								var _this$classFileIndex$;
+								const classNode = (_this$classFileIndex$ = this.classFileIndex.get(baseExpr.text)) === null || _this$classFileIndex$ === void 0 || (_this$classFileIndex$ = _this$classFileIndex$[0]) === null || _this$classFileIndex$ === void 0 ? void 0 : _this$classFileIndex$.node;
+								if (classNode === null || classNode === void 0 ? void 0 : classNode.typeParameters) {
+									const typeMap = /* @__PURE__ */ new Map();
+									classNode.typeParameters.forEach((param, i) => {
+										const typeArg = typeArgs[i];
+										if (typeArg) typeMap.set(param.name.text, this.getTypeNodeToString(typeArg, /* @__PURE__ */ new Map()));
+									});
+									if (typeMap.size > 0) this.transformCallIndex.set(baseExpr.text, typeMap);
+								}
+							}
+						}
+					}
+				}
+				typescript.default.forEachChild(node, visit);
+			};
+			visit(sf);
+		});
+	}
 	transform(cls, sourceOptions) {
-		if (this.classCache.has(cls)) return this.classCache.get(cls);
-		let schema = {
-			type: "object",
-			properties: {}
-		};
 		const result = this.getSourceFileByClass(cls, sourceOptions);
 		if (!result || !(result === null || result === void 0 ? void 0 : result.sourceFile)) {
 			console.warn(`Class ${cls.name} not found in any source file.`);
@@ -958,11 +1053,22 @@ var SchemaTransformer = class SchemaTransformer {
 				}
 			};
 		}
-		const properties = this.getPropertiesByClassDeclaration(result.node);
+		const genericTypeMap = this.transformCallIndex.get(cls.name) ?? /* @__PURE__ */ new Map();
+		const hasGenericArgs = genericTypeMap.size > 0;
+		if (!hasGenericArgs && this.classCache.has(cls)) return this.classCache.get(cls);
+		let schema = {
+			type: "object",
+			properties: {}
+		};
+		const properties = this.getPropertiesByClassDeclaration(result.node, void 0, genericTypeMap);
 		schema = this.getSchemaFromProperties({
 			properties,
 			classDeclaration: result.node
 		});
+		if (!hasGenericArgs) this.classCache.set(cls, {
+			name: cls.name,
+			schema
+		});
 		return {
 			name: cls.name,
 			schema

package/dist/index.mjs

@@ -164,6 +164,7 @@ var SchemaTransformer = class SchemaTransformer {
 	maxCacheSize;
 	autoCleanup;
 	classFileIndex = /* @__PURE__ */ new Map();
+	transformCallIndex = /* @__PURE__ */ new Map();
 	constructor(tsConfigPath = constants.TS_CONFIG_DEFAULT_PATH, options = {}) {
 		this.maxCacheSize = options.maxCacheSize ?? 100;
 		this.autoCleanup = options.autoCleanup ?? true;
@@ -175,19 +176,7 @@ var SchemaTransformer = class SchemaTransformer {
 		const { options: tsOptions, fileNames } = ts.parseJsonConfigFileContent(config, ts.sys, "./");
 		this.program = ts.createProgram(fileNames, tsOptions);
 		this.checker = this.program.getTypeChecker();
-		this.program.getSourceFiles().forEach((sf) => {
-			sf.statements.forEach((stmt) => {
-				if (ts.isClassDeclaration(stmt) && stmt.name) {
-					const name = stmt.name.text;
-					const entry = this.classFileIndex.get(name) || [];
-					entry.push({
-						sourceFile: sf,
-						node: stmt
-					});
-					this.classFileIndex.set(name, entry);
-				}
-			});
-		});
+		this.buildTransformCallIndex();
 	}
 	getPropertiesByClassDeclaration(classNode, visitedDeclarations = /* @__PURE__ */ new Set(), genericTypeMap = /* @__PURE__ */ new Map()) {
 		if (visitedDeclarations.has(classNode)) return [];
@@ -235,6 +224,18 @@ var SchemaTransformer = class SchemaTransformer {
 				const isOptional = !!member.questionToken;
 				const isGeneric = this.isPropertyTypeGeneric(member);
 				const isEnum = this.isEnum(member);
+				const isPrimitive = this.isPrimitiveType(type) || isEnum;
+				const isClassType = this.isClassType(member);
+				const isArray = this.isArrayProperty(member);
+				const isTypeLiteral = this.isTypeLiteral(member);
+				let genericClassReference = void 0;
+				if (isGeneric && !isPrimitive) {
+					const baseTypeName = type.replace(/\[\]$/, "").trim();
+					if (!this.isPrimitiveType(baseTypeName)) {
+						const matches = this.classFileIndex.get(baseTypeName);
+						if (matches && matches.length > 0 && matches[0]) genericClassReference = matches[0].node;
+					}
+				}
 				const property = {
 					name: propertyName,
 					type,
@@ -242,12 +243,13 @@ var SchemaTransformer = class SchemaTransformer {
 					isOptional,
 					isGeneric,
 					originalProperty: member,
-					isPrimitive: this.isPrimitiveType(type) || isEnum,
-					isClassType: this.isClassType(member),
-					isArray: this.isArrayProperty(member),
+					isPrimitive,
+					isClassType,
+					isArray,
 					isEnum,
 					isRef: false,
-					isTypeLiteral: this.isTypeLiteral(member)
+					isTypeLiteral,
+					genericClassReference
 				};
 				if (property.isClassType) {
 					const declaration = this.getDeclarationProperty(property);
@@ -285,6 +287,7 @@ var SchemaTransformer = class SchemaTransformer {
 			if (typeName.toLowerCase() === "uploadfile") return "UploadFile";
 			if (typeName.toLowerCase() === "uploadfiledto") return "UploadFileDto";
 			if (typeNode.typeArguments && typeNode.typeArguments.length > 0) {
+				if (typeName === "Array") return `${this.getTypeNodeToString(typeNode.typeArguments[0], genericTypeMap)}[]`;
 				const firstTypeArg = typeNode.typeArguments[0];
 				if (firstTypeArg && ts.isTypeReferenceNode(firstTypeArg) && ts.isIdentifier(firstTypeArg.typeName)) {
 					if (firstTypeArg.typeName.text.toLowerCase() === "uploadfile") return "UploadFile";
@@ -307,6 +310,11 @@ var SchemaTransformer = class SchemaTransformer {
 				if (types.length > 0 && types[0]) return types[0];
 				return "object";
 			default:
+				if (ts.isIndexedAccessTypeNode(typeNode)) {
+					const resolvedType = this.checker.getTypeAtLocation(typeNode);
+					const resolved = this.checker.typeToString(resolvedType);
+					if (this.isPrimitiveType(resolved)) return resolved;
+				}
 				const typeText = typeNode.getText();
 				if (genericTypeMap && genericTypeMap.has(typeText)) return genericTypeMap.get(typeText);
 				if (typeText.startsWith("Date")) return constants.jsPrimitives.Date.type;
@@ -381,7 +389,7 @@ var SchemaTransformer = class SchemaTransformer {
 	isGenericTypeFromSymbol(type) {
 		if (this.isSimpleArrayType(type)) return false;
 		if (type.aliasTypeArguments && type.aliasTypeArguments.length > 0) return true;
-		if (type.typeArguments && type.typeArguments.length > 0) {
+		if (type.typeArguments && type.typeArguments.length > 0 && type.typeArguments[0].symbol.getName() === "Array") {
 			const symbol = type.getSymbol();
 			if (symbol && symbol.getName() === "Array") {
 				const elementType = type.typeArguments[0];
@@ -426,6 +434,7 @@ var SchemaTransformer = class SchemaTransformer {
 			if (!elementType) return false;
 			if (this.isUtilityTypeFromType(elementType)) return false;
 			if (elementType.typeArguments && elementType.typeArguments.length > 0) return false;
+			if (type.typeArguments && type.typeArguments[0].symbol && type.typeArguments[0].symbol.getName() !== "Array") return false;
 			return true;
 		}
 		return false;
@@ -483,22 +492,6 @@ var SchemaTransformer = class SchemaTransformer {
 		}
 		return matches[0];
 	}
-	checkTypeMatch(value, typeNode) {
-		const runtimeType = typeof value;
-		if (runtimeType === "string" && typeNode.kind === ts.SyntaxKind.StringKeyword) return true;
-		if (runtimeType === "number" && typeNode.kind === ts.SyntaxKind.NumberKeyword) return true;
-		if (runtimeType === "boolean" && typeNode.kind === ts.SyntaxKind.BooleanKeyword) return true;
-		if (Array.isArray(value) && ts.isArrayTypeNode(typeNode)) {
-			if (value.length === 0) return true;
-			const firstItem = value[0];
-			const elementType = typeNode.elementType;
-			return this.checkTypeMatch(firstItem, elementType);
-		}
-		if (runtimeType === "object" && value !== null && !Array.isArray(value)) {
-			if (ts.isTypeReferenceNode(typeNode) || typeNode.kind === ts.SyntaxKind.ObjectKeyword) return true;
-		}
-		return false;
-	}
 	findBestMatch(cls, matches) {
 		let instance = {};
 		try {
@@ -540,7 +533,11 @@ var SchemaTransformer = class SchemaTransformer {
 	isClassType(propertyDeclaration) {
 		if (!propertyDeclaration.type) return false;
 		if (this.isArrayProperty(propertyDeclaration)) {
-			const elementType = propertyDeclaration.type.elementType;
+			var _propertyDeclaration$;
+			let elementType;
+			if (ts.isArrayTypeNode(propertyDeclaration.type)) elementType = propertyDeclaration.type.elementType;
+			else if (ts.isTypeReferenceNode(propertyDeclaration.type) && ((_propertyDeclaration$ = propertyDeclaration.type.typeArguments) === null || _propertyDeclaration$ === void 0 ? void 0 : _propertyDeclaration$[0])) elementType = propertyDeclaration.type.typeArguments[0];
+			if (!elementType) return false;
 			if (ts.isTypeReferenceNode(elementType) && elementType.typeArguments && elementType.typeArguments.length > 0) {
 				const firstTypeArg = elementType.typeArguments[0];
 				if (firstTypeArg) {
@@ -601,7 +598,9 @@ var SchemaTransformer = class SchemaTransformer {
 	}
 	isArrayProperty(propertyDeclaration) {
 		if (!propertyDeclaration.type) return false;
-		return ts.isArrayTypeNode(propertyDeclaration.type);
+		if (ts.isArrayTypeNode(propertyDeclaration.type)) return true;
+		if (ts.isTypeReferenceNode(propertyDeclaration.type) && ts.isIdentifier(propertyDeclaration.type.typeName) && propertyDeclaration.type.typeName.text === "Array") return true;
+		return false;
 	}
 	getSchemaFromProperties({ properties, visitedClass, transformedSchema, classDeclaration }) {
 		let schema = {};
@@ -636,6 +635,17 @@ var SchemaTransformer = class SchemaTransformer {
 			visitedClass,
 			transformedSchema
 		});
+		else if (property.isGeneric) if (property.genericClassReference) schema = this.getSchemaFromClass({
+			isArray: property.isArray,
+			visitedClass,
+			transformedSchema,
+			declaration: property.genericClassReference
+		});
+		else schema = {
+			type: "object",
+			properties: {},
+			additionalProperties: true
+		};
 		else schema = {
 			type: "object",
 			properties: {},
@@ -813,6 +823,11 @@ var SchemaTransformer = class SchemaTransformer {
 		if (decorator.arguments.length === 0) return;
 		const arg = decorator.arguments[0];
 		if (arg && typeof arg === "object" && "kind" in arg) {
+			if (ts.isArrayLiteralExpression(arg)) {
+				const values = this.extractValuesFromArrayLiteral(arg);
+				if (values.length > 0) this.applyEnumValues(values, schema);
+				return;
+			}
 			const type = this.checker.getTypeAtLocation(arg);
 			if (type.symbol && type.symbol.exports) {
 				const values = [];
@@ -824,15 +839,46 @@ var SchemaTransformer = class SchemaTransformer {
 					}
 				});
 				if (values.length > 0) {
-					schema.enum = values;
-					const isString = values.every((v) => typeof v === "string");
-					const isNumber = values.every((v) => typeof v === "number");
-					if (isString) schema.type = "string";
-					else if (isNumber) schema.type = "number";
-					else schema.type = "string";
+					this.applyEnumValues(values, schema);
+					return;
 				}
 			}
+			const values = this.extractValuesFromObjectLiteral(type);
+			if (values.length > 0) this.applyEnumValues(values, schema);
+		}
+	}
+	extractValuesFromArrayLiteral(arrayLiteral) {
+		const values = [];
+		for (const element of arrayLiteral.elements) if (ts.isStringLiteral(element)) values.push(element.text);
+		else if (ts.isNumericLiteral(element)) values.push(Number(element.text));
+		else if (ts.isPrefixUnaryExpression(element) && element.operator === ts.SyntaxKind.MinusToken && ts.isNumericLiteral(element.operand)) values.push(-Number(element.operand.text));
+		return values;
+	}
+	extractValuesFromObjectLiteral(type) {
+		const values = [];
+		const properties = type.getProperties();
+		if (!properties || properties.length === 0) return values;
+		for (const prop of properties) {
+			const propType = this.checker.getTypeOfSymbolAtLocation(prop, prop.valueDeclaration);
+			if (propType.isStringLiteral()) values.push(propType.value);
+			else if (propType.isNumberLiteral()) values.push(propType.value);
+			else if (prop.valueDeclaration && ts.isPropertyAssignment(prop.valueDeclaration)) {
+				const initializer = prop.valueDeclaration.initializer;
+				if (ts.isStringLiteral(initializer)) values.push(initializer.text);
+				else if (ts.isNumericLiteral(initializer)) values.push(Number(initializer.text));
+			}
 		}
+		return values;
+	}
+	applyEnumValues(values, schema) {
+		schema.enum = values;
+		const isString = values.every((v) => typeof v === "string");
+		const isNumber = values.every((v) => typeof v === "number");
+		if (isString) schema.type = "string";
+		else if (isNumber) schema.type = "number";
+		else schema.type = "string";
+		delete schema.properties;
+		delete schema.additionalProperties;
 	}
 	applyDecorators(property, schema) {
 		for (const decorator of property.decorators) switch (decorator.name) {
@@ -910,16 +956,65 @@ var SchemaTransformer = class SchemaTransformer {
 				break;
 			case constants.validatorDecorators.IsEnum.name:
 				if (!property.isArray) this.applyEnumDecorator(decorator, schema);
-				else if (schema.items) this.applyEnumDecorator(decorator, schema.items);
+				else {
+					if (!schema.items) {
+						schema.type = "array";
+						schema.items = {};
+					}
+					this.applyEnumDecorator(decorator, schema.items);
+				}
 				break;
 		}
 	}
+	/**
+	* Scans all non-declaration source files in the program once and records
+	* every call of the form `transform(Foo<Bar, Baz>)`, keyed by class name.
+	* This means generic resolution in transform() is a pure O(1) Map lookup
+	* with no runtime stack inspection.
+	*/
+	buildTransformCallIndex() {
+		this.program.getSourceFiles().forEach((sf) => {
+			if (sf.isDeclarationFile) return;
+			sf.statements.forEach((stmt) => {
+				if (ts.isClassDeclaration(stmt) && stmt.name) {
+					const name = stmt.name.text;
+					const entry = this.classFileIndex.get(name) || [];
+					entry.push({
+						sourceFile: sf,
+						node: stmt
+					});
+					this.classFileIndex.set(name, entry);
+				}
+			});
+			const visit = (node) => {
+				if (ts.isCallExpression(node) && node.arguments.length > 0) {
+					const callee = node.expression;
+					if (ts.isIdentifier(callee) && callee.text === "transform" || ts.isPropertyAccessExpression(callee) && callee.name.text === "transform") {
+						const firstArg = node.arguments[0];
+						const typeArgs = firstArg.typeArguments;
+						if (typeArgs && typeArgs.length > 0) {
+							const baseExpr = firstArg.expression ?? firstArg;
+							if (ts.isIdentifier(baseExpr)) {
+								var _this$classFileIndex$;
+								const classNode = (_this$classFileIndex$ = this.classFileIndex.get(baseExpr.text)) === null || _this$classFileIndex$ === void 0 || (_this$classFileIndex$ = _this$classFileIndex$[0]) === null || _this$classFileIndex$ === void 0 ? void 0 : _this$classFileIndex$.node;
+								if (classNode === null || classNode === void 0 ? void 0 : classNode.typeParameters) {
+									const typeMap = /* @__PURE__ */ new Map();
+									classNode.typeParameters.forEach((param, i) => {
+										const typeArg = typeArgs[i];
+										if (typeArg) typeMap.set(param.name.text, this.getTypeNodeToString(typeArg, /* @__PURE__ */ new Map()));
+									});
+									if (typeMap.size > 0) this.transformCallIndex.set(baseExpr.text, typeMap);
+								}
+							}
+						}
+					}
+				}
+				ts.forEachChild(node, visit);
+			};
+			visit(sf);
+		});
+	}
 	transform(cls, sourceOptions) {
-		if (this.classCache.has(cls)) return this.classCache.get(cls);
-		let schema = {
-			type: "object",
-			properties: {}
-		};
 		const result = this.getSourceFileByClass(cls, sourceOptions);
 		if (!result || !(result === null || result === void 0 ? void 0 : result.sourceFile)) {
 			console.warn(`Class ${cls.name} not found in any source file.`);
@@ -933,11 +1028,22 @@ var SchemaTransformer = class SchemaTransformer {
 				}
 			};
 		}
-		const properties = this.getPropertiesByClassDeclaration(result.node);
+		const genericTypeMap = this.transformCallIndex.get(cls.name) ?? /* @__PURE__ */ new Map();
+		const hasGenericArgs = genericTypeMap.size > 0;
+		if (!hasGenericArgs && this.classCache.has(cls)) return this.classCache.get(cls);
+		let schema = {
+			type: "object",
+			properties: {}
+		};
+		const properties = this.getPropertiesByClassDeclaration(result.node, void 0, genericTypeMap);
 		schema = this.getSchemaFromProperties({
 			properties,
 			classDeclaration: result.node
 		});
+		if (!hasGenericArgs) this.classCache.set(cls, {
+			name: cls.name,
+			schema
+		});
 		return {
 			name: cls.name,
 			schema

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ts-class-to-openapi",
-  "version": "1.3.4",
+  "version": "1.4.1",
   "description": "Transform TypeScript classes into OpenAPI 3.1.0 schema objects, which support class-validator decorators",
   "type": "module",
   "main": "./dist/index.cjs",
@@ -81,9 +81,9 @@
     "test": "node --import tsx --test test/testCases/**/*.test.ts",
     "test:watch": "node --import tsx --inspect --test --watch test/testCases/**/*.test.ts",
     "test:coverage": "node --import tsx --test --experimental-test-coverage test/testCases/**/*.test.ts",
-    "build": "tsc --noEmit && rm -rf dist && tsdown",
+    "build": "tsc --noEmit && tsdown",
     "build:watch": "rm -rf dist && tsdown --watch",
-    "dev": "node --import tsx --test --inspect --watch ./src/run.ts",
+    "dev": "node --import tsx --inspect --watch ./src/run.ts",
     "format": "prettier --write .",
     "format:check": "prettier --check .",
     "prepublish": "pnpm build"