cucumber.runtime.arquillian.jbas7.Jbas7ResourceIterator Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of cukespace-jbas7 Show documentation
There is a newer version: 1.0.0.Beta2
package cucumber.runtime.arquillian.jbas7;

import java.io.IOException;
import java.util.Iterator;
import java.util.NoSuchElementException;

import org.jboss.vfs.VirtualFile;

import cucumber.io.Resource;
import cucumber.runtime.CucumberException;

/**
 * JBoss AS 7 virtual filesystem resource iterator.
 */
public class Jbas7ResourceIterator implements Iterator {
    
    /**
     * The child elements in the virtual resource.
     */
    private final Iterator elements;
    
    /**
     * The next applicable resource.
     */
    private Resource next;
    
    /**
     * The path.
     */
    private final String path;
    
    /**
     * The suffix.
     */
    private final String suffix;
    
    /**
     * Initializes a new instance of the Jbas7ResourceIterator class.
     * 
     * @param virtualFile The virtual file.
     * @param path The path.
     * @param suffix The suffix.
     */
    public Jbas7ResourceIterator(VirtualFile virtualFile, String path, String suffix) {
        
        try {
            this.elements = virtualFile.getChildrenRecursively().iterator();
        } catch (IOException exception) {
            throw new CucumberException(exception);
        }
        
        this.path = path;
        this.suffix = suffix;
        
        this.moveNext();
    }
    
    @Override
    public boolean hasNext() {
        
        return this.next != null;
    }
    
    @Override
    public Resource next() {
        
        try {
            if (this.hasNext()) {
                return this.next;
            }
            throw new NoSuchElementException();
        } finally {
            this.moveNext();
        }
    }
    
    @Override
    public void remove() {
        
        throw new UnsupportedOperationException();
    }
    
    /**
     * Gets the virtual file's path relative to the classpath.
     * 
     * @param virtualFile The virtual file.
     * @return The file path relative to the classpath.
     */
    private String getPathRelativeToClasspath(VirtualFile virtualFile) {
        
        return virtualFile.getPathName().replaceFirst("^.+(\\.jar|\\.war/WEB-INF/classes)/", "");
    }
    
    /**
     * Checks if a resource path has a the specified suffix.
     * 
     * @param path The path to check.
     * @return True, if the path has the suffix.
     */
    private boolean hasSuffix(String path) {
        
        return this.suffix == null || "".equals(this.suffix) || path.endsWith(this.suffix);
    }
    
    /**
     * Moves to the next child in the resource.
     */
    private void moveNext() {
        
        this.next = null;
        
        while(this.elements.hasNext()) {
            VirtualFile virtualFile = this.elements.next();
            String name = this.getPathRelativeToClasspath(virtualFile);
            
            if(name.startsWith(this.path) && this.hasSuffix(name)) {
                this.next = new Jbas7Resource(virtualFile, name);
                break;
            }
        }
    }
}