Fix bug in issue and pr number (#23)

micnncim · Oct 16, 2019 · 0fc44c9 · 0fc44c9
1 parent a82fea0
commit 0fc44c9
Showing 1 changed file with 15 additions and 3 deletions.
diff --git a/cmd/action-lgtm-reaction/main.go b/cmd/action-lgtm-reaction/main.go
@@ -101,9 +101,21 @@ func main() {
 	}
 
 	if needCreateComment {
-		number, err := strconv.Atoi(githubIssueNumber)
-		if err != nil {
-			exit("unable to convert string to int in issue number: %v\n", err)
+		if githubIssueNumber == "" && githubPullRequestNumber == "" {
+			exit("no issue number and pull request number\n")
+		}
+		var number int
+		var err error
+		if githubIssueNumber != "" {
+			number, err = strconv.Atoi(githubIssueNumber)
+			if err != nil {
+				exit("unable to convert string to int in issue number: %v\n", err)
+			}
+		} else if githubPullRequestNumber != "" {
+			number, err = strconv.Atoi(githubPullRequestNumber)
+			if err != nil {
+				exit("unable to convert string to int in pull request number: %v\n", err)
+			}
 		}
 		if err := githubClient.CreateIssueComment(ctx, owner, repo, number, comment); err != nil {
 			exit("unable to create issue comment: %v\n", err)