com.salesforce.dockerfileimageupdate.utils.PullRequests Maven / Gradle / Ivy

Go to download
package com.salesforce.dockerfileimageupdate.utils;

import com.google.common.collect.*;
import com.salesforce.dockerfileimageupdate.model.*;
import com.salesforce.dockerfileimageupdate.process.*;
import net.sourceforge.argparse4j.inf.*;
import org.json.JSONObject;
import org.json.JSONTokener;
import org.json.JSONException;
import org.kohsuke.github.*;
import org.slf4j.*;

import java.io.*;
import java.util.*;

public class PullRequests {
    private static final Logger log = LoggerFactory.getLogger(PullRequests.class);
    public void prepareToCreate(final Namespace ns,
                                            GitHubPullRequestSender pullRequestSender,
                                            PagedSearchIterable contentsFoundWithImage,
                                            GitForkBranch gitForkBranch,
                                            DockerfileGitHubUtil dockerfileGitHubUtil,
                                            RateLimiter rateLimiter) throws IOException {
        Multimap pathToDockerfilesInParentRepo =
                pullRequestSender.forkRepositoriesFoundAndGetPathToDockerfiles(contentsFoundWithImage, gitForkBranch);
        List exceptions = new ArrayList<>();
        List skippedRepos = new ArrayList<>();
        for (String currUserRepo : pathToDockerfilesInParentRepo.keySet()) {
            Optional forkWithContentPaths =
                    pathToDockerfilesInParentRepo.get(currUserRepo).stream().findFirst();
            if (forkWithContentPaths.isPresent()) {
                try {
                    //If the repository has been onboarded to renovate enterprise, skip sending the DFIU PR
                    if(ns.getBoolean(Constants.CHECK_FOR_RENOVATE)
                            && (isRenovateEnabled(Constants.RENOVATE_CONFIG_FILEPATHS, forkWithContentPaths.get()))) {
                        log.info("Found a renovate configuration file in the repo {}. Skip sending DFIU PRs to this repository.", forkWithContentPaths.get().getParent().getFullName());
                    } else {
                        dockerfileGitHubUtil.changeDockerfiles(ns,
                                pathToDockerfilesInParentRepo,
                                forkWithContentPaths.get(), skippedRepos,
                                gitForkBranch, rateLimiter);
                    }
                } catch (IOException | InterruptedException e) {
                    log.error(String.format("Error changing Dockerfile for %s", forkWithContentPaths.get().getParent().getFullName()), e);
                    exceptions.add((IOException) e);
                }
            } else {
                log.warn("Didn't find fork for {} so not changing Dockerfiles", currUserRepo);
            }
        }
        ResultsProcessor.processResults(skippedRepos, exceptions, log);
    }

    /**
     * Check if the repository is onboarded to Renovate. The signal we are looking for are
     * (1) The presence of a file where renovate configurations are stored in the repository
     * (2) Ensuring that the file does not have the key "enabled" set to "false"
     * @param filePaths the list that contains all the names of the files that we are searching for in the repo
     * @param fork A GitHubContentToProcess object that contains the fork repository that is under process
     * @return true if the file is found in the path specified and is not disabled, false otherwise
     */
    protected boolean isRenovateEnabled(List filePaths, GitHubContentToProcess fork) throws IOException {
        for (String filePath : filePaths) {
            try {
                //If the file has the key 'enabled' set to false, it indicates that while the repo has been onboarded to renovate, it has been disabled for some reason
                return readJsonFromContent(fork.getParent().getFileContent(filePath)).optBoolean("enabled", true);
            } catch (FileNotFoundException e) {
                log.debug("The file with name {} not found in the repository.Returning false. Exception: {}", filePath, e.getMessage());
            } catch (IOException e) {
                log.warn("Exception while trying to close a resource. Returning false. Exception: {}", e.getMessage());
            } catch (JSONException e) {
                log.warn("Exception while trying to read the renovate configuration file. Returning false. Exception: {}", e.getMessage());
            }
        }
        return false;
    }

    /**
     * Read the content of a fle from a repository and convert it into a JSON object
     * @param content The GHContent object for the content in the repository
     * @return json object for the content read from the repository
     */
    private JSONObject readJsonFromContent(GHContent content) throws IOException {
        try (BufferedReader bufferedReader = new BufferedReader(new InputStreamReader(content.read()))) {
            JSONTokener tokener = new JSONTokener(bufferedReader);
            return new JSONObject(tokener);
        }
    }
}