org.jboss.windup.graph.dao.ArchiveEntryDao Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of graph-api Show documentation
The newest version!
package org.jboss.windup.graph.dao;

import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.util.zip.ZipEntry;
import java.util.zip.ZipFile;

import org.jboss.windup.graph.model.resource.ArchiveEntryResource;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.thinkaurelius.titan.core.attribute.Text;
import com.thinkaurelius.titan.util.datastructures.IterablesUtil;
import com.tinkerpop.gremlin.java.GremlinPipeline;

public class ArchiveEntryDao extends BaseDao {

	private static Logger LOG = LoggerFactory.getLogger(ArchiveEntryDao.class);
	
	public ArchiveEntryDao() {
		super(ArchiveEntryResource.class);
	}
	
	public Iterable findArchiveEntry(String value) {
		return super.getByProperty("archiveEntry", value);
	}


	public long findArchiveEntryWithExtensionCount(String ... values) {
		GremlinPipeline pipe = new GremlinPipeline();
		return pipe.start(findArchiveEntryWithExtension(values)).count();
	}
	
	public Iterable findArchiveEntryWithExtension(String ... values) {
		//build regex
		if(values.length == 0) {
			return IterablesUtil.emptyIterable();
		}
		
		final String regex;
		if(values.length == 1) {
			regex = ".+\\."+values[0]+"$";
		}
		else {
			StringBuilder builder = new StringBuilder();
			builder.append("\\b(");
			int i=0;
			for(String value : values) {
				if(i>0) {
					builder.append("|");
				}
				builder.append(value);
				i++;
			}
			builder.append(")\\b");
			regex = ".+\\."+builder.toString()+"$";
		}

		LOG.debug("Regex: "+regex);
		return context.getFramed().query().has("type", typeValue).has("archiveEntry", Text.REGEX, regex).vertices(type);
	}
	
	public InputStream asInputStream(ArchiveEntryResource entry) throws IOException {
		//try and read the XML...
		ZipFile zipFile = new ZipFile(new File(entry.getArchive().getFileResource().getFilePath()));
		ZipEntry zipEntry = zipFile.getEntry(entry.getArchiveEntry());
		InputStream is = zipFile.getInputStream(zipEntry);
		
		return is;
	}
	
}