Home | History | Annotate | Download | only in typesolvers
      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 
     17 package com.github.javaparser.symbolsolver.resolution.typesolvers;
     18 
     19 import com.github.javaparser.resolution.UnsolvedSymbolException;
     20 import com.github.javaparser.resolution.declarations.ResolvedReferenceTypeDeclaration;
     21 import com.github.javaparser.symbolsolver.model.resolution.SymbolReference;
     22 import com.github.javaparser.symbolsolver.model.resolution.TypeSolver;
     23 
     24 import java.util.ArrayList;
     25 import java.util.List;
     26 
     27 /**
     28  * @author Federico Tomassetti
     29  */
     30 public class CombinedTypeSolver implements TypeSolver {
     31 
     32     private TypeSolver parent;
     33     private List<TypeSolver> elements = new ArrayList<>();
     34 
     35     public CombinedTypeSolver(TypeSolver... elements) {
     36         for (TypeSolver el : elements) {
     37             add(el);
     38         }
     39     }
     40 
     41     @Override
     42     public TypeSolver getParent() {
     43         return parent;
     44     }
     45 
     46     @Override
     47     public void setParent(TypeSolver parent) {
     48         this.parent = parent;
     49     }
     50 
     51     public void add(TypeSolver typeSolver) {
     52         this.elements.add(typeSolver);
     53         typeSolver.setParent(this);
     54     }
     55 
     56     @Override
     57     public SymbolReference<ResolvedReferenceTypeDeclaration> tryToSolveType(String name) {
     58         for (TypeSolver ts : elements) {
     59             SymbolReference<ResolvedReferenceTypeDeclaration> res = ts.tryToSolveType(name);
     60             if (res.isSolved()) {
     61                 return res;
     62             }
     63         }
     64         return SymbolReference.unsolved(ResolvedReferenceTypeDeclaration.class);
     65     }
     66 
     67     @Override
     68     public ResolvedReferenceTypeDeclaration solveType(String name) throws UnsolvedSymbolException {
     69         SymbolReference<ResolvedReferenceTypeDeclaration> res = tryToSolveType(name);
     70         if (res.isSolved()) {
     71             return res.getCorrespondingDeclaration();
     72         } else {
     73             throw new UnsolvedSymbolException(name);
     74         }
     75     }
     76 }
     77