react on comment in pr with digger commands (#1496)

* react on digger comment

Author: motatoes

backend/controllers/github.go

@@ -607,6 +607,7 @@ func handleIssueCommentEvent(gh utils.GithubClientProvider, payload *github.Issu
 	cloneURL := *payload.Repo.CloneURL
 	issueNumber := *payload.Issue.Number
 	isDraft := payload.Issue.GetDraft()
+	commentId := *payload.GetComment().ID
 
 	link, err := models.DB.GetGithubAppInstallationLink(installationId)
 	if err != nil {
@@ -631,6 +632,11 @@ func handleIssueCommentEvent(gh utils.GithubClientProvider, payload *github.Issu
 		return fmt.Errorf("error getting digger config")
 	}
 
+	err = ghService.CreateCommentReaction(commentId, string(dg_github.GithubCommentEyesReaction))
+	if err != nil {
+		log.Printf("CreateCommentReaction error: %v", err)
+	}
+
 	if !config.AllowDraftPRs && isDraft {
 		log.Printf("AllowDraftPRs is enabled, skipping PR: %v", issueNumber)
 		return nil

cli/pkg/azure/azure.go

@@ -339,6 +339,11 @@ func (a *AzureReposService) EditComment(prNumber int, id interface{}, comment st
 	return err
 }
 
+func (a *AzureReposService) CreateCommentReaction(id interface{}, reaction string) error {
+	// TODO implement me
+	return nil
+}
+
 func (a *AzureReposService) GetBranchName(prNumber int) (string, error) {
 	//TODO implement me
 	return "", nil

cli/pkg/bitbucket/bitbucket.go

@@ -157,6 +157,11 @@ func (b BitbucketAPI) EditComment(prNumber int, id interface{}, comment string)
 	return nil
 }
 
+func (a BitbucketAPI) CreateCommentReaction(id interface{}, reaction string) error {
+	// TODO implement me
+	return nil
+}
+
 type Comment struct {
 	Size     int    `json:"size"`
 	Page     int    `json:"page"`

cli/pkg/digger/digger_test.go

@@ -143,6 +143,11 @@ func (m *MockPRManager) EditComment(prNumber int, id interface{}, comment string
 	return nil
 }
 
+func (m *MockPRManager) CreateCommentReaction(id interface{}, reaction string) error {
+	m.Commands = append(m.Commands, RunInfo{"EditComment", strconv.Itoa(id.(int)) + " " + reaction, time.Now()})
+	return nil
+}
+
 func (m *MockPRManager) GetBranchName(prNumber int) (string, error) {
 	m.Commands = append(m.Commands, RunInfo{"GetBranchName", strconv.Itoa(prNumber), time.Now()})
 	return "", nil

cli/pkg/gitlab/gitlab.go

@@ -243,6 +243,11 @@ func (gitlabService GitLabService) EditComment(prNumber int, id interface{}, com
 	return nil
 }
 
+func (gitlabService GitLabService) CreateCommentReaction(id interface{}, reaction string) error {
+	// TODO implement me
+	return nil
+}
+
 func (gitlabService GitLabService) GetComments(prNumber int) ([]orchestrator.Comment, error) {
 	//TODO implement me
 	return nil, nil

cli/pkg/reporting/reporting_test.go

@@ -93,7 +93,12 @@ func (t MockCiService) EditComment(prNumber int, commentId interface{}, comment
 	return nil
 }
 
-func (t MockCiService) GetBranchName(prNumber int) (string, error) {
+func (svc MockCiService) CreateCommentReaction(id interface{}, reaction string) error {
+	// TODO implement me
+	return nil
+}
+
+func (svc MockCiService) GetBranchName(prNumber int) (string, error) {
 	return "", nil
 }
 

cli/pkg/utils/mocks.go

@@ -127,6 +127,10 @@ func (t MockPullRequestManager) EditComment(prNumber int, commentId interface{},
 	return nil
 }
 
+func (t *MockPullRequestManager) CreateCommentReaction(id interface{}, reaction string) error {
+	return nil
+}
+
 func (t MockPullRequestManager) GetBranchName(prNumber int) (string, error) {
 	return "", nil
 }

cli/pkg/utils/pullrequestmanager_mock.go

@@ -104,6 +104,11 @@ func (mockGithubPullrequestManager *MockGithubPullrequestManager) EditComment(pr
 	return nil
 }
 
+func (mockGithubPullrequestManager *MockGithubPullrequestManager) CreateCommentReaction(id interface{}, reaction string) error {
+	mockGithubPullrequestManager.commands = append(mockGithubPullrequestManager.commands, "CreateCommentReaction")
+	return nil
+}
+
 func (mockGithubPullrequestManager *MockGithubPullrequestManager) GetBranchName(prNumber int) (string, error) {
 	mockGithubPullrequestManager.commands = append(mockGithubPullrequestManager.commands, "GetBranchName")
 	return "", nil

libs/orchestrator/ci.go

@@ -6,6 +6,7 @@ type PullRequestService interface {
 	ListIssues() ([]*Issue, error)
 	PublishIssue(title string, body string) (int64, error)
 	EditComment(prNumber int, id interface{}, comment string) error
+	CreateCommentReaction(id interface{}, reaction string) error
 	GetComments(prNumber int) ([]Comment, error)
 	GetApprovals(prNumber int) ([]string, error)
 	// SetStatus set status of specified pull/merge request, status could be: "pending", "failure", "success"

libs/orchestrator/github/github.go

@@ -172,6 +172,26 @@ func (svc GithubService) EditComment(prNumber int, id interface{}, comment strin
 	return err
 }
 
+type GithubCommentReaction string
+
+const GithubCommentPlusOneReaction GithubCommentReaction = "+1"
+const GithubCommentMinusOneReaction GithubCommentReaction = "-1"
+const GithubCommentLaughReaction GithubCommentReaction = "laugh"
+const GithubCommentConfusedReaction GithubCommentReaction = "confused"
+const GithubCommentHeartReaction GithubCommentReaction = "heart"
+const GithubCommentHoorayReaction GithubCommentReaction = "hooray"
+const GithubCommentRocketReaction GithubCommentReaction = "rocket"
+const GithubCommentEyesReaction GithubCommentReaction = "eyes"
+
+func (svc GithubService) CreateCommentReaction(id interface{}, reaction string) error {
+	_, _, err := svc.Client.Reactions.CreateIssueCommentReaction(context.Background(), svc.Owner, svc.RepoName, id.(int64), reaction)
+	if err != nil {
+		log.Printf("could not addd reaction to comment: %v", err)
+		return fmt.Errorf("could not addd reaction to comment: %v", err)
+	}
+	return nil
+}
+
 func (svc GithubService) SetStatus(prNumber int, status string, statusContext string) error {
 	pr, _, err := svc.Client.PullRequests.Get(context.Background(), svc.Owner, svc.RepoName, prNumber)
 	if err != nil {

libs/orchestrator/github/mocks.go

@@ -93,6 +93,11 @@ func (t MockCiService) EditComment(prNumber int, commentId interface{}, comment
 	return nil
 }
 
+func (t MockCiService) CreateCommentReaction(id interface{}, reaction string) error {
+	// TODO implement me
+	return nil
+}
+
 func (t MockCiService) GetBranchName(prNumber int) (string, error) {
 	return "", nil
 }