Added CSDL filter for entitysets and singletons

Author: darrelmiller

src/Microsoft.OpenApi.Hidi/CsdlFilter.xslt

@@ -0,0 +1,22 @@
+<?xml version="1.0" encoding="utf-8"?>
+<xsl:stylesheet version="1.0" xmlns:xsl="http://www.w3.org/1999/XSL/Transform"
+                xmlns:edm="http://docs.oasis-open.org/odata/ns/edm"
+    xmlns:msxsl="urn:schemas-microsoft-com:xslt" exclude-result-prefixes="msxsl"
+>
+    <xsl:output method="xml" indent="yes"/>
+
+    <xsl:param name="entitySetOrSingleton" select="'serviceappointments'"></xsl:param>
+    
+    <xsl:template match="edm:EntityContainer">
+        <xsl:copy>
+            <xsl:apply-templates select="@* | edm:EntitySet[contains($entitySetOrSingleton,@Name)] | edm:Singleton[contains($entitySetOrSingleton,@Name)]"/>
+        </xsl:copy>
+    </xsl:template>
+
+    <xsl:template match="@* | node()">
+        <xsl:copy>
+            <xsl:apply-templates select="@* | node()"/>
+        </xsl:copy>
+    </xsl:template>
+
+</xsl:stylesheet>

src/Microsoft.OpenApi.Hidi/Microsoft.OpenApi.Hidi.csproj

@@ -31,6 +31,14 @@
     <EmbedUntrackedSources>true</EmbedUntrackedSources>
   </PropertyGroup>
 
+  <ItemGroup>
+    <None Remove="CsdlFilter.xslt" />
+  </ItemGroup>
+
+  <ItemGroup>
+    <EmbeddedResource Include="CsdlFilter.xslt" />
+  </ItemGroup>
+
   <ItemGroup>
     <PackageReference Include="Microsoft.Extensions.Logging" Version="6.0.0" />
     <PackageReference Include="Microsoft.Extensions.Logging.Abstractions" Version="6.0.1" />

src/Microsoft.OpenApi.Hidi/OpenApiService.cs

@@ -24,6 +24,10 @@
 using Microsoft.OpenApi.Writers;
 using static Microsoft.OpenApi.Hidi.OpenApiSpecVersionHelper;
 using System.Threading;
+using System.Xml.Xsl;
+using System.Xml;
+using System.Runtime.CompilerServices;
+using System.Reflection;
 
 namespace Microsoft.OpenApi.Hidi
 {
@@ -35,6 +39,7 @@ public class OpenApiService
         public static async Task<int> TransformOpenApiDocument(
             string openapi,
             string csdl,
+            string csdlFilter,
             FileInfo output,
             bool cleanoutput,
             string? version,
@@ -85,6 +90,13 @@ CancellationToken cancellationToken
                         openApiVersion = version != null ? TryParseOpenApiSpecVersion(version) : OpenApiSpecVersion.OpenApi3_0;
 
                         stream = await GetStream(csdl, logger, cancellationToken);
+
+                        if (!string.IsNullOrEmpty(csdlFilter))
+                        {
+                            XslCompiledTransform transform = GetFilterTransform();
+                            stream = ApplyFilter(csdl, csdlFilter, transform);
+                            stream.Position = 0;
+                        }
                         document = await ConvertCsdlToOpenApi(stream);
                         stopwatch.Stop();
                         logger.LogTrace("{timestamp}ms: Generated OpenAPI with {paths} paths.", stopwatch.ElapsedMilliseconds, document.Paths.Count);
@@ -210,6 +222,31 @@ CancellationToken cancellationToken
             }            
         }
 
