io.fabric8.jube.process.support.DownloadResourcesTask Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of process-manager Show documentation
There is a newer version: 2.2.0
/**
 *  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.jube.process.support;

import java.io.BufferedOutputStream;
import java.io.File;
import java.io.FileOutputStream;
import java.io.InputStream;
import java.net.MalformedURLException;
import java.net.URL;
import java.util.Map;
import java.util.Set;

import io.fabric8.jube.process.InstallContext;
import io.fabric8.jube.process.InstallTask;
import io.fabric8.jube.process.config.ProcessConfig;
import io.fabric8.utils.FileChangeInfo;
import io.fabric8.utils.Files;
import io.fabric8.utils.Strings;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * Downloads the given resources from a map of relative path -> URL mappings
 */
public class DownloadResourcesTask implements InstallTask {
    private static final transient Logger LOG = LoggerFactory.getLogger(DownloadResourcesTask.class);

    private final Map localPathToURLMap;

    public DownloadResourcesTask(Map localPathToURLMap) {
        this.localPathToURLMap = localPathToURLMap;
    }

    @Override
    public void install(InstallContext installContext, ProcessConfig config, String id, File installDir) throws Exception {
        File baseDir = ProcessUtils.findInstallDir(installDir);
        Set> entries = localPathToURLMap.entrySet();
        for (Map.Entry entry : entries) {
            String localPath = entry.getKey();
            String urlText = entry.getValue();
            if (Strings.isNotBlank(urlText)) {
                URL url = null;
                try {
                    url = new URL(urlText);
                } catch (MalformedURLException e) {
                    LOG.warn("Ignoring invalid URL '" + urlText + "' for overlay resource " + localPath + ". " + e, e);
                }
                if (url != null) {
                    File newFile = new File(baseDir, localPath);
                    FileChangeInfo changeInfo = installContext.createChangeInfo(newFile);
                    newFile.getParentFile().mkdirs();
                    InputStream stream = url.openStream();
                    if (stream != null) {
                        Files.copy(stream, new BufferedOutputStream(new FileOutputStream(newFile)));
                        installContext.onFileWrite(newFile, changeInfo);
                    }
                }
            }
        }
    }

}