feat: add issue related funcs, and dummy testing func

instill-ai · Jul 1, 2024 · d1aa87e · d1aa87e
1 parent b953cb7
commit d1aa87e
Show file tree

Hide file tree

Showing 8 changed files with 600 additions and 8 deletions.
diff --git a/operator/github/v0/client.go b/operator/github/v0/client.go
@@ -20,6 +20,7 @@ type GitHubClient struct {
 	*github.Client
 	Repositories RepositoriesService
 	PullRequests PullRequestService
+	Issues IssuesService
 }
 
 func newClient(ctx context.Context, setup *structpb.Struct) Client {
@@ -33,7 +34,14 @@ func newClient(ctx context.Context, setup *structpb.Struct) Client {
 		oauth2Client = oauth2.NewClient(ctx, tokenSource)
 	}
 	client := github.NewClient(oauth2Client)
-	githubClient := Client{client: GitHubClient{Client: client, Repositories: client.Repositories, PullRequests: client.PullRequests} }
+	githubClient := Client{
+		client: GitHubClient{
+			Client: client,
+			Repositories: client.Repositories,
+			PullRequests: client.PullRequests,
+			Issues: client.Issues,
+			},
+		}
 	return githubClient
 }
 

diff --git a/operator/github/v0/commits.go b/operator/github/v0/commits.go
@@ -2,7 +2,6 @@ package github
 
 import (
 	"context"
-	"fmt"
 
 	"github.com/google/go-github/v62/github"
 	"github.com/instill-ai/component/base"
@@ -98,9 +97,6 @@ func  (githubClient *Client) getCommitTask(props *structpb.Struct) (*structpb.St
 	}
 	var resp GetCommitResp
 	resp.Commit = githubClient.extractCommitInformation(commit)
-	fmt.Println("=====================================")
-	fmt.Println("commit: ",resp.Commit)
-	fmt.Println("=====================================")
 	out, err := base.ConvertToStructpb(resp)
 	if err != nil {
 		return nil, err

diff --git a/operator/github/v0/component_test.go b/operator/github/v0/component_test.go
@@ -16,6 +16,7 @@ import (
 var MockGithubClient = GitHubClient{
 	Repositories: &MockRepositoriesService{},
 	PullRequests: &MockPullRequestService{},
+	Issues: &MockIssuesService{},
 }
 
 var fakeHost = "https://fake-github.com"

diff --git a/operator/github/v0/config/definition.json b/operator/github/v0/config/definition.json
@@ -4,7 +4,10 @@
     "TASK_GET_PULL_REQUEST",
     "TASK_GET_COMMIT",
     "TASK_GET_REVIEW_COMMENTS",
-    "TASK_CREATE_REVIEW_COMMENT"
+    "TASK_CREATE_REVIEW_COMMENT",
+    "TASK_GET_ALL_ISSUES",
+    "TASK_GET_ISSUE",
+    "TASK_CREATE_ISSUE"
   ],
   "custom": false,
   "documentationUrl": "https://www.instill.tech/docs/component/application/github",