move track player hooks and mapping into state

This commit is contained in:
austinried 2021-08-17 13:27:24 +09:00
parent f18e4fc811
commit d068288391
12 changed files with 245 additions and 266 deletions

View File

@ -135,7 +135,9 @@ const ListItem: React.FC<{
const starItem = useStore(selectMusic.starItem) const starItem = useStore(selectMusic.starItem)
const toggleStarred = useCallback(() => { const toggleStarred = useCallback(() => {
if (item.itemType !== 'playlist') {
starItem(item.id, item.itemType, starred) starItem(item.id, item.itemType, starred)
}
}, [item.id, item.itemType, starItem, starred]) }, [item.id, item.itemType, starItem, starred])
let title = <></> let title = <></>

View File

@ -1,7 +1,7 @@
import Button from '@app/components/Button' import Button from '@app/components/Button'
import { useSetQueue } from '@app/hooks/trackplayer'
import { Song } from '@app/models/music' import { Song } from '@app/models/music'
import { QueueContextType } from '@app/state/trackplayer' import { useStore } from '@app/state/store'
import { QueueContextType, selectTrackPlayer } from '@app/state/trackplayer'
import colors from '@app/styles/colors' import colors from '@app/styles/colors'
import React, { useState } from 'react' import React, { useState } from 'react'
import { StyleProp, StyleSheet, View, ViewStyle } from 'react-native' import { StyleProp, StyleSheet, View, ViewStyle } from 'react-native'
@ -17,7 +17,7 @@ const ListPlayerControls = React.memo<{
style?: StyleProp<ViewStyle> style?: StyleProp<ViewStyle>
}>(({ songs, typeName, queueName, queueContextType, queueContextId, style }) => { }>(({ songs, typeName, queueName, queueContextType, queueContextId, style }) => {
const [downloaded, setDownloaded] = useState(false) const [downloaded, setDownloaded] = useState(false)
const setQueue = useSetQueue() const setQueue = useStore(selectTrackPlayer.setQueue)
return ( return (
<View style={[styles.controls, style]}> <View style={[styles.controls, style]}>

View File

@ -1,17 +1,6 @@
import { Song } from '@app/models/music'
import { selectCache } from '@app/state/cache'
import { useStore } from '@app/state/store' import { useStore } from '@app/state/store'
import { import { getQueue, selectTrackPlayer, trackPlayerCommands } from '@app/state/trackplayer'
getCurrentTrack, import TrackPlayer from 'react-native-track-player'
getQueue,
getRepeatMode,
QueueContextType,
selectTrackPlayer,
TrackExt,
trackPlayerCommands,
} from '@app/state/trackplayer'
import { useCallback } from 'react'
import TrackPlayer, { RepeatMode } from 'react-native-track-player'
export const usePlay = () => { export const usePlay = () => {
return () => trackPlayerCommands.enqueue(() => TrackPlayer.play()) return () => trackPlayerCommands.enqueue(() => TrackPlayer.play())
@ -67,33 +56,8 @@ export const useSeekTo = () => {
}) })
} }
export const useToggleRepeat = () => {
const setRepeatMode = useStore(selectTrackPlayer.setRepeatMode)
return () =>
trackPlayerCommands.enqueue(async () => {
const repeatMode = await getRepeatMode()
let nextMode = RepeatMode.Off
switch (repeatMode) {
case RepeatMode.Off:
nextMode = RepeatMode.Queue
break
case RepeatMode.Queue:
nextMode = RepeatMode.Track
break
default:
nextMode = RepeatMode.Off
break
}
await TrackPlayer.setRepeatMode(nextMode)
setRepeatMode(nextMode)
})
}
export const useReset = (enqueue = true) => { export const useReset = (enqueue = true) => {
const resetStore = useStore(selectTrackPlayer.reset) const resetStore = useStore(selectTrackPlayer.resetTrackPlayerState)
const reset = async () => { const reset = async () => {
await TrackPlayer.reset() await TrackPlayer.reset()
@ -103,166 +67,6 @@ export const useReset = (enqueue = true) => {
return enqueue ? () => trackPlayerCommands.enqueue(reset) : reset return enqueue ? () => trackPlayerCommands.enqueue(reset) : reset
} }
function shuffleTracks(tracks: TrackExt[], firstTrack?: number) {
if (tracks.length === 0) {
return { tracks, shuffleOrder: [] }
}
const trackIndexes = tracks.map((_t, i) => i)
let shuffleOrder: number[] = []
for (let i = trackIndexes.length; i--; i > 0) {
const randi = Math.floor(Math.random() * (i + 1))
shuffleOrder.push(trackIndexes.splice(randi, 1)[0])
}
if (firstTrack !== undefined) {
shuffleOrder.splice(shuffleOrder.indexOf(firstTrack), 1)
shuffleOrder = [firstTrack, ...shuffleOrder]
}
tracks = shuffleOrder.map(i => tracks[i])
return { tracks, shuffleOrder }
}
function unshuffleTracks(tracks: TrackExt[], shuffleOrder: number[]): TrackExt[] {
if (tracks.length === 0 || shuffleOrder.length === 0) {
return tracks
}
return shuffleOrder.map((_v, i) => tracks[shuffleOrder.indexOf(i)])
}
export const useToggleShuffle = () => {
const setCurrentTrackIdx = useStore(selectTrackPlayer.setCurrentTrackIdx)
const setQueue = useStore(selectTrackPlayer.setQueue)
const setShuffleOrder = useStore(selectTrackPlayer.setShuffleOrder)
const getShuffleOrder = useCallback(() => useStore.getState().shuffleOrder, [])
const setProgress = useStore(selectTrackPlayer.setProgress)
const getProgress = useCallback(() => useStore.getState().progress, [])
return async () => {
return trackPlayerCommands.enqueue(async () => {
const queue = await getQueue()
const current = await getCurrentTrack()
const queueShuffleOrder = getShuffleOrder()
const progress = getProgress()
await TrackPlayer.remove(queue.map((_t, i) => i).filter(i => i !== current))
if (queueShuffleOrder === undefined) {
let { tracks, shuffleOrder } = shuffleTracks(queue, current)
if (tracks.length > 0) {
tracks = tracks.slice(1)
}
await TrackPlayer.add(tracks)
setShuffleOrder(shuffleOrder)
} else {
const tracks = unshuffleTracks(queue, queueShuffleOrder)
if (current !== undefined) {
const shuffledCurrent = queueShuffleOrder[current]
const tracks1 = tracks.slice(0, shuffledCurrent)
const tracks2 = tracks.slice(shuffledCurrent + 1)
await TrackPlayer.add(tracks2)
await TrackPlayer.add(tracks1, 0)
} else {
await TrackPlayer.add(tracks)
}
setShuffleOrder(undefined)
}
setQueue(await getQueue())
setCurrentTrackIdx(await getCurrentTrack())
setProgress(progress)
})
}
}
export const useSetQueue = () => {
const setCurrentTrackIdx = useStore(selectTrackPlayer.setCurrentTrackIdx)
const setQueue = useStore(selectTrackPlayer.setQueue)
const setShuffleOrder = useStore(selectTrackPlayer.setShuffleOrder)
const setQueueName = useStore(selectTrackPlayer.setQueueName)
const getQueueShuffled = useCallback(() => !!useStore.getState().shuffleOrder, [])
const setQueueContextType = useStore(selectTrackPlayer.setQueueContextType)
const setQueueContextId = useStore(selectTrackPlayer.setQueueContextId)
const fetchCoverArtFilePath = useStore(selectCache.fetchCoverArtFilePath)
const buildStreamUri = useStore(selectTrackPlayer.buildStreamUri)
return async (
songs: Song[],
name: string,
contextType: QueueContextType,
contextId: string,
playTrack?: number,
shuffle?: boolean,
) =>
trackPlayerCommands.enqueue(async () => {
const shuffled = shuffle !== undefined ? shuffle : getQueueShuffled()
await TrackPlayer.setupPlayer()
await TrackPlayer.reset()
if (songs.length === 0) {
return
}
const coverArtPaths: { [coverArt: string]: string | undefined } = {}
for (const s of songs) {
if (!s.coverArt) {
continue
}
coverArtPaths[s.coverArt] = await fetchCoverArtFilePath(s.coverArt)
}
let queue = songs.map(s => mapSongToTrack(s, coverArtPaths))
try {
for (const t of queue) {
t.url = buildStreamUri(t.id)
}
} catch {
return
}
if (shuffled) {
const { tracks, shuffleOrder } = shuffleTracks(queue, playTrack)
setShuffleOrder(shuffleOrder)
queue = tracks
playTrack = 0
} else {
setShuffleOrder(undefined)
}
playTrack = playTrack || 0
setQueue(queue)
setCurrentTrackIdx(playTrack)
setQueueName(name)
setQueueContextType(contextType)
setQueueContextId(contextId)
if (playTrack === 0) {
await TrackPlayer.add(queue)
await TrackPlayer.play()
} else {
const tracks1 = queue.slice(0, playTrack)
const tracks2 = queue.slice(playTrack)
await TrackPlayer.add(tracks2)
await TrackPlayer.play()
await TrackPlayer.add(tracks1, 0)
}
})
}
export const useIsPlaying = (contextId: string | undefined, track: number) => { export const useIsPlaying = (contextId: string | undefined, track: number) => {
const queueContextId = useStore(selectTrackPlayer.queueContextId) const queueContextId = useStore(selectTrackPlayer.queueContextId)
const currentTrackIdx = useStore(selectTrackPlayer.currentTrackIdx) const currentTrackIdx = useStore(selectTrackPlayer.currentTrackIdx)
@ -279,34 +83,3 @@ export const useIsPlaying = (contextId: string | undefined, track: number) => {
return contextId === queueContextId && track === currentTrackIdx return contextId === queueContextId && track === currentTrackIdx
} }
function mapSongToTrack(song: Song, coverArtPaths: { [coverArt: string]: string | undefined }): TrackExt {
return {
id: song.id,
title: song.title,
artist: song.artist || 'Unknown Artist',
album: song.album || 'Unknown Album',
url: song.streamUri,
artwork:
song.coverArt && coverArtPaths[song.coverArt] ? coverArtPaths[song.coverArt] : require('@res/fallback.png'),
coverArt: song.coverArt,
duration: song.duration,
artistId: song.artistId,
albumId: song.albumId,
}
}
export function mapTrackExtToSong(track: TrackExt): Song {
return {
itemType: 'song',
id: track.id,
title: track.title as string,
artist: track.artist,
album: track.album,
streamUri: track.url as string,
coverArt: track.coverArt,
duration: track.duration,
artistId: track.artistId,
albumId: track.albumId,
}
}

View File

@ -6,7 +6,7 @@ import NetInfo, { NetInfoStateType } from '@react-native-community/netinfo'
const reset = () => { const reset = () => {
unstable_batchedUpdates(() => { unstable_batchedUpdates(() => {
useStore.getState().reset() useStore.getState().resetTrackPlayerState()
}) })
} }

View File

@ -6,8 +6,9 @@ import Header from '@app/components/Header'
import HeaderBar from '@app/components/HeaderBar' import HeaderBar from '@app/components/HeaderBar'
import ListItem from '@app/components/ListItem' import ListItem from '@app/components/ListItem'
import { useArtistInfo } from '@app/hooks/music' import { useArtistInfo } from '@app/hooks/music'
import { useSetQueue } from '@app/hooks/trackplayer'
import { Album, Song } from '@app/models/music' import { Album, Song } from '@app/models/music'
import { useStore } from '@app/state/store'
import { selectTrackPlayer } from '@app/state/trackplayer'
import colors from '@app/styles/colors' import colors from '@app/styles/colors'
import dimensions from '@app/styles/dimensions' import dimensions from '@app/styles/dimensions'
import font from '@app/styles/font' import font from '@app/styles/font'
@ -52,7 +53,7 @@ const TopSongs = React.memo<{
name: string name: string
artistId: string artistId: string
}>(({ songs, name, artistId }) => { }>(({ songs, name, artistId }) => {
const setQueue = useSetQueue() const setQueue = useStore(selectTrackPlayer.setQueue)
return ( return (
<> <>

View File

@ -1,14 +1,16 @@
import GradientScrollView from '@app/components/GradientScrollView' import GradientScrollView from '@app/components/GradientScrollView'
import ListItem from '@app/components/ListItem' import ListItem from '@app/components/ListItem'
import NowPlayingBar from '@app/components/NowPlayingBar' import NowPlayingBar from '@app/components/NowPlayingBar'
import { mapTrackExtToSong, useSkipTo } from '@app/hooks/trackplayer' import { useSkipTo } from '@app/hooks/trackplayer'
import { useStore } from '@app/state/store' import { useStore } from '@app/state/store'
import { selectTrackPlayer } from '@app/state/trackplayer' import { selectTrackPlayer } from '@app/state/trackplayer'
import { selectTrackPlayerMap } from '@app/state/trackplayermap'
import React from 'react' import React from 'react'
import { StyleSheet, View } from 'react-native' import { StyleSheet, View } from 'react-native'
const NowPlayingQueue = React.memo<{}>(() => { const NowPlayingQueue = React.memo<{}>(() => {
const queue = useStore(selectTrackPlayer.queue) const queue = useStore(selectTrackPlayer.queue)
const mapTrackExtToSong = useStore(selectTrackPlayerMap.mapTrackExtToSong)
const skipTo = useSkipTo() const skipTo = useSkipTo()
return ( return (

View File

@ -4,19 +4,11 @@ import ImageGradientBackground from '@app/components/ImageGradientBackground'
import PressableOpacity from '@app/components/PressableOpacity' import PressableOpacity from '@app/components/PressableOpacity'
import Star from '@app/components/Star' import Star from '@app/components/Star'
import { useStarred } from '@app/hooks/music' import { useStarred } from '@app/hooks/music'
import { import { useNext, usePause, usePlay, usePrevious, useSeekTo } from '@app/hooks/trackplayer'
mapTrackExtToSong,
useNext,
usePause,
usePlay,
usePrevious,
useSeekTo,
useToggleRepeat,
useToggleShuffle,
} from '@app/hooks/trackplayer'
import { selectMusic } from '@app/state/music' import { selectMusic } from '@app/state/music'
import { useStore } from '@app/state/store' import { useStore } from '@app/state/store'
import { QueueContextType, selectTrackPlayer, TrackExt } from '@app/state/trackplayer' import { QueueContextType, selectTrackPlayer, TrackExt } from '@app/state/trackplayer'
import { selectTrackPlayerMap } from '@app/state/trackplayermap'
import colors from '@app/styles/colors' import colors from '@app/styles/colors'
import font from '@app/styles/font' import font from '@app/styles/font'
import formatDuration from '@app/util/formatDuration' import formatDuration from '@app/util/formatDuration'
@ -51,6 +43,7 @@ const NowPlayingHeader = React.memo<{
}>(({ track }) => { }>(({ track }) => {
const queueName = useStore(selectTrackPlayer.queueName) const queueName = useStore(selectTrackPlayer.queueName)
const queueContextType = useStore(selectTrackPlayer.queueContextType) const queueContextType = useStore(selectTrackPlayer.queueContextType)
const mapTrackExtToSong = useStore(selectTrackPlayerMap.mapTrackExtToSong)
if (!track) { if (!track) {
return <></> return <></>
@ -272,9 +265,9 @@ const PlayerControls = () => {
const next = useNext() const next = useNext()
const previous = usePrevious() const previous = usePrevious()
const shuffled = useStore(selectTrackPlayer.shuffled) const shuffled = useStore(selectTrackPlayer.shuffled)
const toggleShuffle = useToggleShuffle() const toggleShuffle = useStore(selectTrackPlayer.toggleShuffle)
const repeatMode = useStore(selectTrackPlayer.repeatMode) const repeatMode = useStore(selectTrackPlayer.repeatMode)
const toggleRepeat = useToggleRepeat() const toggleRepeat = useStore(selectTrackPlayer.toggleRepeatMode)
const navigation = useNavigation() const navigation = useNavigation()
let playPauseIcon: string let playPauseIcon: string

View File

@ -3,10 +3,10 @@ import Header from '@app/components/Header'
import ListItem from '@app/components/ListItem' import ListItem from '@app/components/ListItem'
import NothingHere from '@app/components/NothingHere' import NothingHere from '@app/components/NothingHere'
import { useActiveListRefresh2 } from '@app/hooks/server' import { useActiveListRefresh2 } from '@app/hooks/server'
import { useSetQueue } from '@app/hooks/trackplayer'
import { ListableItem, SearchResults, Song } from '@app/models/music' import { ListableItem, SearchResults, Song } from '@app/models/music'
import { selectMusic } from '@app/state/music' import { selectMusic } from '@app/state/music'
import { useStore } from '@app/state/store' import { useStore } from '@app/state/store'
import { selectTrackPlayer } from '@app/state/trackplayer'
import colors from '@app/styles/colors' import colors from '@app/styles/colors'
import font from '@app/styles/font' import font from '@app/styles/font'
import debounce from 'lodash.debounce' import debounce from 'lodash.debounce'
@ -14,7 +14,7 @@ import React, { useCallback, useMemo, useState } from 'react'
import { ActivityIndicator, StatusBar, StyleSheet, TextInput, View } from 'react-native' import { ActivityIndicator, StatusBar, StyleSheet, TextInput, View } from 'react-native'
const SongItem = React.memo<{ item: Song }>(({ item }) => { const SongItem = React.memo<{ item: Song }>(({ item }) => {
const setQueue = useSetQueue() const setQueue = useStore(selectTrackPlayer.setQueue)
return ( return (
<ListItem <ListItem

View File

@ -5,8 +5,9 @@ import ListItem from '@app/components/ListItem'
import ListPlayerControls from '@app/components/ListPlayerControls' import ListPlayerControls from '@app/components/ListPlayerControls'
import NothingHere from '@app/components/NothingHere' import NothingHere from '@app/components/NothingHere'
import { useAlbumWithSongs, useCoverArtFile, usePlaylistWithSongs } from '@app/hooks/music' import { useAlbumWithSongs, useCoverArtFile, usePlaylistWithSongs } from '@app/hooks/music'
import { useSetQueue } from '@app/hooks/trackplayer'
import { AlbumWithSongs, PlaylistWithSongs, Song } from '@app/models/music' import { AlbumWithSongs, PlaylistWithSongs, Song } from '@app/models/music'
import { useStore } from '@app/state/store'
import { selectTrackPlayer } from '@app/state/trackplayer'
import colors from '@app/styles/colors' import colors from '@app/styles/colors'
import font from '@app/styles/font' import font from '@app/styles/font'
import { useNavigation } from '@react-navigation/native' import { useNavigation } from '@react-navigation/native'
@ -27,7 +28,7 @@ const Songs = React.memo<{
type: SongListType type: SongListType
itemId: string itemId: string
}>(({ songs, name, type, itemId }) => { }>(({ songs, name, type, itemId }) => {
const setQueue = useSetQueue() const setQueue = useStore(selectTrackPlayer.setQueue)
const _songs = [...songs] const _songs = [...songs]
let typeName = '' let typeName = ''

View File

@ -6,11 +6,13 @@ import { persist, StateStorage } from 'zustand/middleware'
import { CacheSlice, createCacheSlice } from './cache' import { CacheSlice, createCacheSlice } from './cache'
import { createMusicMapSlice, MusicMapSlice } from './musicmap' import { createMusicMapSlice, MusicMapSlice } from './musicmap'
import { createTrackPlayerSlice, TrackPlayerSlice } from './trackplayer' import { createTrackPlayerSlice, TrackPlayerSlice } from './trackplayer'
import { createTrackPlayerMapSlice, TrackPlayerMapSlice } from './trackplayermap'
export type Store = SettingsSlice & export type Store = SettingsSlice &
MusicSlice & MusicSlice &
MusicMapSlice & MusicMapSlice &
TrackPlayerSlice & TrackPlayerSlice &
TrackPlayerMapSlice &
CacheSlice & { CacheSlice & {
hydrated: boolean hydrated: boolean
setHydrated: (hydrated: boolean) => void setHydrated: (hydrated: boolean) => void
@ -41,6 +43,7 @@ export const useStore = create<Store>(
...createMusicSlice(set, get), ...createMusicSlice(set, get),
...createMusicMapSlice(set, get), ...createMusicMapSlice(set, get),
...createTrackPlayerSlice(set, get), ...createTrackPlayerSlice(set, get),
...createTrackPlayerMapSlice(set, get),
...createCacheSlice(set, get), ...createCacheSlice(set, get),
hydrated: false, hydrated: false,

View File

@ -1,7 +1,7 @@
import { NoClientError } from '@app/models/error' import { NoClientError } from '@app/models/error'
import { Song } from '@app/models/music'
import PromiseQueue from '@app/util/PromiseQueue' import PromiseQueue from '@app/util/PromiseQueue'
import produce from 'immer' import produce from 'immer'
import { ToastAndroid } from 'react-native'
import TrackPlayer, { RepeatMode, State, Track } from 'react-native-track-player' import TrackPlayer, { RepeatMode, State, Track } from 'react-native-track-player'
import { GetState, SetState } from 'zustand' import { GetState, SetState } from 'zustand'
import { Store } from './store' import { Store } from './store'
@ -32,10 +32,10 @@ export type TrackPlayerSlice = {
setQueueContextId: (queueContextId?: string) => void setQueueContextId: (queueContextId?: string) => void
shuffleOrder?: number[] shuffleOrder?: number[]
setShuffleOrder: (shuffleOrder?: number[]) => void toggleShuffle: () => Promise<void>
repeatMode: RepeatMode repeatMode: RepeatMode
setRepeatMode: (repeatMode: RepeatMode) => void toggleRepeatMode: () => Promise<void>
playerState: State playerState: State
setPlayerState: (playerState: State) => void setPlayerState: (playerState: State) => void
@ -45,7 +45,14 @@ export type TrackPlayerSlice = {
setCurrentTrackIdx: (idx?: number) => void setCurrentTrackIdx: (idx?: number) => void
queue: TrackExt[] queue: TrackExt[]
setQueue: (queue: TrackExt[]) => void setQueue: (
songs: Song[],
name: string,
contextType: QueueContextType,
contextId: string,
playTrack?: number,
shuffle?: boolean,
) => Promise<void>
progress: Progress progress: Progress
setProgress: (progress: Progress) => void setProgress: (progress: Progress) => void
@ -57,7 +64,7 @@ export type TrackPlayerSlice = {
rebuildQueue: () => Promise<void> rebuildQueue: () => Promise<void>
buildStreamUri: (id: string) => string buildStreamUri: (id: string) => string
reset: () => void resetTrackPlayerState: () => void
} }
export const selectTrackPlayer = { export const selectTrackPlayer = {
@ -71,11 +78,11 @@ export const selectTrackPlayer = {
setQueueContextId: (store: TrackPlayerSlice) => store.setQueueContextId, setQueueContextId: (store: TrackPlayerSlice) => store.setQueueContextId,
shuffleOrder: (store: TrackPlayerSlice) => store.shuffleOrder, shuffleOrder: (store: TrackPlayerSlice) => store.shuffleOrder,
setShuffleOrder: (store: TrackPlayerSlice) => store.setShuffleOrder,
shuffled: (store: TrackPlayerSlice) => !!store.shuffleOrder, shuffled: (store: TrackPlayerSlice) => !!store.shuffleOrder,
toggleShuffle: (store: TrackPlayerSlice) => store.toggleShuffle,
repeatMode: (store: TrackPlayerSlice) => store.repeatMode, repeatMode: (store: TrackPlayerSlice) => store.repeatMode,
setRepeatMode: (store: TrackPlayerSlice) => store.setRepeatMode, toggleRepeatMode: (store: TrackPlayerSlice) => store.toggleRepeatMode,
playerState: (store: TrackPlayerSlice) => store.playerState, playerState: (store: TrackPlayerSlice) => store.playerState,
setPlayerState: (store: TrackPlayerSlice) => store.setPlayerState, setPlayerState: (store: TrackPlayerSlice) => store.setPlayerState,
@ -95,7 +102,7 @@ export const selectTrackPlayer = {
setNetState: (store: TrackPlayerSlice) => store.setNetState, setNetState: (store: TrackPlayerSlice) => store.setNetState,
buildStreamUri: (store: TrackPlayerSlice) => store.buildStreamUri, buildStreamUri: (store: TrackPlayerSlice) => store.buildStreamUri,
reset: (store: TrackPlayerSlice) => store.reset, resetTrackPlayerState: (store: TrackPlayerSlice) => store.resetTrackPlayerState,
} }
export const trackPlayerCommands = new PromiseQueue(1) export const trackPlayerCommands = new PromiseQueue(1)
@ -111,10 +118,66 @@ export const createTrackPlayerSlice = (set: SetState<Store>, get: GetState<Store
setQueueContextId: queueContextId => set({ queueContextId }), setQueueContextId: queueContextId => set({ queueContextId }),
shuffleOrder: undefined, shuffleOrder: undefined,
setShuffleOrder: shuffleOrder => set({ shuffleOrder }), toggleShuffle: async () => {
return trackPlayerCommands.enqueue(async () => {
const queue = await getQueue()
const current = await getCurrentTrack()
const queueShuffleOrder = get().shuffleOrder
await TrackPlayer.remove(queue.map((_t, i) => i).filter(i => i !== current))
if (queueShuffleOrder === undefined) {
let { tracks, shuffleOrder } = shuffleTracks(queue, current)
if (tracks.length > 0) {
tracks = tracks.slice(1)
}
await TrackPlayer.add(tracks)
set({ shuffleOrder })
} else {
const tracks = unshuffleTracks(queue, queueShuffleOrder)
if (current !== undefined) {
const shuffledCurrent = queueShuffleOrder[current]
const tracks1 = tracks.slice(0, shuffledCurrent)
const tracks2 = tracks.slice(shuffledCurrent + 1)
await TrackPlayer.add(tracks2)
await TrackPlayer.add(tracks1, 0)
} else {
await TrackPlayer.add(tracks)
}
set({ shuffleOrder: undefined })
}
set({ queue: await getQueue() })
get().setCurrentTrackIdx(await getCurrentTrack())
})
},
repeatMode: RepeatMode.Off, repeatMode: RepeatMode.Off,
setRepeatMode: repeatMode => set({ repeatMode }), toggleRepeatMode: async () => {
return trackPlayerCommands.enqueue(async () => {
const repeatMode = await getRepeatMode()
let nextMode = RepeatMode.Off
switch (repeatMode) {
case RepeatMode.Off:
nextMode = RepeatMode.Queue
break
case RepeatMode.Queue:
nextMode = RepeatMode.Track
break
default:
nextMode = RepeatMode.Off
break
}
await TrackPlayer.setRepeatMode(nextMode)
set({ repeatMode: nextMode })
})
},
playerState: State.None, playerState: State.None,
setPlayerState: playerState => set({ playerState }), setPlayerState: playerState => set({ playerState }),
@ -131,7 +194,64 @@ export const createTrackPlayerSlice = (set: SetState<Store>, get: GetState<Store
}, },
queue: [], queue: [],
setQueue: queue => set({ queue }), setQueue: async (songs, name, contextType, contextId, playTrack, shuffle) => {
return trackPlayerCommands.enqueue(async () => {
const shuffled = shuffle !== undefined ? shuffle : !!get().shuffleOrder
await TrackPlayer.setupPlayer()
await TrackPlayer.reset()
if (songs.length === 0) {
return
}
let queue = await get().mapSongstoTrackExts(songs)
try {
for (const t of queue) {
t.url = get().buildStreamUri(t.id)
}
} catch {
return
}
if (shuffled) {
const { tracks, shuffleOrder } = shuffleTracks(queue, playTrack)
set({ shuffleOrder })
queue = tracks
playTrack = 0
} else {
set({ shuffleOrder: undefined })
}
playTrack = playTrack || 0
try {
set({
queue,
queueName: name,
queueContextType: contextType,
queueContextId: contextId,
})
get().setCurrentTrackIdx(playTrack)
if (playTrack === 0) {
await TrackPlayer.add(queue)
} else {
const tracks1 = queue.slice(0, playTrack)
const tracks2 = queue.slice(playTrack)
await TrackPlayer.add(tracks2)
await TrackPlayer.add(tracks1, 0)
}
await TrackPlayer.play()
} catch {
get().resetTrackPlayerState()
await TrackPlayer.reset()
}
})
},
progress: { position: 0, duration: 0, buffered: 0 }, progress: { position: 0, duration: 0, buffered: 0 },
setProgress: progress => set({ progress }), setProgress: progress => set({ progress }),
@ -194,10 +314,13 @@ export const createTrackPlayerSlice = (set: SetState<Store>, get: GetState<Store
get().setCurrentTrackIdx(currentTrack) get().setCurrentTrackIdx(currentTrack)
await TrackPlayer.add(queue) await TrackPlayer.add(queue)
if (currentTrack) { if (currentTrack) {
await TrackPlayer.skip(currentTrack) await TrackPlayer.skip(currentTrack)
} }
await TrackPlayer.seekTo(position) await TrackPlayer.seekTo(position)
if (state === State.Playing) { if (state === State.Playing) {
await TrackPlayer.play() await TrackPlayer.play()
} }
@ -217,7 +340,7 @@ export const createTrackPlayerSlice = (set: SetState<Store>, get: GetState<Store
}) })
}, },
reset: () => { resetTrackPlayerState: () => {
set({ set({
queueName: undefined, queueName: undefined,
queueContextType: undefined, queueContextType: undefined,
@ -249,3 +372,34 @@ export const getPlayerState = async (): Promise<State> => {
export const getRepeatMode = async (): Promise<RepeatMode> => { export const getRepeatMode = async (): Promise<RepeatMode> => {
return (await TrackPlayer.getRepeatMode()) || RepeatMode.Off return (await TrackPlayer.getRepeatMode()) || RepeatMode.Off
} }
function shuffleTracks(tracks: TrackExt[], firstTrack?: number) {
if (tracks.length === 0) {
return { tracks, shuffleOrder: [] }
}
const trackIndexes = tracks.map((_t, i) => i)
let shuffleOrder: number[] = []
for (let i = trackIndexes.length; i--; i > 0) {
const randi = Math.floor(Math.random() * (i + 1))
shuffleOrder.push(trackIndexes.splice(randi, 1)[0])
}
if (firstTrack !== undefined) {
shuffleOrder.splice(shuffleOrder.indexOf(firstTrack), 1)
shuffleOrder = [firstTrack, ...shuffleOrder]
}
tracks = shuffleOrder.map(i => tracks[i])
return { tracks, shuffleOrder }
}
function unshuffleTracks(tracks: TrackExt[], shuffleOrder: number[]): TrackExt[] {
if (tracks.length === 0 || shuffleOrder.length === 0) {
return tracks
}
return shuffleOrder.map((_v, i) => tracks[shuffleOrder.indexOf(i)])
}

View File

@ -0,0 +1,50 @@
import { Song } from '@app/models/music'
import { GetState, SetState } from 'zustand'
import { Store } from './store'
import { TrackExt } from './trackplayer'
export type TrackPlayerMapSlice = {
mapSongtoTrackExt: (song: Song) => Promise<TrackExt>
mapSongstoTrackExts: (songs: Song[]) => Promise<TrackExt[]>
mapTrackExtToSong: (song: TrackExt) => Song
}
export const selectTrackPlayerMap = {
mapTrackExtToSong: (store: TrackPlayerMapSlice) => store.mapTrackExtToSong,
}
export const createTrackPlayerMapSlice = (set: SetState<Store>, get: GetState<Store>): TrackPlayerMapSlice => ({
mapSongtoTrackExt: async song => {
return {
id: song.id,
title: song.title,
artist: song.artist || 'Unknown Artist',
album: song.album || 'Unknown Album',
url: song.streamUri,
artwork: song.coverArt ? await get().fetchCoverArtFilePath(song.coverArt) : require('@res/fallback.png'),
coverArt: song.coverArt,
duration: song.duration,
artistId: song.artistId,
albumId: song.albumId,
}
},
mapSongstoTrackExts: async songs => {
return await Promise.all(songs.map(get().mapSongtoTrackExt))
},
mapTrackExtToSong: track => {
return {
itemType: 'song',
id: track.id,
title: track.title as string,
artist: track.artist,
album: track.album,
streamUri: track.url as string,
coverArt: track.coverArt,
duration: track.duration,
artistId: track.artistId,
albumId: track.albumId,
}
},
})