001////////////////////////////////////////////////////////////////////////////////
002// checkstyle: Checks Java source code for adherence to a set of rules.
003// Copyright (C) 2001-2020 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.coding;
021
022import com.puppycrawl.tools.checkstyle.StatelessCheck;
023import com.puppycrawl.tools.checkstyle.api.AbstractCheck;
024import com.puppycrawl.tools.checkstyle.api.DetailAST;
025import com.puppycrawl.tools.checkstyle.api.TokenTypes;
026
027/**
028 * <p>
029 * Checks that switch statement has a {@code default} clause.
030 * </p>
031 * <p>
032 * Rationale: It's usually a good idea to introduce a
033 * default case in every switch statement. Even if
034 * the developer is sure that all currently possible
035 * cases are covered, this should be expressed in the
036 * default branch, e.g. by using an assertion. This way
037 * the code is protected against later changes, e.g.
038 * introduction of new types in an enumeration type.
039 * </p>
040 * <p>
041 * To configure the check:
042 * </p>
043 * <pre>
044 * &lt;module name=&quot;MissingSwitchDefault&quot;/&gt;
045 * </pre>
046 * <p>Example of violation:</p>
047 * <pre>
048 * switch (i) {    // violation
049 *  case 1:
050 *    break;
051 *  case 2:
052 *    break;
053 * }
054 * </pre>
055 * <p>Example of correct code:</p>
056 * <pre>
057 * switch (i) {
058 *  case 1:
059 *    break;
060 *  case 2:
061 *    break;
062 *  default: // OK
063 *    break;
064 * }
065 * </pre>
066 *
067 * @since 3.1
068 */
069@StatelessCheck
070public class MissingSwitchDefaultCheck extends AbstractCheck {
071
072    /**
073     * A key is pointing to the warning message text in "messages.properties"
074     * file.
075     */
076    public static final String MSG_KEY = "missing.switch.default";
077
078    @Override
079    public int[] getDefaultTokens() {
080        return getRequiredTokens();
081    }
082
083    @Override
084    public int[] getAcceptableTokens() {
085        return getRequiredTokens();
086    }
087
088    @Override
089    public int[] getRequiredTokens() {
090        return new int[] {TokenTypes.LITERAL_SWITCH};
091    }
092
093    @Override
094    public void visitToken(DetailAST ast) {
095        final DetailAST firstCaseGroupAst = ast.findFirstToken(TokenTypes.CASE_GROUP);
096
097        if (!containsDefaultSwitch(firstCaseGroupAst)) {
098            log(ast, MSG_KEY);
099        }
100    }
101
102    /**
103     * Checks if the case group or its sibling contain the 'default' switch.
104     *
105     * @param caseGroupAst first case group to check.
106     * @return true if 'default' switch found.
107     */
108    private static boolean containsDefaultSwitch(DetailAST caseGroupAst) {
109        DetailAST nextAst = caseGroupAst;
110        boolean found = false;
111
112        while (nextAst != null) {
113            if (nextAst.findFirstToken(TokenTypes.LITERAL_DEFAULT) != null) {
114                found = true;
115                break;
116            }
117
118            nextAst = nextAst.getNextSibling();
119        }
120
121        return found;
122    }
123
124}