+        private static XslCompiledTransform GetFilterTransform()
+        {
+            XslCompiledTransform transform = new();
+            Assembly assembly = typeof(OpenApiService).GetTypeInfo().Assembly;
+            Stream xslt = assembly.GetManifestResourceStream("Microsoft.OpenApi.Hidi.CsdlFilter.xslt");
+            transform.Load(new XmlTextReader(new StreamReader(xslt)));
+            return transform;
+        }
+
+        private static Stream ApplyFilter(string csdl, string entitySetOrSingleton, XslCompiledTransform transform)
+        {
+            Stream stream;
+            StreamReader inputReader = new(csdl);
+            XmlReader inputXmlReader = XmlReader.Create(inputReader);
+            MemoryStream filteredStream = new();
+            StreamWriter writer = new(filteredStream);
+            XsltArgumentList args = new();
+            args.AddParam("entitySetOrSingleton", "", entitySetOrSingleton);
+            transform.Transform(inputXmlReader, args, writer);
+            stream = filteredStream;
+            return stream;
+        }
+
+
+
         /// <summary>
         /// Implementation of the validate command
         /// </summary>
@@ -306,8 +343,8 @@ public static async Task<OpenApiDocument> ConvertCsdlToOpenApi(Stream csdl)
                 EnableDiscriminatorValue = false,
                 EnableDerivedTypesReferencesForRequestBody = false,
                 EnableDerivedTypesReferencesForResponses = false,
-                ShowRootPath = true,
-                ShowLinks = true
+                ShowRootPath = false,
+                ShowLinks = false
             };
             OpenApiDocument document = edmModel.ConvertToOpenApi(settings);
 

src/Microsoft.OpenApi.Hidi/Program.cs

@@ -24,6 +24,9 @@ static async Task Main(string[] args)
             var csdlOption = new Option<string>("--csdl", "Input CSDL file path or URL");
             csdlOption.AddAlias("-cs");
 
+            var csdlFilterOption = new Option<string>("--csdlFilter", "Name of EntitySet or Singleton to filter CSDL on");
+            csdlOption.AddAlias("-csf");
+
             var outputOption = new Option<FileInfo>("--output", () => new FileInfo("./output"), "The output directory path for the generated file.") { Arity = ArgumentArity.ZeroOrOne };
             outputOption.AddAlias("-o");
 
@@ -66,6 +69,7 @@ static async Task Main(string[] args)
             {
                 descriptionOption,
                 csdlOption,
+                csdlFilterOption,
                 outputOption,
                 cleanOutputOption,
                 versionOption,
@@ -78,8 +82,8 @@ static async Task Main(string[] args)
                 resolveExternalOption,
             };
 
-            transformCommand.SetHandler<string, string, FileInfo, bool, string?, OpenApiFormat?, LogLevel, bool, bool, string, string, string, CancellationToken> (
-                OpenApiService.TransformOpenApiDocument, descriptionOption, csdlOption, outputOption, cleanOutputOption, versionOption, formatOption, logLevelOption, inlineOption, resolveExternalOption, filterByOperationIdsOption, filterByTagsOption, filterByCollectionOption);
+            transformCommand.SetHandler<string, string, string, FileInfo, bool, string?, OpenApiFormat?, LogLevel, bool, bool, string, string, string, CancellationToken> (
+                OpenApiService.TransformOpenApiDocument, descriptionOption, csdlOption, csdlFilterOption, outputOption, cleanOutputOption, versionOption, formatOption, logLevelOption, inlineOption, resolveExternalOption, filterByOperationIdsOption, filterByTagsOption, filterByCollectionOption);
 
             rootCommand.Add(transformCommand);
             rootCommand.Add(validateCommand);

test/Microsoft.OpenApi.Tests/Services/OpenApiServiceTests.cs

@@ -22,7 +22,7 @@ public async Task ReturnConvertedCSDLFile()
 
             // Act
             var openApiDoc = await OpenApiService.ConvertCsdlToOpenApi(csdlStream);
-            var expectedPathCount = 6;
+            var expectedPathCount = 5;
 
             // Assert
             Assert.NotNull(openApiDoc);