add ability to load rulesets from directory

This commit is contained in:
Kevin Pham
2023-11-14 15:42:26 -06:00
parent 6bff28e18d
commit e87d19d7f5
5 changed files with 474 additions and 75 deletions

View File

@@ -36,6 +36,11 @@ func main() {
Help: "This will spawn multiple processes listening",
})
ruleset := parser.String("r", "ruleset", &argparse.Options{
Required: false,
Help: "File, Directory or URL to a ruleset.yml. Overrides RULESET environment variable.",
})
err := parser.Parse(os.Args)
if err != nil {
fmt.Print(parser.Usage(err))
@@ -80,7 +85,7 @@ func main() {
app.Get("raw/*", handlers.Raw)
app.Get("api/*", handlers.Api)
app.Get("ruleset", handlers.Raw)
app.Get("/*", handlers.ProxySite)
app.Get("/*", handlers.ProxySite(*ruleset))
log.Fatal(app.Listen(":" + *port))
}