Optimize a few things further.

Signed-off-by: AraHaan <[email protected]>
Elskom · Feb 23, 2024 · 6c156db · 6c156db
1 parent 23d615b
commit 6c156db
Showing 1 changed file with 12 additions and 24 deletions.
diff --git a/src/IDisposableGenerator/DisposableCodeWriter.cs b/src/IDisposableGenerator/DisposableCodeWriter.cs
@@ -24,16 +24,12 @@ Implements IDisposable
         Private isDisposed As Boolean
 ");
 
-                if (classItem.Owns.Count is not 0 && !classItem.Stream)
-                {
-                    _ = sourceBuilder.Append(@"
-        Friend Property IsOwned As Boolean
-");
-                }
-                else if (classItem.Owns.Count is not 0 && classItem.Stream)
+                if (classItem.Owns.Count is not 0)
                 {
-                    _ = sourceBuilder.Append(@"
+                    _ = sourceBuilder.Append(classItem.Stream ? @"
         Friend ReadOnly Property KeepOpen As Boolean
+" : @"
+        Friend Property IsOwned As Boolean
 ");
                 }
 
@@ -135,16 +131,12 @@ namespace {workItem.Namespace};
 {{
     private bool isDisposed;
 ");
-                if (classItem.Owns.Count is not 0 && !classItem.Stream)
-                {
-                    _ = sourceBuilder.Append(@"
-    internal bool IsOwned { get; set; }
-");
-                }
-                else if (classItem.Owns.Count is not 0 && classItem.Stream)
+                if (classItem.Owns.Count is not 0)
                 {
-                    _ = sourceBuilder.Append(@"
+                    _ = sourceBuilder.Append(classItem.Stream ? @"
     internal bool KeepOpen { get; }
+" : @"
+    internal bool IsOwned { get; set; }
 ");
                 }
 
@@ -249,16 +241,12 @@ namespace {workItem.Namespace}
     {{
         private bool isDisposed;
 ");
-                if (classItem.Owns.Count is not 0 && !classItem.Stream)
-                {
-                    _ = sourceBuilder.Append(@"
-        internal bool IsOwned { get; set; }
-");
-                }
-                else if (classItem.Owns.Count is not 0 && classItem.Stream)
+                if (classItem.Owns.Count is not 0)
                 {
-                    _ = sourceBuilder.Append(@"
+                    _ = sourceBuilder.Append(classItem.Stream ? @"
         internal bool KeepOpen { get; }
+" : @"
+        internal bool IsOwned { get; set; }
 ");
                 }