silo-server/internal/userdb/settings.go

package userdb

import (
	"database/sql"
	"fmt"
	"strings"
	"time"

	"github.com/Silo-Server/silo-server/internal/userstore"
)

// GetSetting retrieves a setting value by key. Returns empty string if not found.
func GetSetting(db *sql.DB, key string) (string, error) {
	var value string
	err := db.QueryRow("SELECT value FROM user_settings WHERE key = ?", key).Scan(&value)
	if err == sql.ErrNoRows {
		return "", nil
	}
	if err != nil {
		return "", fmt.Errorf("getting setting %q: %w", key, err)
	}
	return value, nil
}

// SetSetting creates or updates a setting.
func SetSetting(db *sql.DB, key, value string) error {
	_, err := db.Exec(
		"INSERT INTO user_settings (key, value) VALUES (?, ?) ON CONFLICT(key) DO UPDATE SET value = excluded.value",
		key, value,
	)
	if err != nil {
		return fmt.Errorf("setting %q: %w", key, err)
	}
	return nil
}

// DeleteSetting removes a setting by key.
func DeleteSetting(db *sql.DB, key string) error {
	_, err := db.Exec("DELETE FROM user_settings WHERE key = ?", key)
	if err != nil {
		return fmt.Errorf("deleting setting %q: %w", key, err)
	}
	return nil
}

// SettingEntry is an alias for the canonical type in userstore.
type SettingEntry = userstore.SettingEntry

// ListSettings returns all user settings.
func ListSettings(db *sql.DB) ([]SettingEntry, error) {
	rows, err := db.Query("SELECT key, value FROM user_settings ORDER BY key")
	if err != nil {
		return nil, fmt.Errorf("listing settings: %w", err)
	}
	defer rows.Close()

	var entries []SettingEntry
	for rows.Next() {
		var e SettingEntry
		if err := rows.Scan(&e.Key, &e.Value); err != nil {
			return nil, fmt.Errorf("scanning setting: %w", err)
		}
		entries = append(entries, e)
	}
	return entries, rows.Err()
}

func GetDeviceSetting(db *sql.DB, profileID, deviceID, key string) (*userstore.DeviceSettingEntry, error) {
	var entry userstore.DeviceSettingEntry
	err := db.QueryRow(
		`SELECT profile_id, device_id, device_name, device_platform, key, value, updated_at
		 FROM user_device_settings
		 WHERE profile_id = ? AND device_id = ? AND key = ?`,
		profileID, deviceID, key,
	).Scan(&entry.ProfileID, &entry.DeviceID, &entry.DeviceName, &entry.DevicePlatform, &entry.Key, &entry.Value, &entry.UpdatedAt)
	if err == sql.ErrNoRows {
		return nil, nil
	}
	if err != nil {
		return nil, fmt.Errorf("getting device setting %q for device %q: %w", key, deviceID, err)
	}
	return &entry, nil
}

func RegisterDevice(db *sql.DB, entry userstore.DeviceEntry) error {
	if strings.TrimSpace(entry.ProfileID) == "" || strings.TrimSpace(entry.DeviceID) == "" {
		return nil
	}
	lastSeenAt := time.Now().UTC().Format(time.RFC3339Nano)
	_, err := db.Exec(
		`INSERT INTO user_devices
			(profile_id, device_id, device_name, device_platform, last_seen_at)
		 VALUES (?, ?, ?, ?, ?)
		 ON CONFLICT(profile_id, device_id) DO UPDATE SET
			device_name = CASE
				WHEN excluded.device_name <> '' THEN excluded.device_name
				ELSE user_devices.device_name
			END,
			device_platform = CASE
				WHEN excluded.device_platform <> '' THEN excluded.device_platform
				ELSE user_devices.device_platform
			END,
			last_seen_at = excluded.last_seen_at`,
		entry.ProfileID, entry.DeviceID, entry.DeviceName, entry.DevicePlatform, lastSeenAt,
	)
	if err != nil {
		return fmt.Errorf("registering device %q: %w", entry.DeviceID, err)
	}
	return nil
}

func ListDevices(db *sql.DB) ([]userstore.DeviceEntry, error) {
	rows, err := db.Query(
		`SELECT profile_id, device_id, device_name, device_platform, last_seen_at
		 FROM user_devices
		 ORDER BY last_seen_at DESC, profile_id ASC, device_name ASC, device_id ASC`,
	)
	if err != nil {
		return nil, fmt.Errorf("listing devices: %w", err)
	}
	defer rows.Close()

	var entries []userstore.DeviceEntry
	for rows.Next() {
		var entry userstore.DeviceEntry
		if err := rows.Scan(&entry.ProfileID, &entry.DeviceID, &entry.DeviceName, &entry.DevicePlatform, &entry.LastSeenAt); err != nil {
			return nil, fmt.Errorf("scanning device: %w", err)
		}
		entries = append(entries, entry)
	}
	return entries, rows.Err()
}

// DeviceExists carries no user_id predicate: this backend keeps one database
// per user, so the table is already scoped to the account.
func DeviceExists(db *sql.DB, profileID, deviceID string) (bool, error) {
	if strings.TrimSpace(profileID) == "" || strings.TrimSpace(deviceID) == "" {
		return false, nil
	}
	var exists bool
	err := db.QueryRow(
		`SELECT EXISTS(
			SELECT 1 FROM user_devices
			WHERE profile_id = ? AND device_id = ?)`,
		profileID, deviceID,
	).Scan(&exists)
	if err != nil {
		return false, fmt.Errorf("checking device %q: %w", deviceID, err)
	}
	return exists, nil
}

