| | | 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 | | { |
| | 0 | 17 | | var assemblies = GetRelevantAssemblies(); |
| | 0 | 18 | | return new OpenApiComponentSet |
| | 0 | 19 | | { |
| | 0 | 20 | | ParameterTypes = GetTypesWithAttribute(assemblies, typeof(OpenApiParameterComponent)), |
| | 0 | 21 | | SchemaTypes = GetSchemaTypes(assemblies), |
| | 0 | 22 | | ResponseTypes = GetTypesWithAttribute(assemblies, typeof(OpenApiResponseComponent)), |
| | 0 | 23 | | HeaderTypes = GetTypesWithAttribute(assemblies, typeof(OpenApiHeaderComponent)), |
| | 0 | 24 | | ExampleTypes = GetTypesWithAttribute(assemblies, typeof(OpenApiExampleComponent)), |
| | 0 | 25 | | RequestBodyTypes = GetTypesWithAttribute(assemblies, typeof(OpenApiRequestBodyComponent)), |
| | 0 | 26 | | #if EXTENDED_OPENAPI |
| | 0 | 27 | | SecuritySchemeTypes = GetTypesWithAttribute(assemblies, typeof(OpenApiSecuritySchemeComponent)), |
| | 0 | 28 | | LinkTypes = GetTypesWithAttribute(assemblies, typeof(OpenApiLinkComponent)), |
| | 0 | 29 | | CallbackTypes = GetTypesWithKind(assemblies, OpenApiModelKind.Callback), |
| | 0 | 30 | | PathItemTypes = GetTypesWithKind(assemblies, OpenApiModelKind.PathItem) |
| | 0 | 31 | | #endif |
| | 0 | 32 | | }; |
| | | 33 | | } |
| | | 34 | | |
| | | 35 | | private static System.Reflection.Assembly[] GetRelevantAssemblies() |
| | | 36 | | { |
| | 0 | 37 | | return [.. AppDomain.CurrentDomain.GetAssemblies() |
| | 0 | 38 | | .Where(a => a.FullName is not null && |
| | 0 | 39 | | (a.FullName.Contains("PowerShell Class Assembly") || |
| | 0 | 40 | | a.FullName.StartsWith("Kestrun")))]; |
| | | 41 | | } |
| | | 42 | | |
| | | 43 | | private static Type[] GetTypesWithAttribute(System.Reflection.Assembly[] assemblies, Type attributeType) |
| | | 44 | | { |
| | 0 | 45 | | return [.. assemblies.SelectMany(asm => asm.GetTypes()) |
| | 0 | 46 | | .Where(t => t.IsClass && !t.IsAbstract && t.IsDefined(attributeType, true))]; |
| | | 47 | | } |
| | | 48 | | |
| | | 49 | | private static Type[] GetSchemaTypes(System.Reflection.Assembly[] assemblies) |
| | | 50 | | { |
| | 0 | 51 | | return [.. assemblies.SelectMany(asm => asm.GetTypes()) |
| | 0 | 52 | | .Where(t => t.IsClass && !t.IsAbstract && |
| | 0 | 53 | | !typeof(IOpenApiType).IsAssignableFrom(t) && |
| | 0 | 54 | | t.IsDefined(typeof(OpenApiSchemaComponent), true))]; |
| | | 55 | | } |
| | | 56 | | |
| | | 57 | | #if EXTENDED_OPENAPI |
| | | 58 | | private static Type[] GetTypesWithKind(System.Reflection.Assembly[] assemblies, OpenApiModelKind kind) |
| | | 59 | | { |
| | | 60 | | return [.. assemblies.SelectMany(asm => asm.GetTypes()) |
| | | 61 | | .Where(t => t.IsClass && !t.IsAbstract && |
| | | 62 | | t.GetCustomAttributes(typeof(OpenApiModelKindAttribute), true) |
| | | 63 | | .OfType<OpenApiModelKindAttribute>() |
| | | 64 | | .Any(a => a.Kind == kind))]; |
| | | 65 | | } |
| | | 66 | | #endif |
| | | 67 | | } |