const axios = require('axios');
const cheerio = require('cheerio');

async function applemusicsearch(search) {
    const { data: html } = await axios.get(`https://music.apple.com/us/search?term=${encodeURIComponent(search)}`, {
        headers: {
            'accept-language': 'en-US,en;q=0.9',
            'user-agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36'
        }
    });

    const $ = cheerio.load(html);
    const results = {
        topResults: [],
        songs: [],
        albums: [],
        artists: [],
        playlists: [],
        radioEpisodes: [],
        stations: [],
        musicVideos: [],
        videoExtras: []
    };

    const getImage = (srcset) => {
        if (!srcset) return null;
        const match = srcset.split(',')[0].match(/(https:\/\/[^\s]+)/);
        return match && match[1].includes('mzstatic.com') ? match[1] : null;
    };

    const cleanText = (text) => text ? text.replace(/[ ·\s]+/g, ' ').trim() : '';

    $('div[aria-label="Top Results"] .top-search-lockup').each((_, el) => {
        const title = cleanText($(el).find('.top-search-lockup__primary__title').text());
        const subtitle = cleanText($(el).find('.top-search-lockup__secondary').text());
        const url = $(el).find('.click-action').attr('href') || null;
        const image = getImage($(el).find('.artwork-component picture source[type="image/webp"]').attr('srcset'));
        const explicit = $(el).find('[data-testid="explicit-badge"]').length > 0;
        if (title) results.topResults.push({ title, subtitle: subtitle || null, url, image, explicit });
    });

    $('div[aria-label="Songs"] .track-lockup').each((_, el) => {
        const title = cleanText($(el).find('.track-lockup__title a').text());
        const artist = cleanText($(el).find('.track-lockup__subtitle a').text());
        const url = $(el).find('.track-lockup__title a').attr('href') || null;
        const image = getImage($(el).find('.artwork-component picture source[type="image/webp"]').attr('srcset'));
        const explicit = $(el).find('[data-testid="explicit-badge"]').length > 0;
        if (title) results.songs.push({ title, artist: artist || null, url, image, explicit });
    });

    $('div[aria-label="Artists"] .ellipse-lockup').each((_, el) => {
        const name = cleanText($(el).find('.title').text());
        const url = $(el).closest('a').attr('href') || null;
        const image = getImage($(el).find('.artwork-component picture source[type="image/webp"]').attr('srcset'));
        if (name) results.artists.push({ name, url, image });
    });

    $('div[aria-label="Albums"] .product-lockup').each((_, el) => {
        const title = cleanText($(el).find('.product-lockup__title').text());
        const artist = cleanText($(el).find('.product-lockup__subtitle').text());
        const url = $(el).find('.product-lockup__link').attr('href') || null;
        const image = getImage($(el).find('.artwork-component picture source[type="image/webp"]').attr('srcset'));
        const explicit = $(el).find('[data-testid="explicit-badge"]').length > 0;
        if (title) results.albums.push({ title, artist: artist || null, url, image, explicit });
    });

    $('div[aria-label="Playlists"] .product-lockup').each((_, el) => {
        const title = cleanText($(el).find('.product-lockup__title').text());
        const curator = cleanText($(el).find('.product-lockup__subtitle').text());
        const url = $(el).find('.product-lockup__link').attr('href') || null;
        const image = getImage($(el).find('.artwork-component picture source[type="image/webp"]').attr('srcset'));
        if (title) results.playlists.push({ title, curator: curator || 'Apple Music', url, image });
    });

    $('div[aria-label="Radio Episodes"] .product-lockup').each((_, el) => {
        const title = cleanText($(el).find('.product-lockup__title').text());
        const show = cleanText($(el).find('.product-lockup__subtitle').text()) || cleanText($(el).find('.lockup-column__secondary-headline').text());
        const url = $(el).find('.product-lockup__link').attr('href') || null;
        const image = getImage($(el).find('.artwork-component picture source[type="image/webp"]').attr('srcset'));
        if (title) results.radioEpisodes.push({ title, show: show || null, url, image });
    });

    $('div[aria-label="Stations"] .product-lockup').each((_, el) => {
        const title = cleanText($(el).find('.product-lockup__title').text());
        const url = $(el).find('.product-lockup__link').attr('href') || null;
        const image = getImage($(el).find('.artwork-component picture source[type="image/webp"]').attr('srcset'));
        if (title) results.stations.push({ title, url, image });
    });

    $('div[aria-label="Music Videos"] .vertical-video').each((_, el) => {
        const titleEl = $(el).find('.vertical-video__title a');
        const title = cleanText(titleEl.text());
        const url = titleEl.attr('href') || null;
        const allText = cleanText($(el).text());
        const artist = allText.replace(title, '').trim().split('\n')[0] || null;
        const image = getImage($(el).find('.artwork-component picture source[type="image/webp"]').attr('srcset'));
        const explicit = $(el).find('[data-testid="explicit-badge"]').length > 0;
        if (title) results.musicVideos.push({ title, artist: artist || null, url, image, explicit });
    });

    $('div[aria-label="Video Extras"] .vertical-video').each((_, el) => {
        const titleEl = $(el).find('.vertical-video__title a');
        const title = cleanText(titleEl.text());
        const url = titleEl.attr('href') || null;
        const allText = cleanText($(el).text());
        const artist = allText.replace(title, '').trim().split('\n')[0] || null;
        const image = getImage($(el).find('.artwork-component picture source[type="image/webp"]').attr('srcset'));
        if (title) results.videoExtras.push({ title, artist: artist || null, url, image });
    });

    return results;
}

async function applemusic(search) {
    return applemusicsearch(search);
}

applemusic('miguel phonk').then(console.log);