subtracks/app/hooks/music.ts

import { selectCache } from '@app/state/cache'
import { selectMusic } from '@app/state/music'
import { Store, useStore } from '@app/state/store'
import { useCallback, useEffect } from 'react'

export const useArtistInfo = (id: string) => {
  const artistInfo = useStore(useCallback((state: Store) => state.artistInfo[id], [id]))
  const fetchArtistInfo = useStore(selectMusic.fetchArtistInfo)

  useEffect(() => {
    if (!artistInfo) {
      fetchArtistInfo(id)
    }
  })

  return artistInfo
}

export const useAlbumWithSongs = (id: string) => {
  const album = useStore(useCallback((state: Store) => state.albumsWithSongs[id], [id]))
  const fetchAlbum = useStore(selectMusic.fetchAlbumWithSongs)

  useEffect(() => {
    if (!album) {
      fetchAlbum(id)
    }
  })

  return album
}

export const usePlaylistWithSongs = (id: string) => {
  const playlist = useStore(useCallback((state: Store) => state.playlistsWithSongs[id], [id]))
  const fetchPlaylist = useStore(selectMusic.fetchPlaylistWithSongs)

  useEffect(() => {
    if (!playlist) {
      fetchPlaylist(id)
    }
  })

  return playlist
}

export const useStarred = (id: string, type: string) => {
  return useStore(
    useCallback(
      (state: Store) => {
        switch (type) {
          case 'song':
            return state.starredSongs[id]
          case 'album':
            return state.starredAlbums[id]
          case 'artist':
            return state.starredArtists[id]
          default:
            return false
        }
      },
      [type, id],
    ),
  )
}

export const useCoverArtFile = (coverArt: string = '-1') => {
  const file = useStore(useCallback((state: Store) => state.cachedCoverArt[coverArt], [coverArt]))
  const cacheCoverArt = useStore(selectCache.cacheCoverArt)

  useEffect(() => {
    if (!file) {
      cacheCoverArt(coverArt)
    }
  })

  return file
}

export const useArtistCoverArtFile = (artistId: string) => {
  const artistInfo = useArtistInfo(artistId)
  const file = useStore(useCallback((state: Store) => state.cachedArtistArt[artistId], [artistId]))
  const cacheArtistArt = useStore(selectCache.cacheArtistArt)

  useEffect(() => {
    if (!file && artistInfo) {
      cacheArtistArt(artistId, artistInfo.largeImageUrl)
    }
  })

  return file
}