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 package org.crsh.lang.groovy.command;
021
022 import org.codehaus.groovy.ast.ASTNode;
023 import org.codehaus.groovy.ast.ClassHelper;
024 import org.codehaus.groovy.control.CompilePhase;
025 import org.codehaus.groovy.control.SourceUnit;
026 import org.codehaus.groovy.transform.ASTTransformation;
027 import org.codehaus.groovy.transform.GroovyASTTransformation;
028 import org.crsh.cli.Argument;
029 import org.crsh.cli.Command;
030 import org.crsh.cli.Required;
031 import org.crsh.cli.Usage;
032 import org.crsh.cli.Man;
033 import org.crsh.cli.Option;
034 import org.crsh.command.CRaSHCommand;
035 import org.crsh.command.InvocationContext;
036 import org.crsh.command.ScriptException;
037 import org.crsh.text.ui.BorderStyle;
038 import org.crsh.text.Color;
039 import org.crsh.text.Decoration;
040 import org.crsh.text.Style;
041
042 import java.util.logging.Level;
043 import java.util.logging.Logger;
044
045 @GroovyASTTransformation(phase= CompilePhase.CONVERSION)
046 public class DefaultImportTransformer implements ASTTransformation {
047
048 /** . */
049 private static final Logger log = Logger.getLogger(DefaultImportTransformer.class.getName());
050
051 /** . */
052 private static final Class<?>[] defaultImports = {
053 Required.class,
054 Man.class,
055 Usage.class,
056 Argument.class,
057 Option.class,
058 Command.class,
059 ScriptException.class,
060 InvocationContext.class,
061 CRaSHCommand.class,
062 };
063
064 /** . */
065 private static final Class<?>[] defaultStaticImports = {
066 Color.class,
067 Decoration.class,
068 Style.class,
069 BorderStyle.class
070 };
071
072 public void visit(ASTNode[] nodes, final SourceUnit source) {
073 log.log(Level.FINE, "Transforming source to add default import package");
074 for (Class<?> defaultImport : defaultImports) {
075 log.log(Level.FINE, "Adding default import for class " + defaultImport.getName());
076 if (source.getAST().getImport(defaultImport.getSimpleName()) == null) {
077 source.getAST().addImport(defaultImport.getSimpleName(), ClassHelper.make(defaultImport));
078 }
079 }
080 for (Class<?> defaultStaticImport : defaultStaticImports) {
081 log.log(Level.FINE, "Adding default static import for class " + defaultStaticImport.getName());
082 if (!source.getAST().getStaticStarImports().containsKey(defaultStaticImport.getSimpleName())) {
083 source.getAST().addStaticStarImport(defaultStaticImport.getSimpleName(), ClassHelper.make(defaultStaticImport));
084 }
085 }
086 }
087 }