refactor

openflare-server/internal/apps/openflare/agent/access_log_region.go

@@ -8,7 +8,7 @@ import (
 	"net"
 	"strings"
 
-	pkggeoip "github.com/rain-kl/openflare/pkg/geoip"
+	pkggeoip "github.com/Rain-kl/Wavelet/pkg/geoip"
 )
 
 var accessLogGeoProviderFactory = func() (pkggeoip.GeoIPService, error) {

openflare-server/internal/apps/openflare/agent/config.go

@@ -11,7 +11,7 @@ import (
 	"time"
 
 	"github.com/Rain-kl/Wavelet/internal/db"
-	openrestyrender "github.com/rain-kl/openflare/pkg/render/openresty"
+	openrestyrender "github.com/Rain-kl/Wavelet/pkg/render/openresty"
 	"gorm.io/gorm"
 )
 

openflare-server/internal/apps/openflare/agent/config_test.go

@@ -6,7 +6,7 @@ package agent
 import (
 	"testing"
 
-	openrestyrender "github.com/rain-kl/openflare/pkg/render/openresty"
+	openrestyrender "github.com/Rain-kl/Wavelet/pkg/render/openresty"
 )
 
 func TestIsRuntimeGeneratedSupportFile(t *testing.T) {

openflare-server/internal/apps/openflare/agent/helpers_test.go

@@ -8,7 +8,7 @@ import (
 	"testing"
 
 	"github.com/Rain-kl/Wavelet/internal/model"
-	pkggeoip "github.com/rain-kl/openflare/pkg/geoip"
+	pkggeoip "github.com/Rain-kl/Wavelet/pkg/geoip"
 )
 
 type fakeGeoIPProvider struct {

openflare-server/internal/apps/openflare/config_version/renderer.go

@@ -4,7 +4,7 @@
 package config_version
 
 import (
-	openrestyrender "github.com/rain-kl/openflare/pkg/render/openresty"
+	openrestyrender "github.com/Rain-kl/Wavelet/pkg/render/openresty"
 )
 
 // SupportFile is a rendered configuration support artifact.

openflare-server/internal/apps/openflare/config_version/snapshot.go

@@ -13,7 +13,7 @@ import (
 
 	"github.com/Rain-kl/Wavelet/internal/apps/openflare/waf"
 	"github.com/Rain-kl/Wavelet/internal/model"
-	openrestyrender "github.com/rain-kl/openflare/pkg/render/openresty"
+	openrestyrender "github.com/Rain-kl/Wavelet/pkg/render/openresty"
 )
 
 type snapshotRoute struct {

openflare-server/internal/apps/openflare/flared/middleware_test.go

@@ -9,6 +9,7 @@ import (
 	"net/http/httptest"
 	"testing"
 
+	"github.com/Rain-kl/Wavelet/internal/common/response"
 	"github.com/Rain-kl/Wavelet/internal/db"
 	"github.com/Rain-kl/Wavelet/internal/model"
 	"github.com/gin-gonic/gin"
@@ -53,6 +54,7 @@ func TestTunnelAuthMissingToken(t *testing.T) {
 
 	gin.SetMode(gin.TestMode)
 	engine := gin.New()
+	engine.Use(response.ErrorHandlerMiddleware())
 	engine.GET("/flared/test", TunnelAuth(), func(c *gin.Context) {
 		c.Status(http.StatusOK)
 	})
@@ -71,6 +73,7 @@ func TestTunnelAuthRejectsWrongNodeType(t *testing.T) {
 
 	gin.SetMode(gin.TestMode)
 	engine := gin.New()
+	engine.Use(response.ErrorHandlerMiddleware())
 	engine.GET("/flared/test", TunnelAuth(), func(c *gin.Context) {
 		c.Status(http.StatusOK)
 	})

openflare-server/internal/apps/openflare/geoip/lookup.go

@@ -9,7 +9,7 @@ import (
 	"net"
 	"strings"
 
-	pkggeoip "github.com/rain-kl/openflare/pkg/geoip"
+	pkggeoip "github.com/Rain-kl/Wavelet/pkg/geoip"
 )
 
 // LookupView is the legacy OpenFlare GeoIP lookup response shape.

openflare-server/internal/apps/openflare/geoip/lookup_test.go

@@ -7,7 +7,7 @@ import (
 	"net"
 	"testing"
 
-	pkggeoip "github.com/rain-kl/openflare/pkg/geoip"
+	pkggeoip "github.com/Rain-kl/Wavelet/pkg/geoip"
 )
 
 type fakeLookupProvider struct{}

openflare-server/internal/apps/openflare/relay/middleware_test.go

@@ -9,6 +9,7 @@ import (
 	"net/http/httptest"
 	"testing"
 
+	"github.com/Rain-kl/Wavelet/internal/common/response"
 	"github.com/Rain-kl/Wavelet/internal/db"
 	"github.com/Rain-kl/Wavelet/internal/model"
 	"github.com/gin-gonic/gin"
@@ -53,6 +54,7 @@ func TestRelayAuthMissingToken(t *testing.T) {
 
 	gin.SetMode(gin.TestMode)
 	engine := gin.New()
+	engine.Use(response.ErrorHandlerMiddleware())
 	engine.GET("/relay/test", RelayAuth(), func(c *gin.Context) {
 		c.Status(http.StatusOK)
 	})
@@ -71,6 +73,7 @@ func TestRelayAuthRejectsWrongNodeType(t *testing.T) {
 
 	gin.SetMode(gin.TestMode)
 	engine := gin.New()
+	engine.Use(response.ErrorHandlerMiddleware())
 	engine.GET("/relay/test", RelayAuth(), func(c *gin.Context) {
 		c.Status(http.StatusOK)
 	})