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.enhance;
016    
017    import org.apache.tapestry.IComponent;
018    import org.apache.tapestry.spec.IComponentSpecification;
019    
020    /**
021     * Checks to see if the component implements a specific interface, and adds code
022     * to the component's finishLoad() method to register (with the page) for such
023     * notifications.
024     * 
025     * @author Howard M. Lewis Ship
026     * @since 4.0
027     */
028    public class InjectListenerRegistrationWorker implements EnhancementWorker
029    {
030    
031        private Class _listenerInterface;
032    
033        private String _registerMethodName;
034    
035        public void performEnhancement(EnhancementOperation op,
036                IComponentSpecification spec)
037        {
038    
039            if (op.implementsInterface(_listenerInterface))
040            {
041                op.extendMethodImplementation(IComponent.class,
042                        EnhanceUtils.FINISH_LOAD_SIGNATURE, "getPage()."
043                                + _registerMethodName + "(this);");
044            }
045        }
046    
047        public void setListenerInterface(Class listenerInterface)
048        {
049            _listenerInterface = listenerInterface;
050        }
051    
052        public void setRegisterMethodName(String registerMethodName)
053        {
054            _registerMethodName = registerMethodName;
055        }
056    }