org.wildfly.galleon.plugin.Utils 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.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.nio.file.FileAlreadyExistsException;
import java.nio.file.FileSystem;
import java.nio.file.FileSystems;
import java.nio.file.FileVisitOption;
import java.nio.file.FileVisitResult;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.SimpleFileVisitor;
import java.nio.file.attribute.BasicFileAttributes;
import java.util.Collections;
import java.util.EnumSet;
import java.util.HashMap;
import java.util.LinkedHashSet;
import java.util.List;
import java.util.Map;
import java.util.Properties;
import java.util.Set;
import org.jboss.galleon.Errors;
import org.jboss.galleon.ProvisioningException;
import org.jboss.galleon.layout.FeaturePackLayout;
import org.jboss.galleon.layout.ProvisioningLayout;
import org.jboss.galleon.universe.maven.MavenArtifact;
import org.jboss.galleon.util.CollectionUtils;
import org.jboss.galleon.util.IoUtils;
import org.jboss.galleon.util.LayoutUtils;
import org.jboss.galleon.util.StringUtils;
import org.wildfly.galleon.plugin.config.CopyArtifact;
/**
*
* @author Alexey Loubyansky
*/
public class Utils {
public static void readProperties(Path propsFile, Map propsMap) throws ProvisioningException {
try(BufferedReader reader = Files.newBufferedReader(propsFile)) {
String line = reader.readLine();
while(line != null) {
line = line.trim();
if(line.charAt(0) != '#' && !line.isEmpty()) {
final int i = line.indexOf('=');
if(i < 0) {
throw new ProvisioningException("Failed to parse property " + line + " from " + propsFile);
}
propsMap.put(line.substring(0, i), line.substring(i + 1));
}
line = reader.readLine();
}
} catch (IOException e) {
throw new ProvisioningException(Errors.readFile(propsFile), e);
}
}
public static Map readProperties(final Path propsFile) throws ProvisioningException {
final Map propsMap = new HashMap<>();
readProperties(propsFile, propsMap);
return propsMap;
}
public static MavenArtifact toArtifactCoords(Map versionProps, String str, boolean optional) throws ProvisioningException {
String[] parts = str.split(":");
if(parts.length < 2) {
throw new IllegalArgumentException("Unexpected artifact coordinates format: " + str);
}
final MavenArtifact artifact = new MavenArtifact();
artifact.setGroupId(parts[0]);
artifact.setArtifactId(parts[1]);
artifact.setExtension(MavenArtifact.EXT_JAR);
if(parts.length > 2) {
if(!parts[2].isEmpty()) {
artifact.setVersion(parts[2]);
}
if(parts.length > 3) {
artifact.setClassifier(parts[3]);
if(parts.length > 4 && !parts[4].isEmpty()) {
artifact.setExtension(parts[4]);
if (parts.length > 5) {
throw new IllegalArgumentException("Unexpected artifact coordinates format: " + str);
}
}
}
}
if(!artifact.hasVersion()) {
final String resolvedStr = versionProps.get(artifact.getGroupId() + ':' + artifact.getArtifactId());
if (resolvedStr == null) {
if (optional) {
return null;
}
throw new ProvisioningException("Failed to resolve the version of " + artifact.getGroupId() + ':' + artifact.getArtifactId());
}
parts = resolvedStr.split(":");
if (parts.length < 3) {
throw new ProvisioningException("Failed to resolve the version for artifact: " + resolvedStr);
}
artifact.setVersion(parts[2]);
}
return artifact;
}
public static List collectLayersConf(ProvisioningLayout layout) throws ProvisioningException {
List layersConfs = Collections.emptyList();
for(FeaturePackLayout fp : layout.getOrderedFeaturePacks()) {
Path p = LayoutUtils.getPackageContentDir(fp.getDir(), WfConstants.LAYERS_CONF);
if(!Files.exists(p)) {
continue;
}
p = p.resolve(WfConstants.MODULES).resolve(WfConstants.LAYERS_CONF);
if(!Files.exists(p)) {
throw new ProvisioningException(
"Feature-pack " + fp.getFPID() + " package " + WfConstants.LAYERS_CONF + " is expected to contain "
+ WfConstants.MODULES + "/" + WfConstants.LAYERS_CONF + " but it does not");
}
layersConfs = CollectionUtils.add(layersConfs, p);
}
return layersConfs;
}
public static void mergeLayersConfs(List layersConfs, Path distHome) throws ProvisioningException {
if(layersConfs.isEmpty()) {
return;
}
if(layersConfs.size() == 1) {
try {
IoUtils.copy(layersConfs.get(0), distHome.resolve(WfConstants.MODULES).resolve(WfConstants.LAYERS_CONF));
} catch (IOException e) {
throw new ProvisioningException("Failed to install layes.conf to " + distHome, e);
}
return;
}
final Properties props = new Properties();
final Set layers = new LinkedHashSet<>(layersConfs.size());
for(Path p : layersConfs) {
try(BufferedReader reader = Files.newBufferedReader(p)) {
props.load(reader);
} catch (IOException e) {
throw new ProvisioningException("Failed to generate layer.conf", e);
}
String layersProp = props.getProperty(WfConstants.LAYERS);
if (layersProp == null || (layersProp = layersProp.trim()).length() == 0) {
continue;
}
final String[] layerNames = layersProp.split(",");
for(String layerName : layerNames) {
layers.add(layerName);
}
}
if(!layers.isEmpty()) {
final StringBuilder buf = new StringBuilder();
StringUtils.append(buf, layers);
props.setProperty(WfConstants.LAYERS, buf.toString());
}
try (BufferedWriter writer = Files.newBufferedWriter(distHome.resolve(WfConstants.MODULES).resolve(WfConstants.LAYERS_CONF))) {
props.store(writer, "Generated by WildFly Galleon provisioning plugin");
}
catch (IOException e) {
throw new ProvisioningException("Failed to persist generated layers.conf", e);
}
}
public static void extractArtifact(Path artifact, Path target, CopyArtifact copy) throws IOException {
if(!Files.exists(target)) {
Files.createDirectories(target);
}
try (FileSystem zipFS = FileSystems.newFileSystem(artifact, null)) {
for(Path zipRoot : zipFS.getRootDirectories()) {
Files.walkFileTree(zipRoot, EnumSet.of(FileVisitOption.FOLLOW_LINKS), Integer.MAX_VALUE,
new SimpleFileVisitor() {
@Override
public FileVisitResult preVisitDirectory(Path dir, BasicFileAttributes attrs)
throws IOException {
String entry = dir.toString().substring(1);
if(entry.isEmpty()) {
return FileVisitResult.CONTINUE;
}
if(!entry.endsWith("/")) {
entry += '/';
}
if(!copy.includeFile(entry)) {
return FileVisitResult.SKIP_SUBTREE;
}
final Path targetDir = target.resolve(zipRoot.relativize(dir).toString());
try {
Files.copy(dir, targetDir);
} catch (FileAlreadyExistsException e) {
if (!Files.isDirectory(targetDir))
throw e;
}
return FileVisitResult.CONTINUE;
}
@Override
public FileVisitResult visitFile(Path file, BasicFileAttributes attrs)
throws IOException {
if(copy.includeFile(file.toString().substring(1))) {
final Path targetPath = target.resolve(zipRoot.relativize(file).toString());
Files.copy(file, targetPath);
}
return FileVisitResult.CONTINUE;
}
});
}
}
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy