001    /*
002     * Copyright (C) 2012 eXo Platform SAS.
003     *
004     * This is free software; you can redistribute it and/or modify it
005     * under the terms of the GNU Lesser General Public License as
006     * published by the Free Software Foundation; either version 2.1 of
007     * the License, or (at your option) any later version.
008     *
009     * This software is distributed in the hope that it will be useful,
010     * but WITHOUT ANY WARRANTY; without even the implied warranty of
011     * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
012     * Lesser General Public License for more details.
013     *
014     * You should have received a copy of the GNU Lesser General Public
015     * License along with this software; if not, write to the Free
016     * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
017     * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
018     */
019    
020    /*
021     * Copyright (C) 2012 eXo Platform SAS.
022     *
023     * This is free software; you can redistribute it and/or modify it
024     * under the terms of the GNU Lesser General Public License as
025     * published by the Free Software Foundation; either version 2.1 of
026     * the License, or (at your option) any later version.
027     *
028     * This software is distributed in the hope that it will be useful,
029     * but WITHOUT ANY WARRANTY; without even the implied warranty of
030     * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
031     * Lesser General Public License for more details.
032     *
033     * You should have received a copy of the GNU Lesser General Public
034     * License along with this software; if not, write to the Free
035     * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
036     * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
037     */
038    
039    package org.crsh.groovy;
040    
041    import org.codehaus.groovy.ast.ASTNode;
042    import org.codehaus.groovy.ast.AnnotationNode;
043    import org.codehaus.groovy.ast.ClassHelper;
044    import org.codehaus.groovy.ast.ClassNode;
045    import org.codehaus.groovy.ast.MethodNode;
046    import org.codehaus.groovy.control.CompilePhase;
047    import org.codehaus.groovy.control.SourceUnit;
048    import org.codehaus.groovy.transform.ASTTransformation;
049    import org.codehaus.groovy.transform.GroovyASTTransformation;
050    import org.crsh.cli.Command;
051    import org.crsh.command.CRaSHCommand;
052    
053    import java.util.logging.Logger;
054    
055    @GroovyASTTransformation(phase= CompilePhase.SEMANTIC_ANALYSIS)
056    public class CommandTransformer implements ASTTransformation {
057    
058      /** . */
059      private static final ClassNode CRASH_COMMAND = ClassHelper.make(CRaSHCommand.class);
060    
061      /** . */
062      private static final Logger log = Logger.getLogger(CommandTransformer.class.getName());
063    
064      public void visit(ASTNode[] nodes, final SourceUnit source) {
065    
066        for (ClassNode classNode : source.getAST().getClasses()) {
067          out:
068          for (MethodNode method : classNode.getMethods()) {
069            for (AnnotationNode ann : method.getAnnotations()) {
070              if (ann.getClassNode().getName().equals(Command.class.getName())) {
071                ClassNode superClass = classNode.getSuperClass();
072                if (superClass == null || superClass.getName().equals("java.lang.Object")) {
073                  classNode.setSuperClass(CRASH_COMMAND);
074                }
075                break out;
076              }
077            }
078          }
079        }
080      }
081    }