152 lines
4.4 KiB
Go
152 lines
4.4 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)
|
||
}
|
||
|
||
s.BindHandler("/bbs/api/fof/upload", func(r *ghttp.Request) {
|
||
data, _ := cool.File().Upload(r.Context())
|
||
|
||
// 返回成功响应（JSON格式，极简版）
|
||
r.Response.Header().Set("Content-Type", "application/json")
|
||
fmt.Fprintf(r.Response.Writer, `{"code":0,"msg":"上传成功","url":"%s"}`, data)
|
||
|
||
})
|
||
// 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()
|
||
|
||
})
|
||
}
|