package com.shadorc.shadbot.listener.music;
import com.sedmelluq.discord.lavaplayer.player.AudioLoadResultHandler;
import com.sedmelluq.discord.lavaplayer.tools.FriendlyException;
import com.sedmelluq.discord.lavaplayer.track.AudioPlaylist;
import com.sedmelluq.discord.lavaplayer.track.AudioTrack;
import com.sedmelluq.discord.lavaplayer.track.AudioTrackInfo;
import com.shadorc.shadbot.data.Config;
import com.shadorc.shadbot.db.DatabaseManager;
import com.shadorc.shadbot.db.guilds.entity.DBGuild;
import com.shadorc.shadbot.db.guilds.entity.Settings;
import com.shadorc.shadbot.music.GuildMusic;
import com.shadorc.shadbot.music.MusicManager;
import com.shadorc.shadbot.music.TrackScheduler;
import com.shadorc.shadbot.object.Emoji;
import com.shadorc.shadbot.object.ExceptionHandler;
import com.shadorc.shadbot.utils.DiscordUtils;
import com.shadorc.shadbot.utils.FormatUtils;
import com.shadorc.shadbot.utils.ShadbotUtils;
import com.shadorc.shadbot.utils.StringUtils;
import discord4j.common.util.Snowflake;
import discord4j.core.object.entity.User;
import discord4j.core.spec.EmbedCreateSpec;
import reactor.core.publisher.Mono;
import reactor.core.scheduler.Schedulers;
import reactor.function.TupleUtils;
import java.time.Duration;
import java.util.Collections;
import java.util.List;
import java.util.function.Consumer;
import static com.shadorc.shadbot.music.MusicManager.LOGGER;
public class AudioLoadResultListener implements AudioLoadResultHandler {
public static final String YT_SEARCH = "ytsearch: ";
public static final String SC_SEARCH = "scsearch: ";
private static final int MAX_PLAYLIST_NAME_LENGTH = 70;
private final Snowflake guildId;
private final Snowflake djId;
private final String identifier;
private final boolean insertFirst;
private List<AudioTrack> resultTracks;
public AudioLoadResultListener(Snowflake guildId, Snowflake djId, String identifier, boolean insertFirst) {
this.guildId = guildId;
this.djId = djId;
this.identifier = identifier;
this.insertFirst = insertFirst;
}
@Override
public void trackLoaded(AudioTrack audioTrack) {
LOGGER.debug("{Guild ID: {}} Track loaded: {}", this.guildId.asLong(), audioTrack.hashCode());
Mono.justOrEmpty(MusicManager.getInstance().getGuildMusic(this.guildId))
.filter(guildMusic -> !guildMusic.getTrackScheduler().startOrQueue(audioTrack, this.insertFirst))
.flatMap(GuildMusic::getMessageChannel)
.flatMap(channel -> DiscordUtils.sendMessage(String.format(
Emoji.MUSICAL_NOTE + " **%s** has been added to the playlist.",
FormatUtils.trackName(audioTrack.getInfo())), channel))
.then(this.terminate())
.subscribeOn(Schedulers.boundedElastic())
.subscribe(null, ExceptionHandler::handleUnknownError);
}
@Override
public void playlistLoaded(AudioPlaylist audioPlaylist) {
// SoundCloud returns an empty playlist when no results where found
if (audioPlaylist.getTracks().isEmpty()) {
LOGGER.debug("{Guild ID: {}} Empty playlist: {}", this.guildId.asLong(), audioPlaylist.hashCode());
this.onNoMatches();
}
// If a track is specifically selected
else if (audioPlaylist.getSelectedTrack() != null) {
LOGGER.debug("{Guild ID: {}} Playlist loaded, track selected: {}", this.guildId.asLong(), audioPlaylist.hashCode());
this.trackLoaded(audioPlaylist.getSelectedTrack());
}
// The user is searching something
else if (audioPlaylist.isSearchResult()) {
LOGGER.debug("{Guild ID: {}} Playlist loaded, search results: {}", this.guildId.asLong(), audioPlaylist.hashCode());
this.onSearchResult(audioPlaylist);
}
// The user loads a full playlist
else {
LOGGER.debug("{Guild ID: {}} Playlist loaded, full playlist: {}", this.guildId.asLong(), audioPlaylist.hashCode());
this.onPlaylistLoaded(audioPlaylist);
}
}
private void onSearchResult(AudioPlaylist playlist) {
Mono.justOrEmpty(MusicManager.getInstance().getGuildMusic(this.guildId))
.flatMapMany(guildMusic -> {
this.resultTracks = playlist.getTracks()
.subList(0, Math.min(Config.MUSIC_SEARCHES, playlist.getTracks().size()));
guildMusic.setDjId(this.djId);
guildMusic.setWaitingForChoice(true);
return guildMusic.getGateway()
.getUserById(guildMusic.getDjId())
.map(User::getAvatarUrl)
.flatMap(avatarUrl -> this.getPlaylistEmbed(playlist, avatarUrl))
.flatMap(embed -> guildMusic.getMessageChannel()
.flatMap(channel -> DiscordUtils.sendMessage(embed, channel)))
.flatMapMany(ignored ->
AudioLoadResultInputs.create(guildMusic.getGateway(), Duration.ofSeconds(30),
guildMusic.getMessageChannelId(), this)
.waitForInputs()
.then(Mono.fromRunnable(() -> guildMusic.setWaitingForChoice(false))));
})
.then(this.terminate())
.subscribeOn(Schedulers.boundedElastic())
.subscribe(null, ExceptionHandler::handleUnknownError);
}
private void onPlaylistLoaded(AudioPlaylist playlist) {
Mono.justOrEmpty(MusicManager.getInstance().getGuildMusic(this.guildId))
.zipWith(DatabaseManager.getPremium().isPremium(this.guildId, this.djId))
.flatMap(TupleUtils.function((guildMusic, isPremium) -> {
final TrackScheduler trackScheduler = guildMusic.getTrackScheduler();
final StringBuilder strBuilder = new StringBuilder();
int musicsAdded = 0;
for (final AudioTrack track : playlist.getTracks()) {
trackScheduler.startOrQueue(track, this.insertFirst);
musicsAdded++;
// The playlist limit is reached and the user / guild is not premium
if (trackScheduler.getPlaylist().size() >= Config.PLAYLIST_SIZE - 1 && !isPremium) {
strBuilder.append(ShadbotUtils.PLAYLIST_LIMIT_REACHED + "\n");
break;
}
}
strBuilder.append(String.format(Emoji.MUSICAL_NOTE + " %d musics have been added to the playlist.", musicsAdded));
return guildMusic.getMessageChannel()
.flatMap(channel -> DiscordUtils.sendMessage(strBuilder.toString(), channel));
}))
.then(this.terminate())
.subscribeOn(Schedulers.boundedElastic())
.subscribe(null, ExceptionHandler::handleUnknownError);
}
private Mono<Consumer<EmbedCreateSpec>> getPlaylistEmbed(AudioPlaylist playlist, String avatarUrl) {
final String choices = FormatUtils.numberedList(Config.MUSIC_SEARCHES, playlist.getTracks().size(),
count -> {
final AudioTrackInfo info = playlist.getTracks().get(count - 1).getInfo();
return String.format("\t**%d.** [%s](%s)", count, FormatUtils.trackName(info), info.uri);
});
final String playlistName = StringUtils.abbreviate(playlist.getName(), MAX_PLAYLIST_NAME_LENGTH);
return DatabaseManager.getGuilds()
.getDBGuild(this.guildId)
.map(DBGuild::getSettings)
.map(Settings::getPrefix)
.map(prefix -> ShadbotUtils.getDefaultEmbed()
.andThen(embed -> embed.setAuthor(playlistName, null, avatarUrl)
.setThumbnail("https://i.imgur.com/IG3Hj2W.png")
.setDescription("**Select a music by typing the corresponding number.**"
+ "\nYou can choose several musics by separating them with a comma."
+ "\nExample: 1,3,4"
+ "\n\n" + choices)
.setFooter(String.format("Use %scancel to cancel the selection (Automatically " +
"canceled in %ds).", prefix, Config.MUSIC_CHOICE_DURATION), null)));
}
@Override
public void loadFailed(FriendlyException err) {
LOGGER.debug("{Guild ID: {}} Load failed: {}", this.guildId.asLong(), err);
Mono.justOrEmpty(MusicManager.getInstance().getGuildMusic(this.guildId))
.flatMap(guildMusic -> {
final String errMessage = ShadbotUtils.cleanLavaplayerErr(err);
LOGGER.info("{Guild ID: {}} Load failed: {}", this.guildId.asLong(), errMessage);
return guildMusic.getMessageChannel()
.flatMap(channel -> DiscordUtils.sendMessage(
String.format(Emoji.RED_CROSS + " Something went wrong while loading the track: %s",
errMessage.toLowerCase()), channel));
})
.then(this.terminate())
.subscribeOn(Schedulers.boundedElastic())
.subscribe(null, ExceptionHandler::handleUnknownError);
}
@Override
public void noMatches() {
LOGGER.debug("{Guild ID: {}} No matches for identifier: {}", this.guildId.asLong(), this.identifier);
this.onNoMatches();
}
private void onNoMatches() {
Mono.justOrEmpty(MusicManager.getInstance().getGuildMusic(this.guildId))
.flatMap(GuildMusic::getMessageChannel)
.flatMap(channel -> DiscordUtils.sendMessage(String.format(Emoji.MAGNIFYING_GLASS + " No results for `%s`.",
StringUtils.remove(this.identifier, YT_SEARCH, SC_SEARCH)), channel))
.then(this.terminate())
.subscribeOn(Schedulers.boundedElastic())
.subscribe(null, ExceptionHandler::handleUnknownError);
}
private Mono<Void> terminate() {
return Mono.justOrEmpty(MusicManager.getInstance().getGuildMusic(this.guildId))
.flatMap(guildMusic -> guildMusic.removeAudioLoadResultListener(this));
}
public Snowflake getGuildId() {
return this.guildId;
}
public List<AudioTrack> getResultTracks() {
return Collections.unmodifiableList(this.resultTracks);
}
}
