Elm: Fix import for Dict (#685)

apicollective · Oct 11, 2023 · ef4ab4c · ef4ab4c
1 parent 4020ae9
commit ef4ab4c
Showing 1 changed file with 19 additions and 3 deletions.
diff --git a/elm-generator/src/main/scala/generator/elm/Imports.scala b/elm-generator/src/main/scala/generator/elm/Imports.scala
@@ -3,8 +3,14 @@ package generator.elm
 import scala.collection.concurrent.TrieMap
 
 case class Imports() {
+  private[this] sealed trait ExposingAllValue
+  private[this] object ExposingAllValue {
+    case object Wildcard extends ExposingAllValue
+    case class Types(types: Seq[String]) extends  ExposingAllValue
+  }
+
   private[this] val allAs: TrieMap[String, String] = TrieMap[String, String]()
-  private[this] val exposingAll: TrieMap[String, Unit] = TrieMap[String, Unit]()
+  private[this] val exposingAll: TrieMap[String, ExposingAllValue] = TrieMap[String, ExposingAllValue]()
 
   def addAs(name: String, as: String): Unit = {
     allAs.put(name, as).foreach { existing =>
@@ -14,7 +20,14 @@ case class Imports() {
   }
 
   def addExposingAll(name: String): Unit = {
-    exposingAll.put(name, ())
+    exposingAll.put(name, ExposingAllValue.Wildcard)
+    ()
+  }
+
+  def addExposing(name: String, types: String): Unit = addExposing(name, Seq(types))
+
+  private[this] def addExposing(name: String, types: Seq[String]): Unit = {
+    exposingAll.put(name, ExposingAllValue.Types(types))
     ()
   }
 
@@ -23,7 +36,10 @@ case class Imports() {
       allAs.keysIterator.toSeq.sorted.map { name =>
        s"import $name as ${allAs(name)}"
       } ++ exposingAll.keysIterator.toSeq.sorted.map { name =>
-        s"import $name exposing (..)"
+        exposingAll(name) match {
+          case ExposingAllValue.Wildcard => s"import $name exposing (..)"
+          case ExposingAllValue.Types(types) => s"import $name exposing (${types.mkString(", ")})"
+        }
       }
     ).mkString("\n")
   }