✨ Uses v2 configration schema
This commit is contained in:
@ -1,9 +1,10 @@
|
||||
package hypertext
|
||||
|
||||
import (
|
||||
"math/rand"
|
||||
"regexp"
|
||||
|
||||
"code.smartsheep.studio/goatworks/roadnavi/pkg/navi"
|
||||
"code.smartsheep.studio/goatworks/roadsign/pkg/navi"
|
||||
"github.com/gofiber/fiber/v2"
|
||||
"github.com/samber/lo"
|
||||
)
|
||||
@ -16,16 +17,16 @@ func UseProxies(app *fiber.App) {
|
||||
headers := ctx.GetReqHeaders()
|
||||
|
||||
// Filtering sites
|
||||
for _, site := range navi.App.Sites {
|
||||
for _, region := range navi.R.Regions {
|
||||
// Matching rules
|
||||
for _, rule := range site.Rules {
|
||||
if !lo.Contains(rule.Host, host) {
|
||||
for _, location := range region.Locations {
|
||||
if !lo.Contains(location.Host, host) {
|
||||
continue
|
||||
}
|
||||
|
||||
if !func() bool {
|
||||
flag := false
|
||||
for _, pattern := range rule.Path {
|
||||
for _, pattern := range location.Path {
|
||||
if ok, _ := regexp.MatchString(pattern, path); ok {
|
||||
flag = true
|
||||
break
|
||||
@ -38,7 +39,7 @@ func UseProxies(app *fiber.App) {
|
||||
|
||||
// Filter query strings
|
||||
flag := true
|
||||
for rk, rv := range rule.Queries {
|
||||
for rk, rv := range location.Queries {
|
||||
for ik, iv := range queries {
|
||||
if rk != ik && rv != iv {
|
||||
flag = false
|
||||
@ -54,7 +55,7 @@ func UseProxies(app *fiber.App) {
|
||||
}
|
||||
|
||||
// Filter headers
|
||||
for rk, rv := range rule.Headers {
|
||||
for rk, rv := range location.Headers {
|
||||
for ik, iv := range headers {
|
||||
if rk == ik {
|
||||
for _, ov := range iv {
|
||||
@ -76,9 +77,12 @@ func UseProxies(app *fiber.App) {
|
||||
continue
|
||||
}
|
||||
|
||||
idx := rand.Intn(len(location.Destinations))
|
||||
dest := location.Destinations[idx]
|
||||
|
||||
// Passing all the rules means the site is what we are looking for.
|
||||
// Let us respond to our client!
|
||||
return makeResponse(ctx, site)
|
||||
return makeResponse(ctx, &dest)
|
||||
}
|
||||
}
|
||||
|
||||
@ -89,19 +93,19 @@ func UseProxies(app *fiber.App) {
|
||||
})
|
||||
}
|
||||
|
||||
func makeResponse(ctx *fiber.Ctx, site *navi.SiteConfig) error {
|
||||
func makeResponse(ctx *fiber.Ctx, dest *navi.Destination) error {
|
||||
// Modify request
|
||||
for _, transformer := range site.Transformers {
|
||||
for _, transformer := range dest.Transformers {
|
||||
if err := transformer.TransformRequest(ctx); err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
|
||||
// Forward
|
||||
err := navi.App.Forward(ctx, site)
|
||||
err := navi.R.Forward(ctx, dest)
|
||||
|
||||
// Modify response
|
||||
for _, transformer := range site.Transformers {
|
||||
for _, transformer := range dest.Transformers {
|
||||
if err := transformer.TransformResponse(ctx); err != nil {
|
||||
return err
|
||||
}
|
||||
|
Reference in New Issue
Block a user