NapCatQQ/napcat.webui/src/utils/music.ts

import { PlayMode } from '@/const/enum'

import type {
  FinalMusic,
  Music163ListResponse,
  Music163URLResponse
} from '@/types/music'

import { request } from './request'

/**
 * 获取网易云音乐歌单
 * @param id 歌单id
 * @returns 歌单信息
 */
export const get163MusicList = async (id: string) => {
  const res = await request.get<Music163ListResponse>(
    `https://wavesgame.top/playlist/track/all?id=${id}`
  )
  if (res?.data?.code !== 200) {
    throw new Error('获取歌曲列表失败')
  }
  return res.data
}

/**
 * 获取歌曲地址
 * @param ids 歌曲id
 * @returns 歌曲地址
 */
export const getSongsURL = async (ids: number[]) => {
  const _ids = ids.reduce((prev, cur, index) => {
    const groupIndex = Math.floor(index / 10)
    if (!prev[groupIndex]) {
      prev[groupIndex] = []
    }
    prev[groupIndex].push(cur)
    return prev
  }, [] as number[][])
  const res = await Promise.all(
    _ids.map(async (id) => {
      const res = await request.get<Music163URLResponse>(
        `https://wavesgame.top/song/url?id=${id.join(',')}`
      )
      if (res?.data?.code !== 200) {
        throw new Error('获取歌曲地址失败')
      }
      return res.data.data
    })
  )
  const result = res.reduce((prev, cur) => {
    return prev.concat(...cur)
  }, [])
  return result
}

/**
 * 获取网易云音乐歌单歌曲
 * @param id 歌单id
 * @returns 歌曲信息
 */
export const get163MusicListSongs = async (id: string) => {
  const listRes = await get163MusicList(id)
  const songs = listRes.songs.map((song) => song.id)
  const songsRes = await getSongsURL(songs)
  const finalMusic: FinalMusic[] = []
  for (let i = 0; i < listRes.songs.length; i++) {
    const song = listRes.songs[i]
    const music = songsRes.find((s) => s.id === song.id)
    const songURL = music?.url
    if (songURL) {
      finalMusic.push({
        id: song.id,
        url: songURL,
        title: song.name,
        artist: song.ar.map((p) => p.name).join('/'),
        cover: song.al.picUrl
      })
    }
  }
  return finalMusic
}

/**
 * 获取随机音乐
 * @param ids 歌曲id
 * @param currentId 当前音乐id
 * @returns 随机音乐id
 */
export const getRandomMusic = (ids: number[], currentId: number): number => {
  const randomIndex = Math.floor(Math.random() * ids.length)
  const randomId = ids[randomIndex]
  if (randomId === currentId) {
    return getRandomMusic(ids, currentId)
  }
  return randomId
}

/**
 * 获取下一首音乐id
 * @param ids 歌曲id
 * @param currentId 当前音乐ID
 * @param mode 播放模式
 */
export const getNextMusic = (
  musics: FinalMusic[],
  currentId: number,
  mode: PlayMode
): number => {
  const ids = musics.map((music) => music.id)
  if (mode === PlayMode.Loop) {
    const currentIndex = ids.findIndex((id) => id === currentId)
    const nextIndex = currentIndex + 1
    return ids[nextIndex] || ids[0]
  }
  if (mode === PlayMode.Random) {
    return getRandomMusic(ids, currentId)
  }
  return currentId
}