143 lines
4.1 KiB
Go
143 lines
4.1 KiB
Go
package middleware
|
|
|
|
import (
|
|
"blazing/common/rpc"
|
|
"blazing/cool"
|
|
"blazing/modules/base/config"
|
|
"blazing/modules/config/service"
|
|
"fmt"
|
|
"net/http"
|
|
"net/http/httputil"
|
|
"net/url"
|
|
|
|
"github.com/gogf/gf/v2/frame/g"
|
|
"github.com/gogf/gf/v2/net/ghttp"
|
|
"github.com/gogf/gf/v2/util/gconv"
|
|
"github.com/golang-jwt/jwt/v4"
|
|
"github.com/lxzan/gws"
|
|
)
|
|
|
|
const UpStream = "http://43.248.3.21:45632/"
|
|
|
|
func MiddlewareCORS(r *ghttp.Request) {
|
|
r.Response.CORSDefault()
|
|
corsOptions := r.Response.DefaultCORSOptions()
|
|
corsOptions.AllowDomain = []string{"*", "localhost", "tauri.localhost"}
|
|
if !r.Response.CORSAllowedOrigin(corsOptions) {
|
|
r.Response.WriteStatus(http.StatusForbidden)
|
|
return
|
|
}
|
|
r.Response.CORS(corsOptions)
|
|
if r.Response.Request.Method == "OPTIONS" {
|
|
r.Response.WriteStatus(http.StatusOK)
|
|
return
|
|
}
|
|
// fmt.Println(r.Response.Header())
|
|
//g.Dump(r.Response.Server.SetConfig(gtt))
|
|
//r.Response.Header().Del("Server") // 删除Server头
|
|
// r.Response.Header().Set("Server", "blazing")
|
|
r.Middleware.Next()
|
|
}
|
|
func StartServerProxy() {
|
|
s := g.Server()
|
|
// Parse the upstream URL
|
|
u, _ := url.Parse(UpStream)
|
|
// Create a new reverse proxy instance
|
|
proxy := httputil.NewSingleHostReverseProxy(u)
|
|
// Configure error handling for proxy failures
|
|
proxy.ErrorHandler = func(writer http.ResponseWriter, request *http.Request, e error) {
|
|
writer.WriteHeader(http.StatusBadGateway)
|
|
}
|
|
// Handle all requests with path prefix "/proxy/*"
|
|
s.BindHandler("/bbs/*url", func(r *ghttp.Request) {
|
|
var (
|
|
//originalPath = r.Request.URL.Path
|
|
proxyToPath = "/" + r.Get("url").String()
|
|
)
|
|
// Rewrite the request path
|
|
r.Request.URL.Path = proxyToPath
|
|
// Log the proxy operation
|
|
//g.Log().Infof(r.Context(), `proxy:"%s" -> backend:"%s"`, originalPath, proxyToPath)
|
|
// Ensure request body can be read multiple times if needed
|
|
r.MakeBodyRepeatableRead(false)
|
|
// Forward the request to the backend server
|
|
proxy.ServeHTTP(r.Response.Writer, r.Request)
|
|
})
|
|
|
|
}
|
|
|
|
func init() {
|
|
if config.Config.Middleware.Authority.Enable {
|
|
g.Server().BindMiddleware("/admin/*/open/*", BaseAuthorityMiddlewareOpen)
|
|
g.Server().BindMiddleware("/rpc/*", BaseAuthorityMiddlewareOpen)
|
|
g.Server().BindMiddleware("/admin/*/comm/*", BaseAuthorityMiddlewareComm)
|
|
g.Server().BindMiddleware("/admin/*", BaseAuthorityMiddleware)
|
|
// g.Server().BindMiddleware("/*", AutoI18n)
|
|
g.Server().BindMiddleware("/*", MiddlewareCORS)
|
|
|
|
}
|
|
if config.Config.Middleware.Log.Enable {
|
|
g.Server().BindMiddleware("/admin/*", BaseLog)
|
|
}
|
|
StartServerProxy()
|
|
tt := rpc.CServer()
|
|
|
|
g.Server().BindHandler("/rpc/*", func(r *ghttp.Request) {
|
|
|
|
tt.ServeHTTP(r.Response.Writer, r.Request)
|
|
|
|
})
|
|
|
|
g.Server().BindHandler("/server/*", func(r *ghttp.Request) {
|
|
servert := new(ServerHandler)
|
|
id := gconv.Uint16(r.URL.Query().Get("id"))
|
|
servert.isinstall = gconv.Uint32(r.URL.Query().Get("isinstall"))
|
|
servert.ServerList = service.NewServerService().StartUPdate(id, int(servert.isinstall))
|
|
|
|
upgrader := gws.NewUpgrader(servert, &gws.ServerOption{
|
|
|
|
Authorize: func(_ *http.Request, _ gws.SessionStorage) bool {
|
|
|
|
tokenString := r.URL.Query().Get("Authorization")
|
|
token, err := jwt.ParseWithClaims(tokenString, &cool.Claims{}, func(_ *jwt.Token) (interface{}, error) {
|
|
|
|
return []byte(config.Config.Jwt.Secret), nil
|
|
})
|
|
if err != nil {
|
|
return false
|
|
}
|
|
if !token.Valid {
|
|
return false
|
|
}
|
|
admin := token.Claims.(*cool.Claims)
|
|
if admin.UserId != 10001 {
|
|
return false
|
|
}
|
|
|
|
// var name = r.URL.Query().Get("name")
|
|
// if name == "" {
|
|
// return false
|
|
// }
|
|
// t, _ := service.NewBaseSysUserService().Person(admin.UserID)
|
|
|
|
//Loger.Debug(context.TODO(), t.Mimi)
|
|
// session.Store("name", t.Mimi)
|
|
//session.Store("key", r.Header.Get("Sec-WebSocket-Key"))
|
|
return true
|
|
},
|
|
})
|
|
|
|
socket, err := upgrader.Upgrade(r.Response.Writer, r.Request)
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
return
|
|
}
|
|
// ants.Submit(func() {
|
|
// socket.ReadLoop()
|
|
// })
|
|
// ants.Submit(func() { socket.ReadLoop() })
|
|
go socket.ReadLoop()
|
|
|
|
})
|
|
}
|