AppendAction.java

/* 
 Copyright (C) 2013 Raquel Pau and Albert Coroleu.
 
 Walkmod is free software: you can redistribute it and/or modify
 it under the terms of the GNU Lesser General Public License as published by
 the Free Software Foundation, either version 3 of the License, or
 (at your option) any later version.
 
 Walkmod is distributed in the hope that it will be useful,
 but WITHOUT ANY WARRANTY; without even the implied warranty of
 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 GNU Lesser General Public License for more details.
 
 You should have received a copy of the GNU Lesser General Public License
 along with Walkmod.  If not, see <http://www.gnu.org/licenses/>.*/
package org.walkmod.javalang.actions;

import org.walkmod.javalang.ast.Node;
import org.walkmod.javalang.ast.expr.Expression;
import org.walkmod.javalang.ast.stmt.BlockStmt;

public class AppendAction extends Action {

   private Node node;

   private int endLine = -1;

   private int endColumn = -1;

   private int indentationLevel = 0;

   private int indentationSize = 0;

   private int beginPosition = 0;

   private char indentationChar = ' ';

   private String text = null;

   private int diff = 0;

   private int endLines = 0;
   
   
   public AppendAction(int beginLine, int beginPosition, Node node, int level, int indentationSize) {
      this(beginLine, beginPosition, node, level, indentationSize, 0);
   }

   public AppendAction(int beginLine, int beginPosition, Node node, int level, int indentationSize, int endLines) {
      super(beginLine, beginPosition, ActionType.APPEND);

      this.indentationLevel = level;
      this.indentationSize = indentationSize;
      this.node = node;
      this.beginPosition = beginPosition;
      this.endLines = endLines;
      if (beginPosition > 1 && indentationSize > 0) {
         diff = (beginPosition - 1) / indentationSize;
         indentationLevel = indentationLevel - diff;
      }
      getText("", indentationChar);

      String[] lines = text.split("\n");
      endLine = getBeginLine() + lines.length - 1;
      if (endLine == beginLine) {
         endColumn = getBeginColumn() + lines[lines.length - 1].length();
      } else {
         endColumn = lines[lines.length - 1].length();
      }
   }

   public int getEndLines() {
      return endLines;
   }

   public String getText(String indentation, char indentationChar) {

      this.indentationChar = indentationChar;
      text = node.getPrettySource(indentationChar, indentationLevel, indentationSize);

      if (endLines > 0) {
         if (!text.endsWith("\n")) {
            if (text.endsWith(" ")) {
               text = text.substring(0, text.length() - 1);
            }
            text += "\n";
         }
      }

      String[] lines = text.split("\n");

      if (beginPosition > 1 && indentationSize > 0) {
         if (lines.length > 1) {
            StringBuffer aux = new StringBuffer();

            aux.append(FormatterHelper.indent(text, indentation, indentationChar, indentationLevel, indentationSize,
                  requiresExtraIndentationOnFirstLine(node)));

            for (int i = 1; i < endLines; i++) {
               aux.append('\n');
            }

            text = aux.toString();
         }
         if (indentation.length() > 0) {
            text += indentation;
         } else {
            StringBuffer extraString = new StringBuffer("");

            for (int i = 0; i < diff * indentationSize; i++) {
               extraString.append(indentationChar);
            }
            text += extraString;
         }
      }

      return text;
   }

   public int getEndLine() {
      return endLine;
   }

   public int getEndColumn() {
      return endColumn;
   }

   public int getIndentations() {
      return indentationSize;
   }

   private boolean requiresExtraIndentationOnFirstLine(Node node) {

      return !((node instanceof Expression) || (node instanceof BlockStmt));
   }

}