Created
October 7, 2016 11:27
-
-
Save cb372/262a9e935dba5de740a8e26909237419 to your computer and use it in GitHub Desktop.
Prettier AST trees for debugging macros
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| import scala.reflect.macros.blackbox | |
| object PrettyPrinting { | |
| /* | |
| * Print a raw AST, nicely indented. | |
| * Pretty fragile, e.g. will get confused by string literals containing commas or parentheses. | |
| */ | |
| def prettyTree(raw: String): String = { | |
| var level = 0 | |
| def indent = " " * level | |
| val sparse = raw.map { | |
| case ',' => s",\n$indent".dropRight(1) | |
| case '(' => | |
| level += 1 | |
| s"(\n$indent" | |
| case ')' => | |
| level -= 1 | |
| s"\n$indent)" | |
| case other => other | |
| }.mkString | |
| sparse.replaceAll("""\(\s+\)""", "()") | |
| } | |
| def prettyTree(c: blackbox.Context)(tree: c.universe.Tree): String = | |
| prettyTree(c.universe.showRaw(tree)) | |
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Block(List(ClassDef(Modifiers(ABSTRACT | INTERFACE | SEALED | DEFAULTPARAM/TRAIT), TypeName("Foo"), List(), Template(List(Select(Ident(scala), TypeName("AnyRef"))), noSelfType, List())), ModuleDef(Modifiers(), TermName("Foo"), Template(List(Select(Ident(scala), TypeName("AnyRef"))), noSelfType, List(DefDef(Modifiers(), termNames.CONSTRUCTOR, List(), List(List()), TypeTree(), Block(List(pendingSuperCall), Literal(Constant(())))), ValDef(Modifiers(), TermName("a"), TypeTree(), TypeApply(Select(Select(Select(Ident(TermName("monocle")), TermName("macros")), TermName("GenPrism")), TermName("apply")), List(Ident(TypeName("Foo")), Ident(TypeName("A"))))), ValDef(Modifiers(), TermName("b"), TypeTree(), TypeApply(Select(Select(Select(Ident(TermName("monocle")), TermName("macros")), TermName("GenPrism")), TermName("apply")), List(Ident(TypeName("Foo")), Ident(TypeName("B"))))))))), Literal(Constant(()))) |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Block( | |
| List( | |
| ClassDef( | |
| Modifiers( | |
| ABSTRACT | INTERFACE | SEALED | DEFAULTPARAM/TRAIT | |
| ), | |
| TypeName( | |
| "Foo" | |
| ), | |
| List(), | |
| Template( | |
| List( | |
| Select( | |
| Ident( | |
| scala | |
| ), | |
| TypeName( | |
| "AnyRef" | |
| ) | |
| ) | |
| ), | |
| noSelfType, | |
| List() | |
| ) | |
| ), | |
| ModuleDef( | |
| Modifiers(), | |
| TermName( | |
| "Foo" | |
| ), | |
| Template( | |
| List( | |
| Select( | |
| Ident( | |
| scala | |
| ), | |
| TypeName( | |
| "AnyRef" | |
| ) | |
| ) | |
| ), | |
| noSelfType, | |
| List( | |
| DefDef( | |
| Modifiers(), | |
| termNames.CONSTRUCTOR, | |
| List(), | |
| List( | |
| List() | |
| ), | |
| TypeTree(), | |
| Block( | |
| List( | |
| pendingSuperCall | |
| ), | |
| Literal( | |
| Constant( | |
| () | |
| ) | |
| ) | |
| ) | |
| ), | |
| ValDef( | |
| Modifiers(), | |
| TermName( | |
| "a" | |
| ), | |
| TypeTree(), | |
| TypeApply( | |
| Select( | |
| Select( | |
| Select( | |
| Ident( | |
| TermName( | |
| "monocle" | |
| ) | |
| ), | |
| TermName( | |
| "macros" | |
| ) | |
| ), | |
| TermName( | |
| "GenPrism" | |
| ) | |
| ), | |
| TermName( | |
| "apply" | |
| ) | |
| ), | |
| List( | |
| Ident( | |
| TypeName( | |
| "Foo" | |
| ) | |
| ), | |
| Ident( | |
| TypeName( | |
| "A" | |
| ) | |
| ) | |
| ) | |
| ) | |
| ), | |
| ValDef( | |
| Modifiers(), | |
| TermName( | |
| "b" | |
| ), | |
| TypeTree(), | |
| TypeApply( | |
| Select( | |
| Select( | |
| Select( | |
| Ident( | |
| TermName( | |
| "monocle" | |
| ) | |
| ), | |
| TermName( | |
| "macros" | |
| ) | |
| ), | |
| TermName( | |
| "GenPrism" | |
| ) | |
| ), | |
| TermName( | |
| "apply" | |
| ) | |
| ), | |
| List( | |
| Ident( | |
| TypeName( | |
| "Foo" | |
| ) | |
| ), | |
| Ident( | |
| TypeName( | |
| "B" | |
| ) | |
| ) | |
| ) | |
| ) | |
| ) | |
| ) | |
| ) | |
| ) | |
| ), | |
| Literal( | |
| Constant( | |
| () | |
| ) | |
| ) | |
| ) |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
I raise you with an even more hacky implementation that gives slightly more compact output!