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