Initial commit

2024-03-27 23:25:08 +08:00
commit 0884384e91
127 changed files with 9353 additions and 0 deletions
@@ -0,0 +1,151 @@
+package web_gin
+
+import (
+	"errors"
+	"fmt"
+	"github.com/mattn/go-colorable"
+	"net/http"
+	"path/filepath"
+	"regexp"
+	"strings"
+	"time"
+
+	"github.com/gin-gonic/gin"
+	"pet-house.com/core/helper/arr"
+	"pet-house.com/core/helper/dir"
+	"pet-house.com/core/helper/str"
+	"pet-house.com/core/server/web"
+	"pet-house.com/core/server/web/web_gin/middleware"
+)
+
+var ErrAuthDriverEmpty = errors.New("auth driver initialize fail")
+
+// WebServer
+// - app gin.Engine
+// - idleConnsClosed
+// - addr
+// - timeFormat
+// - staticPrefix
+type WebServer struct {
+	app *gin.Engine
+	server
+	addr       string
+	timeFormat string
+	webStatics []WebStatic
+}
+
+type WebStatic struct {
+	Prefix    string
+	IndexFile []byte
+}
+
+// Init
+func Init() *WebServer {
+	gin.SetMode(web.CONFIG.System.Level)
+	app := gin.Default()
+	if web.CONFIG.System.Tls {
+		app.Use(middleware.LoadTls())
+	}
+	app.Use(middleware.Cors())
+	registerValidation()
+
+	gin.DefaultWriter = colorable.NewColorableStdout()
+
+	web.SetDefaultAddrAndTimeFormat()
+
+	return &WebServer{
+		app:        app,
+		addr:       web.CONFIG.System.Addr,
+		timeFormat: web.CONFIG.System.TimeFormat,
+	}
+}
+
+// NoRoute for 404 http status
+func (ws *WebServer) NoRoute() {
+	if len(ws.webStatics) == 0 {
+		return
+	}
+
+	ws.app.NoRoute(func(ctx *gin.Context) {
+		// excepte for /v0 /v1 and so on
+		reg := `^/v[0-9]+$|^(/v[0-9]+)/`
+		ok, _ := regexp.MatchString(reg, ctx.Request.RequestURI)
+		if ok {
+			ctx.Writer.WriteHeader(http.StatusNotFound)
+			ctx.Writer.Flush()
+			return
+		}
+
+		var indexFile []byte
+		for _, wp := range ws.webStatics {
+			// match /admin or /admin/***
+			reg := str.Join("^", wp.Prefix, "$|^(", wp.Prefix, ")/")
+			ok, err := regexp.MatchString(reg, ctx.Request.RequestURI)
+			if err != nil || !ok {
+				continue
+			}
+			indexFile = wp.IndexFile
+		}
+
+		ctx.Writer.WriteHeader(http.StatusOK)
+		ctx.Writer.Write(indexFile)
+
+		ctx.Writer.Header().Add("Accept", "text/html")
+		ctx.Writer.Flush()
+	})
+}
+
+// GetEngine return *gin.Engine
+func (ws *WebServer) GetEngine() *gin.Engine {
+	return ws.app
+}
+
+// AddWebStatic
+func (ws *WebServer) AddWebStatic(staticAbsPath, webPrefix string, paths ...string) {
+	webPrefixs := strings.Split(web.CONFIG.System.WebPrefix, ",")
+	wp := arr.NewCheckArrayType(2)
+	for _, webPrefix := range webPrefixs {
+		wp.Add(webPrefix)
+	}
+	if wp.Check(webPrefix) {
+		return
+	}
+
+	favicon := filepath.Join(staticAbsPath, "favicon.ico")
+	index := filepath.Join(staticAbsPath, "index.html")
+
+	ws.app.Static(str.Join(webPrefix, "/favicon.ico"), favicon)
+	ws.app.StaticFile(webPrefix, index)
+
+	if len(paths) > 0 {
+		for _, path := range paths {
+			static := filepath.Join(staticAbsPath, path)
+			ws.app.Static(path, static)
+		}
+	}
+
+	web.CONFIG.System.WebPrefix = str.Join(web.CONFIG.System.WebPrefix, ",", webPrefix)
+	file, _ := dir.ReadBytes(index)
+	webStatic := WebStatic{
+		Prefix:    webPrefix,
+		IndexFile: file,
+	}
+	ws.webStatics = append(ws.webStatics, webStatic)
+
+}
+
+// AddUploadStatic
+func (ws *WebServer) AddUploadStatic(webPrefix, staticAbsPath string) {
+	ws.app.StaticFS(webPrefix, http.Dir(staticAbsPath))
+	web.CONFIG.System.StaticPrefix = webPrefix
+}
+
+// Run
+func (ws *WebServer) Run() {
+	ws.NoRoute()
+	s := initServer(web.CONFIG.System.Addr, ws.app)
+	time.Sleep(10 * time.Microsecond)
+	fmt.Printf("默认监听地址: http://%s\n", web.CONFIG.System.Addr)
+	s.ListenAndServe()
+
+}