some refactor

Author: mumoshu

cmd/main.go

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

cmd/pullvet/pullvet.go

@@ -3,6 +3,7 @@ package main
 import (
 	"flag"
 	"fmt"
+	"github.com/variantdev/go-actions"
 	"github.com/variantdev/go-actions/pkg/pullvet"
 	"os"
 )
@@ -21,7 +22,7 @@ func main() {
 
 	flag.Parse()
 
-	pr, err := pullvet.GetPullRequest()
+	pr, err := actions.PullRequest()
 	if err != nil {
 		fmt.Fprintf(os.Stderr, "%v\n", err)
 		os.Exit(1)

event.go

@@ -47,3 +47,23 @@ func CheckSuiteEvent() (*github.CheckSuiteEvent, error) {
 	}
 	return evt.(*github.CheckSuiteEvent), nil
 }
+
+func PullRequest() (*github.PullRequest, error) {
+	var pr *github.PullRequest
+	checkSuite, err := CheckSuiteEvent()
+	if err != nil {
+		checkRun, err := CheckRunEvent()
+		if err != nil {
+			pull, err := 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
+}

pkg/pullvet/pullvet.go

@@ -4,7 +4,6 @@ import (
 	"flag"
 	"fmt"
 	"github.com/google/go-github/github"
-	"github.com/variantdev/go-actions"
 	"github.com/variantdev/go-actions/pkg/cmd"
 	"os"
 	"regexp"
@@ -131,22 +130,3 @@ 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
-}