1 /** 2 * This file Copyright (c) 2013-2018 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 java.lang.reflect.Method; 37 import java.util.Collection; 38 import java.util.List; 39 import java.util.Map; 40 41 import com.google.common.base.Function; 42 import com.google.common.collect.Collections2; 43 import com.google.common.collect.Lists; 44 import com.google.common.collect.Maps; 45 import com.thoughtworks.proxy.toys.decorate.Decorator; 46 47 /** 48 * A {@link Decorator} which decorates return values by passing them to {@link ProxytoysI18nizer#decorateChild(Object, Object)}. 49 * If the returned value is a Collection, each item in the collection is decorated. 50 * 51 * @param <P> the decorated type 52 */ 53 class ChildDecorator<P> extends Decorator<P> { 54 private final ProxytoysI18nizer proxyCreator; 55 56 ChildDecorator(ProxytoysI18nizer proxyCreator) { 57 this.proxyCreator = proxyCreator; 58 } 59 60 @Override 61 public Object decorateResult(P proxy, Method method, Object[] args, Object result) { 62 final P parent = proxy; 63 final Object decoratedReturnValue; 64 if (result instanceof Map) { 65 decoratedReturnValue = Maps.transformValues((Map) result, new ChildDecoratorFunction(parent)); 66 } else if (result instanceof Collection) { 67 // We can't use Collections2.transform() or Lists.transform() as these create new immutable collections 68 if (result instanceof List) { 69 //this is wrapped in newArrayList() to make it modifiable since info.magnolia.ui.contentapp.movedialog.MoveDialogPresenterImpl#prepareWorkbenchDefinition relies on it (just for compatibility although it should be fixed there) 70 decoratedReturnValue = Lists.newArrayList(Lists.transform(((List) result), new ChildDecoratorFunction(parent))); 71 } else { 72 decoratedReturnValue = Collections2.transform((Collection) result, new ChildDecoratorFunction(parent)); 73 } 74 } else { 75 decoratedReturnValue = proxyCreator.decorateChild(result, parent); 76 } 77 return decoratedReturnValue; 78 } 79 80 private class ChildDecoratorFunction implements Function { 81 private final P parent; 82 83 public ChildDecoratorFunction(P parent) { 84 this.parent = parent; 85 } 86 87 @Override 88 public Object apply(Object element) { 89 return proxyCreator.decorateChild(element, parent); 90 } 91 } 92 }