    001    // Copyright 2009, 2010, 2011 The Apache Software Foundation
    002    //
    003    // Licensed under the Apache License, Version 2.0 (the "License");
    004    // you may not use this file except in compliance with the License.
    005    // You may obtain a copy of the License at
    006    //
    007    // http://www.apache.org/licenses/LICENSE-2.0
    008    //
    009    // Unless required by applicable law or agreed to in writing, software
    010    // distributed under the License is distributed on an "AS IS" BASIS,
    011    // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    012    // See the License for the specific language governing permissions and
    013    // limitations under the License.
    014
    015    package org.apache.tapestry5.internal.transform;
    016
    017    import org.apache.tapestry5.internal.services.ComponentClassCache;
    018    import org.apache.tapestry5.ioc.ObjectLocator;
    019    import org.apache.tapestry5.ioc.annotations.InjectService;
    020    import org.apache.tapestry5.model.MutableComponentModel;
    021    import org.apache.tapestry5.plastic.PlasticClass;
    022    import org.apache.tapestry5.plastic.PlasticField;
    023    import org.apache.tapestry5.services.transform.ComponentClassTransformWorker2;
    024    import org.apache.tapestry5.services.transform.TransformationSupport;
    025
    026    /**
    027     * Processes the {@link org.apache.tapestry5.ioc.annotations.InjectService} annotation.
    028     *
    029     * @since 5.1.0.0
    030     */
    031    public class InjectServiceWorker implements ComponentClassTransformWorker2
    032    {
    033        private final ObjectLocator locator;
    034
    035        private final ComponentClassCache cache;
    036
    037        public InjectServiceWorker(ObjectLocator locator, ComponentClassCache cache)
    038        {
    039            this.locator = locator;
    040            this.cache = cache;
    041        }
    042
    043        public void transform(PlasticClass plasticClass, TransformationSupport support, MutableComponentModel model)
    044        {
    045            for (PlasticField field : plasticClass.getFieldsWithAnnotation(InjectService.class))
    046            {
    047                InjectService annotation = field.getAnnotation(InjectService.class);
    048
    049                field.claim(annotation);
    050
    051                Class fieldType = cache.forName(field.getTypeName());
    052
    053                Object service = locator.getService(annotation.value(), fieldType);
    054
    055                field.inject(service);
    056            }
    057        }
    058    }