ts-npm-ytmusic-api/src/parsers/SongParser.ts

import { LIST, OBJECT, STRING } from "validate-any"

import { AlbumBasic, ArtistBasic, SongDetailed, SongFull, ThumbnailFull } from "../"
import { ALBUM_BASIC, ARTIST_BASIC, SONG_DETAILED, SONG_FULL, THUMBNAIL_FULL } from "../interfaces"
import checkType from "../utils/checkType"
import traverseList from "../utils/traverseList"
import traverseString from "../utils/traverseString"
import Parser from "./Parser"

export default class SongParser {
	public static parse(data: any): SongFull {
		return checkType<SongFull>(
			{
				type: "SONG",
				videoId: traverseString(data, "videoDetails", "videoId")(),
				name: traverseString(data, "videoDetails", "title")(),
				artists: [
					{
						artistId: traverseString(data, "videoDetails", "channelId")(),
						name: traverseString(data, "author")()
					}
				],
				duration: +traverseString(data, "videoDetails", "lengthSeconds")(),
				thumbnails: traverseList(data, "videoDetails", "thumbnails"),
				description: traverseString(data, "description")(),
				formats: traverseList(data, "streamingData", "formats"),
				adaptiveFormats: traverseList(data, "streamingData", "adaptiveFormats")
			},
			SONG_FULL
		)
	}

	public static parseSearchResult(item: any): SongDetailed {
		const flexColumns = traverseList(item, "flexColumns")

		return checkType<SongDetailed>(
			{
				type: "SONG",
				videoId: traverseString(item, "playlistItemData", "videoId")(),
				name: traverseString(flexColumns[0], "runs", "text")(),
				artists: traverseList(flexColumns[1], "runs")
					.filter(run => "navigationEndpoint" in run)
					.map(run => ({
						artistId: traverseString(run, "browseId")(),
						name: traverseString(run, "text")()
					}))
					.slice(0, -1),
				album: {
					albumId: traverseString(flexColumns[1], "runs", "browseId")(-1),
					name: traverseString(flexColumns[1], "runs", "text")(-3)
				},
				duration: Parser.parseDuration(traverseString(flexColumns[1], "runs", "text")(-1)),
				thumbnails: traverseList(item, "thumbnails")
			},
			SONG_DETAILED
		)
	}

	public static parseArtistSong(item: any): SongDetailed {
		const flexColumns = traverseList(item, "flexColumns")
		const videoId = traverseString(item, "playlistItemData", "videoId")()

		return checkType<SongDetailed>(
			{
				type: "SONG",
				videoId,
				name: traverseString(flexColumns[0], "runs", "text")(),
				artists: traverseList(flexColumns[1], "runs")
					.filter(run => "navigationEndpoint" in run)
					.map(run => ({
						artistId: traverseString(run, "browseId")(),
						name: traverseString(run, "text")()
					})),
				album: {
					albumId: traverseString(flexColumns[2], "browseId")(),
					name: traverseString(flexColumns[2], "runs", "text")()
				},
				duration: Parser.parseDuration(
					traverseString(item, "fixedColumns", "runs", "text")()
				),
				thumbnails: traverseList(item, "thumbnails")
			},
			SONG_DETAILED
		)
	}

	public static parseArtistTopSong(
		item: any,
		artistBasic: ArtistBasic
	): Omit<SongDetailed, "duration"> {
		const flexColumns = traverseList(item, "flexColumns")
		const videoId = traverseString(item, "playlistItemData", "videoId")()

		return checkType<Omit<SongDetailed, "duration">>(
			{
				type: "SONG",
				videoId,
				name: traverseString(flexColumns[0], "runs", "text")(),
				artists: [artistBasic],
				album: {
					albumId: traverseString(flexColumns[2], "browseId")(),
					name: traverseString(flexColumns[2], "runs", "text")()
				},
				thumbnails: traverseList(item, "thumbnails")
			},
			OBJECT({
				type: STRING("SONG"),
				videoId: STRING(),
				name: STRING(),
				artists: LIST(ARTIST_BASIC),
				album: ALBUM_BASIC,
				thumbnails: LIST(THUMBNAIL_FULL)
			})
		)
	}

	public static parseAlbumSong(
		item: any,
		artists: ArtistBasic[],
		albumBasic: AlbumBasic,
		thumbnails: ThumbnailFull[]
	): SongDetailed {
		const flexColumns = traverseList(item, "flexColumns")
		const videoId = traverseString(item, "playlistItemData", "videoId")()

		return checkType<SongDetailed>(
			{
				type: "SONG",
				videoId,
				name: traverseString(flexColumns[0], "runs", "text")(),
				artists,
				album: albumBasic,
				duration: Parser.parseDuration(
					traverseString(item, "fixedColumns", "runs", "text")()
				),
				thumbnails
			},
			SONG_DETAILED
		)
	}
}