我发现了一个可运行的解决方案的努力工作几个小时后:
object Macros {
import scala.reflect.macros.Context
import language.experimental.macros
sealed trait Piece
case class Place(str: String) extends Piece
case class Name(str: String) extends Piece
case class Pos(column: Int, line: Int)
case class LocatedPieces(located: List[(String, Piece, Pos)])
implicit class s2pieces(sc: StringContext) {
def s2(pieces: Piece*) = macro s2impl
}
// pieces contain all the Piece instances passed inside of the string interpolation
def s2impl(c: Context)(pieces: c.Expr[Piece]*): c.Expr[LocatedPieces] = {
import c.universe.{ Name => _, _ }
c.prefix.tree match {
// access data of string interpolation
case Apply(_, List(Apply(_, rawParts))) =>
// helper methods
def typeIdent[A : TypeTag] =
Ident(typeTag[A].tpe.typeSymbol)
def companionIdent[A : TypeTag] =
Ident(typeTag[A].tpe.typeSymbol.companionSymbol)
def identFromString(tpt: String) =
Ident(c.mirror.staticModule(tpt))
// We need to translate the data calculated inside of the macro to an AST
// in order to write it back to the compiler.
def toAST(any: Any) =
Literal(Constant(any))
def toPosAST(column: Tree, line: Tree) =
Apply(
Select(companionIdent[Pos], newTermName("apply")),
List(column, line))
def toTupleAST(t1: Tree, t2: Tree, t3: Tree) =
Apply(
TypeApply(
Select(identFromString("scala.Tuple3"), newTermName("apply")),
List(typeIdent[String], typeIdent[Piece], typeIdent[Pos])),
List(t1, t2, t3))
def toLocatedPiecesAST(located: Tree) =
Apply(
Select(companionIdent[LocatedPieces], newTermName("apply")),
List(located))
def toListAST(xs: List[Tree]) =
Apply(
TypeApply(
Select(identFromString("scala.collection.immutable.List"), newTermName("apply")),
List(AppliedTypeTree(
typeIdent[Tuple3[String, Piece, Pos]],
List(typeIdent[String], typeIdent[Piece], typeIdent[Pos])))),
xs)
// `parts` contain the strings a string interpolation is built of
val parts = rawParts map { case Literal(Constant(const: String)) => const }
// translate compiler positions to a data structure that can live outside of the compiler
val positions = pieces.toList map (_.tree.pos) map (p => Pos(p.column, p.line))
// discard last element of parts, `transpose` does not work otherwise
// trim parts to discard unnecessary white space
val data = List(parts.init map (_.trim), pieces.toList, positions).transpose
// create an AST containing a List[(String, Piece, Pos)]
val tupleAST = data map { case List(part: String, piece: c.Expr[_], Pos(column, line)) =>
toTupleAST(toAST(part), piece.tree, toPosAST(toAST(column), toAST(line)))
}
// create an AST of `LocatedPieces`
val locatedPiecesAST = toLocatedPiecesAST(toListAST(tupleAST))
c.Expr(locatedPiecesAST)
case _ =>
c.abort(c.enclosingPosition, "invalid")
}
}
}
用法:
object StringContextTest {
val place: Piece = Place("world")
val name: Piece = Name("Eric")
val pieces = s2"""
Hello $place
How are you, $name?
"""
pieces.located foreach println
}
结果:
(Hello,Place(world),Pos(12,9))
(How are you,,Name(Eric),Pos(19,10))
我没想到,它可能需要很多时间才能使所有事情都发生呃,但这是一个愉快的时光。我希望代码符合您的要求。如果您需要具体的事情是如何工作的,然后看看其他的问题进行解答关于SO的更多信息:
非常感谢特拉维斯·布朗(见评论),我得到了一个更短的解决方案来编译:
object Macros {
import scala.reflect.macros.Context
import language.experimental.macros
sealed trait Piece
case class Place(str: String) extends Piece
case class Name(str: String) extends Piece
case class Pos(column: Int, line: Int)
case class LocatedPieces(located: Seq[(String, Piece, Pos)])
implicit class s2pieces(sc: StringContext) {
def s2(pieces: Piece*) = macro s2impl
}
def s2impl(c: Context)(pieces: c.Expr[Piece]*): c.Expr[LocatedPieces] = {
import c.universe.{ Name => _, _ }
def toAST[A : TypeTag](xs: Tree*): Tree =
Apply(
Select(Ident(typeOf[A].typeSymbol.companionSymbol), newTermName("apply")),
xs.toList)
val parts = c.prefix.tree match {
case Apply(_, List(Apply(_, rawParts))) =>
rawParts zip (pieces map (_.tree)) map {
case (Literal(Constant(rawPart: String)), piece) =>
val line = c.literal(piece.pos.line).tree
val column = c.literal(piece.pos.column).tree
val part = c.literal(rawPart.trim).tree
toAST[(_, _, _)](part, piece, toAST[Pos](line, column))
}
}
c.Expr(toAST[LocatedPieces](toAST[Seq[_]](parts: _*)))
}
}
它抽象了详细的AST结构,其逻辑有点不同,但几乎相同。如果您在理解代码的工作方式时遇到困难,请首先尝试理解第一个解决方案。它在做什么时更明确。
我已经尝试过各种API,但我还没有能够组装完整的解决方案。任何建议或大方向都会有所帮助。而一个完整的解决方案将得到我永恒的感激之情:-) – Eric 2013-03-10 23:29:39
我不确定它是否有答案,但你的问题提醒我这个帖子:http://hootenannylas.blogspot.com.au/2013/02/syntax -checking-in-scala-string.html – 2013-03-11 02:38:25
我已经阅读过它,并且我的用例稍微多一点。但是我知道Tony和我会请他在本周的下一个ScalaSyd期间帮助我,如果在此期间我没有得到答案:-)。 – Eric 2013-03-11 03:25:21