org.summer.ss.core.richstring.DefaultIndentationHandler.java Source code

Introduction

Here is the source code for org.summer.ss.core.richstring.DefaultIndentationHandler.java

Source

/*******************************************************************************
 * Copyright (c) 2011 itemis AG (http://www.itemis.eu) and others.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *******************************************************************************/
package org.summer.ss.core.richstring;

import java.util.LinkedList;

import org.eclipse.jdt.annotation.NonNullByDefault;

import com.google.common.collect.Lists;

/**
 * Default indentation handler for rich strings. Tries to be graceful with
 * inconsistent indentation.
 * @author Sebastian Zarnekow - Initial contribution and API
 */
@NonNullByDefault
public class DefaultIndentationHandler implements IRichStringIndentationHandler {

    protected static abstract class IndentationData {
        protected CharSequence value;

        protected IndentationData(CharSequence value) {
            this.value = value;
        }

        protected abstract void accept(IRichStringPartAcceptor acceptor);

        @Override
        public String toString() {
            return getClass().getSimpleName() + " [" + value + "]";
        }
    }

    protected static class SemanticIndentationData extends IndentationData {

        protected SemanticIndentationData(CharSequence value) {
            super(value);
        }

        @Override
        protected void accept(IRichStringPartAcceptor acceptor) {
            acceptor.acceptSemanticText(value, null);
        }

    }

    protected static class TemplateIndentationData extends IndentationData {

        protected TemplateIndentationData(CharSequence value) {
            super(value);
        }

        @Override
        protected void accept(IRichStringPartAcceptor acceptor) {
            acceptor.acceptTemplateText(value, null);
        }

    }

    private LinkedList<IndentationData> indentationData;
    private LinkedList<LinkedList<IndentationData>> indentationDataStack;

    public DefaultIndentationHandler() {
        this.indentationData = Lists.newLinkedList();
        this.indentationDataStack = Lists.newLinkedList();
        indentationDataStack.add(indentationData);
    }

    public void popIndentation() {
        indentationData.removeLast();
        if (indentationData.isEmpty() && indentationDataStack.size() > 1) {
            indentationDataStack.removeLast();
            indentationData = indentationDataStack.getLast();
        }
    }

    public void pushTemplateIndentation(CharSequence indentation) {
        if (indentationData.isEmpty()) {
            indentationData.add(new TemplateIndentationData(indentation));
        } else {
            String currentIndentation = getTotalIndentation();
            if (indentation.toString().startsWith(currentIndentation)) {
                String trimmedIndentation = indentation.toString().substring(currentIndentation.length());
                indentationData.add(new TemplateIndentationData(trimmedIndentation));
            } else {
                LinkedList<IndentationData> newIndentationData = Lists.newLinkedList();
                newIndentationData.add(new TemplateIndentationData(indentation));
                indentationDataStack.add(newIndentationData);
                indentationData = newIndentationData;
            }
        }
    }

    public void pushSemanticIndentation(CharSequence indentation) {
        if (indentationData.isEmpty()) {
            indentationData.add(new SemanticIndentationData(indentation));
        } else {
            String currentIndentation = getTotalIndentation();
            if (indentation.toString().startsWith(currentIndentation)) {
                String trimmedIndentation = indentation.toString().substring(currentIndentation.length());
                indentationData.add(new SemanticIndentationData(trimmedIndentation));
            } else {
                LinkedList<IndentationData> newIndentationData = Lists.newLinkedList();
                newIndentationData.add(new SemanticIndentationData(indentation));
                indentationDataStack.add(newIndentationData);
                indentationData = newIndentationData;
            }
        }
    }

    public CharSequence getTotalSemanticIndentation() {
        StringBuilder result = new StringBuilder();
        for (IndentationData data : indentationData) {
            if (data instanceof SemanticIndentationData)
                result.append(data.value);
        }
        return result.toString();
    }

    public String getTotalIndentation() {
        StringBuilder result = new StringBuilder();
        for (IndentationData data : indentationData) {
            result.append(data.value);
        }
        return result.toString();
    }

    public void accept(IRichStringPartAcceptor acceptor) {
        for (IndentationData data : indentationData) {
            data.accept(acceptor);
        }
    }

}