smalidea/src/main/java/org/jf/smalidea/findUsages/SmaliWordScanner.java

/*
 * Copyright 2015, Google Inc.
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are
 * met:
 *
 *     * Redistributions of source code must retain the above copyright
 * notice, this list of conditions and the following disclaimer.
 *     * Redistributions in binary form must reproduce the above
 * copyright notice, this list of conditions and the following disclaimer
 * in the documentation and/or other materials provided with the
 * distribution.
 *     * Neither the name of Google Inc. nor the names of its
 * contributors may be used to endorse or promote products derived from
 * this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

package org.jf.smalidea.findUsages;

import com.intellij.lang.cacheBuilder.WordOccurrence;
import com.intellij.lang.cacheBuilder.WordOccurrence.Kind;
import com.intellij.lang.cacheBuilder.WordsScanner;
import com.intellij.psi.tree.IElementType;
import com.intellij.psi.tree.TokenSet;
import com.intellij.util.Processor;
import org.jetbrains.annotations.NotNull;
import org.jf.smalidea.SmaliLexer;
import org.jf.smalidea.SmaliTokens;

public class SmaliWordScanner implements WordsScanner {

    private static final TokenSet MEMBER_NAME_TOKENS = TokenSet.create(
            SmaliTokens.MEMBER_NAME,
            SmaliTokens.SIMPLE_NAME,
            SmaliTokens.ACCESS_SPEC,
            SmaliTokens.VERIFICATION_ERROR_TYPE,
            SmaliTokens.POSITIVE_INTEGER_LITERAL,
            SmaliTokens.NEGATIVE_INTEGER_LITERAL,
            SmaliTokens.FLOAT_LITERAL_OR_ID,
            SmaliTokens.DOUBLE_LITERAL_OR_ID,
            SmaliTokens.BOOL_LITERAL,
            SmaliTokens.NULL_LITERAL,
            SmaliTokens.REGISTER,
            SmaliTokens.PRIMITIVE_TYPE,
            SmaliTokens.VOID_TYPE,
            SmaliTokens.ANNOTATION_VISIBILITY,
            SmaliTokens.INSTRUCTION_FORMAT10t,
            SmaliTokens.INSTRUCTION_FORMAT10x,
            SmaliTokens.INSTRUCTION_FORMAT10x_ODEX,
            SmaliTokens.INSTRUCTION_FORMAT11x,
            SmaliTokens.INSTRUCTION_FORMAT12x_OR_ID,
            SmaliTokens.INSTRUCTION_FORMAT21c_FIELD,
            SmaliTokens.INSTRUCTION_FORMAT21c_FIELD_ODEX,
            SmaliTokens.INSTRUCTION_FORMAT21c_STRING,
            SmaliTokens.INSTRUCTION_FORMAT21c_TYPE,
            SmaliTokens.INSTRUCTION_FORMAT21t,
            SmaliTokens.INSTRUCTION_FORMAT22c_FIELD,
            SmaliTokens.INSTRUCTION_FORMAT22c_FIELD_ODEX,
            SmaliTokens.INSTRUCTION_FORMAT22c_TYPE,
            SmaliTokens.INSTRUCTION_FORMAT22cs_FIELD,
            SmaliTokens.INSTRUCTION_FORMAT22s_OR_ID,
            SmaliTokens.INSTRUCTION_FORMAT22t,
            SmaliTokens.INSTRUCTION_FORMAT23x,
            SmaliTokens.INSTRUCTION_FORMAT31i_OR_ID,
            SmaliTokens.INSTRUCTION_FORMAT31t,
            SmaliTokens.INSTRUCTION_FORMAT35c_METHOD,
            SmaliTokens.INSTRUCTION_FORMAT35c_METHOD_ODEX,
            SmaliTokens.INSTRUCTION_FORMAT35c_TYPE,
            SmaliTokens.INSTRUCTION_FORMAT35mi_METHOD,
            SmaliTokens.INSTRUCTION_FORMAT35ms_METHOD,
            SmaliTokens.INSTRUCTION_FORMAT51l);

    @Override public void processWords(CharSequence fileText, Processor<WordOccurrence> processor) {
        SmaliLexer lexer = new SmaliLexer();
        lexer.start(fileText);

        IElementType type = lexer.getTokenType();
        while (type != null) {
            if (type == SmaliTokens.CLASS_DESCRIPTOR) {
                processClassDescriptor(fileText, lexer.getTokenStart(), lexer.getTokenEnd(), processor);
            } else if (MEMBER_NAME_TOKENS.contains(type)) {
                processor.process(new WordOccurrence(fileText, lexer.getTokenStart(), lexer.getTokenEnd(), Kind.CODE));
            } else if (type == SmaliTokens.PARAM_LIST_OR_ID_PRIMITIVE_TYPE) {
                int tokenStart = lexer.getTokenStart();
                while (type == SmaliTokens.PARAM_LIST_OR_ID_PRIMITIVE_TYPE) {
                    lexer.advance();
                    type = lexer.getTokenType();
                }
                int tokenEnd = lexer.getTokenStart();
                processor.process(new WordOccurrence(fileText, tokenStart, tokenEnd, Kind.CODE));
            }
            lexer.advance();
            type = lexer.getTokenType();
        }
    }

    private void processClassDescriptor(CharSequence fileText, int tokenStart, int tokenEnd,
                                        @NotNull Processor<WordOccurrence> processor) {
        CharSequence tokenText = fileText.subSequence(tokenStart, tokenEnd);

        assert tokenText.charAt(0) == 'L' && tokenText.charAt(tokenText.length()-1) == ';';
        processor.process(new WordOccurrence(fileText, tokenStart, tokenEnd, Kind.CODE));
    }
}