io.jenkins.updatebot.kind.helm.HelmUpdater Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of updatebot-core Show documentation
There is a newer version: 1.1.7
/*
 * Copyright 2016 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.jenkins.updatebot.kind.helm;

import io.fabric8.utils.Files;
import io.fabric8.utils.IOHelpers;
import io.jenkins.updatebot.commands.CommandContext;
import io.jenkins.updatebot.kind.Kind;
import io.jenkins.updatebot.kind.UpdaterSupport;
import io.jenkins.updatebot.kind.helm.model.Chart;
import io.jenkins.updatebot.kind.helm.model.Requirements;
import io.jenkins.updatebot.model.Dependencies;
import io.jenkins.updatebot.model.DependencyVersionChange;
import io.jenkins.updatebot.support.MarkupHelper;
import io.jenkins.updatebot.support.Strings;

import java.io.File;
import java.io.IOException;
import java.util.List;

import static io.jenkins.updatebot.kind.helm.HelmFiles.CHART_YAML;
import static io.jenkins.updatebot.kind.helm.HelmFiles.REQUIREMENTS_YAML;
import static io.jenkins.updatebot.kind.helm.HelmFiles.VALUES_YAML;
import static io.jenkins.updatebot.support.FileHelper.isFile;

/**
 */
public class HelmUpdater extends UpdaterSupport {

    public static boolean applyRequirementsChanges(CommandContext context, List changes, Requirements requirements, File requirementsFile) throws IOException {
        boolean answer = requirements.applyChanges(changes);
        if (answer) {
            // lets store the updated requirements
            try {
                MarkupHelper.saveYaml(requirements, requirementsFile);
            } catch (IOException e) {
                throw new IOException("Failed to save chart requirements " + requirementsFile + ". " + e, e);
            }
        }
        return answer;
    }

    @Override
    public boolean isApplicable(CommandContext context) {
        boolean answer = isFile(context.file(CHART_YAML));
        if (!answer) {
            answer = fileExistsInDir(context.getDir(), CHART_YAML);
            if (!answer) {
                return isFile(context.file(VALUES_YAML));
            }
        }
        return answer;
    }

    protected boolean fileExistsInDir(File dir, String fileName) {
        if (dir.isFile()) {
            return dir.getName().equals(fileName);
        } else if (dir.isDirectory()) {
            if (isFile(new File(dir, fileName))) {
                return true;
            }
            File[] files = dir.listFiles();
            if (files != null) {
                for (File file : files) {
                    if (Files.isDirectory(file)) {
                        if (fileExistsInDir(file, fileName)) {
                            return true;
                        }
                    }
                }
            }
        }
        return false;
    }

    /**
     * Lets find changes from a local release build of a chart and push them into dependent projects
     *
     * @param context
     * @param dependencyConfig
     * @param list
     * @throws IOException
     */
    @Override
    public void addVersionChangesFromSource(CommandContext context, Dependencies dependencyConfig, List list) throws IOException {
        File file = context.file(CHART_YAML);
        if (isFile(file)) {
            Chart chart;
            try {
                chart = MarkupHelper.loadYaml(file, Chart.class);
            } catch (IOException e) {
                throw new IOException("Failed to load chart " + file + ". " + e, e);
            }
            if (chart != null) {
                String name = chart.getName();
                String version = chart.getVersion();
                if (Strings.notEmpty(name) && Strings.notEmpty(version)) {
                    list.add(new DependencyVersionChange(Kind.HELM, name, version));
                }
            }
        }
    }

    @Override
    public boolean pushVersions(CommandContext context, List changes) throws IOException {
        return pushVersionsForDir(context, changes, context.getDir());
    }

    protected boolean pushVersionsForDir(CommandContext context, List changes, File dir) throws IOException {
        File chartsFile = new File(dir, CHART_YAML);
        boolean answer = false;
        if (isFile(chartsFile)) {
            File valuesFile = new File(dir, VALUES_YAML);
            if (isFile(valuesFile)) {
                if (updateValuesFile(context, changes, valuesFile)) {
                    answer = true;
                }
            }
            File requirementsFile = new File(dir, REQUIREMENTS_YAML);
            if (isFile(requirementsFile)) {
                Requirements requirements;
                try {
                    requirements = MarkupHelper.loadYaml(requirementsFile, Requirements.class);
                } catch (IOException e) {
                    throw new IOException("Failed to load chart requirements " + requirementsFile + ". " + e, e);
                }
                if (requirements != null) {
                    if (applyRequirementsChanges(context, changes, requirements, requirementsFile)) {
                        answer = true;
                    }
                }
            }
        } else {
            File[] files = dir.listFiles();
            if (files != null) {
                for (File file : files) {
                    if (Files.isDirectory(file)) {
                        if (pushVersionsForDir(context, changes, file)) {
                            answer = true;
                        }
                    }
                }
            }
        }
        return answer;
    }

    private boolean updateValuesFile(CommandContext context, List changes, File file) throws IOException {
        List lines = IOHelpers.readLines(file);
        boolean answer = false;
        for (DependencyVersionChange change : changes) {
            String linePrefix = "Image: " + change.getDependency() + ":";
            String value = change.getVersion();

            for (int i = 0, size = lines.size(); i < size; i++) {
                String line = lines.get(i);
                if (line.trim().startsWith(linePrefix)) {
                    String remaining = line.substring(linePrefix.length());
                    if (!remaining.trim().equals(value)) {
                        answer = true;
                        String whitespace = "";
                        int idx = line.indexOf(linePrefix);
                        if (idx > 0) {
                            whitespace = line.substring(0, idx);
                        }
                        lines.set(i, whitespace + linePrefix + value);
                    }
                }
            }
        }
        if (answer) {
            IOHelpers.writeLines(file, lines);
        }
        return answer;
    }
}