use token if provided

rajatjindal · Feb 3, 2020 · 4732921 · 4732921
1 parent a82ada7
commit 4732921
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/go.sum b/go.sum
@@ -44,6 +44,7 @@ github.com/golang/protobuf v1.3.2 h1:6nsPYzhq5kReh6QImI3k5qWzO4PEbvbIW2cwSfR/6xs
 github.com/golang/protobuf v1.3.2/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U=
 github.com/google/go-cmp v0.3.0 h1:crn/baboCvb5fXaQ0IJ1SGTsTVrWpDsCWC8EGETZijY=
 github.com/google/go-cmp v0.3.0/go.mod h1:8QqcDgzrUqlUb/G2PQTWiueGozuR1884gddMywk6iLU=
+github.com/google/go-github v17.0.0+incompatible h1:N0LgJ1j65A7kfXrZnUDaYCs/Sf4rEjNlfyDHW9dolSY=
 github.com/google/go-github/v29 v29.0.2 h1:opYN6Wc7DOz7Ku3Oh4l7prmkOMwEcQxpFtxdU8N8Pts=
 github.com/google/go-github/v29 v29.0.2/go.mod h1:CHKiKKPHJ0REzfwc14QMklvtHwCveD0PxlMjLlzAM5E=
 github.com/google/go-querystring v1.0.0 h1:Xkwi/a1rcvNg1PPYe5vI8GbeBY/jrVuDX5ASuANWTrk=

diff --git a/pkg/source/actions/action_runner.go b/pkg/source/actions/action_runner.go
@@ -14,11 +14,21 @@ import (
 	"github.com/rajatjindal/krew-release-bot/pkg/cicd"
 	"github.com/rajatjindal/krew-release-bot/pkg/source"
 	"github.com/sirupsen/logrus"
+	"golang.org/x/oauth2"
 )
 
+func getHTTPClient() *http.Client {
+	if os.Getenv("GITHUB_TOKEN") != "" {
+		ts := oauth2.StaticTokenSource(&oauth2.Token{AccessToken: os.Getenv("GITHUB_TOKEN")})
+		return oauth2.NewClient(context.TODO(), ts)
+	}
+
+	return nil
+}
+
 //RunAction runs the github action
 func RunAction() error {
-	client := github.NewClient(nil)
+	client := github.NewClient(getHTTPClient())
 	provider := cicd.GetProvider()
 
 	if provider == nil {