org.wickedsource.docxstamper.proxy.ProxyMethodHandler Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of docx-stamper Show documentation

Template engine for .docx documents.

There is a newer version: 1.2.2

package org.wickedsource.docxstamper.proxy;

import javassist.util.proxy.MethodHandler;
import org.wickedsource.docxstamper.api.commentprocessor.ICommentProcessor;

import java.lang.reflect.Method;
import java.util.Map;

public class ProxyMethodHandler implements MethodHandler {

	private final Object contextRoot;

	private final Map, ICommentProcessor> interfacesWithImplementations;


	public ProxyMethodHandler(Object root,
							  Map, ICommentProcessor> interfacesWithImplementations) {
		this.contextRoot = root;
		this.interfacesWithImplementations = interfacesWithImplementations;
		for (Map.Entry, ICommentProcessor> entry : interfacesWithImplementations.entrySet()) {
			Class interfaceClass = entry.getKey();
			ICommentProcessor implementation = entry.getValue();
			if (!interfaceClass.isAssignableFrom(implementation.getClass())) {
				throw new IllegalArgumentException(
						String.format("%s does not implement %s!", implementation, interfaceClass));
			}
		}
	}

	@Override
	public Object invoke(Object o, Method method, Method method2, Object[] args) throws Throwable {
		for (Map.Entry, ICommentProcessor> entry : interfacesWithImplementations.entrySet()) {
			Class interfaceClass = entry.getKey();
			ICommentProcessor implementation = entry.getValue();
			if (methodCanBeHandledByInterface(method, interfaceClass)) {
				return method.invoke(implementation, args);
			}
		}

		return method.invoke(contextRoot, args);
	}

	public boolean methodCanBeHandledByInterface(Method method, Class interfaceClass) {
		try {
			interfaceClass.getMethod(method.getName(), method.getParameterTypes());
			// no exception, so we found the method we're looking for
			return true;
		} catch (NoSuchMethodException e) {
			return false;
		}
	}

}