starred/godoxy-yusing
1
0
Fork 0
mirror of https://github.com/yusing/godoxy.git synced 2026-04-19 23:11:25 +02:00
Code Issues 18 Packages Projects Releases 10 Wiki Activity

feat(handler): add SkipOriginCheckMiddleware and configuration option to bypass origin checks

Browse Source
This commit is contained in:
yusing
2025-08-17 20:46:26 +08:00
parent d3bfb2488b
commit dfc2d5e35c
2 changed files with 19 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
internal/api/handler.go
View File

@@ -4,6 +4,7 @@ import (
"net/http" "net/http"
"github.com/gin-gonic/gin" "github.com/gin-gonic/gin"
"github.com/gorilla/websocket"
"github.com/rs/zerolog/log" "github.com/rs/zerolog/log"
swaggerFiles "github.com/swaggo/files" swaggerFiles "github.com/swaggo/files"
ginSwagger "github.com/swaggo/gin-swagger" ginSwagger "github.com/swaggo/gin-swagger"
@@ -19,6 +20,7 @@ import (
metricsApi "github.com/yusing/go-proxy/internal/api/v1/metrics" metricsApi "github.com/yusing/go-proxy/internal/api/v1/metrics"
routeApi "github.com/yusing/go-proxy/internal/api/v1/route" routeApi "github.com/yusing/go-proxy/internal/api/v1/route"
"github.com/yusing/go-proxy/internal/auth" "github.com/yusing/go-proxy/internal/auth"
"github.com/yusing/go-proxy/internal/common"
) )
// @title GoDoxy API // @title GoDoxy API
@@ -66,6 +68,9 @@ func NewHandler() *gin.Engine {
if auth.IsEnabled() { if auth.IsEnabled() {
v1.Use(AuthMiddleware()) v1.Use(AuthMiddleware())
} }
if common.APISkipOriginCheck {
v1.Use(SkipOriginCheckMiddleware())
}
{ {
v1.GET("/favicon", apiV1.FavIcon) v1.GET("/favicon", apiV1.FavIcon)
v1.GET("/health", apiV1.Health) v1.GET("/health", apiV1.Health)
@@ -152,6 +157,18 @@ func AuthMiddleware() gin.HandlerFunc {
} }
} }
func SkipOriginCheckMiddleware() gin.HandlerFunc {
upgrader := &websocket.Upgrader{
CheckOrigin: func(r *http.Request) bool {
return true
},
}
return func(c *gin.Context) {
c.Set("upgrader", upgrader)
c.Next()
}
}
func ErrorHandler() gin.HandlerFunc { func ErrorHandler() gin.HandlerFunc {
return func(c *gin.Context) { return func(c *gin.Context) {
c.Next() c.Next()

2
internal/common/env.go
View File

@@ -42,6 +42,8 @@ var (
APIUser = GetEnvString("API_USER", "admin") APIUser = GetEnvString("API_USER", "admin")
APIPassword = GetEnvString("API_PASSWORD", "password") APIPassword = GetEnvString("API_PASSWORD", "password")
APISkipOriginCheck = GetEnvBool("API_SKIP_ORIGIN_CHECK", false) // skip this in UI Demo
DebugDisableAuth = GetEnvBool("DEBUG_DISABLE_AUTH", false) DebugDisableAuth = GetEnvBool("DEBUG_DISABLE_AUTH", false)
// OIDC Configuration. // OIDC Configuration.