main.kotlin.org.virtuslab.bazelsteward.app.PullRequestManager.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of bazel-steward Show documentation

Bazel Steward is a bot that helps you keep your library dependencies, Bazel and Bazel rules up-to-date.

There is a newer version: 1.5.2

package org.virtuslab.bazelsteward.app

import mu.KotlinLogging
import org.virtuslab.bazelsteward.app.provider.PostUpdateHookProvider
import org.virtuslab.bazelsteward.core.GitHostClient
import org.virtuslab.bazelsteward.core.GitHostClient.PrStatus.NONE
import org.virtuslab.bazelsteward.core.GitHostClient.PrStatus.OPEN_MERGEABLE
import org.virtuslab.bazelsteward.core.GitHostClient.PrStatus.OPEN_MODIFIED
import org.virtuslab.bazelsteward.core.GitHostClient.PrStatus.OPEN_NOT_MERGEABLE
import org.virtuslab.bazelsteward.core.PullRequest
import org.virtuslab.bazelsteward.core.common.CommandRunner
import org.virtuslab.bazelsteward.core.common.GitOperations
import org.virtuslab.bazelsteward.core.common.HookRunFor
import java.nio.file.Path

private val logger = KotlinLogging.logger {}

data class PullRequestManager(
  private val gitHostClient: GitHostClient,
  private val gitOperations: GitOperations,
  private val provider: PostUpdateHookProvider,
  private val workspaceRoot: Path,
  private val pushToRemote: Boolean,
  private val updateAllPullRequests: Boolean,
) {
  suspend fun applySuggestions(pullRequestSuggestions: List) {
    pullRequestSuggestions.forEach { pr ->
      val prStatus = gitHostClient.checkPrStatus(pr.branch)
      if (canCreateOrUpdate(prStatus)) {
        logger.info { "Creating branch ${pr.branch}, PR status: $prStatus" }
        runCatching {
          val config = provider.resolveForLibrary(pr.oldLibrary)
          gitOperations.createBranchWithChange(pr.branch, pr.commits)
          if (config.commands.isNotEmpty()) {
            config.commands.forEach {
              Thread.sleep(1000)
              CommandRunner.run(listOf("sh", "-c", it), workspaceRoot)
            }
            gitOperations.commitSelectedFiles(config.filesToCommit, config.commitMessage)
            if (config.runFor == HookRunFor.Commit) {
              gitOperations.squashLastTwoCommits(pr.commits.last().message)
            }
          }
          if (pushToRemote) {
            gitOperations.pushBranchToOrigin(pr.branch, force = prStatus != NONE)
            val openPr = if (prStatus == NONE) {
              val oldPrs = gitHostClient.getOpenPrs().filter {
                it.branch.name.startsWith(pr.branchPrefix) && it.branch != pr.branch
              }.filter { gitHostClient.checkPrStatus(it.branch) != OPEN_MODIFIED }
              gitHostClient.closePrs(oldPrs)
              gitHostClient.openNewPr(pr.description)
            } else {
              PullRequest(pr.branch)
            }
            gitHostClient.onPrChange(openPr, prStatus)
          }
        }.onFailure { logger.error(it) { "Failed to create branch ${pr.branch}" } }
        gitOperations.checkoutBaseBranch()
      } else {
        logger.info { "Skipping ${pr.branch}, PR status: $prStatus" }
      }
    }
  }

  private fun canCreateOrUpdate(prStatus: GitHostClient.PrStatus): Boolean =
    when (prStatus) {
      NONE, OPEN_NOT_MERGEABLE -> true
      OPEN_MERGEABLE, OPEN_MODIFIED -> updateAllPullRequests
      else -> false
    }
}