/***
* Copyright (c) 2009 Caelum - www.caelum.com.br/opensource All rights reserved.
*
* Licensed under the Apache License, Version 2.0 (the "License"); you may not
* use this file except in compliance with the License. You may obtain a copy of
* the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
* WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
* License for the specific language governing permissions and limitations under
* the License.
*/
package br.com.caelum.vraptor.view;
import java.lang.reflect.Method;
import java.util.Arrays;
import java.util.Collections;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ConcurrentMap;
import java.util.concurrent.locks.Lock;
import java.util.concurrent.locks.ReentrantLock;
import javassist.CannotCompileException;
import javassist.ClassPool;
import javassist.CtClass;
import javassist.CtMethod;
import javassist.CtNewMethod;
import javax.annotation.PostConstruct;
import javax.enterprise.context.ApplicationScoped;
import javax.inject.Inject;
import javax.inject.Named;
import javax.servlet.ServletContext;
import net.vidageek.mirror.dsl.Mirror;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import br.com.caelum.vraptor.controller.BeanClass;
import br.com.caelum.vraptor.http.route.Router;
import br.com.caelum.vraptor.proxy.MethodInvocation;
import br.com.caelum.vraptor.proxy.Proxifier;
import br.com.caelum.vraptor.proxy.ProxyCreationException;
import br.com.caelum.vraptor.proxy.SuperMethod;
import br.com.caelum.vraptor.util.StringUtils;
import com.google.common.collect.ForwardingMap;
/**
* View helper for generating uris
* @author Lucas Cavalcanti
* @since 3.4.0
*
*/
@Named("linkTo")
@ApplicationScoped
public class LinkToHandler extends ForwardingMap<Class<?>, Object> {
private static final Logger logger = LoggerFactory.getLogger(LinkToHandler.class);
private ServletContext context;
private Router router;
private Proxifier proxifier;
private ConcurrentMap<Class<?>, Class<?>> interfaces = new ConcurrentHashMap<>();
@Deprecated // CDI eyes only
public LinkToHandler() {
}
@Inject
public LinkToHandler(ServletContext context, Router router, Proxifier proxifier) {
this.context = context;
this.router = router;
this.proxifier = proxifier;
}
@PostConstruct
public void start() {
logger.info("Registering linkTo component");
}
@Override
protected Map<Class<?>, Object> delegate() {
return Collections.emptyMap();
}
private Lock lock = new ReentrantLock();
@Override
public Object get(Object key) {
BeanClass beanClass = (BeanClass) key;
final Class<?> controller = beanClass.getType();
Class<?> linkToInterface = interfaces.get(controller);
if (linkToInterface == null) {
lock.lock();
try {
linkToInterface = interfaces.get(controller);
if (linkToInterface == null) {
String interfaceName = controller.getName() + "$linkTo";
linkToInterface = createLinkToInterface(controller, interfaceName);
interfaces.put(controller, linkToInterface);
}
} finally {
lock.unlock();
}
}
return proxifier.proxify(linkToInterface, new MethodInvocation<Object>() {
@Override
public Object intercept(Object proxy, Method method, Object[] args, SuperMethod superMethod) {
String methodName = StringUtils.decapitalize(method.getName().replaceFirst("^get", ""));
List<Object> params = args.length == 0 ? Collections.emptyList() : Arrays.asList((Object[]) args[0]);
return new Linker(controller, methodName, params).getLink();
}
});
}
private Class<?> createLinkToInterface(final Class<?> controller, String interfaceName) {
try {
return Class.forName(interfaceName);
} catch (ClassNotFoundException e1) {
// ok, continue
}
ClassPool pool = ClassPool.getDefault();
CtClass inter = pool.makeInterface(interfaceName);
try {
for(String name : getMethodNames(controller)) {
CtMethod method = CtNewMethod.make(String.format("abstract String %s(Object[] args);", name), inter);
method.setModifiers(method.getModifiers() | 128 /* Modifier.VARARGS */);
inter.addMethod(method);
CtMethod getter = CtNewMethod.make(String.format("abstract String get%s();", StringUtils.capitalize(name)), inter);
inter.addMethod(getter);
}
return inter.toClass();
} catch (CannotCompileException e) {
throw new ProxyCreationException(e);
}
}
private Set<String> getMethodNames(Class<?> controller) {
Set<String> names = new HashSet<>();
for (Method method : new Mirror().on(controller).reflectAll().methods()) {
if (!method.getDeclaringClass().equals(Object.class)) {
names.add(method.getName());
}
}
return names;
}
class Linker {
private final List<Object> args;
private final String methodName;
private final Class<?> controller;
public Linker(Class<?> controller, String methodName, List<Object> args) {
this.controller = controller;
this.methodName = methodName;
this.args = args;
}
public String getLink() {
Method method = null;
if (getMethodsAmountWithSameName() > 1) {
method = new Mirror().on(controller).reflect().method(methodName).withArgs(getClasses(args));
if (method == null && args.isEmpty()) {
throw new IllegalArgumentException("Ambiguous method '" + methodName + "' on " + controller + ". Try to add some parameters to resolve ambiguity, or use different method names.");
}
} else {
method = findMethodWithName(controller, methodName);
}
if(method == null) {
throw new IllegalArgumentException(
String.format("There are no methods on %s named '%s' that receives args of types %s",
controller, methodName, Arrays.toString(getClasses(args))));
}
return context.getContextPath() + router.urlFor(controller, method, getArgs(method));
}
private Object[] getArgs(Method method) {
int methodParamsQuantity = method.getParameterTypes().length;
if (args.size() == methodParamsQuantity)
return args.toArray();
if (args.size() > methodParamsQuantity)
throw new IllegalArgumentException(String.format("linkTo param args must have the same or lower length as method param args. linkTo args: %d | method args: %d", args.size(), methodParamsQuantity));
Object[] noMissingParamsArgs = new Object[methodParamsQuantity];
System.arraycopy(args.toArray(), 0, noMissingParamsArgs, 0, args.size());
return noMissingParamsArgs;
}
private Method findMethodWithName(Class<?> type, String name) {
for (Method method : type.getDeclaredMethods()) {
if (!method.isBridge() && method.getName().equals(name)) {
return method;
}
}
if (type.getSuperclass().equals(Object.class)) {
return null;
}
return findMethodWithName(type.getSuperclass(), name);
}
private int getMethodsAmountWithSameName() {
int amount = 0;
for (Method method : controller.getDeclaredMethods()) {
if (!method.isBridge() && method.getName().equals(methodName)) {
amount++;
}
}
return amount;
}
private Class<?>[] getClasses(List<Object> params) {
Class<?>[] classes = new Class<?>[params.size()];
for(int i = 0; i < params.size(); i ++) {
classes[i] = params.get(i).getClass();
}
return classes;
}
}
}