26 lines
792 B
Go
26 lines
792 B
Go
|
package store
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"fmt"
|
||
|
|
)
|
||
|
|
|
||
|
|
func (s *Store) UserEnabled(ctx context.Context, userID string) (bool, error) {
|
||
|
|
var enabled bool
|
||
|
|
err := s.pool.QueryRow(ctx, `SELECT COALESCE((SELECT enabled FROM user_controls WHERE emby_user_id = $1), true)`, userID).Scan(&enabled)
|
||
|
|
if err != nil {
|
||
|
|
return false, fmt.Errorf("store: read user access: %w", err)
|
||
|
|
}
|
||
|
|
return enabled, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *Store) SetUserEnabled(ctx context.Context, userID string, enabled bool) error {
|
||
|
|
_, err := s.pool.Exec(ctx, `
|
||
|
|
INSERT INTO user_controls (emby_user_id, enabled, updated_at) VALUES ($1, $2, now())
|
||
|
|
ON CONFLICT (emby_user_id) DO UPDATE SET enabled = EXCLUDED.enabled, updated_at = now()`, userID, enabled)
|
||
|
|
if err != nil {
|
||
|
|
return fmt.Errorf("store: set user access: %w", err)
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}
|