feat(pullvet): check_suite support

Author: mumoshu

cmd/main.go

@@ -3,7 +3,6 @@ package main
 import (
 	"flag"
 	"fmt"
-	"github.com/variantdev/go-actions"
 	"github.com/variantdev/go-actions/pkg/pullvet"
 	"os"
 )
@@ -45,7 +44,13 @@ func main() {
 
 		fs.Parse(os.Args[2:])
 
-		if err := cmd.Run(actions.PullRequestEvent().PullRequest); err != nil {
+		pr, err := pullvet.GetPullRequest()
+		if err != nil {
+			fmt.Fprintf(os.Stderr, "%v\n", err)
+			os.Exit(1)
+		}
+
+		if err := cmd.Run(pr); err != nil {
 			fatal("%v\n", err)
 		}
 	default:

cmd/pullvet/pullvet.go

@@ -3,8 +3,6 @@ package main
 import (
 	"flag"
 	"fmt"
-	"github.com/google/go-github/github"
-	"github.com/variantdev/go-actions"
 	"github.com/variantdev/go-actions/pkg/pullvet"
 	"os"
 )
@@ -23,17 +21,10 @@ func main() {
 
 	flag.Parse()
 
-	var pr *github.PullRequest
-
-	checkRun, err := actions.CheckRunEvent()
+	pr, err := pullvet.GetPullRequest()
 	if err != nil {
-		pull, err := actions.PullRequestEvent()
-		if err != nil {
-			fmt.Fprintf(os.Stderr, "%v\n", err)
-		}
-		pr = pull.PullRequest
-	} else {
-		pr = checkRun.CheckRun.PullRequests[0]
+		fmt.Fprintf(os.Stderr, "%v\n", err)
+		os.Exit(1)
 	}
 
 	if err := cmd.Run(pr); err != nil {

event.go

@@ -39,3 +39,11 @@ func CheckRunEvent() (*github.CheckRunEvent, error) {
 	}
 	return evt.(*github.CheckRunEvent), nil
 }
+
+func CheckSuiteEvent() (*github.CheckSuiteEvent, error) {
+	evt, err := github.ParseWebHook("check_suite", Event())
+	if err != nil {
+		return nil, err
+	}
+	return evt.(*github.CheckSuiteEvent), nil
+}

pkg/pullvet/pullvet.go

@@ -4,6 +4,7 @@ import (
 	"flag"
 	"fmt"
 	"github.com/google/go-github/github"
+	"github.com/variantdev/go-actions"
 	"github.com/variantdev/go-actions/pkg/cmd"
 	"os"
 	"regexp"
@@ -129,3 +130,23 @@ func formatFailures(failures []string) string {
 	}
 	return strings.Join(lines, "\n")
 }
+
+func GetPullRequest() (*github.PullRequest, error) {
+	var pr *github.PullRequest
+	checkSuite, err := actions.CheckSuiteEvent()
+	if err != nil {
+		checkRun, err := actions.CheckRunEvent()
+		if err != nil {
+			pull, err := actions.PullRequestEvent()
+			if err != nil {
+				return nil, err
+			}
+			pr = pull.PullRequest
+		} else {
+			pr = checkRun.CheckRun.PullRequests[0]
+		}
+	} else {
+		pr = checkSuite.CheckSuite.PullRequests[0]
+	}
+	return pr, nil
+}