Add vars with SDK

alban-stourbe-wmx · Sep 30, 2024 · d0463e8 · d0463e8
1 parent d161485
commit d0463e8
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/lib/config.go b/lib/config.go
@@ -392,6 +392,20 @@ func WithHeaders(headers []string) NucleiSDKOptions {
 	}
 }
 
+// WithVars allows setting custom variables to use in templates/workflows context
+func WithVars(vars []string) NucleiSDKOptions {
+	// Create a goflags.RuntimeMap
+	runtimeVars := goflags.RuntimeMap{}
+	for _, v := range vars {
+		runtimeVars.Set(v)
+	}
+
+	return func(e *NucleiEngine) error {
+		e.opts.Vars = runtimeVars
+		return nil
+	}
+}
+
 // EnablePassiveMode allows enabling passive HTTP response processing mode
 func EnablePassiveMode() NucleiSDKOptions {
 	return func(e *NucleiEngine) error {