mirror of
https://github.com/anotherhadi/spilltea.git
synced 2026-07-06 20:42:33 +02:00
Hadi
105 lines
2.8 KiB
Go
105 lines
2.8 KiB
Go
package db
|
|
|
|
import (
|
|
"log"
|
|
"time"
|
|
)
|
|
|
|
type ReplayEntry struct {
|
|
ID int64
|
|
Timestamp time.Time
|
|
Scheme string
|
|
Host string
|
|
Path string
|
|
Method string
|
|
OriginalRaw string
|
|
RequestRaw string
|
|
ResponseRaw string
|
|
StatusCode int
|
|
ErrorMsg string
|
|
Flagged bool
|
|
}
|
|
|
|
func (d *DB) InsertReplayEntry(e ReplayEntry) (int64, error) {
|
|
res, err := d.conn.Exec(
|
|
`INSERT INTO replay_entries (timestamp, scheme, host, path, method, original_raw, request_raw, response_raw, status_code, error_msg, flagged)
|
|
VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`,
|
|
e.Timestamp.UTC().Format(time.RFC3339),
|
|
e.Scheme, e.Host, e.Path, e.Method,
|
|
e.OriginalRaw, e.RequestRaw, e.ResponseRaw,
|
|
e.StatusCode, e.ErrorMsg, e.Flagged,
|
|
)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
return res.LastInsertId()
|
|
}
|
|
|
|
func (d *DB) UpdateReplayEntry(e ReplayEntry) error {
|
|
_, err := d.conn.Exec(
|
|
`UPDATE replay_entries SET request_raw=?, response_raw=?, status_code=?, error_msg=?, flagged=? WHERE id=?`,
|
|
e.RequestRaw, e.ResponseRaw, e.StatusCode, e.ErrorMsg, e.Flagged, e.ID,
|
|
)
|
|
return err
|
|
}
|
|
|
|
func (d *DB) ListReplayEntries() ([]ReplayEntry, error) {
|
|
rows, err := d.conn.Query(
|
|
`SELECT id, timestamp, scheme, host, path, method, original_raw, request_raw, response_raw, status_code, error_msg, flagged
|
|
FROM replay_entries ORDER BY id ASC`,
|
|
)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer rows.Close()
|
|
|
|
var entries []ReplayEntry
|
|
for rows.Next() {
|
|
var e ReplayEntry
|
|
var ts string
|
|
var flagged int
|
|
if err := rows.Scan(&e.ID, &ts, &e.Scheme, &e.Host, &e.Path, &e.Method,
|
|
&e.OriginalRaw, &e.RequestRaw, &e.ResponseRaw, &e.StatusCode, &e.ErrorMsg, &flagged); err != nil {
|
|
return nil, err
|
|
}
|
|
var tsErr error
|
|
e.Timestamp, tsErr = time.Parse(time.RFC3339, ts)
|
|
if tsErr != nil {
|
|
log.Printf("db: parse replay timestamp %q: %v", ts, tsErr)
|
|
}
|
|
e.Flagged = flagged != 0
|
|
entries = append(entries, e)
|
|
}
|
|
return entries, rows.Err()
|
|
}
|
|
|
|
func (d *DB) ToggleReplayFlag(id int64) error {
|
|
_, err := d.conn.Exec(`UPDATE replay_entries SET flagged = NOT flagged WHERE id = ?`, id)
|
|
return err
|
|
}
|
|
|
|
func (d *DB) DeleteReplayEntry(id int64) error {
|
|
_, err := d.conn.Exec(`DELETE FROM replay_entries WHERE id = ?`, id)
|
|
return err
|
|
}
|
|
|
|
func (d *DB) DeleteAllReplayEntries() error {
|
|
_, err := d.conn.Exec(`DELETE FROM replay_entries`)
|
|
return err
|
|
}
|
|
|
|
// DeleteAllReplayEntriesExceptFlagged deletes all unflagged replay entries. If
|
|
// none are unflagged (only flagged ones remain), it deletes everything.
|
|
func (d *DB) DeleteAllReplayEntriesExceptFlagged() error {
|
|
var count int
|
|
if err := d.conn.QueryRow(`SELECT COUNT(*) FROM replay_entries WHERE flagged = 0`).Scan(&count); err != nil {
|
|
return err
|
|
}
|
|
if count > 0 {
|
|
_, err := d.conn.Exec(`DELETE FROM replay_entries WHERE flagged = 0`)
|
|
return err
|
|
}
|
|
_, err := d.conn.Exec(`DELETE FROM replay_entries`)
|
|
return err
|
|
}
|