net.lmxm.ute.configuration.ConfigurationInterpolator.java Source code

Java tutorial

Introduction

Here is the source code for net.lmxm.ute.configuration.ConfigurationInterpolator.java

Source

/**
 * Copyright (C) 2011 Shaun Johnson, LMXM LLC
 * 
 * This file is part of Universal Task Executor.
 * 
 * Universal Task Executor is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by the Free
 * Software Foundation, either version 3 of the License, or (at your option) any
 * later version.
 * 
 * Universal Task Executor is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
 * details.
 * 
 * You should have received a copy of the GNU General Public License along with
 * Universal Task Executor. If not, see <http://www.gnu.org/licenses/>.
 */
package net.lmxm.ute.configuration;

import net.lmxm.ute.beans.*;
import net.lmxm.ute.beans.configuration.Configuration;
import net.lmxm.ute.beans.jobs.Job;
import net.lmxm.ute.beans.locations.FileSystemLocation;
import net.lmxm.ute.beans.locations.HttpLocation;
import net.lmxm.ute.beans.locations.MavenRepositoryLocation;
import net.lmxm.ute.beans.locations.SubversionRepositoryLocation;
import net.lmxm.ute.beans.sources.HttpSource;
import net.lmxm.ute.beans.sources.MavenRepositorySource;
import net.lmxm.ute.beans.sources.SubversionRepositorySource;
import net.lmxm.ute.beans.targets.FileSystemTarget;
import net.lmxm.ute.beans.tasks.*;
import net.lmxm.ute.exceptions.ConfigurationException;
import net.lmxm.ute.resources.types.ExceptionResourceType;
import org.apache.commons.lang.SerializationUtils;
import org.apache.commons.lang3.StringUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

import static com.google.common.base.Preconditions.checkNotNull;

/**
 * The Class ConfigurationInterpolator.
 */
public final class ConfigurationInterpolator {

    /**
     * The Enum PropertyType.
     */
    private enum PropertyType {
        NAME, VALUE
    }

    /** The Constant LOGGER. */
    private static final Logger LOGGER = LoggerFactory.getLogger(ConfigurationInterpolator.class);

    /**
     * Generate property maps.
     * 
     * @param configuration the configuration
     * @return the map
     */
    private static Map<PropertyType, String[]> generatePropertyMaps(final Configuration configuration) {
        final String prefix = "generatePropertyMaps() :";

        LOGGER.debug("{} entered, configuration={}", prefix, configuration);

        final List<String> propertyNames = new ArrayList<String>();
        final List<String> propertyValues = new ArrayList<String>();

        for (final Preference preference : configuration.getPreferences()) {
            propertyNames.add("${pref." + preference.getId() + "}");
            propertyValues.add(StringUtils.trimToEmpty(preference.getValue()));
        }

        for (final Property property : configuration.getProperties()) {
            propertyNames.add("${" + property.getId() + "}");
            propertyValues.add(StringUtils.trimToEmpty(property.getValue()));
        }

        for (final Entry<String, String> entry : System.getenv().entrySet()) {
            propertyNames.add("${env." + entry.getKey() + "}");
            propertyValues.add(StringUtils.trimToEmpty(entry.getValue()));
        }

        final Map<PropertyType, String[]> propertyMaps = new HashMap<PropertyType, String[]>();

        propertyMaps.put(PropertyType.NAME, propertyNames.toArray(new String[0]));
        propertyMaps.put(PropertyType.VALUE, propertyValues.toArray(new String[0]));

        LOGGER.debug("{} returning {}", prefix, propertyMaps);

        return propertyMaps;
    }

    /**
     * Interpolate files.
     * 
     * @param files the files
     * @param propertyNames the property names
     * @param propertyValues the property values
     */
    private static void interpolateFiles(final List<FileReference> files, final String[] propertyNames,
            final String[] propertyValues) {
        final String prefix = "interpolateFiles() :";

        LOGGER.debug("{} entered, files={}", prefix, files);

        checkNotNull(files, "Files may not be null");

        for (final FileReference fileReference : files) {
            final String name = fileReference.getName();
            final String targetName = fileReference.getTargetName();

            fileReference.setName(interpolateProperties(name, propertyNames, propertyValues));
            fileReference.setTargetName(interpolateProperties(targetName, propertyNames, propertyValues));
        }

        LOGGER.debug("{} leaving", prefix);
    }

