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