From 8a7a557448ba1b79f6642d3c9734d6c23c27dee9 Mon Sep 17 00:00:00 2001
From: AraHaan <seandhunt_7@yahoo.com>
Date: Fri, 23 Feb 2024 14:33:00 -0500
Subject: [PATCH] Optimize a few things further.

---
 .../DisposableCodeWriter.cs                   | 36 +++++++------------
 1 file changed, 12 insertions(+), 24 deletions(-)

diff --git a/src/IDisposableGenerator/DisposableCodeWriter.cs b/src/IDisposableGenerator/DisposableCodeWriter.cs
index a424d69..0f2a723 100644
--- 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; }
 ");
                 }