org.walkmod.javalang.compiler.reflection.SymbolDataOfMethodReferenceBuilder Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of javalang-compiler Show documentation
Library of compiler components to processs Java code.
There is a newer version: 2.3.10
/*
 Copyright (C) 2015 Raquel Pau and Albert Coroleu.
 
Walkmod is free software: you can redistribute it and/or modify
it under the terms of the GNU Lesser General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.

Walkmod is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU Lesser General Public License for more details.

You should have received a copy of the GNU Lesser General Public License
along with Walkmod.  If not, see .*/
package org.walkmod.javalang.compiler.reflection;

import java.lang.reflect.GenericArrayType;
import java.util.Iterator;
import java.util.List;
import java.util.Map;

import org.walkmod.javalang.ast.MethodSymbolData;
import org.walkmod.javalang.ast.expr.Expression;
import org.walkmod.javalang.ast.expr.MethodCallExpr;
import org.walkmod.javalang.ast.expr.MethodReferenceExpr;
import org.walkmod.javalang.compiler.symbols.SymbolType;
import org.walkmod.javalang.visitors.VoidVisitor;

public class SymbolDataOfMethodReferenceBuilder implements TypeMappingBuilder {

	private Map typeMapping;
	
	private VoidVisitor visitor;
	
	private T ctxt;
	
	public  SymbolDataOfMethodReferenceBuilder(Map typeMapping, VoidVisitor visitor, T ctxt){
		this.typeMapping = typeMapping;
		this.visitor = visitor;
		this.ctxt = ctxt;
	}
	
	@Override
	public MethodCallExpr build(MethodCallExpr n) throws Exception {
		List args = n.getArgs();
		MethodSymbolData st =  n.getSymbolData();
		if (args != null) {
			int i = 0;
			if(st == null){
				Expression scope = n.getScope();
				String scopeName = "empty";
				if(scope!= null){
					scopeName = scope.getSymbolData().toString();
				}
				String argsTypeName = "empty";
				List argExpr = n.getArgs();
				if(argExpr != null){
					argsTypeName = "[";
					Iterator it = argExpr.iterator();
					while(it.hasNext()){
						Expression arg = it.next();
						if(arg != null && arg.getSymbolData() != null){
							argsTypeName += " "+arg.getSymbolData().toString();
						}
						else{
							argsTypeName +=" null";
						}
						if(it.hasNext()){
							argsTypeName+=",";
						}
					}
					argsTypeName+="]";
				}
				throw new Exception("Ops! The method call "+n.toString()+" is not resolved. The scope is ["+scopeName+"] , and the args are : "+argsTypeName);
			}
			java.lang.reflect.Type[] argClasses = st.getMethod()
					.getGenericParameterTypes();
			int paramCount = st.getMethod().getParameterTypes().length;
			for (Expression argument : args) {
				if (argument instanceof MethodReferenceExpr) {
					SymbolType aux = null;
					if (i < paramCount) {
						aux = SymbolType.valueOf(argClasses[i],
								typeMapping);

					} else {
						java.lang.reflect.Type componentType = null;
						java.lang.reflect.Type lastArg = argClasses[argClasses.length - 1];
						if (lastArg instanceof Class) {
							componentType = ((Class) lastArg)
									.getComponentType();
						} else if (lastArg instanceof GenericArrayType) {
							componentType = ((GenericArrayType) lastArg)
									.getGenericComponentType();
						}
						aux = SymbolType.valueOf(componentType,
								typeMapping);
					}
					argument.setSymbolData(aux);
					argument.accept(visitor, ctxt);
				}
				i++;
			}
		}
		return null;
	}

	@Override
	public void setTypeMapping(Map typeMapping) {
		this.typeMapping = typeMapping;
	}

}