org.snapscript.tree.template.TextTemplate Maven / Gradle / Ivy

Go to download

package org.snapscript.tree.template;

import java.io.StringWriter;
import java.util.ArrayList;
import java.util.List;

import org.snapscript.core.Evaluation;
import org.snapscript.core.InternalStateException;
import org.snapscript.core.Scope;
import org.snapscript.core.Value;
import org.snapscript.core.ValueType;
import org.snapscript.parse.StringToken;

public class TextTemplate implements Evaluation {

   private volatile List tokens;
   private volatile StringToken template;
   
   public TextTemplate(StringToken template) {
      this.tokens = new ArrayList();
      this.template = template;
   }

   @Override
   public Value evaluate(Scope scope, Object left) throws Exception {
      String text = template.getValue();
      
      if(text == null) {
         throw new InternalStateException("Text value was null");
      }
      String result = interpolate(scope, text);
   
      return ValueType.getTransient(result);
   }
   
   private String interpolate(Scope scope, String text) throws Exception {
      StringWriter writer = new StringWriter();
            
      if(tokens.isEmpty()) {
         SegmentIterator iterator = new SegmentIterator(text);
         List list = new ArrayList();
         
         while(iterator.hasNext()) {
            Segment token = iterator.next();
            
            if(token != null) {
               list.add(token);  
            }
         }
         tokens = list; // atomic swap
      }
      for(Segment token : tokens) {
         token.process(scope, writer);
      }
      return writer.toString();
   }
}