From 2b9d0e11b868eb282f21c463635fd38f3edcba71 Mon Sep 17 00:00:00 2001 From: yequari Date: Fri, 9 May 2025 23:59:54 -0700 Subject: [PATCH] generic setting update code --- internal/models/user.go | 31 ++++++++++++++++++++++++++++++- 1 file changed, 30 insertions(+), 1 deletion(-) diff --git a/internal/models/user.go b/internal/models/user.go index 3d6432e..5dbad1e 100644 --- a/internal/models/user.go +++ b/internal/models/user.go @@ -31,7 +31,8 @@ type User struct { } type UserModel struct { - DB *sql.DB + DB *sql.DB + Settings map[string]Setting } func (m *UserModel) Insert(shortId uint64, username string, email string, password string, settings UserSettings) error { @@ -187,6 +188,34 @@ func (m *UserModel) initializeUserSettings(userId int64, settings UserSettings) return nil } +func (m *UserModel) UpdateUserSettings(userId int64, settings UserSettings) error { + err := m.UpdateSetting(userId, m.Settings["local_timezone"], settings.LocalTimezone.String()) + if err != nil { + return err + } + return nil +} + +func (m *UserModel) UpdateSetting(userId int64, setting Setting, value string) error { + stmt := `UPDATE user_settings SET + AllowedSettingValueId=(SELECT Id FROM allowed_setting_values WHERE SettingId = user_settings.SettingId AND ItemValue = ?), + UnconstrainedValue=(SELECT ? FROM settings WHERE settings.Id = user_settings.SettingId AND settings.Constrained=0) + WHERE userId = ? + AND SettingId = (SELECT Id from Settings WHERE Description=?);` + result, err := m.DB.Exec(stmt, value, value, userId) + if err != nil { + return err + } + rows, err := result.RowsAffected() + if err != nil { + return err + } + if rows != 1 { + return err + } + return nil +} + func (m *UserModel) SetLocalTimezone(userId int64, timezone string) error { valid, err := validateSetting(m.DB, u_timezone, timezone) if err != nil {