1 /**
2 * This file Copyright (c) 2013 Magnolia International
3 * Ltd. (http://www.magnolia-cms.com). All rights reserved.
4 *
5 *
6 * This file is dual-licensed under both the Magnolia
7 * Network Agreement and the GNU General Public License.
8 * You may elect to use one or the other of these licenses.
9 *
10 * This file is distributed in the hope that it will be
11 * useful, but AS-IS and WITHOUT ANY WARRANTY; without even the
12 * implied warranty of MERCHANTABILITY or FITNESS FOR A
13 * PARTICULAR PURPOSE, TITLE, or NONINFRINGEMENT.
14 * Redistribution, except as permitted by whichever of the GPL
15 * or MNA you select, is prohibited.
16 *
17 * 1. For the GPL license (GPL), you can redistribute and/or
18 * modify this file under the terms of the GNU General
19 * Public License, Version 3, as published by the Free Software
20 * Foundation. You should have received a copy of the GNU
21 * General Public License, Version 3 along with this program;
22 * if not, write to the Free Software Foundation, Inc., 51
23 * Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
24 *
25 * 2. For the Magnolia Network Agreement (MNA), this file
26 * and the accompanying materials are made available under the
27 * terms of the MNA which accompanies this distribution, and
28 * is available at http://www.magnolia-cms.com/mna.html
29 *
30 * Any modifications to this file must keep this entire header
31 * intact.
32 *
33 */
34 package info.magnolia.i18nsystem.proxytoys;
35
36 import com.google.common.base.Predicate;
37 import com.thoughtworks.proxy.toys.decorate.Decorator;
38 import java.lang.reflect.Method;
39
40 /**
41 * A {@link Decorator} which delegates to another only for methods accepted by the given {@link Predicate}.
42 *
43 * TODO: methodSelector is called 4 times, but I couldn't find a way to hook this into ProxyToys without rewriting more code.
44 *
45 * @param <T> the decorated type
46 */
47 public class FilteringMethodDecorator<T> extends Decorator<T> {
48 private final Decorator<T> delegate;
49 private final Predicate<Method> methodSelector;
50
51 public FilteringMethodDecorator(Decorator<T> delegate, Predicate<Method> methodSelector) {
52 this.delegate = delegate;
53 this.methodSelector = methodSelector;
54 }
55
56 @Override
57 public Object[] beforeMethodStarts(T proxy, Method method, Object[] args) {
58 if (methodSelector.apply(method)) {
59 return delegate.beforeMethodStarts(proxy, method, args);
60 } else {
61 return super.beforeMethodStarts(proxy, method, args);
62 }
63 }
64
65 @Override
66 public Object decorateResult(T proxy, Method method, Object[] args, Object result) {
67 if (methodSelector.apply(method)) {
68 return delegate.decorateResult(proxy, method, args, result);
69 } else {
70 return super.decorateResult(proxy, method, args, result);
71 }
72 }
73
74 @Override
75 public Throwable decorateTargetException(T proxy, Method method, Object[] args, Throwable cause) {
76 if (methodSelector.apply(method)) {
77 return delegate.decorateTargetException(proxy, method, args, cause);
78 } else {
79 return super.decorateTargetException(proxy, method, args, cause);
80 }
81 }
82
83 @Override
84 public Exception decorateInvocationException(T proxy, Method method, Object[] args, Exception cause) {
85 if (methodSelector.apply(method)) {
86 return delegate.decorateInvocationException(proxy, method, args, cause);
87 } else {
88 return super.decorateInvocationException(proxy, method, args, cause);
89 }
90 }
91 }