autogits/pr-review/main.go
Adam Majer bd82c134eb .
2024-07-15 21:14:09 +02:00

160 lines
4.4 KiB
Go

package main
import (
"fmt"
"path"
"src.opensuse.org/autogits/common"
"src.opensuse.org/autogits/common/gitea-generated/models"
)
const (
ListenAddr = "[::1]:8001"
GitAuthor = "GiteaBot - AutoStaging"
PrReview = "pr-review"
)
func processPullRequestClosed(h *common.RequestHandler) error {
return nil
}
func processPrjGitPullRequestSync(h *common.RequestHandler) error {
// req := h.Data.(*common.PullRequestAction)
return nil
}
func prGitBranchNameForPR(req *common.PullRequestAction) string {
return fmt.Sprintf("PR_%s#%d", req.Repository.Name, req.Pull_Request.Number)
}
func updateOrCreatePRBranch(h *common.RequestHandler, prjGit *models.Repository, commitMsg, branchName string) {
req := h.Data.(*common.PullRequestAction)
h.GitExec("", "clone", "--depth", "1", prjGit.SSHURL, common.DefaultGitPrj)
h.GitExec(common.DefaultGitPrj, "checkout", "-B", branchName, prjGit.DefaultBranch)
h.GitExec(common.DefaultGitPrj, "submodule", "update", "--init", "--checkout", "--depth", "1", req.Repository.Name)
h.GitExec(path.Join(common.DefaultGitPrj, req.Repository.Name), "fetch", "--depth", "1", "origin", req.Pull_Request.Head.Sha)
h.GitExec(path.Join(common.DefaultGitPrj, req.Repository.Name), "checkout", req.Pull_Request.Head.Sha)
h.GitExec(common.DefaultGitPrj, "commit", "-a", "-m", commitMsg)
h.GitExec(common.DefaultGitPrj, "push", "-f", "origin", branchName)
}
func processPullRequestSync(h *common.RequestHandler) error {
req := h.Data.(*common.PullRequestAction)
if req.Repository.Name == common.DefaultGitPrj {
return processPrjGitPullRequestSync(h)
}
// need to verify that submodule in the PR for prjgit
// is still pointing to the HEAD of the PR
prjPr := h.GetAssociatedPrjGitPR(req)
if h.HasError() {
h.LogError("%v", h.Error)
return h.Error
}
h.Log("associated pr: %v", prjPr)
h.GitExec("", "clone", "--branch", prjPr.Head.Name, "--depth", "1", prjPr.Head.Repo.SSHURL, common.DefaultGitPrj)
commitId, ok := h.GitSubmoduleCommitId(common.DefaultGitPrj, req.Repository.Name, prjPr.Head.Name)
if !ok {
return fmt.Errorf("Cannot fetch submodule commit id in prjgit for '%s'", req.Repository.Name)
}
// nothing changed, still in sync
if commitId == req.Pull_Request.Head.Ref {
return nil
}
commitMsg := fmt.Sprintf(`Sync PR
Update to %s`, req.Pull_Request.Head.Sha)
// we need to update prjgit PR with the new head hash
branchName := prGitBranchNameForPR(req)
updateOrCreatePRBranch(h, prjPr.Base.Repo, commitMsg, branchName)
return h.Error
}
func processPullRequestOpened(h *common.RequestHandler) error {
req := h.Data.(*common.PullRequestAction)
// requests against project are not handled here
if req.Repository.Name == common.DefaultGitPrj {
return nil
}
// create PrjGit branch for buidling the pull request
branchName := prGitBranchNameForPR(req)
commitMsg := fmt.Sprintf(`auto-created for %s
This commit was autocreated by %s
referencing
PullRequest: %s/%s#%d`, req.Repository.Owner.Username,
req.Repository.Name, GitAuthor, req.Repository.Name, req.Pull_Request.Number)
prjGit := h.CreateRepositoryIfNotExist(*req.Repository.Owner, common.DefaultGitPrj)
if h.HasError() {
return h.Error
}
updateOrCreatePRBranch(h, prjGit, commitMsg, branchName)
PR := h.CreatePullRequest(prjGit, branchName, prjGit.DefaultBranch,
fmt.Sprintf("Forwarded PR: %s", req.Repository.Name),
fmt.Sprintf(`This is a forwarded pull request by %s
referencing the following pull request:
`+common.PrPattern,
GitAuthor, req.Repository.Owner.Username, req.Repository.Name, req.Pull_Request.Number),
)
if h.HasError() {
return h.Error
}
// request build review
h.RequestReviews(PR, common.Bot_BuildReview)
return h.Error
}
func processPullRequest(h *common.RequestHandler) error {
req := h.Data.(*common.PullRequestAction)
switch req.Action {
case "opened":
return processPullRequestOpened(h)
case "synchronized":
return processPullRequestSync(h)
case "edited":
// not need to be handled??
return nil
case "closed":
return processPullRequestClosed(h)
}
return fmt.Errorf("Unhandled pull request action: %s", req.Action)
}
func main() {
var defs common.ListenDefinitions
defs.Url = PrReview
defs.GitAuthor = GitAuthor
defs.Handlers = make(map[string]common.RequestProcessor)
defs.Handlers[common.RequestType_PR] = processPullRequest
defs.Handlers[common.RequestType_PR_sync] = processPullRequest
common.RequireGiteaSecretToken()
common.RequireObsSecretToken()
common.StartServerWithAddress(defs, ListenAddr)
}