如何区分宏中的三引号和单引号?

时间:2015-07-03 08:02:00

标签: scala macros

我正在写一个宏expr,其中m("SOME EXPRESSION") m(""" SOME EXPRESSION """) 是某种语言的表达式(不是Scala):

import scala.language.experimental.macros
import scala.reflect.macros.blackbox.Context

object Temp {
  def m(s: String) : String = macro mImpl

  def mImpl(context: Context)(s: context.Expr[String]): context.universe.Tree = {
    import context.universe._

    s match {
      case l @ Literal(Constant(p: String)) =>
        if (l.<TRIPLE QUOTES>) {
          ...
        } else {
          ...
        }

      case _ =>
        context.abort(context.enclosingPosition, "The argument of m must be a string literal")
    }
  }
}

当我解析表达式时,我想在源文件中报告具有适当位置的错误消息。为了实现这一点,我应该知道字符串文字本身的位置和文字(3或1)的引号数。不幸的是,我没有找到任何返回文字引号数的方法:

<TRIPLE QUOTES>

我应该放什么而不是{{1}}?

1 个答案:

答案 0 :(得分:3)

我能想到的唯一方法是访问源文件并检查三重引号:

l.tree.pos.source.content.startsWith("\"\"\"",l.tree.pos.start)

您还需要编辑匹配的案例:

case l @ Expr(Literal(Constant(p: String))) =>

这里的版本有一些解释:

val tree: context.universe.Tree = l.tree
val pos: scala.reflect.api.Position = tree.pos
val source: scala.reflect.internal.util.SourceFile = pos.source
val content: Array[Char] = source.content
val start: Int = pos.start
val isTriple: Boolean = content.startsWith("\"\"\"",start)