subtracks/app/hooks/fetch.ts

import { CacheItemTypeKey } from '@app/models/cache'
import { Album, Playlist, Song } from '@app/models/library'
import { mapAlbum, mapArtist, mapArtistInfo, mapPlaylist, mapSong } from '@app/models/map'
import queryClient from '@app/queryClient'
import { useStore } from '@app/state/store'
import { SubsonicApiClient } from '@app/subsonic/api'
import { GetAlbumList2TypeBase, Search3Params, StarParams } from '@app/subsonic/params'
import { cacheDir } from '@app/util/fs'
import { mapCollectionById } from '@app/util/state'
import userAgent from '@app/util/userAgent'
import cd from 'content-disposition'
import mime from 'mime-types'
import path from 'path'
import ReactNativeBlobUtil, { FetchBlobResponse } from 'react-native-blob-util'
import RNFS from 'react-native-fs'
import qk from './queryKeys'

export const useClient = () => {
  const client = useStore(store => store.client)

  return () => {
    if (!client) {
      throw new Error('no client!')
    }

    return client
  }
}

function cacheStarredData<T extends { id: string; starred?: undefined | any }>(item: T) {
  queryClient.setQueryData<boolean>(qk.starredItems(item.id), !!item.starred)
}

function cacheAlbumCoverArtData<T extends { id: string; coverArt?: string }>(item: T) {
  queryClient.setQueryData<string | undefined>(qk.albumCoverArt(item.id), item.coverArt)
}

export const useFetchArtists = () => {
  const client = useClient()

  return async () => {
    const res = await client().getArtists()

    res.data.artists.forEach(cacheStarredData)

    return mapCollectionById(res.data.artists, mapArtist)
  }
}

export const useFetchArtist = () => {
  const client = useClient()

  return async (id: string) => {
    const res = await client().getArtist({ id })

    cacheStarredData(res.data.artist)
    res.data.albums.forEach(cacheStarredData)

    res.data.albums.forEach(cacheAlbumCoverArtData)

    return {
      artist: mapArtist(res.data.artist),
      albums: res.data.albums.map(mapAlbum),
    }
  }
}

export const useFetchArtistInfo = () => {
  const client = useClient()

  return async (id: string) => {
    const res = await client().getArtistInfo2({ id })
    return mapArtistInfo(id, res.data.artistInfo)
  }
}

export const useFetchArtistTopSongs = () => {
  const client = useClient()

  return async (artistName: string) => {
    const res = await client().getTopSongs({ artist: artistName })

    res.data.songs.forEach(cacheStarredData)

    return res.data.songs.map(mapSong)
  }
}

export const useFetchPlaylists = () => {
  const client = useClient()

  return async () => {
    const res = await client().getPlaylists()
    return mapCollectionById(res.data.playlists, mapPlaylist)
  }
}

export const useFetchPlaylist = () => {
  const client = useClient()

  return async (id: string): Promise<{ playlist: Playlist; songs?: Song[] }> => {
    const res = await client().getPlaylist({ id })

    res.data.playlist.songs.forEach(cacheStarredData)

    return {
      playlist: mapPlaylist(res.data.playlist),
      songs: res.data.playlist.songs.map(mapSong),
    }
  }
}

export async function fetchAlbum(id: string, client: SubsonicApiClient): Promise<{ album: Album; songs?: Song[] }> {
  const res = await client.getAlbum({ id })

  cacheStarredData(res.data.album)
  res.data.songs.forEach(cacheStarredData)

  cacheAlbumCoverArtData(res.data.album)

  return {
    album: mapAlbum(res.data.album),
    songs: res.data.songs.map(mapSong),
  }
}

export const useFetchAlbum = () => {
  const client = useClient()
  return async (id: string) => fetchAlbum(id, client())
}

export const useFetchAlbumList = () => {
  const client = useClient()

  return async (size: number, offset: number, type: GetAlbumList2TypeBase) => {
    const res = await client().getAlbumList2({ size, offset, type })

    res.data.albums.forEach(cacheStarredData)

    res.data.albums.forEach(cacheAlbumCoverArtData)

    return res.data.albums.map(mapAlbum)
  }
}

export const useFetchSong = () => {
  const client = useClient()

  return async (id: string) => {
    const res = await client().getSong({ id })

    cacheStarredData(res.data.song)

    return mapSong(res.data.song)
  }
}

