001////////////////////////////////////////////////////////////////////////////////
002// checkstyle: Checks Java source code for adherence to a set of rules.
003// Copyright (C) 2001-2017 the original author or authors.
004//
005// This library is free software; you can redistribute it and/or
006// modify it under the terms of the GNU Lesser General Public
007// License as published by the Free Software Foundation; either
008// version 2.1 of the License, or (at your option) any later version.
009//
010// This library is distributed in the hope that it will be useful,
011// but WITHOUT ANY WARRANTY; without even the implied warranty of
012// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
013// Lesser General Public License for more details.
014//
015// You should have received a copy of the GNU Lesser General Public
016// License along with this library; if not, write to the Free Software
017// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
018////////////////////////////////////////////////////////////////////////////////
019
020package com.puppycrawl.tools.checkstyle.checks.whitespace;
021
022import java.util.Locale;
023
024import com.puppycrawl.tools.checkstyle.StatelessCheck;
025import com.puppycrawl.tools.checkstyle.api.AbstractCheck;
026import com.puppycrawl.tools.checkstyle.api.DetailAST;
027import com.puppycrawl.tools.checkstyle.api.TokenTypes;
028import com.puppycrawl.tools.checkstyle.utils.CommonUtils;
029
030/**
031 * <p>
032 * Checks line wrapping with separators.
033 * The policy to verify is specified using the {@link WrapOption} class
034 * and defaults to {@link WrapOption#EOL}.
035 * </p>
036 * <p> By default the check will check the following separators:
037 *  {@link TokenTypes#DOT DOT},
038 *  {@link TokenTypes#COMMA COMMA},
039 * Other acceptable tokens are
040 *  {@link TokenTypes#SEMI SEMI},
041 *  {@link TokenTypes#ELLIPSIS ELLIPSIS},
042 *  {@link TokenTypes#AT AT},
043 *  {@link TokenTypes#LPAREN LPAREN},
044 *  {@link TokenTypes#RPAREN RPAREN},
045 *  {@link TokenTypes#ARRAY_DECLARATOR ARRAY_DECLARATOR},
046 *  {@link TokenTypes#RBRACK RBRACK},
047 * </p>
048 * <p>
049 * Code example for comma and dot at the new line:
050 * </p>
051 * <pre>
052 * s
053 *    .isEmpty();
054 * foo(i
055 *    ,s);
056 * </pre>
057 *  <p>
058 * An example of how to configure the check is:
059 * </p>
060 * <pre>
061 * &lt;module name="SeparatorWrap"/&gt;
062 * </pre>
063 * <p>
064 * Code example for comma and dot at the previous line:
065 * </p>
066 * <pre>
067 * s.
068 *    isEmpty();
069 * foo(i,
070 *    s);
071 * </pre>
072 * <p> An example of how to configure the check for comma at the
073 * new line is:
074 * </p>
075 * <pre>
076 * &lt;module name="SeparatorWrap"&gt;
077 *     &lt;property name="tokens" value="COMMA"/&gt;
078 *     &lt;property name="option" value="nl"/&gt;
079 * &lt;/module&gt;
080 * </pre>
081 *
082 * @author maxvetrenko
083 */
084@StatelessCheck
085public class SeparatorWrapCheck
086    extends AbstractCheck {
087
088    /**
089     * A key is pointing to the warning message text in "messages.properties"
090     * file.
091     */
092    public static final String MSG_LINE_PREVIOUS = "line.previous";
093
094    /**
095     * A key is pointing to the warning message text in "messages.properties"
096     * file.
097     */
098    public static final String MSG_LINE_NEW = "line.new";
099
100    /** The policy to enforce. */
101    private WrapOption option = WrapOption.EOL;
102
103    /**
104     * Set the option to enforce.
105     * @param optionStr string to decode option from
106     * @throws IllegalArgumentException if unable to decode
107     */
108    public void setOption(String optionStr) {
109        try {
110            option = WrapOption.valueOf(optionStr.trim().toUpperCase(Locale.ENGLISH));
111        }
112        catch (IllegalArgumentException iae) {
113            throw new IllegalArgumentException("unable to parse " + optionStr, iae);
114        }
115    }
116
117    @Override
118    public int[] getDefaultTokens() {
119        return new int[] {
120            TokenTypes.DOT,
121            TokenTypes.COMMA,
122        };
123    }
124
125    @Override
126    public int[] getAcceptableTokens() {
127        return new int[] {
128            TokenTypes.DOT,
129            TokenTypes.COMMA,
130            TokenTypes.SEMI,
131            TokenTypes.ELLIPSIS,
132            TokenTypes.AT,
133            TokenTypes.LPAREN,
134            TokenTypes.RPAREN,
135            TokenTypes.ARRAY_DECLARATOR,
136            TokenTypes.RBRACK,
137            TokenTypes.METHOD_REF,
138        };
139    }
140
141    @Override
142    public int[] getRequiredTokens() {
143        return CommonUtils.EMPTY_INT_ARRAY;
144    }
145
146    @Override
147    public void visitToken(DetailAST ast) {
148        final String text = ast.getText();
149        final int colNo = ast.getColumnNo();
150        final int lineNo = ast.getLineNo();
151        final String currentLine = getLines()[lineNo - 1];
152        final String substringAfterToken =
153                currentLine.substring(colNo + text.length()).trim();
154        final String substringBeforeToken =
155                currentLine.substring(0, colNo).trim();
156
157        if (option == WrapOption.EOL
158                && substringBeforeToken.isEmpty()) {
159            log(lineNo, colNo, MSG_LINE_PREVIOUS, text);
160        }
161        else if (option == WrapOption.NL
162                 && substringAfterToken.isEmpty()) {
163            log(lineNo, colNo, MSG_LINE_NEW, text);
164        }
165    }
166}