#99 Use the new TemplateBodyPiece interface in ParsedTemplate

gvlasov · Aug 20, 2015 · 728a49e · 728a49e
1 parent 1628bdb
commit 728a49e
Showing 1 changed file with 21 additions and 14 deletions.
diff --git a/src/main/java/org/tendiwa/inflectible/ParsedTemplate.java b/src/main/java/org/tendiwa/inflectible/ParsedTemplate.java
@@ -85,19 +85,19 @@ private ImmutableList<ArgumentName> argumentNames() {
      * @return Template from markup
      */
     private Template delegate() {
-        return new ParseTreeListener(this.argumentNames()).filledUpText();
+        return new TemplateBuilder(this.argumentNames()).template();
     }
 
     /**
      * Walks an ANTLR parse tree and constructs a template.
      */
-    private final class ParseTreeListener
+    private final class TemplateBuilder
         extends TemplateBundleParserBaseListener {
 
         /**
          * Template builder.
          */
-        private transient TextTemplateBuilder builder;
+        private transient ImmutableList.Builder<TemplateBodyPiece> pieces;
 
         /**
          * Argument names in the order as they appear in markup.
@@ -108,7 +108,7 @@ private final class ParseTreeListener
          * Ctor.
          * @param names Argument names in the order as they appear in markup
          */
-        ParseTreeListener(final ImmutableList<ArgumentName> names) {
+        TemplateBuilder(final ImmutableList<ArgumentName> names) {
             super();
             this.arguments = names;
         }
@@ -117,10 +117,12 @@ private final class ParseTreeListener
         public void enterTwoPartPlaceholder(
             final TemplateBundleParser.TwoPartPlaceholderContext context
         ) {
-            this.builder.addPlaceholder(
-                new ParsedTwoPartVariableConceptPlaceholder(
-                    ParsedTemplate.this.grammar,
-                    context
+            this.pieces.add(
+                new PiPlaceholder(
+                    new ParsedTwoPartVariableConceptPlaceholder(
+                        ParsedTemplate.this.grammar,
+                        context
+                    )
                 )
             );
         }
@@ -129,15 +131,17 @@ public void enterTwoPartPlaceholder(
         public void enterRawText(
             final TemplateBundleParser.RawTextContext context
         ) {
-            this.builder.addText(context.getText());
+            this.pieces.add(new PiPlainText(context.getText()));
         }
 
         @Override
         public void enterSinglePartPlaceholder(
             final TemplateBundleParser.SinglePartPlaceholderContext context
         ) {
-            this.builder.addPlaceholder(
-                new ParsedSinglePartPlaceholder(context)
+            this.pieces.add(
+                new PiPlaceholder(
+                    new ParsedSinglePartPlaceholder(context)
+                )
             );
         }
 
@@ -146,10 +150,13 @@ public void enterSinglePartPlaceholder(
          * it.
          * @return Template.
          */
-        private Template filledUpText() {
-            this.builder = new TextTemplateBuilder(this.arguments);
+        private Template template() {
+            this.pieces = ImmutableList.builder();
             ParseTreeWalker.DEFAULT.walk(this, ParsedTemplate.this.ctx);
-            return this.builder.build();
+            return new BasicTemplate(
+                this.arguments,
+                this.pieces.build()
+            );
         }
     }
 }