FieldAccessContext.java revision 9a228915f61892f6ba99fcf1e42e42e1e2622a00 
1/*
2 * Copyright 2016 Federico Tomassetti
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17package com.github.javaparser.symbolsolver.javaparsermodel.contexts;
18
19import com.github.javaparser.ast.expr.Expression;
20import com.github.javaparser.ast.expr.FieldAccessExpr;
21import com.github.javaparser.ast.expr.ThisExpr;
22import com.github.javaparser.resolution.declarations.ResolvedMethodDeclaration;
23import com.github.javaparser.resolution.declarations.ResolvedReferenceTypeDeclaration;
24import com.github.javaparser.resolution.declarations.ResolvedTypeDeclaration;
25import com.github.javaparser.resolution.declarations.ResolvedValueDeclaration;
26import com.github.javaparser.resolution.declarations.ResolvedFieldDeclaration;
27import com.github.javaparser.resolution.types.ResolvedPrimitiveType;
28import com.github.javaparser.resolution.types.ResolvedType;
29import com.github.javaparser.symbolsolver.javaparsermodel.JavaParserFacade;
30import com.github.javaparser.symbolsolver.javaparsermodel.JavaParserFactory;
31import com.github.javaparser.symbolsolver.model.resolution.SymbolReference;
32import com.github.javaparser.symbolsolver.model.resolution.TypeSolver;
33import com.github.javaparser.symbolsolver.model.resolution.Value;
34import com.github.javaparser.symbolsolver.resolution.SymbolSolver;
35
36import java.util.Collection;
37import java.util.List;
38import java.util.Optional;
39
40import static com.github.javaparser.symbolsolver.javaparser.Navigator.getParentNode;
41import static com.github.javaparser.symbolsolver.javaparser.Navigator.requireParentNode;
42
43/**
44 * @author Federico Tomassetti
45 */
46public class FieldAccessContext extends AbstractJavaParserContext<FieldAccessExpr> {
47
48    private static final String ARRAY_LENGTH_FIELD_NAME = "length";
49
50    public FieldAccessContext(FieldAccessExpr wrappedNode, TypeSolver typeSolver) {
51        super(wrappedNode, typeSolver);
52    }
53
54    @Override
55    public SymbolReference<? extends ResolvedValueDeclaration> solveSymbol(String name, TypeSolver typeSolver) {
56        if (wrappedNode.getName().toString().equals(name)) {
57            if (wrappedNode.getScope() instanceof ThisExpr) {
58                ResolvedType typeOfThis = JavaParserFacade.get(typeSolver).getTypeOfThisIn(wrappedNode);
59                return new SymbolSolver(typeSolver).solveSymbolInType(typeOfThis.asReferenceType().getTypeDeclaration(), name);
60            }
61        }
62        return JavaParserFactory.getContext(requireParentNode(wrappedNode), typeSolver).solveSymbol(name, typeSolver);
63    }
64
65    @Override
66    public SymbolReference<ResolvedTypeDeclaration> solveType(String name, TypeSolver typeSolver) {
67        return JavaParserFactory.getContext(requireParentNode(wrappedNode), typeSolver).solveType(name, typeSolver);
68    }
69
70    @Override
71    public SymbolReference<ResolvedMethodDeclaration> solveMethod(String name, List<ResolvedType> parameterTypes, boolean staticOnly, TypeSolver typeSolver) {
72        return JavaParserFactory.getContext(requireParentNode(wrappedNode), typeSolver).solveMethod(name, parameterTypes, false, typeSolver);
73    }
74
75    @Override
76    public Optional<Value> solveSymbolAsValue(String name, TypeSolver typeSolver) {
77        Expression scope = wrappedNode.getScope();
78        if (wrappedNode.getName().toString().equals(name)) {
79            ResolvedType typeOfScope = JavaParserFacade.get(typeSolver).getType(scope);
80            if (typeOfScope.isArray() && name.equals(ARRAY_LENGTH_FIELD_NAME)) {
81                return Optional.of(new Value(ResolvedPrimitiveType.INT, ARRAY_LENGTH_FIELD_NAME));
82            }
83            if (typeOfScope.isReferenceType()) {
84                Optional<ResolvedType> typeUsage = typeOfScope.asReferenceType().getFieldType(name);
85                return typeUsage.map(resolvedType -> new Value(resolvedType, name));
86            } else {
87                return Optional.empty();
88            }
89        } else {
90            return getParent().solveSymbolAsValue(name, typeSolver);
91        }
92    }
93
94    public SymbolReference<ResolvedFieldDeclaration> solveField(String name, TypeSolver typeSolver) {
95        Collection<ResolvedReferenceTypeDeclaration> rrtds = findTypeDeclarations(Optional.of(wrappedNode.getScope()), typeSolver);
96        for (ResolvedReferenceTypeDeclaration rrtd : rrtds) {
97            try {
98                return SymbolReference.solved(rrtd.getField(wrappedNode.getName().getId()));
99            } catch (Throwable t) {
100            }
101        }
102        return SymbolReference.unsolved(ResolvedFieldDeclaration.class);
103    }
104}
105