mirror of
https://github.com/harness/drone.git
synced 2025-05-05 23:11:06 +08:00
5cb824debd
This change is adding git server hook components: - githook CLI implementation - githook api handlers - githook controler
109 lines
3.3 KiB
Go
109 lines
3.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 githook
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"strings"
|
|
|
|
"github.com/harness/gitness/internal/auth"
|
|
events "github.com/harness/gitness/internal/events/git"
|
|
"github.com/harness/gitness/types"
|
|
|
|
"github.com/rs/zerolog/log"
|
|
)
|
|
|
|
const (
|
|
// gitReferenceNamePrefixBranch is the prefix of references of type branch.
|
|
gitReferenceNamePrefixBranch = "refs/heads/"
|
|
|
|
// gitReferenceNamePrefixTag is the prefix of references of type tag.
|
|
gitReferenceNamePrefixTag = "refs/tags/"
|
|
)
|
|
|
|
// PostReceiveInput represents the input of the post-receive git hook.
|
|
type PostReceiveInput struct {
|
|
BaseInput
|
|
// RefUpdates contains all references that got updated as part of the git operation.
|
|
RefUpdates []ReferenceUpdate `json:"ref_updates"`
|
|
}
|
|
|
|
// PostReceive executes the post-receive hook for a git repository.
|
|
func (c *Controller) PostReceive(
|
|
ctx context.Context,
|
|
session *auth.Session,
|
|
in *PostReceiveInput,
|
|
) (*ServerHookOutput, error) {
|
|
if in == nil {
|
|
return nil, fmt.Errorf("input is nil")
|
|
}
|
|
|
|
// report ref events (best effort)
|
|
c.reportReferenceEvents(ctx, in)
|
|
|
|
return &ServerHookOutput{}, nil
|
|
}
|
|
|
|
// reportReferenceEvents is reporting reference events to the event system.
|
|
// NOTE: keep best effort for now as it doesn't change the outcome of the git operation.
|
|
// TODO: in the future we might want to think about propagating errors so user is aware of events not being triggered.
|
|
func (c *Controller) reportReferenceEvents(ctx context.Context, in *PostReceiveInput) {
|
|
for _, refUpdate := range in.RefUpdates {
|
|
switch {
|
|
case strings.HasPrefix(refUpdate.Ref, gitReferenceNamePrefixBranch):
|
|
c.reportBranchEvent(ctx, in.PrincipalID, in.RepoID, refUpdate)
|
|
case strings.HasPrefix(refUpdate.Ref, gitReferenceNamePrefixTag):
|
|
c.reportTagEvent(ctx, in.PrincipalID, in.RepoID, refUpdate)
|
|
default:
|
|
// Ignore any other references in post-receive
|
|
}
|
|
}
|
|
}
|
|
|
|
func (c *Controller) reportBranchEvent(ctx context.Context,
|
|
principalID int64, repoID int64, branchUpdate ReferenceUpdate) {
|
|
switch {
|
|
case branchUpdate.Old == types.NilSHA:
|
|
c.gitReporter.BranchCreated(ctx, &events.BranchCreatedPayload{
|
|
RepoID: repoID,
|
|
PrincipalID: principalID,
|
|
Ref: branchUpdate.Ref,
|
|
SHA: branchUpdate.New,
|
|
})
|
|
case branchUpdate.New == types.NilSHA:
|
|
c.gitReporter.BranchDeleted(ctx, &events.BranchDeletedPayload{
|
|
RepoID: repoID,
|
|
PrincipalID: principalID,
|
|
Ref: branchUpdate.Ref,
|
|
SHA: branchUpdate.Old,
|
|
})
|
|
default:
|
|
c.gitReporter.BranchUpdated(ctx, &events.BranchUpdatedPayload{
|
|
RepoID: repoID,
|
|
PrincipalID: principalID,
|
|
Ref: branchUpdate.Ref,
|
|
OldSHA: branchUpdate.Old,
|
|
NewSHA: branchUpdate.New,
|
|
// Forced: false, // TODO: data not available yet
|
|
})
|
|
}
|
|
}
|
|
|
|
func (c *Controller) reportTagEvent(ctx context.Context,
|
|
principalID int64, repoID int64, tagUpdate ReferenceUpdate) {
|
|
switch {
|
|
case tagUpdate.Old == types.NilSHA:
|
|
// TODO
|
|
case tagUpdate.New == types.NilSHA:
|
|
// TODO
|
|
default:
|
|
// TODO
|
|
}
|
|
|
|
log.Ctx(ctx).Info().Msgf("received a change for tag '%s' in repo %d by principal %d (SHA: %s -> %s)'",
|
|
tagUpdate.Ref, repoID, principalID, tagUpdate.Old, tagUpdate.New)
|
|
}
|