1 files changed, 109 insertions, 0 deletions
diff --git a/internal/store/store.go b/internal/store/store.go
index a3c6b03..211162e 100644
--- a/internal/store/store.go
+++ b/internal/store/store.go
@@ -12,6 +12,8 @@ import (
 	"time"
 
 	_ "modernc.org/sqlite"
+
+	"mitmux/internal/rules"
 )
 
 const schema = `
@@ -36,6 +38,18 @@ CREATE VIRTUAL TABLE IF NOT EXISTS history_fts USING fts5(
 	method, host, path, request_text, response_text,
 	tokenize = 'unicode61 remove_diacritics 2'
 );
+
+CREATE TABLE IF NOT EXISTS rules (
+	id       INTEGER PRIMARY KEY AUTOINCREMENT,
+	enabled  INTEGER NOT NULL DEFAULT 1,
+	name     TEXT NOT NULL DEFAULT '',
+	scope    TEXT NOT NULL,
+	part     TEXT NOT NULL,
+	match    TEXT NOT NULL,
+	replace  TEXT NOT NULL,
+	is_regex INTEGER NOT NULL DEFAULT 0,
+	position INTEGER NOT NULL DEFAULT 0
+);
 `
 
 // Store is a handle to the history database. Safe for concurrent use.
@@ -312,6 +326,101 @@ func prepareFTSQuery(q string) string {
 	return strings.Join(fields, " ")
 }
 
+// ListRules returns every match-and-replace rule, ordered for application.
+func (s *Store) ListRules() ([]rules.Rule, error) {
+	rows, err := s.db.Query(
+		`SELECT id, enabled, name, scope, part, match, replace, is_regex, position
+		 FROM rules ORDER BY position, id`,
+	)
+	if err != nil {
+		return nil, fmt.Errorf("list rules: %w", err)
+	}
+	defer rows.Close()
+
+	var out []rules.Rule
+	for rows.Next() {
+		var r rules.Rule
+		var enabled, isRegex int
+		if err := rows.Scan(&r.ID, &enabled, &r.Name, &r.Scope, &r.Part, &r.Match, &r.Replace, &isRegex, &r.Position); err != nil {
+			return nil, fmt.Errorf("scan rule row: %w", err)
+		}
+		r.Enabled = enabled != 0
+		r.IsRegex = isRegex != 0
+		out = append(out, r)
+	}
+	return out, rows.Err()
+}
+
+// EnabledRules returns enabled rules for scope ("request" or
+// "response"), ordered for application.
+func (s *Store) EnabledRules(scope string) ([]rules.Rule, error) {
+	rows, err := s.db.Query(
+		`SELECT id, enabled, name, scope, part, match, replace, is_regex, position
+		 FROM rules WHERE enabled = 1 AND scope = ? ORDER BY position, id`,
+		scope,
+	)
+	if err != nil {
+		return nil, fmt.Errorf("enabled rules: %w", err)
+	}
+	defer rows.Close()
+
+	var out []rules.Rule
+	for rows.Next() {
+		var r rules.Rule
+		var enabled, isRegex int
+		if err := rows.Scan(&r.ID, &enabled, &r.Name, &r.Scope, &r.Part, &r.Match, &r.Replace, &isRegex, &r.Position); err != nil {
+			return nil, fmt.Errorf("scan rule row: %w", err)
+		}
+		r.Enabled = enabled != 0
+		r.IsRegex = isRegex != 0
+		out = append(out, r)
+	}
+	return out, rows.Err()
+}
+
+// AddRule stores r and returns its assigned ID.
+func (s *Store) AddRule(r rules.Rule) (int64, error) {
+	res, err := s.db.Exec(
+		`INSERT INTO rules (enabled, name, scope, part, match, replace, is_regex, position)
+		 VALUES (?, ?, ?, ?, ?, ?, ?, ?)`,
+		boolToInt(r.Enabled), r.Name, r.Scope, r.Part, r.Match, r.Replace, boolToInt(r.IsRegex), r.Position,
+	)
+	if err != nil {
+		return 0, fmt.Errorf("add rule: %w", err)
+	}
+	return res.LastInsertId()
+}
+
+// UpdateRule replaces the stored rule with the same ID as r.
+func (s *Store) UpdateRule(r rules.Rule) error {
+	_, err := s.db.Exec(
+		`UPDATE rules SET enabled = ?, name = ?, scope = ?, part = ?, match = ?, replace = ?, is_regex = ?, position = ?
+		 WHERE id = ?`,
+		boolToInt(r.Enabled), r.Name, r.Scope, r.Part, r.Match, r.Replace, boolToInt(r.IsRegex), r.Position, r.ID,
+	)
+	if err != nil {
+		return fmt.Errorf("update rule %d: %w", r.ID, err)
+	}
+	return nil
+}
+
+// SetRuleEnabled toggles a rule without touching its other fields.
+func (s *Store) SetRuleEnabled(id int64, enabled bool) error {
+	_, err := s.db.Exec(`UPDATE rules SET enabled = ? WHERE id = ?`, boolToInt(enabled), id)
+	if err != nil {
+		return fmt.Errorf("set rule %d enabled: %w", id, err)
+	}
+	return nil
+}
+
+// DeleteRule removes a rule.
+func (s *Store) DeleteRule(id int64) error {
+	if _, err := s.db.Exec(`DELETE FROM rules WHERE id = ?`, id); err != nil {
+		return fmt.Errorf("delete rule %d: %w", id, err)
+	}
+	return nil
+}
+
 func boolToInt(b bool) int {
 	if b {
 		return 1