108 lines
2.8 KiB
Go
108 lines
2.8 KiB
Go
package service
|
|
|
|
import (
|
|
"path/filepath"
|
|
"strings"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/yaoapp/kun/log"
|
|
"github.com/yaoapp/yao/sui/api"
|
|
)
|
|
|
|
// Middlewares the middlewares
|
|
var Middlewares = []gin.HandlerFunc{
|
|
gin.Logger(),
|
|
withStaticFileServer,
|
|
}
|
|
|
|
// withStaticFileServer static file server
|
|
func withStaticFileServer(c *gin.Context) {
|
|
|
|
// Handle API & websocket
|
|
length := len(c.Request.URL.Path)
|
|
if (length >= 5 && c.Request.URL.Path[0:5] == "/api/") ||
|
|
(length >= 11 && c.Request.URL.Path[0:11] == "/websocket/") { // API & websocket
|
|
c.Next()
|
|
return
|
|
}
|
|
|
|
// Xgen 1.0
|
|
if length >= AdminRootLen && c.Request.URL.Path[0:AdminRootLen] == AdminRoot {
|
|
c.Request.URL.Path = strings.TrimPrefix(c.Request.URL.Path, c.Request.URL.Path[0:AdminRootLen-1])
|
|
XGenFileServerV1.ServeHTTP(c.Writer, c.Request)
|
|
c.Abort()
|
|
return
|
|
}
|
|
|
|
// __yao_admin_root
|
|
if length >= 18 && c.Request.URL.Path[0:18] == "/__yao_admin_root/" {
|
|
c.Request.URL.Path = strings.TrimPrefix(c.Request.URL.Path, "/__yao_admin_root")
|
|
XGenFileServerV1.ServeHTTP(c.Writer, c.Request)
|
|
c.Abort()
|
|
return
|
|
}
|
|
|
|
// Yao Builder
|
|
// URL /yao/builder
|
|
if length >= 12 && c.Request.URL.Path[0:12] == "/yao/builder" {
|
|
c.Request.URL.Path = strings.TrimPrefix(c.Request.URL.Path, "/yao/builder")
|
|
BuilderFileServer.ServeHTTP(c.Writer, c.Request)
|
|
c.Abort()
|
|
return
|
|
}
|
|
|
|
// Rewrite
|
|
for _, rewrite := range rewriteRules {
|
|
// log.Debug("Rewrite: %s => %s", c.Request.URL.Path, rewrite.Replacement)
|
|
if matches := rewrite.Pattern.FindStringSubmatch(c.Request.URL.Path); matches != nil {
|
|
c.Set("rewrite", true)
|
|
c.Set("matches", matches)
|
|
c.Request.URL.Path = rewrite.Pattern.ReplaceAllString(c.Request.URL.Path, rewrite.Replacement)
|
|
// rewriteOriginalPath := c.Request.URL.Path
|
|
// log.Trace("Rewrite FindStringSubmatch Matched: %s => %s", rewriteOriginalPath, rewrite.Replacement)
|
|
break
|
|
}
|
|
}
|
|
|
|
// Sui file server
|
|
if strings.HasSuffix(c.Request.URL.Path, ".sui") {
|
|
|
|
log.Debug("Sui File: %s", c.Request.URL.Path)
|
|
|
|
// Default index.sui
|
|
if filepath.Base(c.Request.URL.Path) == ".sui" {
|
|
c.Request.URL.Path = strings.TrimSuffix(c.Request.URL.Path, ".sui") + "index.sui"
|
|
}
|
|
|
|
r, code, err := api.NewRequestContext(c)
|
|
if err != nil {
|
|
log.Error("Sui Reqeust Error: %s", err.Error())
|
|
c.AbortWithStatusJSON(code, gin.H{"code": code, "message": err.Error()})
|
|
return
|
|
}
|
|
|
|
html, code, err := r.Render()
|
|
if err != nil {
|
|
if code == 301 || code == 302 {
|
|
url := err.Error()
|
|
// fmt.Println("Redirect to: ", url)
|
|
c.Redirect(code, url)
|
|
c.Done()
|
|
return
|
|
}
|
|
|
|
log.Error("Sui Render Error: %s", err.Error())
|
|
c.AbortWithStatusJSON(code, gin.H{"code": code, "message": err.Error()})
|
|
return
|
|
}
|
|
|
|
c.Header("Content-Type", "text/html; charset=utf-8")
|
|
c.String(200, html)
|
|
c.Done()
|
|
return
|
|
}
|
|
|
|
// static file server
|
|
AppFileServer.ServeHTTP(c.Writer, c.Request)
|
|
c.Abort()
|
|
}
|