add api

Author: zmh-program

api.go

@@ -1,38 +1,19 @@
 package main
 
-import (
-	"encoding/json"
-	"github.com/sirupsen/logrus"
-	"io"
-	"net/http"
-)
+import "fmt"
 
-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 "+GetToken())
+func GetUser(username string) (res map[string]interface{}, err error) {
+	return Get(fmt.Sprintf("users/%s", username))
+}
 
-	client := &http.Client{}
-	resp, err := client.Do(req)
-	if err != nil {
-		return nil, err
-	}
+func GetRepos(username string) (res map[string]interface{}, err error) {
+	return Get(fmt.Sprintf("users/%s/repos", username))
+}
 
-	defer func(Body io.ReadCloser) {
-		err := Body.Close()
-		if err != nil {
-			logrus.Infoln()
-		}
-	}(resp.Body)
+func GetRepo(username string, repo string) (res map[string]interface{}, err error) {
+	return Get(fmt.Sprintf("repos/%s/%s", username, repo))
+}
 
-	var data map[string]interface{}
-	err = json.NewDecoder(resp.Body).Decode(&data)
-	if err != nil {
-		return nil, err
-	}
-	return data, nil
+func GetLanguages(username string, repo string) (res map[string]interface{}, err error) {
+	return Get(fmt.Sprintf("repos/%s/%s/languages", username, repo))
 }

main.go

@@ -1,6 +1,7 @@
 package main
 
 import (
+	"fmt"
 	"github.com/sirupsen/logrus"
 )
 
@@ -10,4 +11,5 @@ func main() {
 
 	tokenList = GetTokenFromEnv()
 	ValidateToken()
+	fmt.Println(GetRepos("zmh-program"))
 }

utils.go

@@ -1,6 +1,12 @@
 package main
 
-import "fmt"
+import (
+	"encoding/json"
+	"fmt"
+	"github.com/sirupsen/logrus"
+	"io"
+	"net/http"
+)
 
 func Sum(array []int) int {
 	total := 0
@@ -40,3 +46,33 @@ func ScaleConvert(n float64, useSmallScale bool) string {
 	}
 	return fmt.Sprintf("%.1f%s", n, scaleUnits[idx])
 }
+
+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 "+GetToken())
+
+	client := &http.Client{}
+	resp, err := client.Do(req)
+	if err != nil {
+		return nil, err
+	}
+
+	defer func(Body io.ReadCloser) {
+		err := Body.Close()
+		if err != nil {
+			logrus.Infoln()
+		}
+	}(resp.Body)
+
+	var data map[string]interface{}
+	err = json.NewDecoder(resp.Body).Decode(&data)
+	if err != nil {
+		return nil, err
+	}
+	return data, nil
+}