--- /dev/null
+/*******************************************************************************
+ * Copyright (c) 2000, 2011 IBM Corporation 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
+ *
+ * Contributors:
+ * IBM Corporation - initial API and implementation
+ *******************************************************************************/
+package org.eclipse.jdt.internal.corext.util;
+
+import java.util.Map;
+
+import org.eclipse.core.runtime.Assert;
+
+import org.eclipse.text.edits.TextEdit;
+
+import org.eclipse.jface.text.BadLocationException;
+import org.eclipse.jface.text.Document;
+import org.eclipse.jface.text.IRegion;
+
+import org.eclipse.jdt.core.IJavaProject;
+import org.eclipse.jdt.core.JavaCore;
+import org.eclipse.jdt.core.ToolFactory;
+import org.eclipse.jdt.core.dom.ASTNode;
+import org.eclipse.jdt.core.dom.BodyDeclaration;
+import org.eclipse.jdt.core.dom.Expression;
+import org.eclipse.jdt.core.dom.Statement;
+import org.eclipse.jdt.core.formatter.CodeFormatter;
+import org.eclipse.jdt.core.formatter.DefaultCodeFormatterConstants;
+
+import org.eclipse.jdt.internal.ui.JavaPlugin;
+
+public class CodeFormatterUtil {
+
+ /**
+ * Creates a string that represents the given number of indentation units.
+ * The returned string can contain tabs and/or spaces depending on the core formatter preferences.
+ *
+ * @param indentationUnits
+ * the number of indentation units to generate
+ * @param project
+ * the project from which to get the formatter settings,
+ * <code>null</code> if the workspace default should be used
+ * @return the indent string
+ */
+ public static String createIndentString(int indentationUnits, IJavaProject project) {
+ Map<String, String> options= project != null ? project.getOptions(true) : JavaCore.getOptions();
+ return ToolFactory.createCodeFormatter(options).createIndentationString(indentationUnits);
+ }
+
+ /**
+ * Gets the current tab width.
+ *
+ * @param project
+ * The project where the source is used, used for project specific options or
+ * <code>null</code> if the project is unknown and the workspace default should be used
+ * @return The tab width
+ */
+ public static int getTabWidth(IJavaProject project) {
+ /*
+ * If the tab-char is SPACE, FORMATTER_INDENTATION_SIZE is not used
+ * by the core formatter.
+ * We piggy back the visual tab length setting in that preference in
+ * that case.
+ */
+ String key;
+ if (JavaCore.SPACE.equals(getCoreOption(project, DefaultCodeFormatterConstants.FORMATTER_TAB_CHAR)))
+ key= DefaultCodeFormatterConstants.FORMATTER_INDENTATION_SIZE;
+ else
+ key= DefaultCodeFormatterConstants.FORMATTER_TAB_SIZE;
+
+ return getCoreOption(project, key, 4);
+ }
+
+ /**
+ * Returns the current indent width.
+ *
+ * @param project
+ * the project where the source is used or,
+ * <code>null</code> if the project is unknown and the workspace default should be used
+ * @return the indent width
+ * @since 3.1
+ */
+ public static int getIndentWidth(IJavaProject project) {
+ String key;
+ if (DefaultCodeFormatterConstants.MIXED.equals(getCoreOption(project, DefaultCodeFormatterConstants.FORMATTER_TAB_CHAR)))
+ key= DefaultCodeFormatterConstants.FORMATTER_INDENTATION_SIZE;
+ else
+ key= DefaultCodeFormatterConstants.FORMATTER_TAB_SIZE;
+
+ return getCoreOption(project, key, 4);
+ }
+
+ /**
+ * Returns the possibly <code>project</code>-specific core preference defined under <code>key</code>.
+ *
+ * @param project
+ * the project to get the preference from,
+ * or <code>null</code> to get the global preference
+ * @param key
+ * the key of the preference
+ * @return the value of the preference
+ * @since 3.1
+ */
+ private static String getCoreOption(IJavaProject project, String key) {
+ if (project == null)
+ return JavaCore.getOption(key);
+ return project.getOption(key, true);
+ }
+
+ /**
+ * Returns the possibly <code>project</code>-specific core preference defined under <code>key</code>,
+ * or <code>def</code> if the value is not a integer.
+ *
+ * @param project
+ * the project to get the preference from,
+ * or <code>null</code> to get the global preference
+ * @param key
+ * the key of the preference
+ * @param def
+ * the default value
+ * @return the value of the preference
+ * @since 3.1
+ */
+ private static int getCoreOption(IJavaProject project, String key, int def) {
+ try {
+ return Integer.parseInt(getCoreOption(project, key));
+ } catch (NumberFormatException e) {
+ return def;
+ }
+ }
+
+ // transition code
+
+ /**
+ * Old API. Consider to use format2 (TextEdit)
+ *
+ * @param kind
+ * Use to specify the kind of the code snippet to format.
+ * It can be any of the kind constants defined in {@link CodeFormatter}
+ * @param source
+ * The source to format
+ * @param indentationLevel
+ * The initial indentation level, used to shift left/right the entire source fragment.
+ * An initial indentation level of zero or below has no effect.
+ * @param lineSeparator
+ * The line separator to use in formatted source,
+ * if set to <code>null</code>, then the platform default one will be used.
+ * @param project
+ * The project from which to retrieve the formatter options from
+ * If set to <code>null</code>, then use the current settings from {@link JavaCore#getOptions()}.
+ * @return the formatted source string
+ */
+ public static String format(int kind, String source, int indentationLevel, String lineSeparator, IJavaProject project) {
+ Map<String, String> options= project != null ? project.getOptions(true) : null;
+ return format(kind, source, indentationLevel, lineSeparator, options);
+ }
+
+ /**
+ * Old API. Consider to use format2 (TextEdit)
+ *
+ * @param kind
+ * Use to specify the kind of the code snippet to format.
+ * It can be any of the kind constants defined in {@link CodeFormatter}
+ * @param source
+ * The source to format
+ * @param indentationLevel
+ * The initial indentation level, used to shift left/right the entire source fragment.
+ * An initial indentation level of zero or below has no effect.
+ * @param lineSeparator
+ * The line separator to use in formatted source,
+ * if set to <code>null</code>, then the platform default one will be used.
+ * @param options
+ * The options map to use for formatting with the default code formatter.
+ * Recognized options are documented on {@link JavaCore#getDefaultOptions()}.
+ * If set to <code>null</code>, then use the current settings from {@link JavaCore#getOptions()}.
+ * @return the formatted source string
+ */
+ public static String format(int kind, String source, int indentationLevel, String lineSeparator, Map<String, String> options) {
+ TextEdit edit= format2(kind, source, indentationLevel, lineSeparator, options);
+ if (edit == null) {
+ return source;
+ } else {
+ Document document= new Document(source);
+ try {
+ edit.apply(document, TextEdit.NONE);
+ } catch (BadLocationException e) {
+ JavaPlugin.log(e); // bug in the formatter
+ Assert.isTrue(false, "Formatter created edits with wrong positions: " + e.getMessage()); //$NON-NLS-1$
+ }
+ return document.get();
+ }
+ }
+
+ /**
+ * Creates edits that describe how to format the given string.
+ * Returns <code>null</code> if the code could not be formatted for the given kind.
+ *
+ * @param kind
+ * Use to specify the kind of the code snippet to format.
+ * It can be any of the kind constants defined in {@link CodeFormatter}
+ * @param source
+ * The source to format
+ * @param offset
+ * The given offset to start recording the edits (inclusive).
+ * @param length the given length to stop recording the edits (exclusive).
+ * @param indentationLevel
+ * The initial indentation level, used to shift left/right the entire source fragment.
+ * An initial indentation level of zero or below has no effect.
+ * @param lineSeparator
+ * The line separator to use in formatted source,
+ * if set to <code>null</code>, then the platform default one will be used.
+ * @param options
+ * The options map to use for formatting with the default code formatter.
+ * Recognized options are documented on {@link JavaCore#getDefaultOptions()}.
+ * If set to <code>null</code>, then use the current settings from {@link JavaCore#getOptions()}.
+ * @return an TextEdit describing the changes required to format source
+ * @throws IllegalArgumentException
+ * If the offset and length are not inside the string, a IllegalArgumentException is thrown.
+ */
+ public static TextEdit format2(int kind, String source, int offset, int length, int indentationLevel, String lineSeparator, Map<String, String> options) {
+ if (offset < 0 || length < 0 || offset + length > source.length()) {
+ throw new IllegalArgumentException("offset or length outside of string. offset: " + offset + ", length: " + length + ", string size: " + source.length()); //$NON-NLS-1$//$NON-NLS-2$//$NON-NLS-3$
+ }
+ return ToolFactory.createCodeFormatter(options).format(kind, source, offset, length, indentationLevel, lineSeparator);
+ }
+
+ /**
+ * Creates edits that describe how to format the given string.
+ * Returns <code>null</code> if the code could not be formatted for the given kind.
+ *
+ * @param kind
+ * Use to specify the kind of the code snippet to format.
+ * It can be any of the kind constants defined in {@link CodeFormatter}
+ * @param source
+ * The source to format
+ * @param indentationLevel
+ * The initial indentation level, used to shift left/right the entire source fragment.
+ * An initial indentation level of zero or below has no effect.
+ * @param lineSeparator
+ * The line separator to use in formatted source,
+ * if set to <code>null</code>, then the platform default one will be used.
+ * @param options
+ * The options map to use for formatting with the default code formatter.
+ * Recognized options are documented on {@link JavaCore#getDefaultOptions()}.
+ * If set to <code>null</code>, then use the current settings from {@link JavaCore#getOptions()}.
+ * @return an TextEdit describing the changes required to format source
+ * @throws IllegalArgumentException
+ * If the offset and length are not inside the string, a IllegalArgumentException is thrown.
+ */
+ public static TextEdit format2(int kind, String source, int indentationLevel, String lineSeparator, Map<String, String> options) {
+ return format2(kind, source, 0, source.length(), indentationLevel, lineSeparator, options);
+ }
+
+ /**
+ * Creates edits that describe how to re-format the given string.
+ * This method should be used for formatting existing code.
+ * Returns <code>null</code> if the code could not be formatted for the given kind.
+ *
+ * @param kind
+ * Use to specify the kind of the code snippet to format.
+ * It can be any of the kind constants defined in {@link CodeFormatter}
+ * @param source
+ * The source to format
+ * @param offset
+ * The given offset to start recording the edits (inclusive).
+ * @param length the given length to stop recording the edits (exclusive).
+ * @param indentationLevel
+ * The initial indentation level, used to shift left/right the entire source fragment.
+ * An initial indentation level of zero or below has no effect.
+ * @param lineSeparator
+ * The line separator to use in formatted source,
+ * if set to <code>null</code>, then the platform default one will be used.
+ * @param options
+ * The options map to use for formatting with the default code formatter.
+ * Recognized options are documented on {@link JavaCore#getDefaultOptions()}.
+ * If set to <code>null</code>, then use the current settings from {@link JavaCore#getOptions()}.
+ * @return an TextEdit describing the changes required to format source
+ * @throws IllegalArgumentException
+ * If the offset and length are not inside the string, a IllegalArgumentException is thrown.
+ */
+ public static TextEdit reformat(int kind, String source, int offset, int length, int indentationLevel, String lineSeparator, Map<String, String> options) {
+ if (offset < 0 || length < 0 || offset + length > source.length()) {
+ throw new IllegalArgumentException("offset or length outside of string. offset: " + offset + ", length: " + length + ", string size: " + source.length()); //$NON-NLS-1$//$NON-NLS-2$//$NON-NLS-3$
+ }
+ return ToolFactory.createCodeFormatter(options, ToolFactory.M_FORMAT_EXISTING).format(kind, source, offset, length, indentationLevel, lineSeparator);
+ }
+
+ /**
+ * Creates edits that describe how to re-format the regions in the given string.
+ * This method should be used for formatting existing code.
+ * Returns <code>null</code> if the code could not be formatted for the given kind.
+ *
+ * <p>No region in <code>regions</code> must overlap with any other region in <code>regions</code>.
+ * Each region must be within source. There must be at least one region. Regions must be sorted
+ * by their offsets, smaller offset first.</p>
+ *
+ * @param kind
+ * Use to specify the kind of the code snippet to format.
+ * It can be any of K_EXPRESSION, K_STATEMENTS, K_CLASS_BODY_DECLARATIONS, K_COMPILATION_UNIT, K_UNKNOWN
+ * @param source
+ * The source to format
+ * @param regions
+ * a set of regions in the string to format
+ * @param indentationLevel
+ * The initial indentation level, used to shift left/right the entire source fragment.
+ * An initial indentation level of zero or below has no effect.
+ * @param lineSeparator
+ * The line separator to use in formatted source,
+ * if set to <code>null</code>, then the platform default one will be used.
+ * @param options
+ * The options map to use for formatting with the default code formatter.
+ * Recognized options are documented on {@link JavaCore#getDefaultOptions()}.
+ * If set to <code>null</code>, then use the current settings from {@link JavaCore#getOptions()}.
+ * @return an TextEdit describing the changes required to format source
+ * @throws IllegalArgumentException if there is no region, a region overlaps with another region, or the regions are not
+ * sorted in the ascending order.
+ * @since 3.4
+ */
+ public static TextEdit reformat(int kind, String source, IRegion[] regions, int indentationLevel, String lineSeparator, Map<String, String> options) {
+ return ToolFactory.createCodeFormatter(options, ToolFactory.M_FORMAT_EXISTING).format(kind, source, regions, indentationLevel, lineSeparator);
+ }
+
+ /**
+ * Creates edits that describe how to re-format the given string.
+ * This method should be used for formatting existing code.
+ * Returns <code>null</code> if the code could not be formatted for the given kind.
+ *
+ * @param kind
+ * Use to specify the kind of the code snippet to format.
+ * It can be any of the kind constants defined in {@link CodeFormatter}
+ * @param source
+ * The source to format
+ * @param indentationLevel
+ * The initial indentation level, used to shift left/right the entire source fragment.
+ * An initial indentation level of zero or below has no effect.
+ * @param lineSeparator
+ * The line separator to use in formatted source,
+ * if set to <code>null</code>, then the platform default one will be used.
+ * @param options
+ * The options map to use for formatting with the default code formatter.
+ * Recognized options are documented on {@link JavaCore#getDefaultOptions()}.
+ * If set to <code>null</code>, then use the current settings from {@link JavaCore#getOptions()}.
+ * @return an TextEdit describing the changes required to format source
+ * @throws IllegalArgumentException
+ * If the offset and length are not inside the string, a IllegalArgumentException is thrown.
+ */
+ public static TextEdit reformat(int kind, String source, int indentationLevel, String lineSeparator, Map<String, String> options) {
+ return reformat(kind, source, 0, source.length(), indentationLevel, lineSeparator, options);
+ }
+
+ /**
+ * Creates edits that describe how to format the given string.
+ * The given node is used to infer the kind to use to format the string.
+ * Consider to use {@link #format2(int, String, int, String, Map)} if the kind is already known.
+ * Returns <code>null</code> if the code could not be formatted for the given kind.
+ *
+ * @param node
+ * Use to infer the kind of the code snippet to format.
+ * @param source
+ * The source to format
+ * @param indentationLevel
+ * The initial indentation level, used to shift left/right the entire source fragment.
+ * An initial indentation level of zero or below has no effect.
+ * @param lineSeparator
+ * The line separator to use in formatted source,
+ * if set to <code>null</code>, then the platform default one will be used.
+ * @param options
+ * The options map to use for formatting with the default code formatter.
+ * Recognized options are documented on {@link JavaCore#getDefaultOptions()}.
+ * If set to <code>null</code>, then use the current settings from {@link JavaCore#getOptions()}.
+ * @return an TextEdit describing the changes required to format source
+ * @throws IllegalArgumentException
+ * If the offset and length are not inside the string, a IllegalArgumentException is thrown.
+ */
+ public static TextEdit format2(ASTNode node, String source, int indentationLevel, String lineSeparator, Map<String, String> options) {
+ int code;
+ String prefix= ""; //$NON-NLS-1$
+ String suffix= ""; //$NON-NLS-1$
+ if (node instanceof Statement) {
+ code= CodeFormatter.K_STATEMENTS;
+ if (node.getNodeType() == ASTNode.SWITCH_CASE) {
+ prefix= "switch(1) {"; //$NON-NLS-1$
+ suffix= "}"; //$NON-NLS-1$
+ code= CodeFormatter.K_STATEMENTS;
+ }
+ } else if (node instanceof Expression && node.getNodeType() != ASTNode.VARIABLE_DECLARATION_EXPRESSION) {
+ code= CodeFormatter.K_EXPRESSION;
+ } else if (node instanceof BodyDeclaration) {
+ code= CodeFormatter.K_CLASS_BODY_DECLARATIONS;
+ } else {
+ switch (node.getNodeType()) {
+ case ASTNode.ARRAY_TYPE:
+ case ASTNode.PARAMETERIZED_TYPE:
+ case ASTNode.PRIMITIVE_TYPE:
+ case ASTNode.QUALIFIED_TYPE:
+ case ASTNode.SIMPLE_TYPE:
+ suffix= " x;"; //$NON-NLS-1$
+ code= CodeFormatter.K_CLASS_BODY_DECLARATIONS;
+ break;
+ case ASTNode.WILDCARD_TYPE:
+ prefix= "A<"; //$NON-NLS-1$
+ suffix= "> x;"; //$NON-NLS-1$
+ code= CodeFormatter.K_CLASS_BODY_DECLARATIONS;
+ break;
+ case ASTNode.COMPILATION_UNIT:
+ code= CodeFormatter.K_COMPILATION_UNIT;
+ break;
+ case ASTNode.VARIABLE_DECLARATION_EXPRESSION:
+ case ASTNode.SINGLE_VARIABLE_DECLARATION:
+ suffix= ";"; //$NON-NLS-1$
+ code= CodeFormatter.K_STATEMENTS;
+ break;
+ case ASTNode.VARIABLE_DECLARATION_FRAGMENT:
+ prefix= "A "; //$NON-NLS-1$
+ suffix= ";"; //$NON-NLS-1$
+ code= CodeFormatter.K_STATEMENTS;
+ break;
+ case ASTNode.PACKAGE_DECLARATION:
+ case ASTNode.IMPORT_DECLARATION:
+ suffix= "\nclass A {}"; //$NON-NLS-1$
+ code= CodeFormatter.K_COMPILATION_UNIT;
+ break;
+ case ASTNode.JAVADOC:
+ suffix= "void foo();"; //$NON-NLS-1$
+ code= CodeFormatter.K_CLASS_BODY_DECLARATIONS;
+ break;
+ case ASTNode.CATCH_CLAUSE:
+ prefix= "try {}"; //$NON-NLS-1$
+ code= CodeFormatter.K_STATEMENTS;
+ break;
+ case ASTNode.ANONYMOUS_CLASS_DECLARATION:
+ prefix= "new A()"; //$NON-NLS-1$
+ suffix= ";"; //$NON-NLS-1$
+ code= CodeFormatter.K_STATEMENTS;
+ break;
+ case ASTNode.MEMBER_VALUE_PAIR:
+ prefix= "@Author("; //$NON-NLS-1$
+ suffix= ") class x {}"; //$NON-NLS-1$
+ code= CodeFormatter.K_COMPILATION_UNIT;
+ break;
+ case ASTNode.MODIFIER:
+ suffix= " class x {}"; //$NON-NLS-1$
+ code= CodeFormatter.K_COMPILATION_UNIT;
+ break;
+ case ASTNode.TYPE_PARAMETER:
+ prefix= "class X<"; //$NON-NLS-1$
+ suffix= "> {}"; //$NON-NLS-1$
+ code= CodeFormatter.K_COMPILATION_UNIT;
+ break;
+ case ASTNode.MEMBER_REF:
+ case ASTNode.METHOD_REF:
+ case ASTNode.METHOD_REF_PARAMETER:
+ case ASTNode.TAG_ELEMENT:
+ case ASTNode.TEXT_ELEMENT:
+ // Javadoc formatting not yet supported:
+ return null;
+ default:
+ //Assert.isTrue(false, "Node type not covered: " + node.getClass().getName()); //$NON-NLS-1$
+ return null;
+ }
+ }
+
+ String concatStr= prefix + source + suffix;
+ TextEdit edit= format2(code, concatStr, prefix.length(), source.length(), indentationLevel, lineSeparator, options);
+ if (edit != null && prefix.length() > 0) {
+ edit.moveTree(-prefix.length());
+ }
+ return edit;
+ }
+
+}