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 }