added API to add/remove issue label

eddycharly · Jul 8, 2019 · 8a85339 · 8a85339
1 parent a359145
commit 8a85339
Show file tree

Hide file tree

Showing 8 changed files with 61 additions and 0 deletions.
diff --git a/scm/const.go b/scm/const.go
@@ -144,6 +144,7 @@ const (
 	DriverBitbucket
 	DriverStash
 	DriverCoding
+	DriverFake
 )
 
 // String returns the string representation of Driver.
@@ -163,6 +164,8 @@ func (d Driver) String() (s string) {
 		return "stash"
 	case DriverCoding:
 		return "coding"
+	case DriverFake:
+		return "fake"
 	default:
 		return "unknown"
 	}

diff --git a/scm/driver/bitbucket/issue.go b/scm/driver/bitbucket/issue.go
@@ -14,6 +14,14 @@ type issueService struct {
 	client *wrapper
 }
 
+func (s *issueService) AddLabel(ctx context.Context, repo string, number int, label string) (*scm.Response, error) {
+	return nil, scm.ErrNotSupported
+}
+
+func (s *issueService) DeleteLabel(ctx context.Context, repo string, number int, label string) (*scm.Response, error) {
+	return nil, scm.ErrNotSupported
+}
+
 func (s *issueService) Find(ctx context.Context, repo string, number int) (*scm.Issue, *scm.Response, error) {
 	return nil, nil, scm.ErrNotSupported
 }

diff --git a/scm/driver/gitea/issue.go b/scm/driver/gitea/issue.go
@@ -16,6 +16,14 @@ type issueService struct {
 	client *wrapper
 }
 
+func (s *issueService) AddLabel(ctx context.Context, repo string, number int, label string) (*scm.Response, error) {
+	return nil, scm.ErrNotSupported
+}
+
+func (s *issueService) DeleteLabel(ctx context.Context, repo string, number int, label string) (*scm.Response, error) {
+	return nil, scm.ErrNotSupported
+}
+
 func (s *issueService) Find(ctx context.Context, repo string, number int) (*scm.Issue, *scm.Response, error) {
 	path := fmt.Sprintf("api/v1/repos/%s/issues/%d", repo, number)
 	out := new(issue)

diff --git a/scm/driver/github/issue.go b/scm/driver/github/issue.go
@@ -55,6 +55,18 @@ func (s *issueService) Create(ctx context.Context, repo string, input *scm.Issue
 	return convertIssue(out), res, err
 }
 
+func (s *issueService) AddLabel(ctx context.Context, repo string, number int, label string) (*scm.Response, error) {
+	path := fmt.Sprintf("repos/%s/issues/%d/labels", repo, number)
+	in := []string{label}
+	res, err := s.client.do(ctx, "POST", path, in, nil)
+	return res, err
+}
+
+func (s *issueService) DeleteLabel(ctx context.Context, repo string, number int, label string) (*scm.Response, error) {
+	path := fmt.Sprintf("repos/%s/issues/labels/%s", repo, label)
+	return s.client.do(ctx, "DELETE", path, nil, nil)
+}
+
 func (s *issueService) CreateComment(ctx context.Context, repo string, number int, input *scm.CommentInput) (*scm.Comment, *scm.Response, error) {
 	path := fmt.Sprintf("repos/%s/issues/%d/comments", repo, number)
 	in := &issueCommentInput{

diff --git a/scm/driver/gitlab/issue.go b/scm/driver/gitlab/issue.go
@@ -18,6 +18,14 @@ type issueService struct {
 	client *wrapper
 }
 
+func (s *issueService) AddLabel(ctx context.Context, repo string, number int, label string) (*scm.Response, error) {
+	return nil, scm.ErrNotSupported
+}
+
+func (s *issueService) DeleteLabel(ctx context.Context, repo string, number int, label string) (*scm.Response, error) {
+	return nil, scm.ErrNotSupported
+}
+
 func (s *issueService) Find(ctx context.Context, repo string, number int) (*scm.Issue, *scm.Response, error) {
 	path := fmt.Sprintf("api/v4/projects/%s/issues/%d", encode(repo), number)
 	out := new(issue)

diff --git a/scm/driver/gogs/issue.go b/scm/driver/gogs/issue.go
@@ -16,6 +16,14 @@ type issueService struct {
 	client *wrapper
 }
 
+func (s *issueService) AddLabel(ctx context.Context, repo string, number int, label string) (*scm.Response, error) {
+	return nil, scm.ErrNotSupported
+}
+
+func (s *issueService) DeleteLabel(ctx context.Context, repo string, number int, label string) (*scm.Response, error) {
+	return nil, scm.ErrNotSupported
+}
+
 func (s *issueService) Find(ctx context.Context, repo string, number int) (*scm.Issue, *scm.Response, error) {
 	path := fmt.Sprintf("api/v1/repos/%s/issues/%d", repo, number)
 	out := new(issue)

diff --git a/scm/driver/stash/issue.go b/scm/driver/stash/issue.go
@@ -14,6 +14,14 @@ type issueService struct {
 	client *wrapper
 }
 
+func (s *issueService) AddLabel(ctx context.Context, repo string, number int, label string) (*scm.Response, error) {
+	return nil, scm.ErrNotSupported
+}
+
+func (s *issueService) DeleteLabel(ctx context.Context, repo string, number int, label string) (*scm.Response, error) {
+	return nil, scm.ErrNotSupported
+}
+
 func (s *issueService) Find(ctx context.Context, repo string, number int) (*scm.Issue, *scm.Response, error) {
 	return nil, nil, scm.ErrNotSupported
 }

diff --git a/scm/issue.go b/scm/issue.go
@@ -90,5 +90,11 @@ type (
 
 		// Unlock unlocks an issue discussion.
 		Unlock(context.Context, string, int) (*Response, error)
+
+		// AddLabel adds a label to an issue
+		AddLabel(ctx context.Context, repo string, number int, label string) (*Response, error)
+
+		// DeleteLabel deletes a label from an issue
+		DeleteLabel(ctx context.Context, repo string, number int, label string) (*Response, error)
 	}
 )