Merge pull request #61 from everywall/proxy_v2/cleanup-main

cleanup main

cmd/main.go

@@ -1,12 +1,11 @@
 package main
 
 import (
-	"embed"
+	_ "embed"
 	"fmt"
 	"html/template"
 	"log"
 	"os"
-	"strings"
 
 	"ladder/handlers"
 	"ladder/internal/cli"
@@ -14,20 +13,9 @@ import (
 
 	"github.com/akamensky/argparse"
 	"github.com/gofiber/fiber/v2"
-	"github.com/gofiber/fiber/v2/middleware/basicauth"
-	"github.com/gofiber/fiber/v2/middleware/favicon"
 	"github.com/gofiber/template/html/v2"
 )
 
-//go:embed favicon.ico
-var faviconData string
-
-//go:embed styles.css
-var cssData embed.FS
-
-//go:embed script.js
-var scriptData embed.FS
-
 //go:embed VERSION
 var version string
 
@@ -151,23 +139,8 @@ func main() {
 		},
 	)
 
-	// TODO: move to cmd/auth.go
-	userpass := os.Getenv("USERPASS")
-	if userpass != "" {
-		userpass := strings.Split(userpass, ":")
-
-		app.Use(basicauth.New(basicauth.Config{
-			Users: map[string]string{
-				userpass[0]: userpass[1],
-			},
-		}))
-	}
-
-	// TODO: move to handlers/favicon.go
-	app.Use(favicon.New(favicon.Config{
-		Data: []byte(faviconData),
-		URL:  "/favicon.ico",
-	}))
+	app.Use(handlers.Auth())
+	app.Use(handlers.Favicon())
 
 	if os.Getenv("NOLOGS") != "true" {
 		app.Use(func(c *fiber.Ctx) error {
@@ -179,29 +152,8 @@ func main() {
 
 	app.Get("/", handlers.Form)
 
-	app.Get("/styles.css", func(c *fiber.Ctx) error {
-		cssData, err := cssData.ReadFile("styles.css")
-		if err != nil {
-			return c.Status(fiber.StatusInternalServerError).SendString("Internal Server Error")
-		}
-
-		c.Set("Content-Type", "text/css")
-
-		return c.Send(cssData)
-	})
-
-	// TODO: move to handlers/script.go
-	app.Get("/script.js", func(c *fiber.Ctx) error {
-		scriptData, err := scriptData.ReadFile("script.js")
-		if err != nil {
-			return c.Status(fiber.StatusInternalServerError).SendString("Internal Server Error")
-		}
-
-		c.Set("Content-Type", "text/javascript")
-
-		return c.Send(scriptData)
-	})
-
+	app.Get("styles.css", handlers.Styles)
+	app.Get("script.js", handlers.Script)
 	app.Get("ruleset", handlers.Ruleset)
 	app.Get("raw/*", handlers.Raw)
 

handlers/auth.go

@@ -0,0 +1,25 @@
+package handlers
+
+import (
+	"os"
+	"strings"
+
+	"github.com/gofiber/fiber/v2"
+	"github.com/gofiber/fiber/v2/middleware/basicauth"
+)
+
+func Auth() fiber.Handler {
+	userpass := os.Getenv("USERPASS")
+	if userpass != "" {
+		userpass := strings.Split(userpass, ":")
+		return basicauth.New(basicauth.Config{
+			Users: map[string]string{
+				userpass[0]: userpass[1],
+			},
+		})
+	}
+
+	return func(c *fiber.Ctx) error {
+		return c.Next()
+	}
+}

handlers/favicon.go

@@ -0,0 +1,18 @@
+package handlers
+
+import (
+	_ "embed"
+
+	"github.com/gofiber/fiber/v2"
+	"github.com/gofiber/fiber/v2/middleware/favicon"
+)
+
+//go:embed favicon.ico
+var faviconData string
+
+func Favicon() fiber.Handler {
+	return favicon.New(favicon.Config{
+		Data: []byte(faviconData),
+		URL:  "/favicon.ico",
+	})
+}

handlers/proxy.go

@@ -33,13 +33,13 @@ func NewProxySiteHandler(opts *ProxyOptions) fiber.Handler {
 			SetRequestModifications(
 				// rx.SpoofJA3fingerprint(ja3, "Googlebot"),
 				// rx.MasqueradeAsFacebookBot(),
-				//rx.MasqueradeAsGoogleBot(),
+				// rx.MasqueradeAsGoogleBot(),
 				rx.DeleteOutgoingCookies(),
 				rx.ForwardRequestHeaders(),
-				//rx.SpoofReferrerFromGoogleSearch(),
+				// rx.SpoofReferrerFromGoogleSearch(),
 				rx.SpoofReferrerFromLinkedInPost(),
-				//rx.RequestWaybackMachine(),
-				//rx.RequestArchiveIs(),
+				// rx.RequestWaybackMachine(),
+				// rx.RequestArchiveIs(),
 			).
 			AddResponseModifications(
 				tx.ForwardResponseHeaders(),

handlers/script.go

@@ -0,0 +1,23 @@
+package handlers
+
+import (
+	"embed"
+
+	"github.com/gofiber/fiber/v2"
+)
+
+//go:embed script.js
+var scriptData embed.FS
+
+func Script(c *fiber.Ctx) error {
+
+	scriptData, err := scriptData.ReadFile("script.js")
+	if err != nil {
+		return c.Status(fiber.StatusInternalServerError).SendString("Internal Server Error")
+	}
+
+	c.Set("Content-Type", "text/javascript")
+
+	return c.Send(scriptData)
+
+}

handlers/styles.go

@@ -0,0 +1,23 @@
+package handlers
+
+import (
+	"embed"
+
+	"github.com/gofiber/fiber/v2"
+)
+
+//go:embed styles.css
+var cssData embed.FS
+
+func Styles(c *fiber.Ctx) error {
+
+	cssData, err := cssData.ReadFile("styles.css")
+	if err != nil {
+		return c.Status(fiber.StatusInternalServerError).SendString("Internal Server Error")
+	}
+
+	c.Set("Content-Type", "text/css")
+
+	return c.Send(cssData)
+
+}

package.json

@@ -1,6 +1,6 @@
 {
 	"scripts": {
-		"build": "pnpx tailwindcss -i ./styles/input.css -o ./styles/output.css --build && pnpx minify ./styles/output.css > ./cmd/styles.css"
+		"build": "pnpx tailwindcss -i ./styles/input.css -o ./styles/output.css --build && pnpx minify ./styles/output.css > ./handlers/styles.css"
 	},
 	"devDependencies": {
 		"minify": "^10.5.2",

proxychain/codegen/codegen.go

@@ -83,7 +83,7 @@ func init() {
 
 	%s
 }`, strings.Join(factoryMaps, "\n"))
-	//fmt.Println(code)
+	// fmt.Println(code)
 	return code, nil
 }
 
@@ -155,7 +155,7 @@ func init() {
 
 	%s
 }`, strings.Join(factoryMaps, "\n"))
-	//fmt.Println(code)
+	// fmt.Println(code)
 	return code, nil
 }
 
@@ -177,7 +177,7 @@ func main() {
 	if err != nil {
 		panic(err)
 	}
-	//fmt.Println(rqmCode)
+	// fmt.Println(rqmCode)
 
 	fq, err := os.Create("../ruleset/rule_reqmod_types.gen.go")
 	if err != nil {
@@ -192,7 +192,7 @@ func main() {
 	if err != nil {
 		panic(err)
 	}
-	//fmt.Println(rsmCode)
+	// fmt.Println(rsmCode)
 
 	fs, err := os.Create("../ruleset/rule_resmod_types.gen.go")
 	if err != nil {

proxychain/requestmodifiers/modify_path_with_regex.go

@@ -2,8 +2,9 @@ package requestmodifiers
 
 import (
 	"fmt"
-	"ladder/proxychain"
 	"regexp"
+
+	"ladder/proxychain"
 )
 
 func ModifyPathWithRegex(matchRegex string, replacement string) proxychain.RequestModification {

proxychain/responsemodifiers/patch_google_analytics.go

@@ -3,8 +3,9 @@ package responsemodifiers
 import (
 	_ "embed"
 	"io"
-	"ladder/proxychain"
 	"strings"
+
+	"ladder/proxychain"
 )
 
 //go:embed patch_google_analytics.js

proxychain/responsemodifiers/patch_tracker_scripts.go

@@ -4,9 +4,10 @@ import (
 	"embed"
 	"encoding/json"
 	"io"
-	"ladder/proxychain"
 	"log"
 	"regexp"
+
+	"ladder/proxychain"
 )
 
 //go:embed vendor/ddg-tracker-surrogates/mapping.json

proxychain/ruleset/rule.go

@@ -3,6 +3,7 @@ package ruleset_v2
 import (
 	"encoding/json"
 	"fmt"
+
 	"gopkg.in/yaml.v3"
 	"ladder/proxychain"
 )
@@ -84,7 +85,6 @@ func (rule *Rule) MarshalJSON() ([]byte, error) {
 
 // implement type yaml marshaller
 func (rule *Rule) UnmarshalYAML(unmarshal func(interface{}) error) error {
-
 	type Aux struct {
 		Domains               []string `yaml:"domains"`
 		RequestModifications  []_rqm   `yaml:"requestmodifications"`
@@ -122,7 +122,6 @@ func (rule *Rule) UnmarshalYAML(unmarshal func(interface{}) error) error {
 }
 
 func (rule *Rule) MarshalYAML() (interface{}, error) {
-
 	type Aux struct {
 		Domains               []string `yaml:"domains"`
 		RequestModifications  []_rqm   `yaml:"requestmodifications"`

proxychain/ruleset/rule_reqmod_types.gen.go

@@ -1,5 +1,5 @@
-
 package ruleset_v2
+
 // DO NOT EDIT THIS FILE. It is automatically generated by ladder/proxychain/codegen/codegen.go
 // The purpose of this is serialization of rulesets from JSON or YAML into functional options suitable
 // for use in proxychains.
@@ -16,168 +16,167 @@ var rqmModMap map[string]RequestModifierFactory
 func init() {
 	rqmModMap = make(map[string]RequestModifierFactory)
 
-	  rqmModMap["ForwardRequestHeaders"] = func(_ ...string) proxychain.RequestModification {
-    return rx.ForwardRequestHeaders()
-  }
+	rqmModMap["ForwardRequestHeaders"] = func(_ ...string) proxychain.RequestModification {
+		return rx.ForwardRequestHeaders()
+	}
 
-  rqmModMap["MasqueradeAsGoogleBot"] = func(_ ...string) proxychain.RequestModification {
-    return rx.MasqueradeAsGoogleBot()
-  }
+	rqmModMap["MasqueradeAsGoogleBot"] = func(_ ...string) proxychain.RequestModification {
+		return rx.MasqueradeAsGoogleBot()
+	}
 
-  rqmModMap["MasqueradeAsBingBot"] = func(_ ...string) proxychain.RequestModification {
-    return rx.MasqueradeAsBingBot()
-  }
+	rqmModMap["MasqueradeAsBingBot"] = func(_ ...string) proxychain.RequestModification {
+		return rx.MasqueradeAsBingBot()
+	}
 
-  rqmModMap["MasqueradeAsWaybackMachineBot"] = func(_ ...string) proxychain.RequestModification {
-    return rx.MasqueradeAsWaybackMachineBot()
-  }
+	rqmModMap["MasqueradeAsWaybackMachineBot"] = func(_ ...string) proxychain.RequestModification {
+		return rx.MasqueradeAsWaybackMachineBot()
+	}
 
-  rqmModMap["MasqueradeAsFacebookBot"] = func(_ ...string) proxychain.RequestModification {
-    return rx.MasqueradeAsFacebookBot()
-  }
+	rqmModMap["MasqueradeAsFacebookBot"] = func(_ ...string) proxychain.RequestModification {
+		return rx.MasqueradeAsFacebookBot()
+	}
 
-  rqmModMap["MasqueradeAsYandexBot"] = func(_ ...string) proxychain.RequestModification {
-    return rx.MasqueradeAsYandexBot()
-  }
+	rqmModMap["MasqueradeAsYandexBot"] = func(_ ...string) proxychain.RequestModification {
+		return rx.MasqueradeAsYandexBot()
+	}
 
-  rqmModMap["MasqueradeAsBaiduBot"] = func(_ ...string) proxychain.RequestModification {
-    return rx.MasqueradeAsBaiduBot()
-  }
+	rqmModMap["MasqueradeAsBaiduBot"] = func(_ ...string) proxychain.RequestModification {
+		return rx.MasqueradeAsBaiduBot()
+	}
 
-  rqmModMap["MasqueradeAsDuckDuckBot"] = func(_ ...string) proxychain.RequestModification {
-    return rx.MasqueradeAsDuckDuckBot()
-  }
+	rqmModMap["MasqueradeAsDuckDuckBot"] = func(_ ...string) proxychain.RequestModification {
+		return rx.MasqueradeAsDuckDuckBot()
+	}
 
-  rqmModMap["MasqueradeAsYahooBot"] = func(_ ...string) proxychain.RequestModification {
-    return rx.MasqueradeAsYahooBot()
-  }
+	rqmModMap["MasqueradeAsYahooBot"] = func(_ ...string) proxychain.RequestModification {
+		return rx.MasqueradeAsYahooBot()
+	}
 
-  rqmModMap["ModifyDomainWithRegex"] = func(params ...string) proxychain.RequestModification {
-    return rx.ModifyDomainWithRegex(params[0], params[1])
-  }
+	rqmModMap["ModifyDomainWithRegex"] = func(params ...string) proxychain.RequestModification {
+		return rx.ModifyDomainWithRegex(params[0], params[1])
+	}
 
-  rqmModMap["SetOutgoingCookie"] = func(params ...string) proxychain.RequestModification {
-    return rx.SetOutgoingCookie(params[0], params[1])
-  }
+	rqmModMap["SetOutgoingCookie"] = func(params ...string) proxychain.RequestModification {
+		return rx.SetOutgoingCookie(params[0], params[1])
+	}
 
-  rqmModMap["SetOutgoingCookies"] = func(params ...string) proxychain.RequestModification {
-    return rx.SetOutgoingCookies(params[0])
-  }
+	rqmModMap["SetOutgoingCookies"] = func(params ...string) proxychain.RequestModification {
+		return rx.SetOutgoingCookies(params[0])
+	}
 
-  rqmModMap["DeleteOutgoingCookie"] = func(params ...string) proxychain.RequestModification {
-    return rx.DeleteOutgoingCookie(params[0])
-  }
+	rqmModMap["DeleteOutgoingCookie"] = func(params ...string) proxychain.RequestModification {
+		return rx.DeleteOutgoingCookie(params[0])
+	}
 
-  rqmModMap["DeleteOutgoingCookies"] = func(_ ...string) proxychain.RequestModification {
-    return rx.DeleteOutgoingCookies()
-  }
+	rqmModMap["DeleteOutgoingCookies"] = func(_ ...string) proxychain.RequestModification {
+		return rx.DeleteOutgoingCookies()
+	}
 
-  rqmModMap["DeleteOutgoingCookiesExcept"] = func(params ...string) proxychain.RequestModification {
-    return rx.DeleteOutgoingCookiesExcept(params[0])
-  }
+	rqmModMap["DeleteOutgoingCookiesExcept"] = func(params ...string) proxychain.RequestModification {
+		return rx.DeleteOutgoingCookiesExcept(params[0])
+	}
 
-  rqmModMap["ModifyPathWithRegex"] = func(params ...string) proxychain.RequestModification {
-    return rx.ModifyPathWithRegex(params[0], params[1])
-  }
+	rqmModMap["ModifyPathWithRegex"] = func(params ...string) proxychain.RequestModification {
+		return rx.ModifyPathWithRegex(params[0], params[1])
+	}
 
-  rqmModMap["ModifyQueryParams"] = func(params ...string) proxychain.RequestModification {
-    return rx.ModifyQueryParams(params[0], params[1])
-  }
+	rqmModMap["ModifyQueryParams"] = func(params ...string) proxychain.RequestModification {
+		return rx.ModifyQueryParams(params[0], params[1])
+	}
 
-  rqmModMap["SetRequestHeader"] = func(params ...string) proxychain.RequestModification {
-    return rx.SetRequestHeader(params[0], params[1])
-  }
+	rqmModMap["SetRequestHeader"] = func(params ...string) proxychain.RequestModification {
+		return rx.SetRequestHeader(params[0], params[1])
+	}
 
-  rqmModMap["DeleteRequestHeader"] = func(params ...string) proxychain.RequestModification {
-    return rx.DeleteRequestHeader(params[0])
-  }
+	rqmModMap["DeleteRequestHeader"] = func(params ...string) proxychain.RequestModification {
+		return rx.DeleteRequestHeader(params[0])
+	}
 
-  rqmModMap["RequestArchiveIs"] = func(_ ...string) proxychain.RequestModification {
-    return rx.RequestArchiveIs()
-  }
+	rqmModMap["RequestArchiveIs"] = func(_ ...string) proxychain.RequestModification {
+		return rx.RequestArchiveIs()
+	}
 
-  rqmModMap["RequestGoogleCache"] = func(_ ...string) proxychain.RequestModification {
-    return rx.RequestGoogleCache()
-  }
+	rqmModMap["RequestGoogleCache"] = func(_ ...string) proxychain.RequestModification {
+		return rx.RequestGoogleCache()
+	}
 
-  rqmModMap["RequestWaybackMachine"] = func(_ ...string) proxychain.RequestModification {
-    return rx.RequestWaybackMachine()
-  }
+	rqmModMap["RequestWaybackMachine"] = func(_ ...string) proxychain.RequestModification {
+		return rx.RequestWaybackMachine()
+	}
 
-  rqmModMap["ResolveWithGoogleDoH"] = func(_ ...string) proxychain.RequestModification {
-    return rx.ResolveWithGoogleDoH()
-  }
+	rqmModMap["ResolveWithGoogleDoH"] = func(_ ...string) proxychain.RequestModification {
+		return rx.ResolveWithGoogleDoH()
+	}
 
-  rqmModMap["SpoofOrigin"] = func(params ...string) proxychain.RequestModification {
-    return rx.SpoofOrigin(params[0])
-  }
+	rqmModMap["SpoofOrigin"] = func(params ...string) proxychain.RequestModification {
+		return rx.SpoofOrigin(params[0])
+	}
 
-  rqmModMap["HideOrigin"] = func(_ ...string) proxychain.RequestModification {
-    return rx.HideOrigin()
-  }
+	rqmModMap["HideOrigin"] = func(_ ...string) proxychain.RequestModification {
+		return rx.HideOrigin()
+	}
 
-  rqmModMap["SpoofReferrer"] = func(params ...string) proxychain.RequestModification {
-    return rx.SpoofReferrer(params[0])
-  }
+	rqmModMap["SpoofReferrer"] = func(params ...string) proxychain.RequestModification {
+		return rx.SpoofReferrer(params[0])
+	}
 
-  rqmModMap["HideReferrer"] = func(_ ...string) proxychain.RequestModification {
-    return rx.HideReferrer()
-  }
+	rqmModMap["HideReferrer"] = func(_ ...string) proxychain.RequestModification {
+		return rx.HideReferrer()
+	}
 
-  rqmModMap["SpoofReferrerFromBaiduSearch"] = func(_ ...string) proxychain.RequestModification {
-    return rx.SpoofReferrerFromBaiduSearch()
-  }
+	rqmModMap["SpoofReferrerFromBaiduSearch"] = func(_ ...string) proxychain.RequestModification {
+		return rx.SpoofReferrerFromBaiduSearch()
+	}
 
-  rqmModMap["SpoofReferrerFromBingSearch"] = func(_ ...string) proxychain.RequestModification {
-    return rx.SpoofReferrerFromBingSearch()
-  }
+	rqmModMap["SpoofReferrerFromBingSearch"] = func(_ ...string) proxychain.RequestModification {
+		return rx.SpoofReferrerFromBingSearch()
+	}
 
-  rqmModMap["SpoofReferrerFromGoogleSearch"] = func(_ ...string) proxychain.RequestModification {
-    return rx.SpoofReferrerFromGoogleSearch()
-  }
+	rqmModMap["SpoofReferrerFromGoogleSearch"] = func(_ ...string) proxychain.RequestModification {
+		return rx.SpoofReferrerFromGoogleSearch()
+	}
 
-  rqmModMap["SpoofReferrerFromLinkedInPost"] = func(_ ...string) proxychain.RequestModification {
-    return rx.SpoofReferrerFromLinkedInPost()
-  }
+	rqmModMap["SpoofReferrerFromLinkedInPost"] = func(_ ...string) proxychain.RequestModification {
+		return rx.SpoofReferrerFromLinkedInPost()
+	}
 
-  rqmModMap["SpoofReferrerFromNaverSearch"] = func(_ ...string) proxychain.RequestModification {
-    return rx.SpoofReferrerFromNaverSearch()
-  }
+	rqmModMap["SpoofReferrerFromNaverSearch"] = func(_ ...string) proxychain.RequestModification {
+		return rx.SpoofReferrerFromNaverSearch()
+	}
 
-  rqmModMap["SpoofReferrerFromPinterestPost"] = func(_ ...string) proxychain.RequestModification {
-    return rx.SpoofReferrerFromPinterestPost()
-  }
+	rqmModMap["SpoofReferrerFromPinterestPost"] = func(_ ...string) proxychain.RequestModification {
+		return rx.SpoofReferrerFromPinterestPost()
+	}
 
-  rqmModMap["SpoofReferrerFromQQPost"] = func(_ ...string) proxychain.RequestModification {
-    return rx.SpoofReferrerFromQQPost()
-  }
+	rqmModMap["SpoofReferrerFromQQPost"] = func(_ ...string) proxychain.RequestModification {
+		return rx.SpoofReferrerFromQQPost()
+	}
 
-  rqmModMap["SpoofReferrerFromRedditPost"] = func(_ ...string) proxychain.RequestModification {
-    return rx.SpoofReferrerFromRedditPost()
-  }
+	rqmModMap["SpoofReferrerFromRedditPost"] = func(_ ...string) proxychain.RequestModification {
+		return rx.SpoofReferrerFromRedditPost()
+	}
 
-  rqmModMap["SpoofReferrerFromTumblrPost"] = func(_ ...string) proxychain.RequestModification {
-    return rx.SpoofReferrerFromTumblrPost()
-  }
+	rqmModMap["SpoofReferrerFromTumblrPost"] = func(_ ...string) proxychain.RequestModification {
+		return rx.SpoofReferrerFromTumblrPost()
+	}
 
-  rqmModMap["SpoofReferrerFromTwitterPost"] = func(_ ...string) proxychain.RequestModification {
-    return rx.SpoofReferrerFromTwitterPost()
-  }
+	rqmModMap["SpoofReferrerFromTwitterPost"] = func(_ ...string) proxychain.RequestModification {
+		return rx.SpoofReferrerFromTwitterPost()
+	}
 
-  rqmModMap["SpoofReferrerFromVkontaktePost"] = func(_ ...string) proxychain.RequestModification {
-    return rx.SpoofReferrerFromVkontaktePost()
-  }
+	rqmModMap["SpoofReferrerFromVkontaktePost"] = func(_ ...string) proxychain.RequestModification {
+		return rx.SpoofReferrerFromVkontaktePost()
+	}
 
-  rqmModMap["SpoofReferrerFromWeiboPost"] = func(_ ...string) proxychain.RequestModification {
-    return rx.SpoofReferrerFromWeiboPost()
-  }
+	rqmModMap["SpoofReferrerFromWeiboPost"] = func(_ ...string) proxychain.RequestModification {
+		return rx.SpoofReferrerFromWeiboPost()
+	}
 
-  rqmModMap["SpoofUserAgent"] = func(params ...string) proxychain.RequestModification {
-    return rx.SpoofUserAgent(params[0])
-  }
-
-  rqmModMap["SpoofXForwardedFor"] = func(params ...string) proxychain.RequestModification {
-    return rx.SpoofXForwardedFor(params[0])
-  }
+	rqmModMap["SpoofUserAgent"] = func(params ...string) proxychain.RequestModification {
+		return rx.SpoofUserAgent(params[0])
+	}
 
+	rqmModMap["SpoofXForwardedFor"] = func(params ...string) proxychain.RequestModification {
+		return rx.SpoofXForwardedFor(params[0])
+	}
 }

proxychain/ruleset/rule_resmod_types.gen.go

@@ -1,5 +1,5 @@
-
 package ruleset_v2
+
 // DO NOT EDIT THIS FILE. It is automatically generated by ladder/proxychain/codegen/codegen.go
 // The purpose of this is serialization of rulesets from JSON or YAML into functional options suitable
 // for use in proxychains.
@@ -16,84 +16,83 @@ var rsmModMap map[string]ResponseModifierFactory
 func init() {
 	rsmModMap = make(map[string]ResponseModifierFactory)
 
-	  rsmModMap["APIContent"] = func(_ ...string) proxychain.ResponseModification {
-    return tx.APIContent()
-  }
+	rsmModMap["APIContent"] = func(_ ...string) proxychain.ResponseModification {
+		return tx.APIContent()
+	}
 
-  rsmModMap["BlockElementRemoval"] = func(params ...string) proxychain.ResponseModification {
-    return tx.BlockElementRemoval(params[0])
-  }
+	rsmModMap["BlockElementRemoval"] = func(params ...string) proxychain.ResponseModification {
+		return tx.BlockElementRemoval(params[0])
+	}
 
-  rsmModMap["BypassCORS"] = func(_ ...string) proxychain.ResponseModification {
-    return tx.BypassCORS()
-  }
+	rsmModMap["BypassCORS"] = func(_ ...string) proxychain.ResponseModification {
+		return tx.BypassCORS()
+	}
 
-  rsmModMap["BypassContentSecurityPolicy"] = func(_ ...string) proxychain.ResponseModification {
-    return tx.BypassContentSecurityPolicy()
-  }
+	rsmModMap["BypassContentSecurityPolicy"] = func(_ ...string) proxychain.ResponseModification {
+		return tx.BypassContentSecurityPolicy()
+	}
 
-  rsmModMap["SetContentSecurityPolicy"] = func(params ...string) proxychain.ResponseModification {
-    return tx.SetContentSecurityPolicy(params[0])
-  }
+	rsmModMap["SetContentSecurityPolicy"] = func(params ...string) proxychain.ResponseModification {
+		return tx.SetContentSecurityPolicy(params[0])
+	}
 
-  rsmModMap["ForwardResponseHeaders"] = func(_ ...string) proxychain.ResponseModification {
-    return tx.ForwardResponseHeaders()
-  }
+	rsmModMap["ForwardResponseHeaders"] = func(_ ...string) proxychain.ResponseModification {
+		return tx.ForwardResponseHeaders()
+	}
 
-  rsmModMap["GenerateReadableOutline"] = func(_ ...string) proxychain.ResponseModification {
-    return tx.GenerateReadableOutline()
-  }
+	rsmModMap["GenerateReadableOutline"] = func(_ ...string) proxychain.ResponseModification {
+		return tx.GenerateReadableOutline()
+	}
 
-  rsmModMap["InjectScriptBeforeDOMContentLoaded"] = func(params ...string) proxychain.ResponseModification {
-    return tx.InjectScriptBeforeDOMContentLoaded(params[0])
-  }
+	rsmModMap["InjectScriptBeforeDOMContentLoaded"] = func(params ...string) proxychain.ResponseModification {
+		return tx.InjectScriptBeforeDOMContentLoaded(params[0])
+	}
 
-  rsmModMap["InjectScriptAfterDOMContentLoaded"] = func(params ...string) proxychain.ResponseModification {
-    return tx.InjectScriptAfterDOMContentLoaded(params[0])
-  }
+	rsmModMap["InjectScriptAfterDOMContentLoaded"] = func(params ...string) proxychain.ResponseModification {
+		return tx.InjectScriptAfterDOMContentLoaded(params[0])
+	}
 
-  rsmModMap["InjectScriptAfterDOMIdle"] = func(params ...string) proxychain.ResponseModification {
-    return tx.InjectScriptAfterDOMIdle(params[0])
-  }
+	rsmModMap["InjectScriptAfterDOMIdle"] = func(params ...string) proxychain.ResponseModification {
+		return tx.InjectScriptAfterDOMIdle(params[0])
+	}
 
-  rsmModMap["DeleteIncomingCookies"] = func(params ...string) proxychain.ResponseModification {
-    return tx.DeleteIncomingCookies(params[0])
-  }
+	rsmModMap["DeleteIncomingCookies"] = func(params ...string) proxychain.ResponseModification {
+		return tx.DeleteIncomingCookies(params[0])
+	}
 
-  rsmModMap["DeleteIncomingCookiesExcept"] = func(params ...string) proxychain.ResponseModification {
-    return tx.DeleteIncomingCookiesExcept(params[0])
-  }
+	rsmModMap["DeleteIncomingCookiesExcept"] = func(params ...string) proxychain.ResponseModification {
+		return tx.DeleteIncomingCookiesExcept(params[0])
+	}
 
-  rsmModMap["SetIncomingCookies"] = func(params ...string) proxychain.ResponseModification {
-    return tx.SetIncomingCookies(params[0])
-  }
+	rsmModMap["SetIncomingCookies"] = func(params ...string) proxychain.ResponseModification {
+		return tx.SetIncomingCookies(params[0])
+	}
 
-  rsmModMap["SetIncomingCookie"] = func(params ...string) proxychain.ResponseModification {
-    return tx.SetIncomingCookie(params[0], params[1])
-  }
+	rsmModMap["SetIncomingCookie"] = func(params ...string) proxychain.ResponseModification {
+		return tx.SetIncomingCookie(params[0], params[1])
+	}
 
-  rsmModMap["SetResponseHeader"] = func(params ...string) proxychain.ResponseModification {
-    return tx.SetResponseHeader(params[0], params[1])
-  }
+	rsmModMap["SetResponseHeader"] = func(params ...string) proxychain.ResponseModification {
+		return tx.SetResponseHeader(params[0], params[1])
+	}
 
-  rsmModMap["DeleteResponseHeader"] = func(params ...string) proxychain.ResponseModification {
-    return tx.DeleteResponseHeader(params[0])
-  }
+	rsmModMap["DeleteResponseHeader"] = func(params ...string) proxychain.ResponseModification {
+		return tx.DeleteResponseHeader(params[0])
+	}
 
-  rsmModMap["PatchDynamicResourceURLs"] = func(_ ...string) proxychain.ResponseModification {
-    return tx.PatchDynamicResourceURLs()
-  }
+	rsmModMap["PatchDynamicResourceURLs"] = func(_ ...string) proxychain.ResponseModification {
+		return tx.PatchDynamicResourceURLs()
+	}
 
-  rsmModMap["PatchGoogleAnalytics"] = func(_ ...string) proxychain.ResponseModification {
-    return tx.PatchGoogleAnalytics()
-  }
+	rsmModMap["PatchGoogleAnalytics"] = func(_ ...string) proxychain.ResponseModification {
+		return tx.PatchGoogleAnalytics()
+	}
 
-  rsmModMap["PatchTrackerScripts"] = func(_ ...string) proxychain.ResponseModification {
-    return tx.PatchTrackerScripts()
-  }
-
-  rsmModMap["RewriteHTMLResourceURLs"] = func(_ ...string) proxychain.ResponseModification {
-    return tx.RewriteHTMLResourceURLs()
-  }
+	rsmModMap["PatchTrackerScripts"] = func(_ ...string) proxychain.ResponseModification {
+		return tx.PatchTrackerScripts()
+	}
 
+	rsmModMap["RewriteHTMLResourceURLs"] = func(_ ...string) proxychain.ResponseModification {
+		return tx.RewriteHTMLResourceURLs()
+	}
 }

proxychain/ruleset/rule_test.go

@@ -3,8 +3,9 @@ package ruleset_v2
 import (
 	"encoding/json"
 	"fmt"
-	"gopkg.in/yaml.v3"
 	"testing"
+
+	"gopkg.in/yaml.v3"
 )
 
 // unmarshalRule is a helper function to unmarshal a Rule from a JSON string.