maintain-flagfit-expiration-issue.main.kts

@file:Repository("https://repo.maven.apache.org/maven2")
@file:DependsOn("org.json:json:20200518")
@file:DependsOn("org.kohsuke:github-api:1.315")

import org.json.JSONObject
import org.kohsuke.github.GHIssueState
import org.kohsuke.github.GitHub
import java.io.File
import java.util.regex.Pattern

class FlagExpirationIssueMaintainer(
  private val githubToken: String,
  private val repoName: String,
  private val headSha: String,
) {

  fun maintain(
    locationSarif: String,
    labelName: String = "",
    assignerAlternative: String,
  ) {
    val gitHub = GitHub.connectUsingOAuth(githubToken)
    val repo = gitHub.getRepository(repoName)
    val targetRuleIdList = listOf(FLAGFIT_DEADLINE_SOON, FLAGFIT_DEADLINE_EXPIRED)
    val file = File(locationSarif)
    val content = file.readText()
    val jsonData = JSONObject(content)
    val runs = jsonData
      .getJSONArray("runs")
    val results = runs.getJSONObject(0).getJSONArray("results")
    val limitIssue = 50
    val existingIssues =
      repo.queryIssues()
        .apply { if (labelName.isNotEmpty()) label(labelName) }
        .state(GHIssueState.OPEN)
        .pageSize(limitIssue)
        .list()
        .take(limitIssue)
        .toMutableList()

    if (existingIssues.size >= limitIssue) {
      val messageLimitIssue = if (labelName.isNotEmpty()) {
        "Found more than $limitIssue Issues with $labelName set, " +
          "please make sure it is less than ${limitIssue}!"
      } else {
        "Found more than $limitIssue Issues set, " +
          "please make sure it is less than ${limitIssue}!"
      }
      throw IllegalStateException(messageLimitIssue)
    }

    for (i in 0 until results.length()) {
      val result = results.getJSONObject(i)
      val ruleId = result.getString("ruleId")
      if (ruleId in targetRuleIdList) {
        val message = result.getJSONObject("message").getString("markdown")

        val keyPatternRegex = "`key: (.*?)`"
        val ownerPatternRegex = "`owner: (.*?)`"
        val key = matchText(text = message, patternRegex = keyPatternRegex)
        val owner = matchText(text = message, patternRegex = ownerPatternRegex)

        val issueTitle = "Expiration status of the $key flag"
        val warningMessage = """
          |$message
          |
          |<!--
          |DO NOT CHANGE
          |This metadata is used for issue management.
          |
          |`ruleId: $ruleId`
          |`key: $key`
          |`owner: $owner`
          |-->
        """.trimMargin()
        val locations = result.getJSONArray("locations").getJSONObject(0)
        val physicalLocation = locations.getJSONObject("physicalLocation")
        val artifactLocation = physicalLocation.getJSONObject("artifactLocation")
        val baseUrl = repo.htmlUrl
        val uri = artifactLocation.getString("uri")
        val contextRegion = physicalLocation.getJSONObject("contextRegion")
        val artifactUri = "${baseUrl}/blob/${headSha}/${uri}" +
          "#L${contextRegion.getInt("startLine")}-L${
            contextRegion.getInt("endLine")
          }"

        val existingIssue = existingIssues.firstOrNull { issue ->
          val body = issue.body
          key == matchText(text = body, patternRegex = keyPatternRegex)
        }

        val collaborator = repo.collaboratorNames
        val assignee = if (owner in collaborator) owner else assignerAlternative

        if (existingIssue == null) {
          val issue = repo.createIssue(issueTitle)
            .body(warningMessage)
            .assignee(assignee)
            .apply { if (labelName.isNotEmpty()) label(labelName) }
            .create()
          issue.comment(artifactUri)
          existingIssues.add(issue)
        } else {
          val ruleIdPatternRegex = "`ruleId: (.*?)`"
          if (ruleId != matchText(
              text = existingIssue.body,
              patternRegex = ruleIdPatternRegex
            )) {
            existingIssue.comments.forEach {
              if (ruleId == matchText(text = it.body, patternRegex = ruleIdPatternRegex)) {
                it.delete()
              }
            }
            val warningComment = "## Warning\n$warningMessage"
            existingIssue.comment(warningComment)
          }
        }
      }
    }
  }

  private fun matchText(text: String, patternRegex: String): String {
    val pattern = Pattern.compile(patternRegex)
    val matcher = pattern.matcher(text)
    return if (matcher.find()) matcher.group(1) else ""
  }

  companion object {
    const val FLAGFIT_DEADLINE_SOON = "FlagfitDeadlineSoon"
    const val FLAGFIT_DEADLINE_EXPIRED = "FlagfitDeadlineExpired"
  }
}

FlagExpirationIssueMaintainer(
  githubToken = System.getenv("GITHUB_TOKEN"),
  repoName = System.getenv("GITHUB_REPOSITORY"),
  headSha = System.getenv("HEAD_SHA")
).maintain(
  locationSarif = "./lint-results.sarif",
  labelName = "",
  assignerAlternative = "{Alternative assignor's GitHub UserId}"
)