/**
* Copyright 2005-2014 Red Hat, Inc.
*
* Red Hat licenses this file to you 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 io.fabric8.process.manager.support;
import com.google.common.base.Charsets;
import com.google.common.collect.Maps;
import com.google.common.io.Files;
import io.fabric8.utils.FileChangeInfo;
import io.fabric8.process.manager.InstallContext;
import io.fabric8.process.manager.service.ProcessManagerService;
import io.fabric8.process.manager.support.mvel.MvelPredicate;
import io.fabric8.process.manager.support.mvel.MvelTemplateRendering;
import io.fabric8.process.manager.InstallTask;
import io.fabric8.process.manager.config.ProcessConfig;
import java.io.File;
import java.io.IOException;
import java.util.Map;
public class ApplyConfigurationTask implements InstallTask {
private final Map<String, Object> variables;
private final Map<String, String> configuration;
private final MvelPredicate isTemplate = new MvelPredicate();
public ApplyConfigurationTask(Map<String, String> configuration, Map<String, Object> variables) {
this.configuration = configuration;
this.variables = variables;
}
@Override
public void install(InstallContext installContext, ProcessConfig config, String id, File installDir) throws Exception {
Map<String, String> templates = Maps.filterKeys(configuration, isTemplate);
Map<String, String> plainFiles = Maps.difference(configuration, templates).entriesOnlyOnLeft();
ProcessManagerService.substituteEnvironmentVariableExpressions((Map)variables, config.getEnvironment());
Map<String, String> renderedTemplates = Maps.transformValues(templates, new MvelTemplateRendering(variables));
File baseDir = ProcessUtils.findInstallDir(installDir);
applyTemplates(installContext, renderedTemplates, baseDir);
applyPlainConfiguration(installContext, plainFiles, baseDir);
}
private void applyTemplates(InstallContext installContext, Map<String, String> templates, File installDir) throws IOException {
for (Map.Entry<String, String> entry : templates.entrySet()) {
String path = entry.getKey();
String content = entry.getValue();
String resourcePath = path.substring(path.indexOf("/"));
resourcePath = resourcePath.substring(0, resourcePath.lastIndexOf(MvelPredicate.MVEN_EXTENTION));
copyToContent(installContext, installDir, resourcePath, content);
}
}
private void applyPlainConfiguration(InstallContext installContext, Map<String, String> configuration, File installDir) throws IOException {
for (Map.Entry<String, String> entry : configuration.entrySet()) {
String path = entry.getKey();
String content = entry.getValue();
int slashIndex = path.indexOf("/");
String resourcePath = slashIndex > 0 ? path.substring(slashIndex): path;
copyToContent(installContext, installDir, resourcePath, content);
}
}
private void copyToContent(InstallContext installContext, File baseDir, String name, String content) throws IOException {
File target = new File(baseDir, name);
FileChangeInfo changeInfo = installContext.createChangeInfo(target);
if (!target.exists() && !target.getParentFile().exists() && !target.getParentFile().mkdirs()) {
throw new IOException("Directory: " + target.getParentFile().getAbsolutePath() + " can't be created");
} else if (target.isDirectory()) {
throw new IOException("Can't write to : " + target.getAbsolutePath() + ". It's a directory");
} else if (!target.exists() && !target.createNewFile()) {
throw new IOException("Failed to create file: " + target.getAbsolutePath() + ".");
}
Files.write(content.getBytes(Charsets.UTF_8), target);
String lowerName = name.toLowerCase();
if (lowerName.endsWith(".sh") || lowerName.endsWith(".bat") || lowerName.endsWith(".cmd")) {
// lets ensure its executable
target.setExecutable(true);
}
installContext.onFileWrite(target, changeInfo);
}
}