001    // Copyright 2005 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.tapestry.annotations;
016    
017    import java.lang.reflect.Method;
018    
019    import org.apache.hivemind.Location;
020    import org.apache.tapestry.enhance.EnhancementOperation;
021    import org.apache.tapestry.spec.IComponentSpecification;
022    import org.apache.tapestry.spec.InjectSpecification;
023    import org.apache.tapestry.spec.InjectSpecificationImpl;
024    
025    /**
026     * 
027     * @author Howard Lewis Ship
028     */
029    public class InjectStateFlagAnnotationWorker implements MethodAnnotationEnhancementWorker
030    {
031    
032        public void performEnhancement(EnhancementOperation op, IComponentSpecification spec,
033                Method method, Location location)
034        {
035            InjectStateFlag isv = method.getAnnotation(InjectStateFlag.class);
036            
037            String keyName = isv.value();
038    
039            if (keyName.equals(""))
040            {
041                keyName = AnnotationUtils.convertMethodNameToKeyName(method.getName());
042                if (keyName.endsWith("-exists")) // strip it
043                {
044                    int length = keyName.length();
045                    keyName = keyName.substring(0, length - 7);
046                }
047            }
048    
049            String propertyName = AnnotationUtils.getPropertyName(method);
050    
051            InjectSpecification inject = new InjectSpecificationImpl();
052    
053            inject.setType("state-flag");
054            inject.setProperty(propertyName);
055            inject.setObject(keyName);
056            inject.setLocation(location);
057    
058            spec.addInjectSpecification(inject);
059        }
060    
061    }