add conf

Author: zmh-program

api.go

@@ -5,23 +5,16 @@ import (
 	"github.com/sirupsen/logrus"
 	"io"
 	"net/http"
-	"os"
-	"strings"
 )
 
-func GetToken() []string {
-	data := os.Getenv("TOKEN")
-	return strings.Split(data, "|")
-}
-
-func Get(uri string, token string) (res map[string]interface{}, err error) {
+func Get(uri string) (res map[string]interface{}, err error) {
 	req, err := http.NewRequest("GET", "https://api.github.com/"+uri, nil)
 	if err != nil {
 		return nil, err
 	}
 
 	req.Header.Set("Accept", "application/json")
-	req.Header.Set("Authorization", "Bearer "+token)
+	req.Header.Set("Authorization", "Bearer "+GetToken())
 
 	client := &http.Client{}
 	resp, err := client.Do(req)

conf.go

@@ -0,0 +1,37 @@
+package main
+
+import (
+	"fmt"
+	"log"
+	"math/rand"
+	"os"
+	"strings"
+	"time"
+)
+
+var tokenList []string
+
+func GetToken() string {
+	source := rand.NewSource(time.Now().UnixNano())
+	idx := rand.New(source).Intn(len(tokenList))
+	return tokenList[idx]
+}
+
+func GetTokenFromEnv() []string {
+	data := strings.TrimSpace(os.Getenv("TOKEN"))
+	tokenList = strings.Split(data, "|")
+	result := make([]string, 0)
+	for _, token := range tokenList {
+		if strings.HasPrefix(token, "ghp_") {
+			result = append(result, token)
+		}
+	}
+	return result
+}
+
+func ValidateToken() {
+	if len(tokenList) == 0 {
+		log.Fatal("No token found!")
+	}
+	logger.Debug(fmt.Sprintf("Found %d available token(s).", len(tokenList)))
+}

logger.go

@@ -7,6 +7,8 @@ import (
 	"time"
 )
 
+var logger = logrus.New()
+
 type Formatter struct {
 }
 

main.go

@@ -1,15 +1,14 @@
 package main
 
 import (
-	"fmt"
 	"github.com/sirupsen/logrus"
 )
 
 func main() {
-	logger := logrus.New()
 	logger.SetLevel(logrus.DebugLevel)
 	logger.SetFormatter(&Formatter{})
 
-	tokens := GetToken()
-	fmt.Println(tokens)
+	tokenList = GetTokenFromEnv()
+	ValidateToken()
+
 }