feat(output) add using to output for libs

nuetzliches · May 17, 2022 · ba60640 · ba60640
1 parent 23759a9
commit ba60640
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/src/Generator.cs b/src/Generator.cs
@@ -111,7 +111,6 @@ public SourceText GetInputTextForStoredProcedure(Definition.Schema schema, Defin
             var nsNode = (NamespaceDeclarationSyntax)root.Members[0];
             var classNode = (ClassDeclarationSyntax)nsNode.Members[0];
 
-
             // Create obsolete constructor
             var obsoleteContructor = classNode.CreateConstructor($"{storedProcedure.Name}Input");
             root = root.AddObsoleteAttribute(ref obsoleteContructor, "This empty contructor will be removed in vNext. Please use constructor with parameters.");
@@ -240,6 +239,13 @@ public SourceText GetOutputTextForStoredProcedure(Definition.Schema schema, Defi
             var tree = CSharpSyntaxTree.ParseText(fileContent);
             var root = tree.GetCompilationUnitRoot();
 
+            // Add Usings
+            if (_configFile.Config.Project.Role.Kind == ERoleKind.Lib)
+            {
+                var outputUsingDirective = SyntaxFactory.UsingDirective(SyntaxFactory.ParseName($"{_configFile.Config.Project.Output.Namespace}.Outputs"));
+                root = root.AddUsings(outputUsingDirective.NormalizeWhitespace().WithLeadingTrivia(SyntaxFactory.CarriageReturnLineFeed)).WithTrailingTrivia(SyntaxFactory.CarriageReturnLineFeed);
+            }
+
             // Replace Namespace
             if (_configFile.Config.Project.Role.Kind == ERoleKind.Lib)
             {

diff --git a/src/SpocR.csproj b/src/SpocR.csproj
@@ -1,7 +1,7 @@
 <Project Sdk="Microsoft.NET.Sdk">
 
   <PropertyGroup Label="Build">
-    <Version>3.0.2</Version>    
+    <Version>3.0.3</Version>    
     <TargetFramework>net5.0</TargetFramework>
   </PropertyGroup>