#!/bin/bash

if [ "$#" -lt 6 ] || [ "$#" -gt 7 ]; then
    echo "Usage: $0 <base_url> <user> <artist_count> <album_count> <song_count> <batch_size> [musicFolderId]"
    exit 1
fi

# make any command failure exit
set -e

read -r -s -p "Enter password: " user_password
echo

# Retrieve the argument from the command line
base_url="$1"
user="$2"
artist_count="$3"
album_count="$4"
song_count="$5"
batch_size="$6"

music_folder_id=""
if [ "$#" -eq 7 ]; then
    music_folder_id="$7"
fi

append_music_folder() {
    local url="$1"
    if [ -n "$music_folder_id" ]; then
        url="$url&musicFolderId=$music_folder_id"
    fi
    echo "$url"
}

start_time=$(date +%s.%3N)

# artists
echo "Fetching $artist_count artists..."
for ((i = 0; i < artist_count; i += $batch_size)); do
    wget -q -O - "$(append_music_folder "$base_url/rest/search3.view?u=$user&p=$user_password&v=1.13.0&c=benchmark&f=json&query=&artistCount=$batch_size&artistOffset=$i&albumCount=0&songCount=0")" > /dev/null
done

# albums
echo "Fetching $album_count albums..."
for ((i = 0; i < album_count; i += $batch_size)); do
	wget -q -O - "$(append_music_folder "$base_url/rest/search3.view?u=$user&p=$user_password&v=1.13.0&c=benchmark&f=json&query=&artistCount=0&albumCount=$batch_size&albumOffset=$i&songCount=0")" > /dev/null
done

# songs
echo "Fetching $song_count songs..."
for ((i = 0; i < song_count; i += $batch_size)); do
	wget -q -O - "$(append_music_folder "$base_url/rest/search3.view?u=$user&p=$user_password&v=1.13.0&c=benchmark&f=json&query=&artistCount=0&albumCount=0&songCount=$batch_size&songOffset=$i")" > /dev/null
done

end_time=$(date +%s.%3N)

elapsed_time=$(echo "$end_time - $start_time" | bc)

echo "Fetch time: $elapsed_time seconds"