    /**
     * Interpolate Maven artifacts.
     *
     * @param artifacts the artifacts
     * @param propertyNames the property names
     * @param propertyValues the property values
     */
    private static void interpolateMavenArtifacts(final List<MavenArtifact> artifacts, final String[] propertyNames,
            final String[] propertyValues) {
        final String prefix = "interpolateMavenArtifacts() :";

        LOGGER.debug("{} entered, artifacts={}", prefix, artifacts);

        checkNotNull(artifacts, "Artifacts may not be null");

        for (final MavenArtifact artifact : artifacts) {
            final String coordinates = artifact.getCoordinates();
            final String targetName = artifact.getTargetName();

            artifact.setCoordinates(interpolateProperties(coordinates, propertyNames, propertyValues));
            artifact.setTargetName(interpolateProperties(targetName, propertyNames, propertyValues));
        }

        LOGGER.debug("{} leaving", prefix);
    }

    /**
     * Interpolate file system location.
     * 
     * @param target the target
     * @param propertyNames the property names
     * @param propertyValues the property values
     */
    private static void interpolateFileSystemTarget(final FileSystemTarget target, final String[] propertyNames,
            final String[] propertyValues) {
        final String prefix = "interpolateFileSystemTarget() :";

        LOGGER.debug("{} entered, target={}", prefix, target);

        if (target == null) {
            LOGGER.debug("{} skipping, target is null", prefix);
        } else {
            final FileSystemLocation location = target.getLocation();

            if (location == null) {
                LOGGER.debug("{} skipping, target.location is null", prefix);
            } else {
                final String path = location.getPath();

                location.setPath(interpolateProperties(path, propertyNames, propertyValues));
            }

            target.setRelativePath(interpolateProperties(target.getRelativePath(), propertyNames, propertyValues));
        }

        LOGGER.debug("{} leaving", prefix);
    }

    /**
     * Interpolate http query params.
     * 
     * @param source the source
     * @param propertyNames the property names
     * @param propertyValues the property values
     */
    private static void interpolateHttpQueryParams(final HttpSource source, final String[] propertyNames,
            final String[] propertyValues) {
        final String prefix = "interpolateHttpQueryParams() :";

        LOGGER.debug("{} entered, source={}", prefix, source);

        if (source == null) {
            LOGGER.debug("{} skipping, source is null", prefix);
        } else if (source.getQueryParams() == null) {
            LOGGER.debug("{} skipping, query params is null", prefix);
        } else {
            final Map<String, String> queryParams = new HashMap<String, String>();

            for (final Entry<String, String> queryParam : source.getQueryParams().entrySet()) {
                final String name = interpolateProperties(queryParam.getKey(), propertyNames, propertyValues);
                final String value = interpolateProperties(queryParam.getValue(), propertyNames, propertyValues);

                queryParams.put(name, value);
            }

            source.setQueryParams(queryParams);
        }

        LOGGER.debug("{} leaving", prefix);
    }

    /**
     * Interpolate http source.
     * 
     * @param source the source
     * @param propertyNames the property names
     * @param propertyValues the property values
     */
    private static void interpolateHttpSource(final HttpSource source, final String[] propertyNames,
            final String[] propertyValues) {
        final String prefix = "interpolateHttpSource() :";

        LOGGER.debug("{} entered, source={}", prefix, source);

        if (source == null) {
            LOGGER.debug("{} skipping, source is null", prefix);
        } else {
            final HttpLocation location = source.getLocation();

            if (location == null) {
                LOGGER.debug("{} skipping, source is null", prefix);
            } else {
                location.setUrl(interpolateProperties(location.getUrl(), propertyNames, propertyValues));
            }

            source.setRelativePath(interpolateProperties(source.getRelativePath(), propertyNames, propertyValues));

            interpolateHttpQueryParams(source, propertyNames, propertyValues);
        }

        LOGGER.debug("{} leaving", prefix);
    }

