AppleMusicDecrypt/src/rip.py

import asyncio

from loguru import logger

from src.api import get_info_from_adam, get_song_lyrics, get_meta, download_song
from src.config import Config, Device
from src.decrypt import decrypt
from src.metadata import SongMetadata
from src.mp4 import extract_media, extract_song, encapsulate, write_metadata
from src.save import save
from src.types import GlobalAuthParams, Codec
from src.url import Song, Album, URLType
from src.utils import check_song_exists


@logger.catch
async def rip_song(song: Song, auth_params: GlobalAuthParams, codec: str, config: Config, device: Device,
                   force_save: bool = False):
    logger.debug(f"Task of song id {song.id} was created")
    token = auth_params.anonymousAccessToken
    song_data = await get_info_from_adam(song.id, token, song.storefront)
    song_metadata = SongMetadata.parse_from_song_data(song_data)
    logger.info(f"Ripping song: {song_metadata.artist} - {song_metadata.title}")
    if not force_save and check_song_exists(song_metadata, config.download, codec):
        logger.info(f"Song: {song_metadata.artist} - {song_metadata.title} already exists")
        return
    await song_metadata.get_cover(config.download.coverFormat)
    if song_data.attributes.hasTimeSyncedLyrics:
        lyrics = await get_song_lyrics(song.id, song.storefront, auth_params.accountAccessToken,
                                       auth_params.dsid, auth_params.accountToken)
        song_metadata.lyrics = lyrics
    song_uri, keys, selected_codec = await extract_media(song_data.attributes.extendedAssetUrls.enhancedHls, codec)
    logger.info(f"Selected codec: {selected_codec} for song: {song_metadata.artist} - {song_metadata.title}")
    logger.info(f"Downloading song: {song_metadata.artist} - {song_metadata.title}")
    raw_song = await download_song(song_uri)
    song_info = extract_song(raw_song, codec)
    decrypted_song = await decrypt(song_info, keys, song_data, device)
    song = encapsulate(song_info, decrypted_song, config.download.atmosConventToM4a)
    if codec != Codec.EC3 or (codec == Codec.EC3 and config.download.atmosConventToM4a):
        song = write_metadata(song, song_metadata, config.metadata.embedMetadata, config.download.coverFormat)
    if codec != Codec.AC3 or (codec == Codec.AC3 and config.download.atmosConventToM4a):
        song = write_metadata(song, song_metadata, config.metadata.embedMetadata, config.download.coverFormat)
    save(song, codec, song_metadata, config.download)
    logger.info(f"Song {song_metadata.artist} - {song_metadata.title} saved!")


async def rip_album(album: Album, auth_params: GlobalAuthParams, codec: str, config: Config, device: Device,
                    force_save: bool = False):
    album_info = await get_meta(album.id, auth_params.anonymousAccessToken, album.storefront)
    logger.info(f"Ripping Album: {album_info.data[0].attributes.artistName} - {album_info.data[0].attributes.name}")
    async with asyncio.TaskGroup() as tg:
        for track in album_info.data[0].relationships.tracks.data:
            song = Song(id=track.id, storefront=album.storefront, url="", type=URLType.Song)
            tg.create_task(rip_song(song, auth_params, codec, config, device, force_save))
    logger.info(
        f"Album: {album_info.data[0].attributes.artistName} - {album_info.data[0].attributes.name} finished ripping")


async def rip_playlist():
    pass


async def rip_artist():
    pass