From a986be8e0a874535c30e57d198c4b47ff4a11a3a Mon Sep 17 00:00:00 2001
From: reimei <34594195+Reimei1213@users.noreply.github.com>
Date: Thu, 20 Jul 2023 19:12:36 +0900
Subject: [PATCH] chore: set env & add message (#41)

---
 main.go | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

diff --git a/main.go b/main.go
index a11e118..88e69cd 100644
--- a/main.go
+++ b/main.go
@@ -4,6 +4,7 @@ import (
 	"bytes"
 	"context"
 	"fmt"
+	"os"
 	"os/exec"
 	"regexp"
 	"strconv"
@@ -40,6 +41,12 @@ func main() {
 		fmt.Printf("failed to load inputs: %s\n", err.Error())
 		panic(err.Error())
 	}
+	err = os.Setenv("GITHUB_TOKEN", e.GithubToken)
+	if err != nil {
+		fmt.Printf("failed to set env: %s\n", err.Error())
+		panic(err.Error())
+	}
+
 	ctx, f := context.WithTimeout(context.Background(), jobTimeout)
 	defer f()
 	client := newGHClient(e.GithubToken)
@@ -59,7 +66,12 @@ func main() {
 		fmt.Printf("failed to merge: %v", err)
 		panic(err.Error())
 	}
-	successMsg := "Merged PR #" + fmt.Sprintf("%d", e.PRNumber) + " successfully!"
+	var successMsg string
+	if e.EnableAutoMerge {
+		successMsg = "Enabled auto merge #" + fmt.Sprintf("%d", e.PRNumber) + " \nIf CI fails, fix problems and retry."
+	} else {
+		successMsg = "Merged PR #" + fmt.Sprintf("%d", e.PRNumber) + " successfully!"
+	}
 	if err := client.sendMsg(ctx, e.Owner, e.Repo, e.PRNumber, successMsg); err != nil {
 		fmt.Printf("failed to send message: %v", err)
 		panic(err.Error())