475 lines
17 KiB
Java
475 lines
17 KiB
Java
package me.chayapak1.chomens_bot.plugins;
|
||
|
||
import me.chayapak1.chomens_bot.Bot;
|
||
import me.chayapak1.chomens_bot.data.BotBossBar;
|
||
import me.chayapak1.chomens_bot.data.PlayerEntry;
|
||
import me.chayapak1.chomens_bot.song.Loop;
|
||
import me.chayapak1.chomens_bot.song.Note;
|
||
import me.chayapak1.chomens_bot.song.Song;
|
||
import me.chayapak1.chomens_bot.song.SongLoaderThread;
|
||
import me.chayapak1.chomens_bot.util.ColorUtilities;
|
||
import me.chayapak1.chomens_bot.util.LoggerUtilities;
|
||
import me.chayapak1.chomens_bot.util.MathUtilities;
|
||
import net.kyori.adventure.text.Component;
|
||
import net.kyori.adventure.text.format.NamedTextColor;
|
||
import org.cloudburstmc.math.vector.Vector3f;
|
||
import org.geysermc.mcprotocollib.network.event.session.DisconnectedEvent;
|
||
import org.geysermc.mcprotocollib.protocol.data.game.BossBarColor;
|
||
import org.geysermc.mcprotocollib.protocol.data.game.BossBarDivision;
|
||
|
||
import java.io.IOException;
|
||
import java.net.URL;
|
||
import java.nio.file.Files;
|
||
import java.nio.file.Path;
|
||
import java.util.ArrayList;
|
||
import java.util.List;
|
||
import java.util.concurrent.TimeUnit;
|
||
|
||
// Author: _ChipMC_ & chayapak <3
|
||
public class MusicPlayerPlugin extends Bot.Listener {
|
||
public static final String SELECTOR = "@a[tag=!nomusic,tag=!chomens_bot_nomusic,tag=!custompitch]";
|
||
public static final String CUSTOM_PITCH_SELECTOR = "@a[tag=!nomusic,tag=!chomens_bot_nomusic,tag=custompitch]";
|
||
public static final String BOTH_SELECTOR = "@a[tag=!nomusic,tag=!chomens_bot_nomusic]";
|
||
|
||
private final Bot bot;
|
||
|
||
public static final Path SONG_DIR = Path.of("songs");
|
||
static {
|
||
try {
|
||
if (!Files.exists(SONG_DIR)) Files.createDirectory(SONG_DIR);
|
||
} catch (IOException e) {
|
||
LoggerUtilities.error(e);
|
||
}
|
||
}
|
||
|
||
public Song currentSong;
|
||
public final List<Song> songQueue = new ArrayList<>();
|
||
public SongLoaderThread loaderThread = null;
|
||
public Loop loop = Loop.OFF;
|
||
|
||
// sus nightcore stuff,..,.,.
|
||
public float pitch = 0;
|
||
public float speed = 1;
|
||
|
||
public int amplify = 1;
|
||
|
||
public String instrument = "off";
|
||
|
||
private int urlLimit = 0;
|
||
|
||
public boolean locked = false; // this can be set through servereval
|
||
|
||
private final String bossBarName = "music";
|
||
public BossBarColor bossBarColor;
|
||
|
||
public String currentLyrics = "";
|
||
|
||
public MusicPlayerPlugin (Bot bot) {
|
||
this.bot = bot;
|
||
|
||
bot.addListener(this);
|
||
|
||
bot.core.addListener(new CorePlugin.Listener() {
|
||
@Override
|
||
public void ready() {
|
||
onCoreReady();
|
||
}
|
||
});
|
||
|
||
bot.executor.scheduleAtFixedRate(this::onTick, 0, 50, TimeUnit.MILLISECONDS);
|
||
bot.executor.scheduleAtFixedRate(() -> urlLimit = 0, 0, bot.config.music.urlRatelimit.seconds, TimeUnit.SECONDS);
|
||
}
|
||
|
||
public void loadSong (Path location, PlayerEntry sender) {
|
||
if (songQueue.size() > 500) return;
|
||
|
||
loaderThread = new SongLoaderThread(location, bot, sender.profile.getName());
|
||
|
||
bot.chat.tellraw(
|
||
Component
|
||
.translatable(
|
||
"Loading %s",
|
||
Component.text(location.getFileName().toString(), ColorUtilities.getColorByString(bot.config.colorPalette.secondary))
|
||
)
|
||
.color(ColorUtilities.getColorByString(bot.config.colorPalette.defaultColor))
|
||
);
|
||
|
||
loaderThread.start();
|
||
}
|
||
|
||
public void loadSong (URL location, PlayerEntry sender) {
|
||
if (songQueue.size() > 500) return;
|
||
|
||
urlLimit++;
|
||
|
||
if (urlLimit > bot.config.music.urlRatelimit.limit) {
|
||
bot.chat.tellraw(Component.text("ohio").color(NamedTextColor.RED));
|
||
return;
|
||
}
|
||
|
||
loaderThread = new SongLoaderThread(location, bot, sender.profile.getName());
|
||
|
||
bot.chat.tellraw(
|
||
Component
|
||
.translatable(
|
||
"Loading %s",
|
||
Component.text(location.toString(), ColorUtilities.getColorByString(bot.config.colorPalette.secondary))
|
||
)
|
||
.color(ColorUtilities.getColorByString(bot.config.colorPalette.defaultColor))
|
||
);
|
||
|
||
loaderThread.start();
|
||
}
|
||
|
||
public void loadSong (byte[] data, PlayerEntry sender) {
|
||
if (songQueue.size() > 500) return;
|
||
|
||
loaderThread = new SongLoaderThread(data, bot, sender.profile.getName());
|
||
|
||
bot.chat.tellraw(
|
||
Component
|
||
.translatable(
|
||
"Loading %s",
|
||
Component.text(sender.profile.getName() + "'s song item", ColorUtilities.getColorByString(bot.config.colorPalette.secondary))
|
||
)
|
||
.color(ColorUtilities.getColorByString(bot.config.colorPalette.defaultColor))
|
||
);
|
||
|
||
loaderThread.start();
|
||
}
|
||
|
||
private void onCoreReady () {
|
||
if (currentSong != null) currentSong.play();
|
||
}
|
||
|
||
// this needs a separate ticker because we need
|
||
// the song to be playing without lag
|
||
private void onTick () {
|
||
try {
|
||
if (!bot.loggedIn) return;
|
||
|
||
if (currentSong == null) {
|
||
if (songQueue.isEmpty()) return; // this line
|
||
|
||
addBossBar();
|
||
|
||
currentSong = songQueue.getFirst(); // songQueue.poll();
|
||
bot.chat.tellraw(
|
||
Component.translatable(
|
||
"Now playing %s",
|
||
Component.empty().append(Component.text(currentSong.name)).color(ColorUtilities.getColorByString(bot.config.colorPalette.secondary))
|
||
).color(ColorUtilities.getColorByString(bot.config.colorPalette.defaultColor))
|
||
);
|
||
currentSong.play();
|
||
}
|
||
|
||
if (currentSong.paused) return;
|
||
|
||
if (!currentSong.finished()) {
|
||
handleLyrics();
|
||
|
||
BotBossBar bossBar = bot.bossbar.get(bossBarName);
|
||
|
||
if (bossBar == null && bot.bossbar.enabled) bossBar = addBossBar();
|
||
|
||
if (bot.bossbar.enabled && bot.options.useCore) {
|
||
bossBar.setTitle(generateBossBar());
|
||
bossBar.setColor(bossBarColor);
|
||
bossBar.setValue((int) Math.floor(((double) (currentSong.time * speed) / 1000)));
|
||
bossBar.setMax((long) (currentSong.length * speed) / 1000);
|
||
}
|
||
|
||
if (currentSong.paused || bot.core.isRateLimited()) return;
|
||
|
||
handlePlaying();
|
||
} else {
|
||
currentLyrics = "";
|
||
|
||
if (loop == Loop.CURRENT) {
|
||
currentSong.loop();
|
||
return;
|
||
}
|
||
|
||
bot.chat.tellraw(
|
||
Component.translatable(
|
||
"Finished playing %s",
|
||
Component.empty().append(Component.text(currentSong.name)).color(ColorUtilities.getColorByString(bot.config.colorPalette.secondary))
|
||
).color(ColorUtilities.getColorByString(bot.config.colorPalette.defaultColor))
|
||
);
|
||
|
||
if (loop == Loop.ALL) {
|
||
skip();
|
||
return;
|
||
}
|
||
|
||
songQueue.removeFirst();
|
||
|
||
if (songQueue.isEmpty()) {
|
||
stopPlaying();
|
||
return;
|
||
}
|
||
|
||
if (currentSong.size() > 0) {
|
||
currentSong = songQueue.getFirst();
|
||
currentSong.setTime(0);
|
||
currentSong.play();
|
||
}
|
||
}
|
||
} catch (Exception e) {
|
||
bot.logger.error(e);
|
||
}
|
||
}
|
||
|
||
public void skip () {
|
||
if (loop == Loop.ALL) {
|
||
songQueue.add(songQueue.removeFirst()); // bot.music.queue.push(bot.music.queue.shift()) in js
|
||
} else {
|
||
songQueue.removeFirst();
|
||
}
|
||
if (songQueue.isEmpty()) {
|
||
stopPlaying();
|
||
return;
|
||
}
|
||
currentSong = songQueue.getFirst();
|
||
currentSong.setTime(0);
|
||
currentSong.play();
|
||
}
|
||
|
||
public BotBossBar addBossBar () {
|
||
if (currentSong == null) return null;
|
||
|
||
final BotBossBar bossBar = new BotBossBar(
|
||
Component.empty(),
|
||
BOTH_SELECTOR,
|
||
BossBarColor.LIME,
|
||
BossBarDivision.NONE,
|
||
true,
|
||
(int) currentSong.length / 1000,
|
||
0,
|
||
bot
|
||
);
|
||
|
||
bot.bossbar.add(bossBarName, bossBar);
|
||
|
||
return bossBar;
|
||
}
|
||
|
||
private void handleLyrics () {
|
||
// please help, this is many attempts trying to get this working
|
||
// midi lyrics are very weird
|
||
// i need some karaoke players to see how this works
|
||
|
||
// final Map<Long, String> lyrics = currentSong.lyrics;
|
||
//
|
||
// if (lyrics.isEmpty()) return;
|
||
//
|
||
// final List<String> lyricsList = new ArrayList<>();
|
||
//
|
||
// for (Map.Entry<Long, String> entry : lyrics.entrySet()) {
|
||
// final long time = entry.getKey();
|
||
// String _lyric = entry.getValue();
|
||
//
|
||
// if (time > currentSong.time) continue;
|
||
//
|
||
//// StringBuilder lyric = new StringBuilder();
|
||
////
|
||
//// for (char character : _lyric.toCharArray()) {
|
||
//// if ((character != '\n' && character != '\r' && character < ' ') || character == '<27>') continue;
|
||
////
|
||
//// lyric.append(character);
|
||
//// }
|
||
////
|
||
//// String stringLyric = lyric.toString();
|
||
////
|
||
//// if (stringLyric.startsWith("\\") || stringLyric.startsWith("/")) {
|
||
//// lyricsList.clear();
|
||
////
|
||
//// stringLyric = stringLyric.substring(1);
|
||
//// }
|
||
//
|
||
// lyricsList.add(_lyric);
|
||
// }
|
||
//
|
||
// final String joined = String.join("", lyricsList);
|
||
// currentLyrics = joined.substring(Math.max(0, joined.length() - 25));
|
||
}
|
||
|
||
public void removeBossBar() {
|
||
final BotBossBar bossBar = bot.bossbar.get(bossBarName);
|
||
|
||
if (bossBar != null) bossBar.setTitle(Component.text("No song is currently playing"));
|
||
|
||
bot.bossbar.remove(bossBarName);
|
||
}
|
||
|
||
public Component generateBossBar () {
|
||
NamedTextColor nameColor;
|
||
if (pitch > 0) {
|
||
nameColor = NamedTextColor.LIGHT_PURPLE;
|
||
bossBarColor = BossBarColor.PURPLE;
|
||
} else if (pitch < 0) {
|
||
nameColor = NamedTextColor.AQUA;
|
||
bossBarColor = BossBarColor.CYAN;
|
||
} else {
|
||
nameColor = NamedTextColor.GREEN;
|
||
bossBarColor = BossBarColor.YELLOW;
|
||
}
|
||
|
||
Component component = Component.empty()
|
||
.append(Component.empty().append(Component.text(currentSong.name)).color(nameColor))
|
||
.append(Component.text(" | ").color(NamedTextColor.DARK_GRAY))
|
||
.append(
|
||
Component
|
||
.translatable("%s / %s",
|
||
formatTime((long) (currentSong.time * speed)).color(NamedTextColor.GRAY),
|
||
formatTime((long) (currentSong.length * speed)).color(NamedTextColor.GRAY)).color(NamedTextColor.DARK_GRAY)
|
||
);
|
||
|
||
if (!bot.core.hasRateLimit() && !currentLyrics.isEmpty()) {
|
||
component = component
|
||
.append(Component.text(" | ", NamedTextColor.DARK_GRAY))
|
||
.append(Component.text(currentLyrics).color(NamedTextColor.BLUE));
|
||
}
|
||
|
||
if (currentSong.paused) {
|
||
return component
|
||
.append(Component.text(" | ", NamedTextColor.DARK_GRAY))
|
||
.append(Component.text("Paused", NamedTextColor.LIGHT_PURPLE));
|
||
}
|
||
|
||
if (loop != Loop.OFF) {
|
||
return component
|
||
.append(Component.translatable(" | ", NamedTextColor.DARK_GRAY))
|
||
.append(Component.translatable("Looping " + ((loop == Loop.CURRENT) ? "current" : "all"), NamedTextColor.LIGHT_PURPLE));
|
||
}
|
||
|
||
return component;
|
||
}
|
||
|
||
public Component formatTime (long millis) {
|
||
final int seconds = (int) millis / 1000;
|
||
|
||
final String minutePart = String.valueOf(seconds / 60);
|
||
final String unpaddedSecondPart = String.valueOf(seconds % 60);
|
||
|
||
return Component.translatable(
|
||
"%s:%s",
|
||
Component.text(minutePart),
|
||
Component.text(unpaddedSecondPart.length() < 2 ? "0" + unpaddedSecondPart : unpaddedSecondPart)
|
||
);
|
||
}
|
||
|
||
public void stopPlaying () {
|
||
removeBossBar();
|
||
currentSong = null;
|
||
}
|
||
|
||
@Override
|
||
public void disconnected (DisconnectedEvent event) {
|
||
if (currentSong != null) currentSong.pause(); // nice.
|
||
}
|
||
|
||
public void handlePlaying () {
|
||
if (currentSong == null) return;
|
||
|
||
try {
|
||
currentSong.advanceTime();
|
||
while (currentSong.reachedNextNote()) {
|
||
final Note note = currentSong.getNextNote();
|
||
|
||
if (note.volume == 0) continue;
|
||
|
||
float key = note.shiftedPitch;
|
||
|
||
final Vector3f blockPosition = getBlockPosition(note);
|
||
|
||
final float notShiftedFloatingPitch = (float) Math.pow(2.0, ((note.pitch + (pitch / 10)) - 12) / 12.0);
|
||
|
||
key += 33;
|
||
|
||
final boolean isMoreOrLessOctave = key < 33 || key > 57;
|
||
|
||
final boolean shouldCustomPitch = currentSong.nbs ?
|
||
isMoreOrLessOctave :
|
||
note.pitch != note.shiftedPitch ||
|
||
note.shiftedInstrument != note.instrument;
|
||
|
||
final double volume = note.volume;
|
||
|
||
if (shouldCustomPitch) {
|
||
bot.core.run(
|
||
"minecraft:execute as " +
|
||
CUSTOM_PITCH_SELECTOR +
|
||
" at @s run playsound " +
|
||
(!instrument.equals("off") ? instrument : note.instrument.sound) + ".pitch." + notShiftedFloatingPitch +
|
||
" record @s ^" + blockPosition.getX() + " ^" + blockPosition.getY() + " ^" + blockPosition.getZ() + " " +
|
||
volume +
|
||
" " +
|
||
0 +
|
||
" 1"
|
||
);
|
||
}
|
||
|
||
// these 2 lines are totallynotskidded from https://github.com/OpenNBS/OpenNoteBlockStudio/blob/master/scripts/selection_transpose/selection_transpose.gml
|
||
// so huge thanks to them uwu
|
||
while (key < 33) key += 12; // 1 octave has 12 notes, so we just keep moving octaves here
|
||
while (key > 57) key -= 12;
|
||
|
||
key -= 33;
|
||
|
||
float floatingPitch = (float) (0.5 * Math.pow(2, (key + (pitch / 10)) / 12));
|
||
|
||
for (int i = 0; i < amplify; i++) {
|
||
bot.core.run(
|
||
"minecraft:execute as " +
|
||
(shouldCustomPitch ? SELECTOR : BOTH_SELECTOR) +
|
||
" at @s run playsound " +
|
||
(!instrument.equals("off") ? instrument : note.shiftedInstrument.sound) +
|
||
" record @s ^" + blockPosition.getX() + " ^" + blockPosition.getY() + " ^" + blockPosition.getZ() + " " +
|
||
volume +
|
||
" " +
|
||
MathUtilities.clamp(floatingPitch, 0, 2) +
|
||
" 1"
|
||
);
|
||
}
|
||
}
|
||
} catch (Exception e) {
|
||
bot.logger.error(e);
|
||
}
|
||
}
|
||
|
||
private Vector3f getBlockPosition (Note note) {
|
||
Vector3f blockPosition;
|
||
|
||
if (currentSong.nbs) {
|
||
// https://github.com/OpenNBS/OpenNoteBlockStudio/blob/45f35ea193268fb541c1297d0b656f4964339d97/scripts/dat_generate/dat_generate.gml#L22C18-L22C31
|
||
final int average = (note.stereo + note.panning) / 2;
|
||
|
||
float pos;
|
||
|
||
if (average > 100) pos = (float) (average - 100) / -100;
|
||
else if (average == 100) pos = 0;
|
||
else pos = (float) ((average - 100) * -1) / 100;
|
||
|
||
blockPosition = Vector3f.from(pos * 2, 0, 0);
|
||
} else {
|
||
// i wrote this part
|
||
|
||
final int originalPitch = note.originalPitch;
|
||
|
||
float xPos = -(float) originalPitch / 768;
|
||
if (originalPitch > 25) xPos = Math.abs(xPos);
|
||
|
||
float yPos = -(float) originalPitch / 35;
|
||
if (originalPitch < 75) yPos = -yPos;
|
||
|
||
float zPos = -(float) originalPitch / 40;
|
||
if (originalPitch < 75) zPos = -zPos;
|
||
|
||
blockPosition = Vector3f.from(xPos, yPos, zPos);
|
||
}
|
||
|
||
return blockPosition;
|
||
}
|
||
}
|