LittleSheep
f31d35c86c
All checks were successful
release-nightly / build-docker (push) Successful in 1m4s
60 lines
1.2 KiB
Go
60 lines
1.2 KiB
Go
package transformers
|
|
|
|
import (
|
|
"encoding/json"
|
|
"github.com/gofiber/fiber/v2"
|
|
)
|
|
|
|
// Definitions
|
|
|
|
type RequestTransformer struct {
|
|
ModifyRequest func(options any, ctx *fiber.Ctx) error
|
|
ModifyResponse func(options any, ctx *fiber.Ctx) error
|
|
}
|
|
|
|
type RequestTransformerConfig struct {
|
|
Type string `json:"type" yaml:"type"`
|
|
Options any `json:"options" yaml:"options"`
|
|
}
|
|
|
|
func (v *RequestTransformerConfig) TransformRequest(ctx *fiber.Ctx) error {
|
|
for k, f := range Transformers {
|
|
if k == v.Type {
|
|
if f.ModifyRequest != nil {
|
|
return f.ModifyRequest(v.Options, ctx)
|
|
}
|
|
break
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (v *RequestTransformerConfig) TransformResponse(ctx *fiber.Ctx) error {
|
|
for k, f := range Transformers {
|
|
if k == v.Type {
|
|
if f.ModifyResponse != nil {
|
|
return f.ModifyResponse(v.Options, ctx)
|
|
}
|
|
break
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// Helpers
|
|
|
|
func DeserializeOptions[T any](data any) T {
|
|
var out T
|
|
raw, _ := json.Marshal(data)
|
|
_ = json.Unmarshal(raw, &out)
|
|
return out
|
|
}
|
|
|
|
// Map of Transformers
|
|
// Every transformer need to be mapped here so that they can get work.
|
|
|
|
var Transformers = map[string]RequestTransformer{
|
|
"replacePath": ReplacePath,
|
|
"compressResponse": CompressResponse,
|
|
}
|