nits

LumexUI · Feb 19, 2025 · 80945f2 · 80945f2
1 parent 420d921
commit 80945f2
Show file tree

Hide file tree

Showing 3 changed files with 42 additions and 21 deletions.
diff --git a/src/LumexUI.Variants/SlotCollection.cs b/src/LumexUI.Variants/SlotCollection.cs
@@ -6,20 +6,27 @@
 
 namespace LumexUI.Variants;
 
-public class SlotCollection : IEnumerable<KeyValuePair<string, string>>
+public class SlotCollection : IEnumerable<KeyValuePair<string, string?>>
 {
-	private readonly Dictionary<string, string> _dictionary = [];
+	private readonly Dictionary<string, string?> _dictionary = [];
 
-	public string this[string slotName]
+	public string? this[string slotName]
 	{
-		get => _dictionary[slotName];
-		set => _dictionary[slotName] = value;
+		get
+		{
+			_dictionary.TryGetValue( slotName, out var classes );
+			return classes;
+		}
+		set
+		{
+			_dictionary[slotName] = value;
+		}
 	}
 
 	public IEnumerable<string> Slots => _dictionary.Keys;
-	public IEnumerable<string> Values => _dictionary.Values;
+	public IEnumerable<string?> Values => _dictionary.Values;
 
-	public IEnumerator<KeyValuePair<string, string>> GetEnumerator()
+	public IEnumerator<KeyValuePair<string, string?>> GetEnumerator()
 		=> _dictionary.GetEnumerator();
 
 	IEnumerator IEnumerable.GetEnumerator()

diff --git a/src/LumexUI.Variants/VariantCollection.cs b/src/LumexUI.Variants/VariantCollection.cs
@@ -6,20 +6,27 @@
 
 namespace LumexUI.Variants;
 
-public class VariantCollection : IEnumerable<KeyValuePair<string, VariantValueCollection>>
+public class VariantCollection : IEnumerable<KeyValuePair<string, VariantValueCollection?>>
 {
-	private readonly Dictionary<string, VariantValueCollection> _dictionary = [];
+	private readonly Dictionary<string, VariantValueCollection?> _dictionary = [];
 
-	public VariantValueCollection this[string variantName]
+	public VariantValueCollection? this[string variantName]
 	{
-		get => _dictionary[variantName];
-		set => _dictionary[variantName] = value;
+		get
+		{
+			_dictionary.TryGetValue( variantName, out var variantValues );
+			return variantValues;
+		}
+		set
+		{
+			_dictionary[variantName] = value;
+		}
 	}
 
 	public IEnumerable<string> Variants => _dictionary.Keys;
-	public IEnumerable<VariantValueCollection> Values => _dictionary.Values;
+	public IEnumerable<VariantValueCollection?> Values => _dictionary.Values;
 
-	public IEnumerator<KeyValuePair<string, VariantValueCollection>> GetEnumerator()
+	public IEnumerator<KeyValuePair<string, VariantValueCollection?>> GetEnumerator()
 		=> _dictionary.GetEnumerator();
 
 	IEnumerator IEnumerable.GetEnumerator()

diff --git a/src/LumexUI.Variants/VariantValueCollection.cs b/src/LumexUI.Variants/VariantValueCollection.cs
@@ -6,20 +6,27 @@
 
 namespace LumexUI.Variants;
 
-public class VariantValueCollection : IEnumerable<KeyValuePair<string, SlotCollection>>
+public class VariantValueCollection : IEnumerable<KeyValuePair<string, SlotCollection?>>
 {
-	private readonly Dictionary<string, SlotCollection> _dictionary = [];
+	private readonly Dictionary<string, SlotCollection?> _dictionary = [];
 
-	public SlotCollection this[string variantValueName]
+	public SlotCollection? this[string variantValueName]
 	{
-		get => _dictionary[variantValueName];
-		set => _dictionary[variantValueName] = value;
+		get
+		{
+			_dictionary.TryGetValue( variantValueName, out var slots );
+			return slots;
+		}
+		set
+		{
+			_dictionary[variantValueName] = value;
+		}
 	}
 
 	public IEnumerable<string> VariantValues => _dictionary.Keys;
-	public IEnumerable<SlotCollection> Values => _dictionary.Values;
+	public IEnumerable<SlotCollection?> Values => _dictionary.Values;
 
-	public IEnumerator<KeyValuePair<string, SlotCollection>> GetEnumerator()
+	public IEnumerator<KeyValuePair<string, SlotCollection?>> GetEnumerator()
 		=> _dictionary.GetEnumerator();
 
 	IEnumerator IEnumerable.GetEnumerator()