    /**
     * Interpolate job values.
     * 
     * @param originalJob the original job
     * @param configuration the configuration
     * @return the job
     */
    public static Job interpolateJobValues(final Job originalJob, final Configuration configuration) {
        final String prefix = "interpolateJobValues() :";

        LOGGER.debug("{} entered, originalJob={}", prefix, originalJob);

        final Job job = (Job) SerializationUtils.clone(originalJob);

        LOGGER.debug("{} cloned job={}", prefix, job);

        final Map<PropertyType, String[]> propertyMaps = generatePropertyMaps(configuration);
        final String[] propertyNames = propertyMaps.get(PropertyType.NAME);
        final String[] propertyValues = propertyMaps.get(PropertyType.VALUE);

        for (final Task task : job.getTasks()) {
            if (task instanceof FileSystemDeleteTask) {
                final FileSystemTargetTask fileSystemDeleteTask = (FileSystemTargetTask) task;

                interpolateFileSystemTarget(fileSystemDeleteTask.getTarget(), propertyNames, propertyValues);
            } else if (task instanceof FindReplaceTask) {
                final FindReplaceTask findReplaceTask = (FindReplaceTask) task;

                interpolateFileSystemTarget(findReplaceTask.getTarget(), propertyNames, propertyValues);
                interpolatePatterns(findReplaceTask.getPatterns(), propertyNames, propertyValues);
            } else if (task instanceof GroovyTask) {
                final GroovyTask groovyTask = (GroovyTask) task;

                interpolateFileSystemTarget(groovyTask.getTarget(), propertyNames, propertyValues);
            } else if (task instanceof HttpDownloadTask) {
                final HttpDownloadTask httpDownloadTask = (HttpDownloadTask) task;

                interpolateHttpSource(httpDownloadTask.getSource(), propertyNames, propertyValues);
                interpolateFileSystemTarget(httpDownloadTask.getTarget(), propertyNames, propertyValues);
                interpolateFiles(httpDownloadTask.getFiles(), propertyNames, propertyValues);
            } else if (task instanceof MavenRepositoryDownloadTask) {
                final MavenRepositoryDownloadTask mavenRepositoryDownloadTask = (MavenRepositoryDownloadTask) task;

                interpolateMavenRepositorySource(mavenRepositoryDownloadTask.getSource(), propertyNames,
                        propertyValues);
                interpolateFileSystemTarget(mavenRepositoryDownloadTask.getTarget(), propertyNames, propertyValues);
                interpolateMavenArtifacts(mavenRepositoryDownloadTask.getArtifacts(), propertyNames,
                        propertyValues);
            } else if (task instanceof SubversionExportTask) {
                final SubversionExportTask subversionExportTask = (SubversionExportTask) task;

                interpolateSubversionRepositorySource(subversionExportTask.getSource(), propertyNames,
                        propertyValues);
                interpolateFileSystemTarget(subversionExportTask.getTarget(), propertyNames, propertyValues);
                interpolateFiles(subversionExportTask.getFiles(), propertyNames, propertyValues);
            } else if (task instanceof SubversionUpdateTask) {
                final SubversionUpdateTask subversionUpdateTask = (SubversionUpdateTask) task;

                interpolateSubversionRepositorySource(subversionUpdateTask.getSource(), propertyNames,
                        propertyValues);
                interpolateFileSystemTarget(subversionUpdateTask.getTarget(), propertyNames, propertyValues);
                interpolateFiles(subversionUpdateTask.getFiles(), propertyNames, propertyValues);
            } else {
                LOGGER.error("{} Unsupported task type {}", prefix, task.getClass());
                throw new ConfigurationException(ExceptionResourceType.UNSUPPORTED_TASK_TYPE, task.getClass());
            }
        }

        LOGGER.debug("{} returning {}", prefix, job);

        return job;
    }

