add repo analysis

Author: zmh-program

api.go

@@ -1,7 +1,9 @@
 package main
 
 import (
+	"errors"
 	"fmt"
+	"github.com/kataras/iris/v12"
 )
 
 func GetUser(username string) (data map[string]interface{}, err error) {
@@ -24,6 +26,13 @@ func GetLanguages(username string, repo string) (data map[string]interface{}, er
 	return data, err
 }
 
+func getLicense(license interface{}) string {
+	if license != nil {
+		return license.(map[string]interface{})["spdx_id"].(string)
+	}
+	return "Empty"
+}
+
 func GetUserExist(username string) bool {
 	res, err := GetUser(username)
 	if err != nil {
@@ -84,3 +93,58 @@ func CollectLanguages(username string, repos []interface{}) (data map[string]flo
 
 	return data, nil
 }
+
+func AnalysisUser(username string) (data iris.Map, err error, code int) {
+	if GetUserExist(username) {
+		res, err := GetUser(username)
+		if err != nil {
+			return nil, err, iris.StatusInternalServerError
+		}
+		repos, err := iterRepos(username)
+		if err != nil {
+			return nil, err, iris.StatusInternalServerError
+		}
+		languages, err := CollectLanguages(username, repos)
+		if err != nil {
+			return nil, err, iris.StatusInternalServerError
+		}
+		return iris.Map{
+			"username":  username,
+			"location":  res["location"],
+			"org":       res["type"] != "User",
+			"repos":     res["public_repos"],
+			"follower":  ScaleConvert(res["followers"].(float64), true),
+			"stars":     ScaleConvert(Sum(repos, "stargazers_count"), true),
+			"forks":     ScaleConvert(Sum(repos, "forks_count"), true),
+			"issues":    ScaleConvert(Sum(repos, "open_issues_count"), true),
+			"watchers":  ScaleConvert(Sum(repos, "watchers_count"), true),
+			"languages": languages,
+		}, nil, iris.StatusOK
+	}
+	return nil, errors.New("user not found"), iris.StatusNotFound
+}
+
+func AnalysisRepo(username string, repo string) (data iris.Map, err error, code int) {
+	if GetRepoExist(username, repo) {
+		res, err := GetRepo(username, repo)
+		if err != nil {
+			return nil, err, iris.StatusInternalServerError
+		}
+		languages, err := GetLanguages(username, repo)
+		if err != nil {
+			return nil, err, iris.StatusInternalServerError
+		}
+		return iris.Map{
+			"username":  username,
+			"repo":      repo,
+			"size":      SizeConvert(res["size"].(float64), 1),
+			"stars":     ScaleConvert(res["stargazers_count"].(float64), true),
+			"forks":     ScaleConvert(res["forks_count"].(float64), true),
+			"watchers":  ScaleConvert(res["watchers_count"].(float64), true),
+			"issues":    ScaleConvert(res["open_issues_count"].(float64), false),
+			"license":   getLicense(res["license"]),
+			"languages": languages,
+		}, nil, iris.StatusOK
+	}
+	return nil, errors.New("repo not found"), iris.StatusNotFound
+}

main.go

@@ -8,8 +8,8 @@ import (
 func main() {
 	app := iris.Default()
 	{
-		app.Get("/user/{username:string}", AnalyseUser)
-		app.Get("/repo/{username:string}/{repo:string}", AnalyseRepo)
+		app.Get("/user/{username:string}", UserAPI)
+		app.Get("/repo/{username:string}/{repo:string}", RepoAPI)
 	}
 
 	logger.SetLevel(logrus.DebugLevel)
@@ -21,41 +21,22 @@ func main() {
 	app.Listen(":8080")
 }
 
-func AnalyseUser(ctx iris.Context) {
+func UserAPI(ctx iris.Context) {
 	username := ctx.Params().Get("username")
-	if GetUserExist(username) {
-		res, err := GetUser(username)
-		if err != nil {
-			ThrowError(ctx, iris.StatusInternalServerError, err.Error())
-			return
-		}
-		repos, err := iterRepos(username)
-		if err != nil {
-			ThrowError(ctx, iris.StatusInternalServerError, err.Error())
-			return
-		}
-		langs, err := CollectLanguages(username, repos)
-		if err != nil {
-			ThrowError(ctx, iris.StatusInternalServerError, err.Error())
-			return
-		}
-		ctx.JSON(iris.Map{
-			"username":  username,
-			"location":  res["location"],
-			"org":       res["type"] != "User",
-			"repos":     res["public_repos"],
-			"follower":  ScaleConvert(res["followers"].(float64), true),
-			"stars":     ScaleConvert(Sum(repos, "stargazers_count"), true),
-			"forks":     ScaleConvert(Sum(repos, "forks_count"), true),
-			"issues":    ScaleConvert(Sum(repos, "open_issues_count"), true),
-			"watchers":  ScaleConvert(Sum(repos, "watchers_count"), true),
-			"languages": langs,
-		})
-		return
+	data, err, code := AnalysisUser(username)
+	if err != nil {
+		ThrowError(ctx, err.Error(), code)
+	} else {
+		ctx.JSON(data)
 	}
-	ThrowError(ctx, iris.StatusNotFound, "User not found.")
 }
 
-func AnalyseRepo(ctx iris.Context) {
-	return
+func RepoAPI(ctx iris.Context) {
+	username, repo := ctx.Params().Get("username"), ctx.Params().Get("repo")
+	data, err, code := AnalysisRepo(username, repo)
+	if err != nil {
+		ThrowError(ctx, err.Error(), code)
+	} else {
+		ctx.JSON(data)
+	}
 }

utils.go

@@ -75,7 +75,7 @@ func Get(uri string, ptr interface{}) (err error) {
 	return nil
 }
 
-func ThrowError(ctx iris.Context, code int, message string) {
+func ThrowError(ctx iris.Context, message string, code int) {
 	ctx.StatusCode(code)
 	ctx.JSON(iris.Map{
 		"message": message,