| | | 1 | | // OpenApiSchemaDiscovery.cs |
| | | 2 | | namespace Kestrun.OpenApi; |
| | | 3 | | |
| | | 4 | | /// <summary> |
| | | 5 | | /// Helper to discover OpenAPI schema types in loaded assemblies. |
| | | 6 | | /// </summary> |
| | | 7 | | public static class OpenApiSchemaDiscovery |
| | | 8 | | { |
| | | 9 | | /// <summary> |
| | | 10 | | /// Discover all OpenAPI component types from dynamic PowerShell classes. |
| | | 11 | | /// Scans all loaded assemblies for classes decorated with OpenApiModelKindAttribute |
| | | 12 | | /// or having any OpenApi* attributes. |
| | | 13 | | /// </summary> |
| | | 14 | | /// <returns> The discovered OpenAPI component types.</returns> |
| | | 15 | | public static OpenApiComponentSet GetOpenApiTypesAuto() |
| | | 16 | | { |
| | 3 | 17 | | var assemblies = GetRelevantAssemblies(); |
| | 3 | 18 | | return new OpenApiComponentSet |
| | 3 | 19 | | { |
| | 3 | 20 | | SchemaTypes = GetSchemaTypes(assemblies), |
| | 3 | 21 | | #if EXTENDED_OPENAPI |
| | 3 | 22 | | RequestBodyTypes = GetTypesWithAttribute(assemblies, typeof(OpenApiRequestBodyComponentAttribute)), |
| | 3 | 23 | | ResponseTypes = GetTypesWithAttribute(assemblies, typeof(OpenApiResponseComponentAttribute)), |
| | 3 | 24 | | ParameterTypes = GetTypesWithAttribute(assemblies, typeof(OpenApiParameterComponentAttribute)), |
| | 3 | 25 | | HeaderTypes = GetTypesWithAttribute(assemblies, typeof(OpenApiHeaderAttribute)), |
| | 3 | 26 | | SecuritySchemeTypes = GetTypesWithAttribute(assemblies, typeof(OpenApiSecuritySchemeComponent)), |
| | 3 | 27 | | CallbackTypes = GetTypesWithKind(assemblies, OpenApiModelKind.Callback), |
| | 3 | 28 | | PathItemTypes = GetTypesWithKind(assemblies, OpenApiModelKind.PathItem) |
| | 3 | 29 | | #endif |
| | 3 | 30 | | }; |
| | | 31 | | } |
| | | 32 | | |
| | | 33 | | private static System.Reflection.Assembly[] GetRelevantAssemblies() |
| | | 34 | | { |
| | 3 | 35 | | return [.. AppDomain.CurrentDomain.GetAssemblies() |
| | 972 | 36 | | .Where(a => a.FullName is not null && |
| | 972 | 37 | | (a.FullName.Contains("PowerShell Class Assembly") || |
| | 972 | 38 | | a.FullName.StartsWith("Kestrun")))]; |
| | | 39 | | } |
| | | 40 | | |
| | | 41 | | private static Type[] GetSchemaTypes(System.Reflection.Assembly[] assemblies) |
| | | 42 | | { |
| | 3 | 43 | | var primitivesAssembly = typeof(OpenApiString).Assembly; |
| | | 44 | | |
| | 12 | 45 | | return [.. assemblies.SelectMany(asm => asm.GetTypes()) |
| | 5529 | 46 | | .Where(t => t.IsClass && !t.IsAbstract && |
| | 5529 | 47 | | t.IsDefined(typeof(OpenApiSchemaComponent), true) && |
| | 5529 | 48 | | // Exclude built-in OpenApi* primitives from auto-discovered components, |
| | 5529 | 49 | | // but keep user-defined schema components that inherit those primitives. |
| | 5529 | 50 | | !(t.Assembly == primitivesAssembly && typeof(IOpenApiScalar).IsAssignableFrom(t)))]; |
| | | 51 | | } |
| | | 52 | | |
| | | 53 | | #if EXTENDED_OPENAPI |
| | | 54 | | |
| | | 55 | | private static Type[] GetTypesWithAttribute(System.Reflection.Assembly[] assemblies, Type attributeType) |
| | | 56 | | { |
| | | 57 | | return [.. assemblies.SelectMany(asm => asm.GetTypes()) |
| | | 58 | | .Where(t => t.IsClass && !t.IsAbstract && t.IsDefined(attributeType, true))]; |
| | | 59 | | } |
| | | 60 | | private static Type[] GetTypesWithKind(System.Reflection.Assembly[] assemblies, OpenApiModelKind kind) |
| | | 61 | | { |
| | | 62 | | return [.. assemblies.SelectMany(asm => asm.GetTypes()) |
| | | 63 | | .Where(t => t.IsClass && !t.IsAbstract && |
| | | 64 | | t.GetCustomAttributes(typeof(OpenApiModelKindAttribute), true) |
| | | 65 | | .OfType<OpenApiModelKindAttribute>() |
| | | 66 | | .Any(a => a.Kind == kind))]; |
| | | 67 | | } |
| | | 68 | | #endif |
| | | 69 | | } |