Dependent tuples in Scala 3 / Dotty

308 views Asked by At

I'm trying to encode a dependent map using a list of dependent tuples. Here is what I have that does not work:

  class DTuple[Key, ValueMap[_ <: Key]](val first: Key)(val second: ValueMap[first.type])
  
  type DKey = "Tag" | "Versions" | "Author"

  type DMapping[X <: DKey] = X match {
      case "Tag" => String
      case "Versions" => Array[String]
      case "Author" => String
    }
  
  def mkString(d: DTuple[DKey, DMapping]) = d.first match {
    case _: "Tag" => "#" + d.second
    case _: "Versions" => d.second.mkString(",")
    case _: "Author" => "@" + d.second
  }

All I get is

[error] -- [E008] Not Found Error: Main.scala:21:35
[error] 21 |    case _: "Versions" => d.second.mkString(",")
[error]    |                          ^^^^^^^^^^^^^^^^^
[error]    |      value mkString is not a member of Main.DMapping[(d.first : Main.DKey)]

I can't think of a good way to pattern match d.second so that its type depends on d.first. I could add .asInstanceOf[Array[String]] and .asInstanceOf[String] but that is not the goal here, I'm trying to type-check code.

1

There are 1 answers

2
Marth On

There's probably a better and easier way to do this, but:

import java.time.ZonedDateTime
import java.time.format.DateTimeFormatter.ISO_ZONED_DATE_TIME

// This is a case case solely for the unapply method, you could implement it on your own
case class DTuple[Key, ValueMap[_ <: Key]](first: Key)(val second: ValueMap[first.type])

type DKey = "Tag" | "Versions" | "Author" | "BuildTime"

type DMapping[X <: DKey] = X match {
  case "Tag" => String
  case "Versions" => Array[String]
  case "Author" => String
  case "BuildTime" => ZonedDateTime
}

// the DTuple("<value>") is used at runtime to check the string (DKey) value
// the DTuple["<value>", DMapping] type hint makes dotty see the `d` value as the correct type, hence infering the type of d.second too
def mkString(dt: DTuple[DKey, DMapping]): String = dt match {
  // this would fail at runtime as e.g DTuple("Tag") would enter this case (the `DTuple["BuildTime", DMapping]` is unchecked at runtime
  // case d: DTuple["BuildTime", DMapping] => d.second.format(ISO_ZONED_DATE_TIME)

  // this doesn't compile because `d.second`'s type is still 'DMapping[(d.first : DKey)]', not 'DMapping["BuildTime"]'
  // case d@DTuple("BuildTime") => d.second.format(ISO_ZONED_DATE_TIME)

  case d@DTuple("Tag"): DTuple["Tag", DMapping] => d.second
  case d@DTuple("Versions"): DTuple["Versions", DMapping] => d.second.mkString(", ")
  case d@DTuple("Author"): DTuple["Author", DMapping] => d.second.toString
  case d@DTuple("BuildTime"): DTuple["BuildTime", DMapping] => d.second.format(ISO_ZONED_DATE_TIME)
}

object Main extends App {
  List(
    DTuple[DKey, DMapping]("Versions")(Array("1.0", "2.0")),
    DTuple[DKey, DMapping]("Tag")("env=SO"),
    DTuple[DKey, DMapping]("Author")("MK"),
    DTuple[DKey, DMapping]("BuildTime")(ZonedDateTime.now())
  ).foreach { dt =>
    println(mkString(dt))
  }
}

prints

1.0, 2.0
env=SO
MK
2020-10-23T21:04:06.696+02:00[Europe/Paris]