navidrome/ui/src/common/useRating.jsx

import { useState, useCallback, useEffect, useRef } from 'react'
import { useDataProvider, useNotify } from 'react-admin'
import subsonic from '../subsonic'

export const useRating = (resource, record) => {
  const [loading, setLoading] = useState(false)
  const notify = useNotify()
  const dataProvider = useDataProvider()
  const mountedRef = useRef(false)
  const rating = record.rating

  useEffect(() => {
    mountedRef.current = true
    return () => {
      mountedRef.current = false
    }
  }, [])

  const refreshRating = useCallback(() => {
    // For playlist tracks, refresh both resources to keep data in sync
    if (record.mediaFileId) {
      // This is a playlist track - refresh both the playlist track and the song
      const promises = [
        dataProvider.getOne('song', { id: record.mediaFileId }),
        dataProvider.getOne('playlistTrack', {
          id: record.id,
          filter: { playlist_id: record.playlistId },
        }),
      ]

      Promise.all(promises)
        .catch((e) => {
          // eslint-disable-next-line no-console
          console.log('Error encountered: ' + e)
        })
        .finally(() => {
          if (mountedRef.current) {
            setLoading(false)
          }
        })
    } else {
      // Regular song or other resource
      dataProvider
        .getOne(resource, { id: record.id })
        .catch((e) => {
          // eslint-disable-next-line no-console
          console.log('Error encountered: ' + e)
        })
        .finally(() => {
          if (mountedRef.current) {
            setLoading(false)
          }
        })
    }
  }, [dataProvider, record.id, record.mediaFileId, record.playlistId, resource])

  const rate = (val, id) => {
    setLoading(true)
    subsonic
      .setRating(id, val)
      .then(refreshRating)
      .catch((e) => {
        // eslint-disable-next-line no-console
        console.log('Error setting star rating: ', e)
        notify('ra.page.error', 'warning')
        if (mountedRef.current) {
          setLoading(false)
        }
      })
  }

  return [rate, rating, loading]
}