cleanup handlers directory

handlers/api_ruleset.go

@@ -0,0 +1,96 @@
+package handlers
+
+import (
+	"encoding/json"
+	"fmt"
+	"github.com/gofiber/fiber/v2"
+	"gopkg.in/yaml.v3"
+	"net/url"
+	"strings"
+)
+
+func NewRulesetSiteHandler(opts *ProxyOptions) fiber.Handler {
+
+	return func(c *fiber.Ctx) error {
+		if opts == nil {
+			c.SendStatus(404)
+			c.SendString("No ruleset specified. Set the RULESET environment variable or use the --ruleset flag.")
+		}
+
+		// no specific rule requested, return the entire ruleset
+		if c.Params("*") == "" {
+			switch c.Get("accept") {
+			case "application/json":
+				jsn, err := opts.Ruleset.JSON()
+				if err != nil {
+					return err
+				}
+				c.Set("content-type", "application/json")
+				return c.Send([]byte(jsn))
+
+			default:
+				yml, err := opts.Ruleset.YAML()
+				if err != nil {
+					return err
+				}
+				c.Set("content-type", "text/yaml")
+				return c.Send([]byte(yml))
+			}
+		}
+
+		// a specific rule was requested by path /ruleset/https://example.com
+		// return only that particular rule
+		reqURL, err := extractURLFromContext(c, "api/ruleset/")
+		if err != nil {
+			c.SendStatus(404)
+			return c.SendString(fmt.Sprintf("A rule that matches '%s' was not found in the ruleset. Possible URL formatting issue.", c.Params("*")))
+		}
+		rule, exists := opts.Ruleset.GetRule(reqURL)
+		if !exists {
+			c.SendStatus(404)
+			return c.SendString(fmt.Sprintf("A rule that matches '%s' was not found in the ruleset.", reqURL))
+		}
+
+		switch c.Get("accept") {
+		case "application/json":
+			jsn, err := json.MarshalIndent(rule, "", "  ")
+			if err != nil {
+				return err
+			}
+			c.Set("content-type", "application/json")
+			return c.Send(jsn)
+		default:
+			yml, err := yaml.Marshal(rule)
+			if err != nil {
+				return err
+			}
+			c.Set("content-type", "text/yaml")
+			return c.Send(yml)
+		}
+	}
+}
+
+// extractURLFromContext extracts a URL from the request ctx.
+func extractURLFromContext(ctx *fiber.Ctx, apiPrefix string) (*url.URL, error) {
+	reqURL := ctx.Params("*")
+
+	reqURL = strings.TrimPrefix(reqURL, apiPrefix)
+	if !strings.HasPrefix(reqURL, "http") {
+		reqURL = "https://" + reqURL
+	}
+
+	// sometimes client requests doubleroot '//'
+	// there is a bug somewhere else, but this is a workaround until we find it
+	if strings.HasPrefix(reqURL, "/") || strings.HasPrefix(reqURL, `%2F`) {
+		reqURL = strings.TrimPrefix(reqURL, "/")
+		reqURL = strings.TrimPrefix(reqURL, `%2F`)
+	}
+
+	// unescape url query
+	uReqURL, err := url.QueryUnescape(reqURL)
+	if err == nil {
+		reqURL = uReqURL
+	}
+
+	return url.Parse(reqURL)
+}