scala.meta.internal.prettyprinters.Enquote.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of prettyprinters_2.11 Show documentation

Scala.meta's baseline prettyprinters

The newest version!

package scala.meta
package internal
package prettyprinters

sealed trait QuoteStyle
case object SingleQuotes extends QuoteStyle { override def toString = "'" }
case object DoubleQuotes extends QuoteStyle { override def toString = "\"" }
case object TripleQuotes extends QuoteStyle { override def toString = "\"\"\"" }

object enquote {
  def apply(s: String, style: QuoteStyle): String = {
    // TODO: comprehensive handling (e.g. escape triple quotes in triple quotes)
    val codepage = Map(
      "\"" -> (if (style == DoubleQuotes) "\\\"" else "\""),
      "\'" -> (if (style == SingleQuotes) "\\\'" else "\'")
    )
    val escaped = if (style != TripleQuotes) escape(s).flatMap(c => codepage.getOrElse(c.toString, c.toString)) else s
    style.toString + escaped + style.toString
  }
}