001// Licensed under the Apache License, Version 2.0 (the "License");
002// you may not use this file except in compliance with the License.
003// You may obtain a copy of the License at
004//
005// http://www.apache.org/licenses/LICENSE-2.0
006//
007// Unless required by applicable law or agreed to in writing, software
008// distributed under the License is distributed on an "AS IS" BASIS,
009// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
010// See the License for the specific language governing permissions and
011// limitations under the License.
012
013package org.apache.tapestry5.internal.transform;
014
015import org.apache.tapestry5.commons.ObjectLocator;
016import org.apache.tapestry5.commons.util.ExceptionUtils;
017import org.apache.tapestry5.func.F;
018import org.apache.tapestry5.func.Predicate;
019import org.apache.tapestry5.ioc.OperationTracker;
020import org.apache.tapestry5.ioc.annotations.Inject;
021import org.apache.tapestry5.model.MutableComponentModel;
022import org.apache.tapestry5.plastic.PlasticClass;
023import org.apache.tapestry5.plastic.PlasticField;
024import org.apache.tapestry5.services.transform.ComponentClassTransformWorker2;
025import org.apache.tapestry5.services.transform.InjectionProvider2;
026import org.apache.tapestry5.services.transform.TransformationSupport;
027
028/**
029 * Performs injection triggered by any field annotated with the {@link org.apache.tapestry5.ioc.annotations.Inject}
030 * annotation or the {@link javax.inject.Inject} annotation.
031 *
032 * The implementation of this worker mostly delegates to a chain of command of {@link InjectionProvider2}.
033 */
034public class InjectWorker implements ComponentClassTransformWorker2
035{
036    private final ObjectLocator locator;
037
038    // Really, a chain of command
039
040    private final InjectionProvider2 injectionProvider;
041
042    private final OperationTracker tracker;
043
044    private final Predicate<PlasticField> MATCHER = new Predicate<PlasticField>()
045    {
046        public boolean accept(PlasticField field)
047        {
048            return field.hasAnnotation(Inject.class) ||
049                    field.hasAnnotation(javax.inject.Inject.class);
050        }
051    };
052
053    public InjectWorker(ObjectLocator locator, InjectionProvider2 injectionProvider, OperationTracker tracker)
054    {
055        this.locator = locator;
056        this.injectionProvider = injectionProvider;
057        this.tracker = tracker;
058    }
059
060    public void transform(final PlasticClass plasticClass, TransformationSupport support, final MutableComponentModel model)
061    {
062        for (final PlasticField field : F.flow(plasticClass.getUnclaimedFields()).filter(MATCHER))
063        {
064            final String fieldName = field.getName();
065
066            tracker.run(String.format("Injecting field  %s.%s", plasticClass.getClassName(), fieldName), new Runnable()
067            {
068                public void run()
069                {
070                    try
071                    {
072                        boolean success = injectionProvider.provideInjection(field, locator, model);
073
074                        if (success)
075                        {
076                            field.claim("@Inject");
077                        }
078                    } catch (RuntimeException ex)
079                    {
080                        throw new RuntimeException(String.format("Error obtaining injected value for field %s.%s: %s", plasticClass.getClassName(), fieldName, ExceptionUtils.toMessage(ex)), ex);
081                    }
082                }
083            });
084        }
085    }
086}