1 files changed, 72 insertions, 0 deletions
diff --git a/internal/store/intruder.go b/internal/store/intruder.go
new file mode 100644
index 0000000..4968f5d
--- /dev/null
+++ b/internal/store/intruder.go
@@ -0,0 +1,72 @@
+package store
+
+import "fmt"
+
+type IntruderConfig struct {
+	DelayMs         int
+	StopOnStatus    int
+	MaxRedirects    int
+	FollowRedirects string // "never", "always", "in-scope"
+	TimeoutMs       int
+	RawRequest      string
+	Payloads        string
+}
+
+func DefaultIntruderConfig() IntruderConfig {
+	return IntruderConfig{
+		DelayMs:         0,
+		StopOnStatus:    0,
+		MaxRedirects:    10,
+		FollowRedirects: "never",
+		TimeoutMs:       30000,
+		RawRequest:      "",
+		Payloads:        "",
+	}
+}
+
+func (s *Store) LoadIntruderConfig() IntruderConfig {
+	config := DefaultIntruderConfig()
+	err := s.db.QueryRow(`
+		SELECT delay_ms, stop_on_status, max_redirects, follow_redirects, timeout_ms, raw_request, payloads
+		FROM intruder_config WHERE id = 1`).Scan(
+		&config.DelayMs,
+		&config.StopOnStatus,
+		&config.MaxRedirects,
+		&config.FollowRedirects,
+		&config.TimeoutMs,
+		&config.RawRequest,
+		&config.Payloads,
+	)
+	if err != nil {
+		return config
+	}
+	return config
+}
+
+func (s *Store) SaveIntruderConfig(config IntruderConfig) error {
+	return s.write(func() error {
+		_, err := s.db.Exec(`
+			INSERT INTO intruder_config (id, delay_ms, stop_on_status, max_redirects, follow_redirects, timeout_ms, raw_request, payloads)
+			VALUES (1, ?, ?, ?, ?, ?, ?, ?)
+			ON CONFLICT(id) DO UPDATE SET
+				delay_ms = excluded.delay_ms,
+				stop_on_status = excluded.stop_on_status,
+				max_redirects = excluded.max_redirects,
+				follow_redirects = excluded.follow_redirects,
+				timeout_ms = excluded.timeout_ms,
+				raw_request = excluded.raw_request,
+				payloads = excluded.payloads`,
+			config.DelayMs,
+			config.StopOnStatus,
+			config.MaxRedirects,
+			config.FollowRedirects,
+			config.TimeoutMs,
+			config.RawRequest,
+			config.Payloads,
+		)
+		if err != nil {
+			return fmt.Errorf("store: save intruder config: %w", err)
+		}
+		return nil
+	})
+}