export const useFetchSearchResults = () => {
  const client = useClient()

  return async (params: Search3Params) => {
    const res = await client().search3(params)

    res.data.artists.forEach(cacheStarredData)
    res.data.albums.forEach(cacheStarredData)
    res.data.songs.forEach(cacheStarredData)

    res.data.albums.forEach(cacheAlbumCoverArtData)

    return {
      artists: res.data.artists.map(mapArtist),
      albums: res.data.albums.map(mapAlbum),
      songs: res.data.songs.map(mapSong),
    }
  }
}

export const useFetchStar = () => {
  const client = useClient()

  return async (params: StarParams) => {
    await client().star(params)
    return
  }
}

export const useFetchUnstar = () => {
  const client = useClient()

  return async (params: StarParams) => {
    await client().unstar(params)
    return
  }
}

export type FetchExisingFileOptions = {
  itemType: CacheItemTypeKey
  itemId: string
}

export async function fetchExistingFile(
  options: FetchExisingFileOptions,
  serverId: string | undefined,
): Promise<string | undefined> {
  const { itemType, itemId } = options
  const fileDir = cacheDir(serverId, itemType, itemId)

  try {
    const dir = await RNFS.readDir(fileDir)
    console.log('existing file:', dir[0].path)
    return dir[0].path
  } catch {}
}

export const useFetchExistingFile = () => {
  const serverId = useStore(store => store.settings.activeServerId)
  return async (options: FetchExisingFileOptions) => fetchExistingFile(options, serverId)
}

function assertMimeType(expected?: string, actual?: string) {
  expected = expected?.toLowerCase()
  actual = actual?.toLowerCase()

  if (!expected || expected === actual) {
    return
  }

  if (!expected.includes(';')) {
    actual = actual?.split(';')[0]
  }

  if (!expected.includes('/')) {
    actual = actual?.split('/')[0]
  }

  if (expected !== actual) {
    throw new Error(`Request does not satisfy expected content type. Expected: ${expected} Actual: ${actual}`)
  }
}

export type FetchFileOptions = FetchExisingFileOptions & {
  fromUrl: string
  useCacheBuster?: boolean
  expectedContentType?: string
  progress?: (received: number, total: number) => void
}

export async function fetchFile(options: FetchFileOptions, serverId: string | undefined): Promise<string> {
  let { itemType, itemId, fromUrl, useCacheBuster, expectedContentType, progress } = options
  useCacheBuster = useCacheBuster === undefined ? true : useCacheBuster

  const fileDir = cacheDir(serverId, itemType, itemId)
  const filePathNoExt = path.join(fileDir, useCacheBuster ? useStore.getState().settings.cacheBuster : itemType)

  try {
    await RNFS.unlink(fileDir)
  } catch {}

  const headers = { 'User-Agent': userAgent }

  // we send a HEAD first for two reasons:
  // 1. to follow any redirects and get the actual URL (DownloadManager does not support redirects)
  // 2. to obtain the mime-type up front so we can use it for the file extension/validation
  const headRes = await fetch(fromUrl, { method: 'HEAD', headers })

  if (headRes.status > 399) {
    throw new Error(`HTTP status error ${headRes.status}. File: ${itemType} ID: ${itemId}`)
  }

  const contentType = headRes.headers.get('content-type') || undefined
  assertMimeType(expectedContentType, contentType)

  const contentDisposition = headRes.headers.get('content-disposition') || undefined
  const filename = contentDisposition ? cd.parse(contentDisposition).parameters.filename : undefined

  let extension: string | undefined
  if (filename) {
    extension = path.extname(filename) || undefined
    if (extension) {
      extension = extension.substring(1)
    }
  } else if (contentType) {
    extension = mime.extension(contentType) || undefined
  }

  const config = ReactNativeBlobUtil.config({
    addAndroidDownloads: {
      useDownloadManager: true,
      notification: false,
      mime: contentType,
      description: 'subtracks',
      path: extension ? `${filePathNoExt}.${extension}` : filePathNoExt,
    },
  })

  const fetchParams: Parameters<typeof config['fetch']> = ['GET', headRes.url, headers]

  let res: FetchBlobResponse
  if (progress) {
    res = await config.fetch(...fetchParams).progress(progress)
  } else {
    res = await config.fetch(...fetchParams)
  }

  const downloadPath = res.path()
  queryClient.setQueryData<string>(qk.existingFiles(itemType, itemId), downloadPath)

  console.log('downloaded file:', downloadPath)
  return downloadPath
}

export const useFetchFile = () => {
  const serverId = useStore(store => store.settings.activeServerId)
  return async (options: FetchFileOptions) => fetchFile(options, serverId)
}