custom collection converters

7mind · Jan 9, 2025 · 6995554 · 6995554
1 parent d691292
commit 6995554
Show file tree

Hide file tree

Showing 2 changed files with 27 additions and 2 deletions.
diff --git a/src/main/resources/baboon-runtime/cs/BaboonRuntimeShared.cs b/src/main/resources/baboon-runtime/cs/BaboonRuntimeShared.cs
@@ -19,6 +19,31 @@
 // ReSharper disable UnusedType.Global
 
 namespace Baboon.Runtime.Shared {
+    public static class BaboonEnumerable
+    {
+        public static List<TSource> BbnToList<TSource>(this IEnumerable<TSource> source)
+        {
+            if (source is List<TSource> l)
+            {
+                return l;
+            }
+
+            return source.ToList();
+        }
+
+        public static Dictionary<TKey, TElement> BbnToDictionary<TKey, TElement>(
+            this IEnumerable<KeyValuePair<TKey, TElement>> source)
+            where TKey : notnull
+        {
+            if (source is Dictionary<TKey, TElement> d)
+            {
+                return d;
+            }
+
+            return new Dictionary<TKey, TElement>(source);
+        }
+    }
+
     public interface IBaboonGenerated {
         public string BaboonDomainVersion();
         public string BaboonDomainIdentifier();

diff --git a/src/main/scala/io/septimalmind/baboon/translator/csharp/CSTypes.scala b/src/main/scala/io/septimalmind/baboon/translator/csharp/CSTypes.scala
@@ -136,7 +136,7 @@ object CSTypes {
 
   val debug: CSType = CSType(csDiagnosticsPkg, "Debug", fq = false)
 
-  val mkDict = q"ToDictionary(v => v.Key, v => v.Value)"
-  val mkList = q"ToList()"
+  val mkDict = q"BbnToDictionary()"
+  val mkList = q"BbnToList()"
   val mkSet  = q"ToImmutableHashSet()"
 }