Add user settings validation

This commit is contained in:
2024-09-27 14:56:43 +02:00
parent f99c93ac23
commit 3a651ce4d8
7 changed files with 105 additions and 9 deletions

View File

@@ -15,10 +15,9 @@ var migrations = []Migration{
Version: 1,
SQL: `CREATE TABLE IF NOT EXISTS settings (
user_id INTEGER PRIMARY KEY,
settings TEXT
settings JSON NOT NULL
)`,
},
// Add new migrations here as your schema evolves
}
func (db *DB) Migrate() error {

View File

@@ -9,17 +9,20 @@ import (
func (db *DB) GetSettings(userID int) (models.Settings, error) {
var settings models.Settings
var settingsJSON string
var settingsJSON []byte
err := db.QueryRow("SELECT user_id, settings FROM settings WHERE user_id = ?", userID).Scan(&settings.UserID, &settingsJSON)
if err != nil {
if err == sql.ErrNoRows {
// If no settings found, return default settings
settings.UserID = userID
settings.Settings = models.UserSettings{} // This will be filled with defaults later
return settings, nil
}
return settings, err
}
err = json.Unmarshal([]byte(settingsJSON), &settings.Settings)
err = json.Unmarshal(settingsJSON, &settings.Settings)
if err != nil {
return settings, err
}
@@ -28,11 +31,15 @@ func (db *DB) GetSettings(userID int) (models.Settings, error) {
}
func (db *DB) SaveSettings(settings models.Settings) error {
if err := settings.Validate(); err != nil {
return err
}
settingsJSON, err := json.Marshal(settings.Settings)
if err != nil {
return err
}
_, err = db.Exec("INSERT OR REPLACE INTO settings (user_id, settings) VALUES (?, ?)", settings.UserID, string(settingsJSON))
_, err = db.Exec("INSERT OR REPLACE INTO settings (user_id, settings) VALUES (?, json(?))", settings.UserID, string(settingsJSON))
return err
}