909 lines · 33.4 KB · Go
RawDownload
// Copyright 2019 The Gitea Authors. All rights reserved.// SPDX-License-Identifier: MIT package webhook import (	"context" 	issues_model "forgejo.org/models/issues"	packages_model "forgejo.org/models/packages"	"forgejo.org/models/perm"	access_model "forgejo.org/models/perm/access"	repo_model "forgejo.org/models/repo"	user_model "forgejo.org/models/user"	"forgejo.org/modules/git"	"forgejo.org/modules/log"	"forgejo.org/modules/repository"	"forgejo.org/modules/setting"	api "forgejo.org/modules/structs"	webhook_module "forgejo.org/modules/webhook"	"forgejo.org/services/convert"	notify_service "forgejo.org/services/notify") func init() {	notify_service.RegisterNotifier(&webhookNotifier{})} type webhookNotifier struct {	notify_service.NullNotifier} var _ notify_service.Notifier = &webhookNotifier{} // NewNotifier create a new webhookNotifier notifierfunc NewNotifier() notify_service.Notifier {	return &webhookNotifier{}} func (m *webhookNotifier) IssueClearLabels(ctx context.Context, doer *user_model.User, issue *issues_model.Issue) {	if err := issue.LoadPoster(ctx); err != nil {		log.Error("LoadPoster: %v", err)		return	} 	if err := issue.LoadRepo(ctx); err != nil {		log.Error("LoadRepo: %v", err)		return	} 	permission, _ := access_model.GetUserRepoPermission(ctx, issue.Repo, issue.Poster)	var err error	if issue.IsPull {		if err = issue.LoadPullRequest(ctx); err != nil {			log.Error("LoadPullRequest: %v", err)			return		} 		err = PrepareWebhooks(ctx, EventSource{Repository: issue.Repo}, webhook_module.HookEventPullRequestLabel, &api.PullRequestPayload{			Action:      api.HookIssueLabelCleared,			Index:       issue.Index,			PullRequest: convert.ToAPIPullRequest(ctx, issue.PullRequest, doer),			Repository:  convert.ToRepo(ctx, issue.Repo, permission),			Sender:      convert.ToUser(ctx, doer, nil),		})	} else {		err = PrepareWebhooks(ctx, EventSource{Repository: issue.Repo}, webhook_module.HookEventIssueLabel, &api.IssuePayload{			Action:     api.HookIssueLabelCleared,			Index:      issue.Index,			Issue:      convert.ToAPIIssue(ctx, doer, issue),			Repository: convert.ToRepo(ctx, issue.Repo, permission),			Sender:     convert.ToUser(ctx, doer, nil),		})	}	if err != nil {		log.Error("PrepareWebhooks [is_pull: %v]: %v", issue.IsPull, err)	}} func (m *webhookNotifier) ForkRepository(ctx context.Context, doer *user_model.User, oldRepo, repo *repo_model.Repository) {	oldPermission, _ := access_model.GetUserRepoPermission(ctx, oldRepo, doer)	permission, _ := access_model.GetUserRepoPermission(ctx, repo, doer) 	// forked webhook	if err := PrepareWebhooks(ctx, EventSource{Repository: oldRepo}, webhook_module.HookEventFork, &api.ForkPayload{		Forkee: convert.ToRepo(ctx, oldRepo, oldPermission),		Repo:   convert.ToRepo(ctx, repo, permission),		Sender: convert.ToUser(ctx, doer, nil),	}); err != nil {		log.Error("PrepareWebhooks [repo_id: %d]: %v", oldRepo.ID, err)	} 	u := repo.MustOwner(ctx) 	// Add to hook queue for created repo after session commit.	if u.IsOrganization() {		if err := PrepareWebhooks(ctx, EventSource{Repository: repo}, webhook_module.HookEventRepository, &api.RepositoryPayload{			Action:       api.HookRepoCreated,			Repository:   convert.ToRepo(ctx, repo, access_model.Permission{AccessMode: perm.AccessModeOwner}),			Organization: convert.ToUser(ctx, u, nil),			Sender:       convert.ToUser(ctx, doer, nil),		}); err != nil {			log.Error("PrepareWebhooks [repo_id: %d]: %v", repo.ID, err)		}	}} func (m *webhookNotifier) CreateRepository(ctx context.Context, doer, u *user_model.User, repo *repo_model.Repository) {	// Add to hook queue for created repo after session commit.	if err := PrepareWebhooks(ctx, EventSource{Repository: repo}, webhook_module.HookEventRepository, &api.RepositoryPayload{		Action:       api.HookRepoCreated,		Repository:   convert.ToRepo(ctx, repo, access_model.Permission{AccessMode: perm.AccessModeOwner}),		Organization: convert.ToUser(ctx, u, nil),		Sender:       convert.ToUser(ctx, doer, nil),	}); err != nil {		log.Error("PrepareWebhooks [repo_id: %d]: %v", repo.ID, err)	}} func (m *webhookNotifier) DeleteRepository(ctx context.Context, doer *user_model.User, repo *repo_model.Repository) {	if err := PrepareWebhooks(ctx, EventSource{Repository: repo}, webhook_module.HookEventRepository, &api.RepositoryPayload{		Action:       api.HookRepoDeleted,		Repository:   convert.ToRepo(ctx, repo, access_model.Permission{AccessMode: perm.AccessModeOwner}),		Organization: convert.ToUser(ctx, repo.MustOwner(ctx), nil),		Sender:       convert.ToUser(ctx, doer, nil),	}); err != nil {		log.Error("PrepareWebhooks [repo_id: %d]: %v", repo.ID, err)	}} func (m *webhookNotifier) MigrateRepository(ctx context.Context, doer, u *user_model.User, repo *repo_model.Repository) {	// Add to hook queue for created repo after session commit.	if err := PrepareWebhooks(ctx, EventSource{Repository: repo}, webhook_module.HookEventRepository, &api.RepositoryPayload{		Action:       api.HookRepoCreated,		Repository:   convert.ToRepo(ctx, repo, access_model.Permission{AccessMode: perm.AccessModeOwner}),		Organization: convert.ToUser(ctx, u, nil),		Sender:       convert.ToUser(ctx, doer, nil),	}); err != nil {		log.Error("PrepareWebhooks [repo_id: %d]: %v", repo.ID, err)	}} func (m *webhookNotifier) IssueChangeAssignee(ctx context.Context, doer *user_model.User, issue *issues_model.Issue, assignee *user_model.User, removed bool, comment *issues_model.Comment) {	if issue.IsPull {		permission, _ := access_model.GetUserRepoPermission(ctx, issue.Repo, doer) 		if err := issue.LoadPullRequest(ctx); err != nil {			log.Error("LoadPullRequest failed: %v", err)			return		}		apiPullRequest := &api.PullRequestPayload{			Index:       issue.Index,			PullRequest: convert.ToAPIPullRequest(ctx, issue.PullRequest, doer),			Repository:  convert.ToRepo(ctx, issue.Repo, permission),			Sender:      convert.ToUser(ctx, doer, nil),		}		if removed {			apiPullRequest.Action = api.HookIssueUnassigned		} else {			apiPullRequest.Action = api.HookIssueAssigned		}		// Assignee comment triggers a webhook		if err := PrepareWebhooks(ctx, EventSource{Repository: issue.Repo}, webhook_module.HookEventPullRequestAssign, apiPullRequest); err != nil {			log.Error("PrepareWebhooks [is_pull: %v, remove_assignee: %v]: %v", issue.IsPull, removed, err)			return		}	} else {		permission, _ := access_model.GetUserRepoPermission(ctx, issue.Repo, doer)		apiIssue := &api.IssuePayload{			Index:      issue.Index,			Issue:      convert.ToAPIIssue(ctx, doer, issue),			Repository: convert.ToRepo(ctx, issue.Repo, permission),			Sender:     convert.ToUser(ctx, doer, nil),		}		if removed {			apiIssue.Action = api.HookIssueUnassigned		} else {			apiIssue.Action = api.HookIssueAssigned		}		// Assignee comment triggers a webhook		if err := PrepareWebhooks(ctx, EventSource{Repository: issue.Repo}, webhook_module.HookEventIssueAssign, apiIssue); err != nil {			log.Error("PrepareWebhooks [is_pull: %v, remove_assignee: %v]: %v", issue.IsPull, removed, err)			return		}	}} func (m *webhookNotifier) IssueChangeTitle(ctx context.Context, doer *user_model.User, issue *issues_model.Issue, oldTitle string) {	permission, _ := access_model.GetUserRepoPermission(ctx, issue.Repo, issue.Poster)	var err error	if issue.IsPull {		if err = issue.LoadPullRequest(ctx); err != nil {			log.Error("LoadPullRequest failed: %v", err)			return		}		err = PrepareWebhooks(ctx, EventSource{Repository: issue.Repo}, webhook_module.HookEventPullRequest, &api.PullRequestPayload{			Action: api.HookIssueEdited,			Index:  issue.Index,			Changes: &api.ChangesPayload{				Title: &api.ChangesFromPayload{					From: oldTitle,				},			},			PullRequest: convert.ToAPIPullRequest(ctx, issue.PullRequest, doer),			Repository:  convert.ToRepo(ctx, issue.Repo, permission),			Sender:      convert.ToUser(ctx, doer, nil),		})	} else {		err = PrepareWebhooks(ctx, EventSource{Repository: issue.Repo}, webhook_module.HookEventIssues, &api.IssuePayload{			Action: api.HookIssueEdited,			Index:  issue.Index,			Changes: &api.ChangesPayload{				Title: &api.ChangesFromPayload{					From: oldTitle,				},			},			Issue:      convert.ToAPIIssue(ctx, doer, issue),			Repository: convert.ToRepo(ctx, issue.Repo, permission),			Sender:     convert.ToUser(ctx, doer, nil),		})	} 	if err != nil {		log.Error("PrepareWebhooks [is_pull: %v]: %v", issue.IsPull, err)	}} func (m *webhookNotifier) IssueChangeStatus(ctx context.Context, doer *user_model.User, commitID string, issue *issues_model.Issue, actionComment *issues_model.Comment, isClosed bool) {	permission, _ := access_model.GetUserRepoPermission(ctx, issue.Repo, issue.Poster)	var err error	if issue.IsPull {		if err = issue.LoadPullRequest(ctx); err != nil {			log.Error("LoadPullRequest: %v", err)			return		}		// Merge pull request calls issue.changeStatus so we need to handle separately.		apiPullRequest := &api.PullRequestPayload{			Index:       issue.Index,			PullRequest: convert.ToAPIPullRequest(ctx, issue.PullRequest, doer),			Repository:  convert.ToRepo(ctx, issue.Repo, permission),			Sender:      convert.ToUser(ctx, doer, nil),			CommitID:    commitID,		}		if isClosed {			apiPullRequest.Action = api.HookIssueClosed		} else {			apiPullRequest.Action = api.HookIssueReOpened		}		err = PrepareWebhooks(ctx, EventSource{Repository: issue.Repo}, webhook_module.HookEventPullRequest, apiPullRequest)	} else {		apiIssue := &api.IssuePayload{			Index:      issue.Index,			Issue:      convert.ToAPIIssue(ctx, doer, issue),			Repository: convert.ToRepo(ctx, issue.Repo, permission),			Sender:     convert.ToUser(ctx, doer, nil),			CommitID:   commitID,		}		if isClosed {			apiIssue.Action = api.HookIssueClosed		} else {			apiIssue.Action = api.HookIssueReOpened		}		err = PrepareWebhooks(ctx, EventSource{Repository: issue.Repo}, webhook_module.HookEventIssues, apiIssue)	}	if err != nil {		log.Error("PrepareWebhooks [is_pull: %v, is_closed: %v]: %v", issue.IsPull, isClosed, err)	}} func (m *webhookNotifier) NewIssue(ctx context.Context, issue *issues_model.Issue, mentions []*user_model.User) {	if err := issue.LoadRepo(ctx); err != nil {		log.Error("issue.LoadRepo: %v", err)		return	}	if err := issue.LoadPoster(ctx); err != nil {		log.Error("issue.LoadPoster: %v", err)		return	} 	permission, _ := access_model.GetUserRepoPermission(ctx, issue.Repo, issue.Poster)	if err := PrepareWebhooks(ctx, EventSource{Repository: issue.Repo}, webhook_module.HookEventIssues, &api.IssuePayload{		Action:     api.HookIssueOpened,		Index:      issue.Index,		Issue:      convert.ToAPIIssue(ctx, issue.Poster, issue),		Repository: convert.ToRepo(ctx, issue.Repo, permission),		Sender:     convert.ToUser(ctx, issue.Poster, nil),	}); err != nil {		log.Error("PrepareWebhooks: %v", err)	}} func (m *webhookNotifier) NewPullRequest(ctx context.Context, pull *issues_model.PullRequest, mentions []*user_model.User) {	if err := pull.LoadIssue(ctx); err != nil {		log.Error("pull.LoadIssue: %v", err)		return	}	if err := pull.Issue.LoadRepo(ctx); err != nil {		log.Error("pull.Issue.LoadRepo: %v", err)		return	}	if err := pull.Issue.LoadPoster(ctx); err != nil {		log.Error("pull.Issue.LoadPoster: %v", err)		return	} 	permission, _ := access_model.GetUserRepoPermission(ctx, pull.Issue.Repo, pull.Issue.Poster)	if err := PrepareWebhooks(ctx, EventSource{Repository: pull.Issue.Repo}, webhook_module.HookEventPullRequest, &api.PullRequestPayload{		Action:      api.HookIssueOpened,		Index:       pull.Issue.Index,		PullRequest: convert.ToAPIPullRequest(ctx, pull, pull.Issue.Poster),		Repository:  convert.ToRepo(ctx, pull.Issue.Repo, permission),		Sender:      convert.ToUser(ctx, pull.Issue.Poster, nil),	}); err != nil {		log.Error("PrepareWebhooks: %v", err)	}} func (m *webhookNotifier) IssueChangeContent(ctx context.Context, doer *user_model.User, issue *issues_model.Issue, oldContent string) {	if err := issue.LoadRepo(ctx); err != nil {		log.Error("LoadRepo: %v", err)		return	} 	permission, _ := access_model.GetUserRepoPermission(ctx, issue.Repo, issue.Poster)	var err error	if issue.IsPull {		if err := issue.LoadPullRequest(ctx); err != nil {			log.Error("LoadPullRequest: %v", err)			return		}		err = PrepareWebhooks(ctx, EventSource{Repository: issue.Repo}, webhook_module.HookEventPullRequest, &api.PullRequestPayload{			Action: api.HookIssueEdited,			Index:  issue.Index,			Changes: &api.ChangesPayload{				Body: &api.ChangesFromPayload{					From: oldContent,				},			},			PullRequest: convert.ToAPIPullRequest(ctx, issue.PullRequest, doer),			Repository:  convert.ToRepo(ctx, issue.Repo, permission),			Sender:      convert.ToUser(ctx, doer, nil),		})	} else {		err = PrepareWebhooks(ctx, EventSource{Repository: issue.Repo}, webhook_module.HookEventIssues, &api.IssuePayload{			Action: api.HookIssueEdited,			Index:  issue.Index,			Changes: &api.ChangesPayload{				Body: &api.ChangesFromPayload{					From: oldContent,				},			},			Issue:      convert.ToAPIIssue(ctx, doer, issue),			Repository: convert.ToRepo(ctx, issue.Repo, permission),			Sender:     convert.ToUser(ctx, doer, nil),		})	}	if err != nil {		log.Error("PrepareWebhooks [is_pull: %v]: %v", issue.IsPull, err)	}} func (m *webhookNotifier) UpdateComment(ctx context.Context, doer *user_model.User, c *issues_model.Comment, oldContent string) {	if err := c.LoadPoster(ctx); err != nil {		log.Error("LoadPoster: %v", err)		return	}	if err := c.LoadIssue(ctx); err != nil {		log.Error("LoadIssue: %v", err)		return	} 	if err := c.Issue.LoadAttributes(ctx); err != nil {		log.Error("LoadAttributes: %v", err)		return	} 	var eventType webhook_module.HookEventType	var pullRequest *api.PullRequest	if c.Issue.IsPull {		eventType = webhook_module.HookEventPullRequestComment		pullRequest = convert.ToAPIPullRequest(ctx, c.Issue.PullRequest, doer)	} else {		eventType = webhook_module.HookEventIssueComment	} 	permission, _ := access_model.GetUserRepoPermission(ctx, c.Issue.Repo, doer)	if err := PrepareWebhooks(ctx, EventSource{Repository: c.Issue.Repo}, eventType, &api.IssueCommentPayload{		Action:      api.HookIssueCommentEdited,		Issue:       convert.ToAPIIssue(ctx, doer, c.Issue),		PullRequest: pullRequest,		Comment:     convert.ToAPIComment(ctx, c.Issue.Repo, c),		Changes: &api.ChangesPayload{			Body: &api.ChangesFromPayload{				From: oldContent,			},		},		Repository: convert.ToRepo(ctx, c.Issue.Repo, permission),		Sender:     convert.ToUser(ctx, doer, nil),		IsPull:     c.Issue.IsPull,	}); err != nil {		log.Error("PrepareWebhooks [comment_id: %d]: %v", c.ID, err)	}} func (m *webhookNotifier) CreateIssueComment(ctx context.Context, doer *user_model.User, repo *repo_model.Repository,	issue *issues_model.Issue, comment *issues_model.Comment, mentions []*user_model.User,) {	var eventType webhook_module.HookEventType	var pullRequest *api.PullRequest	if issue.IsPull {		eventType = webhook_module.HookEventPullRequestComment		if err := issue.LoadPullRequest(ctx); err != nil {			log.Error("LoadPullRequest: %v", err)			return		}		pullRequest = convert.ToAPIPullRequest(ctx, issue.PullRequest, doer)	} else {		eventType = webhook_module.HookEventIssueComment	} 	permission, _ := access_model.GetUserRepoPermission(ctx, repo, doer)	if err := PrepareWebhooks(ctx, EventSource{Repository: issue.Repo}, eventType, &api.IssueCommentPayload{		Action:      api.HookIssueCommentCreated,		Issue:       convert.ToAPIIssue(ctx, doer, issue),		PullRequest: pullRequest,		Comment:     convert.ToAPIComment(ctx, repo, comment),		Repository:  convert.ToRepo(ctx, repo, permission),		Sender:      convert.ToUser(ctx, doer, nil),		IsPull:      issue.IsPull,	}); err != nil {		log.Error("PrepareWebhooks [comment_id: %d]: %v", comment.ID, err)	}} func (m *webhookNotifier) DeleteComment(ctx context.Context, doer *user_model.User, comment *issues_model.Comment) {	var err error 	if err = comment.LoadPoster(ctx); err != nil {		log.Error("LoadPoster: %v", err)		return	}	if err = comment.LoadIssue(ctx); err != nil {		log.Error("LoadIssue: %v", err)		return	} 	if err = comment.Issue.LoadAttributes(ctx); err != nil {		log.Error("LoadAttributes: %v", err)		return	} 	var eventType webhook_module.HookEventType	var pullRequest *api.PullRequest	if comment.Issue.IsPull {		eventType = webhook_module.HookEventPullRequestComment		pullRequest = convert.ToAPIPullRequest(ctx, comment.Issue.PullRequest, doer)	} else {		eventType = webhook_module.HookEventIssueComment	} 	permission, _ := access_model.GetUserRepoPermission(ctx, comment.Issue.Repo, doer)	if err := PrepareWebhooks(ctx, EventSource{Repository: comment.Issue.Repo}, eventType, &api.IssueCommentPayload{		Action:      api.HookIssueCommentDeleted,		Issue:       convert.ToAPIIssue(ctx, doer, comment.Issue),		PullRequest: pullRequest,		Comment:     convert.ToAPIComment(ctx, comment.Issue.Repo, comment),		Repository:  convert.ToRepo(ctx, comment.Issue.Repo, permission),		Sender:      convert.ToUser(ctx, doer, nil),		IsPull:      comment.Issue.IsPull,	}); err != nil {		log.Error("PrepareWebhooks [comment_id: %d]: %v", comment.ID, err)	}} func (m *webhookNotifier) NewWikiPage(ctx context.Context, doer *user_model.User, repo *repo_model.Repository, page, comment string) {	// Add to hook queue for created wiki page.	if err := PrepareWebhooks(ctx, EventSource{Repository: repo}, webhook_module.HookEventWiki, &api.WikiPayload{		Action:     api.HookWikiCreated,		Repository: convert.ToRepo(ctx, repo, access_model.Permission{AccessMode: perm.AccessModeOwner}),		Sender:     convert.ToUser(ctx, doer, nil),		Page:       page,		Comment:    comment,	}); err != nil {		log.Error("PrepareWebhooks [repo_id: %d]: %v", repo.ID, err)	}} func (m *webhookNotifier) EditWikiPage(ctx context.Context, doer *user_model.User, repo *repo_model.Repository, page, comment string) {	// Add to hook queue for edit wiki page.	if err := PrepareWebhooks(ctx, EventSource{Repository: repo}, webhook_module.HookEventWiki, &api.WikiPayload{		Action:     api.HookWikiEdited,		Repository: convert.ToRepo(ctx, repo, access_model.Permission{AccessMode: perm.AccessModeOwner}),		Sender:     convert.ToUser(ctx, doer, nil),		Page:       page,		Comment:    comment,	}); err != nil {		log.Error("PrepareWebhooks [repo_id: %d]: %v", repo.ID, err)	}} 

Lines 1–500 of 909.