    /**
     * Interpolate Maven repository source.
     *
     * @param source the source
     * @param propertyNames the property names
     * @param propertyValues the property values
     */
    private static void interpolateMavenRepositorySource(final MavenRepositorySource source,
            final String[] propertyNames, final String[] propertyValues) {
        final String prefix = "interpolateMavenRepositorySource() :";

        LOGGER.debug("{} entered, source={}", prefix, source);

        if (source == null) {
            LOGGER.debug("{} skipping, source is null", prefix);
        } else {
            final MavenRepositoryLocation location = source.getLocation();

            if (location == null) {
                LOGGER.debug("{} skipping, source is null", prefix);
            } else {
                location.setUrl(interpolateProperties(location.getUrl(), propertyNames, propertyValues));
            }

            source.setRelativePath(interpolateProperties(source.getRelativePath(), propertyNames, propertyValues));
        }

        LOGGER.debug("{} leaving", prefix);
    }

    /**
     * Interpolate patterns.
     * 
     * @param patterns the patterns
     * @param propertyNames the property names
     * @param propertyValues the property values
     */
    private static void interpolatePatterns(final List<FindReplacePattern> patterns, final String[] propertyNames,
            final String[] propertyValues) {
        final String prefix = "interpolatePatterns() :";

        LOGGER.debug("{} entered, patterns={}", prefix, patterns);

        checkNotNull(patterns, "Patterns may not be null");

        for (final FindReplacePattern pattern : patterns) {
            final String find = pattern.getFind();
            final String replace = pattern.getReplace();

            pattern.setFind(interpolateProperties(find, propertyNames, propertyValues));
            pattern.setReplace(interpolateProperties(replace, propertyNames, propertyValues));
        }

        LOGGER.debug("{} leaving", prefix);
    }

    /**
     * Interpolate properties.
     * 
     * @param string the string
     * @param propertyNames the property names
     * @param propertyValues the property values
     * @return the string
     */
    private static String interpolateProperties(final String string, final String[] propertyNames,
            final String[] propertyValues) {
        final String newString = StringUtils.replaceEachRepeatedly(string, propertyNames, propertyValues);

        validateDoesNotContainProperties(newString);

        return newString;
    }

    /**
     * Interpolate subversion repository source.
     * 
     * @param source the source
     * @param propertyNames the property names
     * @param propertyValues the property values
     */
    private static void interpolateSubversionRepositorySource(final SubversionRepositorySource source,
            final String[] propertyNames, final String[] propertyValues) {
        final String prefix = "interpolateSubversionRepositorySource() :";

        LOGGER.debug("{} entered, source={}", prefix, source);

        if (source == null) {
            LOGGER.debug("{} skipping, source is null", prefix);
        } else {
            final SubversionRepositoryLocation location = source.getLocation();

            if (location == null) {
                LOGGER.debug("{} skipping, source is null", prefix);
            } else {
                location.setUrl(interpolateProperties(location.getUrl(), propertyNames, propertyValues));
                location.setUsername(interpolateProperties(location.getUsername(), propertyNames, propertyValues));
                location.setPassword(interpolateProperties(location.getPassword(), propertyNames, propertyValues));
            }

            source.setRelativePath(interpolateProperties(source.getRelativePath(), propertyNames, propertyValues));
        }

        LOGGER.debug("{} leaving", prefix);
    }

    /**
     * Validate does not contain properties.
     * 
     * @param string the string
     */
    protected static void validateDoesNotContainProperties(final String string) {
        final String prefix = "validateDoesNotContainProperties() :";

        LOGGER.debug("{} entered, string={}", prefix, string);

        if (StringUtils.isBlank(string)) {
            LOGGER.debug("{} leaving, string is blank", prefix);

            return;
        }

        final Pattern pattern = Pattern.compile("^(.*)(\\$\\{)(.*)(\\})(.*)$");
        final Matcher matcher = pattern.matcher(string);

        if (matcher.find()) {
            LOGGER.error("{} undefined property with name \"{}\"", prefix, matcher.group(3));
            throw new ConfigurationException(ExceptionResourceType.UNDEFINED_PROPERTY, matcher.group(3));
        } else {
            LOGGER.debug("{} no property references found", prefix);
        }

        LOGGER.debug("{} leaving", prefix);
    }

    /**
     * Instantiates a new configuration interpolator.
     */
    private ConfigurationInterpolator() {
        throw new AssertionError();
    }
}