Skip to content

Instantly share code, notes, and snippets.

@eed3si9n
Forked from xuwei-k/Macros.scala
Last active December 13, 2015 23:29
Show Gist options
  • Select an option

  • Save eed3si9n/4992129 to your computer and use it in GitHub Desktop.

Select an option

Save eed3si9n/4992129 to your computer and use it in GitHub Desktop.
import scala.reflect.macros.Context
import scala.util.matching.Regex
import java.util.regex.PatternSyntaxException
object Macros{
implicit class RegexContext(val c: StringContext) {
def r(): Regex = macro regexImpl
}
def regexImpl(c: Context)(): c.Expr[Regex] = {
import c.universe._
c.prefix.tree match {
case Apply(_,List(Apply(_,List(Literal(Constant(str: String)))))) =>
try{
str.r
}catch{
case e: PatternSyntaxException => c.abort(c.enclosingPosition, e.toString)
}
val Apply(fun, _) = reify(new Regex("")).tree
c.Expr[Regex](Apply.apply(fun, c.literal(str).tree :: Nil))
}
}
}
import Macros._
object Main extends App{
println(r"foo|bar") // compile success !
println(r"[foo") // compile error !
println(r"foo$") // error: invalid string interpolation: `$$', `$'ident or `$'BlockExpr expected
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment