Merge pull request #125 from shanielh/master

Allow extending user defined traits by adding option 'trait' to messages
SandroGrzicic · Mar 8, 2016 · 984edef · 984edef
2 parents 7928eb3 + a342e6b
commit 984edef
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/scalabuff-compiler/src/main/net/sandrogrzicic/scalabuff/compiler/Generator.scala b/scalabuff-compiler/src/main/net/sandrogrzicic/scalabuff/compiler/Generator.scala
@@ -149,6 +149,10 @@ class Generator protected (sourceName: String, importedSymbols: Map[String, Impo
       }
       out.append('\n').append(indent1).append("with net.sandrogrzicic.scalabuff.Parser[").append(name).append("]")
 
+      for (OptionValue(_, value) <- body.options.filter(_.key == "trait")) {
+        out.append('\n').append(indent1).append("with ").append(value.replace("$name", name).stripQuotes)
+      }
+
       out.append(" {\n\n")
 
       // setters