func ForgetDevice(db *sql.DB, profileID, deviceID string) error {
	if strings.TrimSpace(profileID) == "" || strings.TrimSpace(deviceID) == "" {
		return nil
	}
	_, err := db.Exec(
		`DELETE FROM user_devices WHERE profile_id = ? AND device_id = ?`,
		profileID, deviceID,
	)
	if err != nil {
		return fmt.Errorf("forgetting device %q: %w", deviceID, err)
	}
	return nil
}

func SetDeviceSetting(db *sql.DB, entry userstore.DeviceSettingEntry) error {
	if err := RegisterDevice(db, userstore.DeviceEntry{
		ProfileID:      entry.ProfileID,
		DeviceID:       entry.DeviceID,
		DeviceName:     entry.DeviceName,
		DevicePlatform: entry.DevicePlatform,
	}); err != nil {
		return err
	}
	updatedAt := time.Now().UTC().Format(time.RFC3339Nano)
	_, err := db.Exec(
		`INSERT INTO user_device_settings
			(profile_id, device_id, key, value, device_name, device_platform, updated_at)
		 VALUES (?, ?, ?, ?, ?, ?, ?)
		 ON CONFLICT(profile_id, device_id, key) DO UPDATE SET
			value = excluded.value,
			device_name = excluded.device_name,
			device_platform = excluded.device_platform,
			updated_at = excluded.updated_at`,
		entry.ProfileID, entry.DeviceID, entry.Key, entry.Value, entry.DeviceName, entry.DevicePlatform, updatedAt,
	)
	if err != nil {
		return fmt.Errorf("setting device setting %q for device %q: %w", entry.Key, entry.DeviceID, err)
	}
	return nil
}

func DeleteDeviceSetting(db *sql.DB, profileID, deviceID, key string) error {
	_, err := db.Exec("DELETE FROM user_device_settings WHERE profile_id = ? AND device_id = ? AND key = ?", profileID, deviceID, key)
	if err != nil {
		return fmt.Errorf("deleting device setting %q for device %q: %w", key, deviceID, err)
	}
	return nil
}

// DeleteAllDeviceSettings clears everything one device holds for one profile.
// It is the forget-device path, so it drops the canonical profile_device values
// alongside the legacy string overrides: this database declares no foreign keys,
// so nothing else would.
func DeleteAllDeviceSettings(db *sql.DB, profileID, deviceID string) error {
	tx, err := db.Begin()
	if err != nil {
		return fmt.Errorf("beginning transaction to forget device %q: %w", deviceID, err)
	}
	defer tx.Rollback() //nolint:errcheck

	if _, err := tx.Exec(
		"DELETE FROM user_device_settings WHERE profile_id = ? AND device_id = ?",
		profileID, deviceID,
	); err != nil {
		return fmt.Errorf("deleting all device settings for device %q: %w", deviceID, err)
	}
	if _, err := tx.Exec(
		"DELETE FROM user_setting_values WHERE scope = 'profile_device' AND profile_id = ? AND device_id = ?",
		profileID, deviceID,
	); err != nil {
		return fmt.Errorf("deleting setting values for device %q: %w", deviceID, err)
	}
	return tx.Commit()
}

func DeleteDeviceSettingsByKey(db *sql.DB, key string) error {
	_, err := db.Exec("DELETE FROM user_device_settings WHERE key = ?", key)
	if err != nil {
		return fmt.Errorf("deleting device settings for key %q: %w", key, err)
	}
	return nil
}

func ListDeviceSettings(db *sql.DB, key string) ([]userstore.DeviceSettingEntry, error) {
	rows, err := db.Query(
		`SELECT profile_id, device_id, device_name, device_platform, key, value, updated_at
		 FROM user_device_settings
		 WHERE key = ?
		 ORDER BY updated_at DESC, profile_id ASC, device_name ASC, device_id ASC`,
		key,
	)
	if err != nil {
		return nil, fmt.Errorf("listing device settings for key %q: %w", key, err)
	}
	defer rows.Close()

	var entries []userstore.DeviceSettingEntry
	for rows.Next() {
		var entry userstore.DeviceSettingEntry
		if err := rows.Scan(&entry.ProfileID, &entry.DeviceID, &entry.DeviceName, &entry.DevicePlatform, &entry.Key, &entry.Value, &entry.UpdatedAt); err != nil {
			return nil, fmt.Errorf("scanning device setting: %w", err)
		}
		entries = append(entries, entry)
	}
	return entries, rows.Err()
}

func ListAllDeviceSettings(db *sql.DB) ([]userstore.DeviceSettingEntry, error) {
	rows, err := db.Query(
		`SELECT profile_id, device_id, device_name, device_platform, key, value, updated_at
		 FROM user_device_settings
		 ORDER BY updated_at DESC, profile_id ASC, key ASC, device_name ASC, device_id ASC`,
	)
	if err != nil {
		return nil, fmt.Errorf("listing all device settings: %w", err)
	}
	defer rows.Close()

	var entries []userstore.DeviceSettingEntry
	for rows.Next() {
		var entry userstore.DeviceSettingEntry
		if err := rows.Scan(&entry.ProfileID, &entry.DeviceID, &entry.DeviceName, &entry.DevicePlatform, &entry.Key, &entry.Value, &entry.UpdatedAt); err != nil {
			return nil, fmt.Errorf("scanning device setting: %w", err)
		}
		entries = append(entries, entry)
	}
	return entries, rows.Err()
}