update backend structure

Author: zmh-program

.github/workflows/cd.yml .github/workflows/deploy.yml.github/workflows/cd.yml renamed to .github/workflows/deploy.yml

@@ -1,8 +1,7 @@
 name: Deployment
 on:
-  push:
-    branches:
-      - main
+  release:
+    types: [published]
 
 jobs:
   release:
@@ -43,14 +42,16 @@ jobs:
           mkdir deploy/app
           mv app deploy/app/
 
-      - name: Commit Migration
+      - name: Pack Application
         run: |
-          git config --global user.name "GitHub Actions"
-          git config --global user.email "actions@github.com"
-          cd deploy
-          git checkout -b deploy
-          git add .
-          git commit -m "deploy"
-          git push --force --quiet "https://${{ secrets.GITHUB_TOKEN }}@github.com/zmh-program/code-statistic.git"
-    env:
-      GITHUB_TOKEN: ${{ secrets.TOKEN }}
+          zip -r deployment.zip ./deploy/*
+
+      - name: Upload Package
+        uses: actions/upload-release-asset@v1.0.2
+        with:
+          upload_url: ${{ github.event.release.upload_url }}
+          asset_path: ./deployment.zip
+          asset_name: deployment.zip
+          asset_content_type: application/zip
+        env:
+          GITHUB_TOKEN: ${{ secrets.TOKEN }}

conf.go

@@ -1,9 +1,68 @@
 package main
 
+import (
+	"fmt"
+	"github.com/kataras/iris/v12"
+	"github.com/sirupsen/logrus"
+	"github.com/spf13/viper"
+	"os"
+	"strings"
+	"time"
+)
+
+var logger = logrus.New()
+
+type Formatter struct {
+}
+
+func (f *Formatter) Format(entry *logrus.Entry) ([]byte, error) {
+	timestamp := time.Now().Format("2006/01/02 15:04")
+	level := strings.ToUpper(entry.Level.String())
+	message := entry.Message
+
+	var color string
+	switch entry.Level {
+	case logrus.DebugLevel:
+		color = "\u001B[33m"
+	case logrus.InfoLevel:
+		color = "\033[32m"
+	case logrus.WarnLevel:
+		color = "\033[33m"
+	case logrus.ErrorLevel, logrus.FatalLevel, logrus.PanicLevel:
+		color = "\033[31m"
+	default:
+		color = "\033[0m"
+	}
+
+	return []byte(fmt.Sprintf("%s[%s]\u001B[0m %s - %s\n", color, level, timestamp, message)), nil
+}
+
 func main() {
-	SetupLogger()
-	ReadConfig()
-	ReadToken()
+	logger.SetLevel(logrus.DebugLevel)
+	logger.SetFormatter(&Formatter{})
+
+	_, err := os.Stat("config.yaml")
+	if err != nil && os.IsNotExist(err) {
+		logger.Fatal("Config file (config.yaml) not found!")
+	}
+
+	viper.SetConfigFile("config.yaml")
+	if err := viper.ReadInConfig(); err != nil {
+		logger.Fatalf("Failed to read config file: %v", err)
+	}
+	if err := viper.Unmarshal(&conf); err != nil {
+		logger.Fatalf("Failed to unmarshal config file: %v", err)
+	}
+
+	tokenList = GetTokenFromEnv()
+	ValidateToken()
 	SetupCache()
-	RunServer()
+
+	app := iris.Default()
+	{
+		app.Get("/api/user/{username:string}", CachedHandler(UserAPI))
+		app.Get("/api/repo/{username:string}/{repo:string}", CachedHandler(RepoAPI))
+		app.Get("/api/contributor/{username:string}/{repo:string}", CachedHandler(ContributorAPI))
+	}
+	app.Listen(fmt.Sprintf(":%d", conf.Server.Port))
 }

logger.go

@@ -1,20 +1,9 @@
 package main
 
 import (
-	"fmt"
 	"github.com/kataras/iris/v12"
 )
 
-func RunServer() {
-	app := iris.Default()
-	{
-		app.Get("/api/user/{username:string}", CachedHandler(UserAPI))
-		app.Get("/api/repo/{username:string}/{repo:string}", CachedHandler(RepoAPI))
-		app.Get("/api/contributor/{username:string}/{repo:string}", CachedHandler(ContributorAPI))
-	}
-	app.Listen(fmt.Sprintf(":%d", conf.Server.Port))
-}
-
 func UserAPI(ctx iris.Context) {
 	username := ctx.Params().Get("username")
 	data := AnalysisUser(username)