BracesNode.java

/*******************************************************************************
 * Copyright (c) 2009 xored software, Inc.  
 *
 * 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  
 *
 * Contributors:
 *     xored software, Inc. - initial API and Implementation (Vladimir Belov)
 *******************************************************************************/

package org.eclipse.dltk.javascript.formatter.internal.nodes;

import java.util.List;

import org.eclipse.core.runtime.Assert;
import org.eclipse.dltk.formatter.IFormatterContext;
import org.eclipse.dltk.formatter.IFormatterDocument;
import org.eclipse.dltk.formatter.IFormatterNode;
import org.eclipse.dltk.formatter.IFormatterTextNode;
import org.eclipse.dltk.formatter.IFormatterWriter;
import org.eclipse.dltk.javascript.formatter.JavaScriptFormatterConstants;

public class BracesNode extends FormatterBlockWithBeginEndNode {

	private IBracesConfiguration configuration;

	public BracesNode(IFormatterDocument document,
			IBracesConfiguration configuration) {
		super(document);

		Assert.isNotNull(configuration);

		this.configuration = configuration;
	}

	@Override
	public void accept(IFormatterContext context, IFormatterWriter visitor)
			throws Exception {

		context.setBlankLines(getBlankLinesBefore(context));

		printBeforeOpenBrace(context, visitor);

		if (configuration.isBracesIndenting())
			context.incIndent();

		// print "{"
		if (getBegin() != null) {
			IFormatterNode[] nodes = getBegin();
			for (int i = 0; i < nodes.length; i++) {
				nodes[i].accept(context, visitor);
			}
		}
		// special support for comments on the same line as the start brace
		// if a formatter node is found before any other node (or textnodes that don't contain new lines)
		// skip the printAfterOpenBrace but just put an empty space.
		boolean printAfterOpenBrace = true;
		List<IFormatterNode> children = getChildren();
		for (IFormatterNode formatterNode : children) {
			if (formatterNode instanceof FormatterCommentNode) {
				printAfterOpenBrace = false;
			}
			else if (formatterNode instanceof IFormatterTextNode) {
				if (((IFormatterTextNode) formatterNode).getText().indexOf('\n') == -1) {
					continue;
				}
			}
			break;
		}

		if (printAfterOpenBrace) printAfterOpenBrace(context, visitor);
		else visitor.writeText(context, JSLiterals.SPACE);

		// print body
		final boolean indenting = isIndenting();
		if (indenting) {
			context.incIndent();
		}
		acceptBody(context, visitor);
		if (indenting) {
			context.decIndent();
		}

		printBeforeCloseBrace(context, visitor);

		// print "}"
		if (getEnd() != null) {
			visitor.write(context, getEnd().getStartOffset(), getEnd()
					.getEndOffset());
		}

		printAfterCloseBrace(context, visitor);

		if (configuration.isBracesIndenting())
			context.decIndent();

	}

	private boolean isStatementNewLine() {
		return getDocument().getBoolean(
				JavaScriptFormatterConstants.STATEMENT_NEW_LINE);
	}

	private void printBeforeCloseBrace(IFormatterContext context,
			IFormatterWriter visitor) throws Exception {

		switch (configuration.insertBeforeCloseBrace()) {
		case IBracesConfiguration.LINE_BREAK:
			if (isStatementNewLine()) {
				context.setBlankLines(-1);
				visitor.writeLineBreak(context);
			}
			break;

		case IBracesConfiguration.EMPTY_SPACE:
			visitor.appendToPreviousLine(context, JSLiterals.EMPTY);
			visitor.skipNextLineBreaks(context);
			break;

		case IBracesConfiguration.ONE_SPACE:
			visitor.appendToPreviousLine(context, JSLiterals.EMPTY);
			visitor.writeText(context, JSLiterals.SPACE);
			visitor.skipNextLineBreaks(context);
			break;
		}
	}

	private void printAfterOpenBrace(IFormatterContext context,
			IFormatterWriter visitor) throws Exception {

		switch (configuration.insertAfterOpenBrace()) {

		case IBracesConfiguration.LINE_BREAK:
			if (isStatementNewLine()) {
				context.setBlankLines(-1);
				visitor.writeLineBreak(context);
			}
			break;

		case IBracesConfiguration.ONE_SPACE:
			visitor.appendToPreviousLine(context, JSLiterals.EMPTY);
			visitor.writeText(context, JSLiterals.SPACE);
			visitor.skipNextLineBreaks(context);
			break;

		default:
			if (configuration.insertBeforeOpenBrace() != IBracesConfiguration.LINE_BREAK) {
				visitor.appendToPreviousLine(context, JSLiterals.EMPTY);
				if (configuration.insertBeforeCloseBrace() != IBracesConfiguration.EMPTY_SPACE) {
					visitor.writeText(context, JSLiterals.SPACE);
				}
				visitor.skipNextLineBreaks(context);
			}
			break;

		}
	}

	private void printBeforeOpenBrace(IFormatterContext context,
			IFormatterWriter visitor) throws Exception {
		switch (configuration.insertBeforeOpenBrace()) {
		case IBracesConfiguration.LINE_BREAK:
			if (isStatementNewLine()) {
				context.setBlankLines(-1);
				visitor.writeLineBreak(context);
			}
			break;
		case IBracesConfiguration.ONE_SPACE:
			visitor.appendToPreviousLine(context, JSLiterals.EMPTY);
			visitor.writeText(context, JSLiterals.SPACE);
			visitor.skipNextLineBreaks(context);
			break;
		case IBracesConfiguration.EMPTY_SPACE:
			visitor.appendToPreviousLine(context, JSLiterals.EMPTY);
			visitor.skipNextLineBreaks(context);
			break;
		}
	}

	private void printAfterCloseBrace(IFormatterContext context,
			IFormatterWriter visitor) throws Exception {

		switch (configuration.insertAfterCloseBrace()) {

		case IBracesConfiguration.LINE_BREAK:
			if (isStatementNewLine()) {
				context.setBlankLines(-1);
				visitor.writeLineBreak(context);
			}
			break;

		case IBracesConfiguration.ONE_SPACE:
			visitor.appendToPreviousLine(context, JSLiterals.SPACE);
			visitor.skipNextLineBreaks(context);
			break;

		}
	}

	@Override
	protected boolean isIndenting() {
		return configuration.isIndenting();
	}

}