Server changes/Sonarr

2026-07-27 21:06:51 +12:00
parent 62f6345a40
commit 8d6cf2f5a1
42 changed files with 2388 additions and 284 deletions
@@ -138,6 +138,51 @@ func (s *Store) LibraryCandidates(ctx context.Context, genres []string, limit in
 	return collectPayloads(rows)
 }

+// CuratedCandidates filters the imported catalogue for a server-authored shelf. Arrays
+// are matched case-insensitively because Emby studio capitalisation is not consistent.
+func (s *Store) CuratedCandidates(
+	ctx context.Context,
+	itemTypes, genres, studios []string,
+	limit int,
+) ([]json.RawMessage, error) {
+	if len(itemTypes) == 0 || (len(genres) == 0 && len(studios) == 0) {
+		return nil, nil
+	}
+	rows, err := s.pool.Query(ctx, `
+		SELECT payload
+		FROM library_items
+		WHERE type = ANY($1)
+		  AND (
+			cardinality($2::text[]) = 0 OR
+			EXISTS (
+				SELECT 1 FROM unnest(genres) AS genre
+				WHERE lower(genre) = ANY($2)
+			)
+		  )
+		  AND (
+			cardinality($3::text[]) = 0 OR
+			EXISTS (
+				SELECT 1 FROM unnest(studios) AS studio
+				WHERE lower(studio) = ANY($3)
+			)
+		  )
+		ORDER BY community_rating DESC NULLS LAST, date_created DESC NULLS LAST
+		LIMIT $4`,
+		itemTypes, lowerStrings(genres), lowerStrings(studios), limit)
+	if err != nil {
+		return nil, fmt.Errorf("store: curated candidates: %w", err)
+	}
+	return collectPayloads(rows)
+}
+
+func lowerStrings(values []string) []string {
+	out := make([]string, 0, len(values))
+	for _, value := range values {
+		out = append(out, strings.ToLower(strings.TrimSpace(value)))
+	}
+	return out
+}
+
 func (s *Store) LibraryStats(ctx context.Context) (LibraryStats, error) {
 	stats := LibraryStats{ByType: map[string]int64{}}

@@ -10,12 +10,28 @@ CREATE TABLE IF NOT EXISTS sessions (
    username     TEXT        NOT NULL,
    server_id    TEXT        NOT NULL DEFAULT '',
    device_id    TEXT        NOT NULL DEFAULT '',
+    device_name  TEXT        NOT NULL DEFAULT 'Memby TV',
    created_at   TIMESTAMPTZ NOT NULL DEFAULT now(),
    last_seen_at TIMESTAMPTZ NOT NULL DEFAULT now()
 );

+ALTER TABLE sessions ADD COLUMN IF NOT EXISTS device_name TEXT NOT NULL DEFAULT 'Memby TV';
+
+-- Older builds could create more than one token for the same physical TV. Keep the most
+-- recently used row before adding the identity constraint.
+DELETE FROM sessions older
+USING sessions newer
+WHERE older.emby_user_id = newer.emby_user_id
+  AND older.device_id = newer.device_id
+  AND (
+      older.last_seen_at < newer.last_seen_at
+      OR (older.last_seen_at = newer.last_seen_at AND older.token_hash < newer.token_hash)
+  );
+
 CREATE INDEX IF NOT EXISTS sessions_emby_user_idx ON sessions (emby_user_id);
 CREATE INDEX IF NOT EXISTS sessions_last_seen_idx ON sessions (last_seen_at);
+CREATE UNIQUE INDEX IF NOT EXISTS sessions_user_device_idx
+    ON sessions (emby_user_id, device_id);

 -- The imported library.
 --
@@ -17,6 +17,7 @@ var schema string

 // ErrNotFound is returned when a token does not match a live session.
 var ErrNotFound = errors.New("store: session not found")
+var ErrDeviceLimit = errors.New("store: device limit reached")

 type Session struct {
 	TokenHash  []byte
@@ -25,6 +26,7 @@ type Session struct {
 	Username   string
 	ServerID   string
 	DeviceID   string
+	DeviceName string
 	LastSeenAt time.Time
 }

@@ -56,30 +58,73 @@ func (s *Store) Migrate(ctx context.Context) error {
 	return nil
 }

-func (s *Store) CreateSession(ctx context.Context, sess Session) error {
-	_, err := s.pool.Exec(ctx, `
-		INSERT INTO sessions (token_hash, emby_user_id, emby_token, username, server_id, device_id)
-		VALUES ($1, $2, $3, $4, $5, $6)
-		ON CONFLICT (token_hash) DO UPDATE SET
+// CreateSession enforces a user's device allowance under a per-user transaction lock.
+// Re-authenticating the same stable device replaces its token and never consumes a slot.
+// The replaced hash is returned so its Redis entry can be invalidated immediately.
+func (s *Store) CreateSession(ctx context.Context, sess Session, maxClients int) ([]byte, int, error) {
+	tx, err := s.pool.Begin(ctx)
+	if err != nil {
+		return nil, 0, fmt.Errorf("store: begin session: %w", err)
+	}
+	defer func() { _ = tx.Rollback(ctx) }()
+
+	if _, err := tx.Exec(ctx, `SELECT pg_advisory_xact_lock(hashtextextended($1, 0))`, sess.EmbyUserID); err != nil {
+		return nil, 0, fmt.Errorf("store: lock user sessions: %w", err)
+	}
+
+	var previousHash []byte
+	err = tx.QueryRow(ctx, `
+		SELECT token_hash FROM sessions
+		WHERE emby_user_id = $1 AND device_id = $2`,
+		sess.EmbyUserID, sess.DeviceID).Scan(&previousHash)
+	if err != nil && !errors.Is(err, pgx.ErrNoRows) {
+		return nil, 0, fmt.Errorf("store: find device session: %w", err)
+	}
+	existingDevice := err == nil
+
+	var activeClients int
+	if err := tx.QueryRow(ctx,
+		`SELECT count(*) FROM sessions WHERE emby_user_id = $1`,
+		sess.EmbyUserID).Scan(&activeClients); err != nil {
+		return nil, 0, fmt.Errorf("store: count user sessions: %w", err)
+	}
+	if !existingDevice && activeClients >= maxClients {
+		return nil, activeClients, ErrDeviceLimit
+	}
+
+	_, err = tx.Exec(ctx, `
+		INSERT INTO sessions (
+			token_hash, emby_user_id, emby_token, username, server_id, device_id, device_name
+		)
+		VALUES ($1, $2, $3, $4, $5, $6, $7)
+		ON CONFLICT (emby_user_id, device_id) DO UPDATE SET
+			token_hash   = EXCLUDED.token_hash,
 			emby_token   = EXCLUDED.emby_token,
 			username     = EXCLUDED.username,
 			server_id    = EXCLUDED.server_id,
-			device_id    = EXCLUDED.device_id,
+			device_name  = EXCLUDED.device_name,
 			last_seen_at = now()`,
-		sess.TokenHash, sess.EmbyUserID, sess.EmbyToken, sess.Username, sess.ServerID, sess.DeviceID)
+		sess.TokenHash, sess.EmbyUserID, sess.EmbyToken, sess.Username,
+		sess.ServerID, sess.DeviceID, sess.DeviceName)
 	if err != nil {
-		return fmt.Errorf("store: create session: %w", err)
+		return nil, 0, fmt.Errorf("store: create session: %w", err)
 	}
-	return nil
+	if !existingDevice {
+		activeClients++
+	}
+	if err := tx.Commit(ctx); err != nil {
+		return nil, 0, fmt.Errorf("store: commit session: %w", err)
+	}
+	return previousHash, activeClients, nil
 }

 func (s *Store) SessionByTokenHash(ctx context.Context, hash []byte) (Session, error) {
 	var sess Session
 	err := s.pool.QueryRow(ctx, `
-		SELECT token_hash, emby_user_id, emby_token, username, server_id, device_id, last_seen_at
+		SELECT token_hash, emby_user_id, emby_token, username, server_id, device_id, device_name, last_seen_at
 		FROM sessions WHERE token_hash = $1`, hash).
 		Scan(&sess.TokenHash, &sess.EmbyUserID, &sess.EmbyToken, &sess.Username,
-			&sess.ServerID, &sess.DeviceID, &sess.LastSeenAt)
+			&sess.ServerID, &sess.DeviceID, &sess.DeviceName, &sess.LastSeenAt)
 	if errors.Is(err, pgx.ErrNoRows) {
 		return Session{}, ErrNotFound
 	}
@@ -111,3 +156,51 @@ func (s *Store) DeleteIdleSessions(ctx context.Context, idle time.Duration) (int
 	}
 	return tag.RowsAffected(), nil
 }
+
+// TrimSessionsToLimit brings data created under an older, more generous policy back
+// within the current allowance. The most recently active devices survive.
+func (s *Store) TrimSessionsToLimit(ctx context.Context, maxClients int) ([]Session, error) {
+	rows, err := s.pool.Query(ctx, `
+		WITH ranked AS (
+			SELECT token_hash,
+				row_number() OVER (
+					PARTITION BY emby_user_id
+					ORDER BY last_seen_at DESC, created_at DESC, token_hash DESC
+				) AS device_rank
+			FROM sessions
+		),
+		retired AS (
+			DELETE FROM sessions current
+			USING ranked
+			WHERE current.token_hash = ranked.token_hash
+			  AND ranked.device_rank > $1
+			RETURNING current.token_hash, current.emby_user_id, current.emby_token,
+				current.username, current.server_id, current.device_id,
+				current.device_name, current.last_seen_at
+		)
+		SELECT token_hash, emby_user_id, emby_token, username, server_id,
+			device_id, device_name, last_seen_at
+		FROM retired`,
+		maxClients,
+	)
+	if err != nil {
+		return nil, fmt.Errorf("store: trim sessions: %w", err)
+	}
+	defer rows.Close()
+
+	var retired []Session
+	for rows.Next() {
+		var sess Session
+		if err := rows.Scan(
+			&sess.TokenHash, &sess.EmbyUserID, &sess.EmbyToken, &sess.Username,
+			&sess.ServerID, &sess.DeviceID, &sess.DeviceName, &sess.LastSeenAt,
+		); err != nil {
+			return nil, fmt.Errorf("store: scan trimmed session: %w", err)
+		}
+		retired = append(retired, sess)
+	}
+	if err := rows.Err(); err != nil {
+		return nil, fmt.Errorf("store: trim sessions rows: %w", err)
+	}
+	return retired, nil
+}