org.chromattic.groovy.ChromatticAnnotationMover Maven / Gradle / Ivy

Go to download
/*
* Copyright (C) 2003-2009 eXo Platform SAS.
*
* This is free software; you can redistribute it and/or modify it
* under the terms of the GNU Lesser General Public License as
* published by the Free Software Foundation; either version 2.1 of
* the License, or (at your option) any later version.
*
* This software is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this software; if not, write to the Free
* Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
* 02110-1301 USA, or see the FSF site: http://www.fsf.org.
*/

package org.chromattic.groovy;

import org.chromattic.groovy.annotations.ChromatticDelegation;
import org.chromattic.groovy.annotations.GroovyInstrumentor;
import org.chromattic.groovy.exceptions.NoSuchGetterException;
import org.chromattic.groovy.exceptions.NoSuchSetterException;
import org.codehaus.groovy.ast.*;

import java.util.Iterator;

/**
 * @author Alain Defrance
 * @version $Revision$
 */
public class ChromatticAnnotationMover {
  
  public void addFieldAnnotationToMethod(ClassNode classNode, FieldNode fieldNode, AnnotationNode annotationNode) throws NoSuchGetterException {
    MethodNode getterNode = GroovyUtils.getGetter(classNode, fieldNode);
    if (getterNode == null || !getterNode.getDeclaringClass().equals(classNode)) throw new NoSuchGetterException("Cannot apply chromattic annotations because getter don't exist for : " + fieldNode.getName());
    getterNode.addAnnotation(annotationNode);
  }

  public void addSetterDelegationAnnotation(ClassNode classNode, FieldNode fieldNode) throws NoSuchSetterException {
    MethodNode setterNode = GroovyUtils.getSetter(classNode, fieldNode);
    if (setterNode == null || !setterNode.getDeclaringClass().equals(classNode)) throw new NoSuchSetterException("Cannot apply annotation @ChromatticDelegation because setter don't exist for : " + fieldNode.getName());
    setterNode.addAnnotation(new AnnotationNode(new ClassNode(ChromatticDelegation.class)));
  }

  public void generateGetter(ClassNode classNode, FieldNode fieldNode) {
      GroovyUtils.createGetter(classNode, fieldNode);
  }

  public void generateGetter(ClassNode classNode, FieldNode fieldNode, AnnotationNode defaultAnnotatedNode) {
    generateGetter(classNode, fieldNode);
    try {
      addFieldAnnotationToMethod(classNode, fieldNode, defaultAnnotatedNode);
    } catch (NoSuchGetterException ignore) { }
  }

  public void generateSetter(ClassNode classNode, FieldNode fieldNode) {
    GroovyUtils.createSetter(classNode, fieldNode);
    try {
      addSetterDelegationAnnotation(classNode, fieldNode);
    } catch (NoSuchSetterException ignore) { }
  }

  public void removeChromatticAnnotation(FieldNode fieldNode) {
    Iterator it = fieldNode.getAnnotations().iterator();
    while(it.hasNext())
      if (it.next().getClassNode().getName().startsWith(GroovyUtils.ANNOTATIONS_PACKAGE))
        it.remove();
  }

  public void applyGroovyInstrumentor(ClassNode classNode) {
    classNode.addAnnotation(new AnnotationNode(new ClassNode(GroovyInstrumentor.class)));
  }
}