Start line:  
End line:  

Snippet Preview

Snippet HTML Code

Stack Overflow Questions
  /*
   * JBoss, Home of Professional Open Source.
   * Copyright 2012, Red Hat, Inc., and individual contributors
   * as indicated by the @author tags. See the copyright.txt file in the
   * distribution for a full listing of individual contributors.
   *
   * 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.jboss.as.test.module.util;
 
 import java.io.File;
 import java.util.List;
 

Utility class with some convenience methods to create and remove modules.

Author(s):
Pedro Igor
 
 public class TestModule {
 
     private static final Logger log  = Logger.getLogger(TestModule.class);
 
     private final String moduleName;
     private final File moduleXml;
     private final String[] dependencies;
     private final List<ClassCallbackclassCallbacks = new ArrayList<TestModule.ClassCallback>();
     private final List<JavaArchiveresources = new ArrayList<JavaArchive>();


    

Creates a new module with the given name and module definition.

Parameters:
moduleName The name of the module.
moduleXml The module definition file.
 
     public TestModule(String moduleNameFile moduleXml) {
         if (!moduleXml.exists()) {
             throw new IllegalArgumentException("The module definition must exists.");
         }
 
         this. = moduleName;
         this. = moduleXml;
         this. = null;
     }

    

Creates a new module with the given name and module dependencies. The module.xml will be generated

Parameters:
moduleName The name of the module.
moduleXml The module definition file.
 
     public TestModule(String moduleNameString...dependencies) {
         this. = moduleName;
         this. = null;
         this. = dependencies;
     }

    

Creates the module directory. If the module already exists, it will deleted first.

Throws:
java.io.IOException If any error occurs during the module creation.
 
     public void create() throws IOException {
         create(true);
     }


    
Add a callback to handle classes being added

Parameters:
callback the call back to add
Returns:
this
    public TestModule addClassCallback(ClassCallback callback) {
        .add(callback);
        return this;
    }

    

Creates the module directory.

Parameters:
deleteFirst If the module already exists, this argument specifies if it should be deleted before continuing.
Throws:
java.io.IOException
    public void create(boolean deleteFirstthrows IOException {
        File moduleDirectory = getModuleDirectory();
        if (moduleDirectory.exists()) {
            if (!deleteFirst) {
                throw new IllegalArgumentException(moduleDirectory + " already exists.");
            }
            remove();
        }
        File mainDirectory = new File(moduleDirectory"main");
        .infof("creating module in: %s",mainDirectory);
        if (!mainDirectory.mkdirs()) {
            throw new IllegalArgumentException("Could not create " + mainDirectory);
        }
        try {
            if ( != null) {
                try (FileInputStream fis = new FileInputStream(this.)) {
                    copyFile(new File(mainDirectory"module.xml"), fis);
                }
            } else {
                generateModuleXml(mainDirectory);
            }
        } catch (IOException e) {
            throw new RuntimeException("Could not create module definition."e);
        }
        for (JavaArchive resourceRoot : this.) {
            FileOutputStream jarFile = null;
            try {
                Indexer indexer = new Indexer();
                List<Class<?>> classes = new ArrayList<Class<?>>();
                for (Node content : resourceRoot.getContent().values()) {
                    final String path = content.getPath().get();
                    if (path.endsWith(".class")) {
                        indexer.index(content.getAsset().openStream());
                        if (.size() > 0) {
                            //TODO load class
                            String usePath = path.startsWith("/") ? path.substring(1, path.length() - 6) : path.substring(0, path.length() - 6);
                            usePath = usePath.replaceAll("/"".");
                            Class<?> clazz = Class.forName(usePath);
                            classes.add(clazz);
                        }
                    }
                }
                for (ClassCallback callback : ) {
                    callback.classesAdded(resourceRootclasses);
                }
                Index index = indexer.complete();
                ByteArrayOutputStream data = new ByteArrayOutputStream();
                IndexWriter writer = new IndexWriter(data);
                writer.write(index);
                resourceRoot.addAsManifestResource(new ByteArrayAsset(data.toByteArray()), "jandex.idx");
                jarFile = new FileOutputStream(new File(mainDirectoryresourceRoot.getName()));
                resourceRoot.as(ZipExporter.class).exportTo(jarFile);
            } catch (Exception e) {
                throw new RuntimeException("Could not create module resource [" + resourceRoot.getName() + "."e);
            } finally {
                if (jarFile != null) {
                    jarFile.flush();
                    jarFile.close();
                }
            }
        }
    }

    

Removes the module from the modules directory. This operation can not be reverted.

    public void remove() {
        remove(getModuleDirectory());
    }

    

Creates a org.jboss.shrinkwrap.api.spec.JavaArchive instance that will be used to create a jar file inside the module's directory.

The name of the archive must match one of the resource-root definitions defined in the module definition.

Parameters:
fileName The name of the archive.
Returns:
    public JavaArchive addResource(String fileName) {
        JavaArchive resource = ShrinkWrap.create(JavaArchive.classfileName);
        if (.size() == 0) {
            //Add the test module to the first jar in the module to avoid having to do that from the tests
            resource.addClass(TestModule.class);
        }
        .add(resource);
        return resource;
    }
    private void remove(File file) {
        if (file.exists()) {
            if (file.isDirectory()) {
                for (String name : file.list()) {
                    remove(new File(filename));
                }
            }
            if (!file.delete()) {
                //This will often not work on Windows
                .error("Could not delete [" + file.getPath() + ".");
            }
        } else {
            throw new IllegalStateException("Module [" + this. + "] does not exists.");
        }
    }
    private File getModuleDirectory() {
        return new File(getModulesDirectory(), this..replace('.'.));
    }
    private File getModulesDirectory() {
        String modulePath = System.getProperty("module.path"null);
        if (modulePath == null) {
            String jbossHome = System.getProperty("jboss.home"null);
            if (jbossHome == null) {
                throw new IllegalStateException("Neither -Dmodule.path nor -Djboss.home were set");
            }
            modulePath = jbossHome + . + "modules";
        } else {
            modulePath = modulePath.split(.)[0];
        }
        File moduleDir = new File(modulePath);
        if (!moduleDir.exists()) {
            throw new IllegalStateException("Determined module path does not exist");
        }
        if (!moduleDir.isDirectory()) {
            throw new IllegalStateException("Determined module path is not a dir");
        }
        return moduleDir;
    }
    private static void copyFile(File targetInputStream srcthrows IOException {
        final BufferedOutputStream out = new BufferedOutputStream(new FileOutputStream(target));
        try {
            int i = src.read();
            while (i != -1) {
                out.write(i);
                i = src.read();
            }
        } finally {
            out.close();
        }
    }
    public String getName() {
        return ;
    }
    private void generateModuleXml(File mainDirectorythrows IOException {
        BufferedWriter writer = new BufferedWriter(new FileWriter(new File(mainDirectory"module.xml")));
        try {
            writer.write("<module xmlns=\"urn:jboss:module:1.1\" name=\"" +  + "\">");
            writer.write("<resources>");
            for (JavaArchive jar : ) {
                writer.write("<resource-root path=\"" + jar.getName() + "\"/>");
            }
            writer.write("</resources>");
            writer.write("<dependencies>");
            for (String dependency : ) {
                writer.write("<module name=\"" + dependency + "\"/>");
            }
            writer.write("</dependencies>");
            writer.write("</module>");
        } finally {
            writer.close();
        }
    }
    public interface ClassCallback {
        void classesAdded(JavaArchive jarfinal List<Class<?>> classes);
    }
New to GrepCode? Check out our FAQ X