mirror of
https://github.com/BaaRai/ARTEX.git
synced 2026-11-13 11:40:53 +08:00
BaaRai
e233c02314
Comments, doc comments, UI labels, log and error messages, LLM prompt strings, markdown docs, test data and assertions. Identifiers, enum keys, routes, placeholders and code structure are unchanged. Notes: - docs/漏洞流量证据.md renamed to docs/vulnerability-traffic-evidence.md - tests that exercised CJK specific handling (GBK zip names, Chinese full text search) now use ASCII equivalents; go build passes and the test suite matches the pre translation baseline (remaining failures need a PostgreSQL DSN and fail identically on the original) - frontend mention syntax now uses English kinds (finding, asset, ...) matching the backend, chat mentions tests pass
64 lines
2.0 KiB
Go
64 lines
2.0 KiB
Go
package db
|
|
|
|
import "database/sql"
|
|
|
|
// Settings is a tiny key-value store for global app config the UI toggles at
|
|
// runtime (e.g. traffic_capture). Missing keys fall back to caller defaults.
|
|
|
|
// GetSetting returns the stored value and ok=false when the key is unset.
|
|
func (d *DB) GetSetting(key string) (value string, ok bool, err error) {
|
|
err = d.QueryRow(`SELECT value FROM settings WHERE key=$1`, key).Scan(&value)
|
|
if err == sql.ErrNoRows {
|
|
return "", false, nil
|
|
}
|
|
if err != nil {
|
|
return "", false, err
|
|
}
|
|
return value, true, nil
|
|
}
|
|
|
|
// SetSetting upserts a setting value.
|
|
func (d *DB) SetSetting(key, value string) error {
|
|
_, err := d.Exec(`
|
|
INSERT INTO settings(key, value) VALUES ($1, $2)
|
|
ON CONFLICT (key) DO UPDATE SET value = EXCLUDED.value, updated_at = now()`, key, value)
|
|
return err
|
|
}
|
|
|
|
// InsertSettingIfAbsent writes only when the key does not exist yet; an existing key is
|
|
// kept as-is and inserted=false is returned.
|
|
// For keys like auth.password_hash that "may only be set on first run": the judgment rests
|
|
// on the database primary-key constraint. A caller doing GetSetting-then-write is just a
|
|
// fast-fail path — on a read error or a concurrent race, the existing value will not be
|
|
// silently overwritten by ON CONFLICT DO UPDATE.
|
|
func (d *DB) InsertSettingIfAbsent(key, value string) (inserted bool, err error) {
|
|
res, err := d.Exec(`
|
|
INSERT INTO settings(key, value) VALUES ($1, $2)
|
|
ON CONFLICT (key) DO NOTHING`, key, value)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
n, err := res.RowsAffected()
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
return n > 0, nil
|
|
}
|
|
|
|
// GetBool returns the boolean setting, or def when unset/unparseable.
|
|
func (d *DB) GetBool(key string, def bool) bool {
|
|
v, ok, err := d.GetSetting(key)
|
|
if err != nil || !ok {
|
|
return def
|
|
}
|
|
return v == "true" || v == "1"
|
|
}
|
|
|
|
// SetBool stores a boolean setting as "true"/"false".
|
|
func (d *DB) SetBool(key string, val bool) error {
|
|
if val {
|
|
return d.SetSetting(key, "true")
|
|
}
|
|
return d.SetSetting(key, "false")
|
|
}
|