1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
|
package store
import (
"fmt"
)
// RepeaterTab represents a saved repeater tab.
type RepeaterTab struct {
ID int64
Name string
Host string
Port int
TLS bool
TabType string // "http" or "websocket"
RawRequest string
LastResponse string
Position int
}
// AllRepeaterTabs returns all saved repeater tabs ordered by position.
func (s *Store) AllRepeaterTabs() ([]RepeaterTab, error) {
rows, err := s.db.Query(`
SELECT id, name, host, port, tls, COALESCE(tab_type, 'http'), raw_request, COALESCE(last_response, ''), position
FROM repeater_tabs
ORDER BY position ASC`)
if err != nil {
return nil, fmt.Errorf("store: query repeater tabs: %w", err)
}
defer rows.Close()
var tabs []RepeaterTab
for rows.Next() {
var tab RepeaterTab
var tlsFlag int
if err := rows.Scan(&tab.ID, &tab.Name, &tab.Host, &tab.Port, &tlsFlag, &tab.TabType, &tab.RawRequest, &tab.LastResponse, &tab.Position); err != nil {
return nil, fmt.Errorf("store: scan repeater tab: %w", err)
}
tab.TLS = tlsFlag == 1
if tab.TabType == "" {
tab.TabType = "http"
}
tabs = append(tabs, tab)
}
return tabs, rows.Err()
}
// SaveRepeaterTab inserts a new repeater tab and returns its ID.
func (s *Store) SaveRepeaterTab(tab RepeaterTab) (int64, error) {
tabType := tab.TabType
if tabType == "" {
tabType = "http"
}
var id int64
err := s.write(func() error {
result, err := s.db.Exec(`
INSERT INTO repeater_tabs (name, host, port, tls, tab_type, raw_request, last_response, position)
VALUES (?, ?, ?, ?, ?, ?, ?, (SELECT COALESCE(MAX(position), 0) + 1 FROM repeater_tabs))`,
tab.Name, tab.Host, tab.Port, boolToInt(tab.TLS), tabType, tab.RawRequest, tab.LastResponse,
)
if err != nil {
return fmt.Errorf("store: save repeater tab: %w", err)
}
id, err = result.LastInsertId()
return err
})
return id, err
}
// UpdateRepeaterTab updates the request and response for an existing tab.
func (s *Store) UpdateRepeaterTab(id int64, rawRequest, lastResponse string) error {
return s.write(func() error {
_, err := s.db.Exec(`
UPDATE repeater_tabs SET raw_request = ?, last_response = ? WHERE id = ?`,
rawRequest, lastResponse, id,
)
if err != nil {
return fmt.Errorf("store: update repeater tab: %w", err)
}
return nil
})
}
// RenameRepeaterTab updates the name of a tab.
func (s *Store) RenameRepeaterTab(id int64, name string) error {
return s.write(func() error {
_, err := s.db.Exec(`UPDATE repeater_tabs SET name = ? WHERE id = ?`, name, id)
if err != nil {
return fmt.Errorf("store: rename repeater tab: %w", err)
}
return nil
})
}
// DeleteRepeaterTab removes a repeater tab by ID.
func (s *Store) DeleteRepeaterTab(id int64) error {
return s.write(func() error {
_, err := s.db.Exec(`DELETE FROM repeater_tabs WHERE id = ?`, id)
if err != nil {
return fmt.Errorf("store: delete repeater tab: %w", err)
}
return nil
})
}
|
