import os
import json
import traceback
from deezspot.spotloader import SpoLogin
from deezspot.deezloader import DeeLogin

def download_playlist(service, url, main, fallback=None):
    try:
        if service == 'spotify':
            if fallback:
                # First attempt: use DeeLogin's download_playlistspo with the 'main' (Deezer credentials)
                try:
                    # Load Deezer credentials from 'main' under deezer directory
                    deezer_creds_dir = os.path.join('./creds/deezer', main)
                    deezer_creds_path = os.path.abspath(os.path.join(deezer_creds_dir, 'credentials.json'))
                    with open(deezer_creds_path, 'r') as f:
                        deezer_creds = json.load(f)
                    # Initialize DeeLogin with Deezer credentials
                    dl = DeeLogin(
                        arl=deezer_creds.get('arl', ''),
                        email=deezer_creds.get('email', ''),
                        password=deezer_creds.get('password', '')
                    )
                    # Download using download_playlistspo
                    dl.download_playlistspo(
                        link_playlist=url,
                        output_dir="./downloads",
                        quality_download="FLAC",
                        recursive_quality=True,
                        recursive_download=False,
                        not_interface=False,
                        make_zip=False,
                        method_save=1
                    )
                except Exception as e:
                    # If the first attempt fails, use the fallback Spotify main
                    print(f"Failed to download via Deezer fallback: {e}. Trying Spotify fallback main.")
                    # Load fallback Spotify credentials and attempt download
                    try:
                        spo_creds_dir = os.path.join('./creds/spotify', fallback)
                        spo_creds_path = os.path.abspath(os.path.join(spo_creds_dir, 'credentials.json'))
                        spo = SpoLogin(credentials_path=spo_creds_path)
                        spo.download_playlist(
                            link_playlist=url,
                            output_dir="./downloads",
                            quality_download="HIGH",
                            recursive_quality=True,
                            recursive_download=False,
                            not_interface=False,
                            method_save=1,
                            make_zip=False
                        )
                    except Exception as e2:
                        # If fallback also fails, raise an error indicating both attempts failed
                        raise RuntimeError(
                            f"Both main (Deezer) and fallback (Spotify) attempts failed. "
                            f"Deezer error: {e}, Spotify error: {e2}"
                        ) from e2
            else:
                # Original behavior: use Spotify main
                creds_dir = os.path.join('./creds/spotify', main)
                credentials_path = os.path.abspath(os.path.join(creds_dir, 'credentials.json'))
                spo = SpoLogin(credentials_path=credentials_path)
                spo.download_playlist(
                    link_playlist=url,
                    output_dir="./downloads",
                    quality_download="HIGH",
                    recursive_quality=True,
                    recursive_download=False,
                    not_interface=False,
                    method_save=1,
                    make_zip=False
                )
        elif service == 'deezer':
            # Existing code for Deezer, using main as Deezer account
            creds_dir = os.path.join('./creds/deezer', main)
            creds_path = os.path.abspath(os.path.join(creds_dir, 'credentials.json'))
            with open(creds_path, 'r') as f:
                creds = json.load(f)
            dl = DeeLogin(
                arl=creds.get('arl', ''),
                email=creds.get('email', ''),
                password=creds.get('password', '')
            )
            dl.download_playlistdee(
                link_playlist=url,
                output_dir="./downloads",
                quality_download="FLAC",
                recursive_quality=False,
                recursive_download=False,
                method_save=1,
                make_zip=False
            )
        else:
            raise ValueError(f"Unsupported service: {service}")
    except Exception as e:
        traceback.print_exc()
        raise  # Re-raise the exception after logging