RoadSign/pkg/navi/struct.go

package navi

import (
	"fmt"
	"net/url"
	"strings"

	"code.smartsheep.studio/goatworks/roadsign/pkg/navi/transformers"
	"code.smartsheep.studio/goatworks/roadsign/pkg/warden"
	"github.com/gofiber/fiber/v2"
	"github.com/samber/lo"
)

type Region struct {
	ID           string               `json:"id" toml:"id"`
	Disabled     bool                 `json:"disabled" toml:"disabled"`
	Locations    []Location           `json:"locations" toml:"locations"`
	Applications []warden.Application `json:"applications" toml:"applications"`
}

type Location struct {
	ID           string              `json:"id" toml:"id"`
	Hosts        []string            `json:"hosts" toml:"hosts"`
	Paths        []string            `json:"paths" toml:"paths"`
	Queries      map[string]string   `json:"queries" toml:"queries"`
	Headers      map[string][]string `json:"headers" toml:"headers"`
	Destinations []Destination       `json:"destinations" toml:"destinations"`
}

type DestinationType = int8

const (
	DestinationHypertext = DestinationType(iota)
	DestinationStaticFile
	DestinationUnknown
)

type Destination struct {
	ID           string                           `json:"id" toml:"id"`
	Uri          string                           `json:"uri" toml:"uri"`
	Helmet       *HelmetConfig                    `json:"helmet" toml:"helmet"`
	Transformers []transformers.TransformerConfig `json:"transformers" toml:"transformers"`
}

func (v *Destination) GetProtocol() string {
	return strings.SplitN(v.Uri, "://", 2)[0]
}

func (v *Destination) GetType() DestinationType {
	protocol := v.GetProtocol()
	switch protocol {
	case "http", "https":
		return DestinationHypertext
	case "file", "files":
		return DestinationStaticFile
	}
	return DestinationUnknown
}

func (v *Destination) GetRawUri() (string, url.Values) {
	uri := strings.SplitN(v.Uri, "://", 2)[1]
	data := strings.SplitN(uri, "?", 2)
	data = append(data, " ") // Make data array least have two element
	qs, _ := url.ParseQuery(data[1])

	return data[0], qs
}

func (v *Destination) MakeUri(ctx *fiber.Ctx) string {
	var queries []string
	for k, v := range ctx.Queries() {
		parsed, _ := url.QueryUnescape(v)
		value := url.QueryEscape(parsed)
		queries = append(queries, fmt.Sprintf("%s=%s", k, value))
	}

	path := string(ctx.Request().URI().Path())
	hash := string(ctx.Request().URI().Hash())
	protocol := v.GetProtocol()
	uri, _ := v.GetRawUri()

	return protocol + "://" + uri + path +
		lo.Ternary(len(queries) > 0, "?"+strings.Join(queries, "&"), "") +
		lo.Ternary(len(hash) > 0, "#"+hash, "")
}

func (v *Destination) MakeWebsocketUri(ctx *fiber.Ctx) string {
	return strings.Replace(v.MakeUri(ctx), "http", "ws", 1)
}