drone/internal/services/webhook/pullreq.go
2023-01-31 13:07:48 -08:00

66 lines
2.3 KiB
Go

// Copyright 2022 Harness Inc. All rights reserved.
// Use of this source code is governed by the Polyform Free Trial License
// that can be found in the LICENSE.md file for this repository.
package webhook
import (
"context"
"github.com/harness/gitness/events"
pullreqevents "github.com/harness/gitness/internal/events/pullreq"
"github.com/harness/gitness/types"
"github.com/harness/gitness/types/enum"
)
const (
// gitReferenceNamePrefixBranch is the prefix of references of type branch.
gitReferenceNamePrefixBranch = "refs/heads/"
)
// PullReqBranchBody describes the body of the pullreq branch related webhook trigger.
// NOTE: Embed ReferenceBody to make it easier for consumers!
// TODO: move in separate package for small import?
type PullReqBranchBody struct {
ReferenceBody
TargetRef ReferenceInfo `json:"target_ref"`
PullReq PullReqInfo `json:"pull_req"`
}
// handleEventPullReqBranchUpdated handles branch updated events for pull requests
// and triggers pullreq branch updated webhooks for the source repo.
func (s *Service) handleEventPullReqBranchUpdated(ctx context.Context,
event *events.Event[*pullreqevents.BranchUpdatedPayload]) error {
return s.triggerForEventWithPullReq(ctx, enum.WebhookTriggerPullReqBranchUpdated,
event.ID, event.Payload.PrincipalID, event.Payload.PullReqID,
func(principal *types.Principal, pr *types.PullReq, targetRepo, sourceRepo *types.Repository) (any, error) {
commitInfo, err := s.fetchCommitInfoForEvent(ctx, sourceRepo.GitUID, event.Payload.NewSHA)
if err != nil {
return nil, err
}
targetRepoInfo := repositoryInfoFrom(targetRepo, s.urlProvider)
sourceRepoInfo := repositoryInfoFrom(sourceRepo, s.urlProvider)
return &PullReqBranchBody{
ReferenceBody: ReferenceBody{
Trigger: enum.WebhookTriggerPullReqBranchUpdated,
Repo: targetRepoInfo,
Principal: principalInfoFrom(principal),
Ref: ReferenceInfo{
Name: gitReferenceNamePrefixBranch + pr.SourceBranch,
Repo: sourceRepoInfo,
},
Before: event.Payload.OldSHA,
After: event.Payload.NewSHA,
Commit: &commitInfo,
// Forced: true/false, // TODO: data not available yet
},
TargetRef: ReferenceInfo{
Name: gitReferenceNamePrefixBranch + pr.TargetBranch,
Repo: targetRepoInfo,
},
PullReq: pullReqInfoFrom(pr),
}, nil
})
}