org.wildfly.galleon.plugin.JandexIndexer Maven / Gradle / Ivy
/*
* Copyright 2016-2018 Red Hat, Inc. and/or its affiliates
* and other contributors as indicated by the @author tags.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.wildfly.galleon.plugin;
import java.io.Closeable;
import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.util.Enumeration;
import java.util.jar.JarEntry;
import java.util.jar.JarFile;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;
import org.jboss.galleon.MessageWriter;
import org.jboss.jandex.Index;
import org.jboss.jandex.IndexWriter;
import org.jboss.jandex.Indexer;
/**
* @author Stuart Douglas
*/
class JandexIndexer {
public static void createIndex(File jarFile, OutputStream target, MessageWriter log) throws IOException {
ZipOutputStream zo;
Indexer indexer = new Indexer();
JarFile jar = new JarFile(jarFile);
zo = new ZipOutputStream(target);
try {
Enumeration entries = jar.entries();
while (entries.hasMoreElements()) {
JarEntry entry = entries.nextElement();
if (entry.getName().endsWith(".class")) {
try {
final InputStream stream = jar.getInputStream(entry);
try {
indexer.index(stream);
} finally {
safeClose(stream, log);
}
} catch (Exception e) {
String message = e.getMessage() == null ? e.getClass().getSimpleName() : e.getMessage();
log.error("Could not index " + entry.getName() + ": " + message, e);
}
}
}
zo.putNextEntry(new ZipEntry("META-INF/jandex.idx"));
IndexWriter writer = new IndexWriter(zo);
Index index = indexer.complete();
writer.write(index);
} finally {
safeClose(zo, log);
safeClose(jar, log);
safeClose(target, log);
}
}
private static void safeClose(Closeable closeable, MessageWriter log) {
if (closeable != null) {
try {
closeable.close();
} catch (IOException e) {
log.error("Failed to close", e);
}
}
}
}