Arcturus-Community/src/main/java/com/eu/habbo/habbohotel/rooms/TraxManager.java
2020-07-02 15:44:13 -05:00

445 lines
17 KiB
Java

package com.eu.habbo.habbohotel.rooms;
import com.eu.habbo.Emulator;
import com.eu.habbo.core.Disposable;
import com.eu.habbo.habbohotel.achievements.AchievementManager;
import com.eu.habbo.habbohotel.gameclients.GameClient;
import com.eu.habbo.habbohotel.items.SoundTrack;
import com.eu.habbo.habbohotel.items.interactions.InteractionJukeBox;
import com.eu.habbo.habbohotel.items.interactions.InteractionMusicDisc;
import com.eu.habbo.habbohotel.users.Habbo;
import com.eu.habbo.habbohotel.users.HabboItem;
import com.eu.habbo.messages.ServerMessage;
import com.eu.habbo.messages.outgoing.generic.alerts.BubbleAlertComposer;
import com.eu.habbo.messages.outgoing.inventory.AddHabboItemComposer;
import com.eu.habbo.messages.outgoing.inventory.InventoryRefreshComposer;
import com.eu.habbo.messages.outgoing.inventory.RemoveHabboItemComposer;
import com.eu.habbo.messages.outgoing.rooms.items.jukebox.JukeBoxMySongsComposer;
import com.eu.habbo.messages.outgoing.rooms.items.jukebox.JukeBoxNowPlayingMessageComposer;
import com.eu.habbo.messages.outgoing.rooms.items.jukebox.JukeBoxPlayListComposer;
import gnu.trove.map.hash.THashMap;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.List;
import java.util.stream.Collectors;
public class TraxManager implements Disposable {
public static int NORMAL_JUKEBOX_LIMIT = 10;
public static int LARGE_JUKEBOX_LIMIT = 20;
private static final Logger LOGGER = LoggerFactory.getLogger(TraxManager.class);
private final Room room;
private InteractionJukeBox jukeBox;
private final List<InteractionMusicDisc> songs = new ArrayList<>(0);
private int totalLength = 0;
private int startedTimestamp = 0;
private InteractionMusicDisc currentlyPlaying = null;
private int playingIndex = 0;
private int cycleStartedTimestamp = 0;
private Habbo starter = null;
private int songsLimit = 0;
private boolean disposed = false;
public TraxManager(Room room) {
this.room = room;
//Check if room has a Jukebox already on DB
this.jukeBox = this.loadRoomJukebox();
if (this.jukeBox == null) {
//Check again if there's a jukebox on room but has not been saved on DB before
for (HabboItem item : room.getRoomSpecialTypes().getItemsOfType(InteractionJukeBox.class))
{
this.jukeBox = (InteractionJukeBox) item;
}
if(this.jukeBox != null)
{
this.loadPlaylist();
this.songsLimit = this.getSongsLimit(this.jukeBox);
}
} else {
this.loadPlaylist();
this.songsLimit = this.getSongsLimit(this.jukeBox);
}
}
public InteractionJukeBox loadRoomJukebox() {
try (Connection connection = Emulator.getDatabase().getDataSource().getConnection(); PreparedStatement statement = connection.prepareStatement("SELECT * FROM room_trax WHERE room_id = ?")) {
statement.setInt(1, this.room.getId());
try (ResultSet set = statement.executeQuery()) {
if (set.next()) {
HabboItem jukebox = Emulator.getGameEnvironment().getItemManager().loadHabboItem(set.getInt("trax_item_id"));
if(jukebox != null) {
if (!(jukebox instanceof InteractionJukeBox)) {
return null;
} else {
return (InteractionJukeBox) jukebox;
}
}
}
}
} catch (SQLException e) {
LOGGER.error("Caught SQL exception", e);
}
return null;
}
public void loadPlaylist() {
if(this.jukeBox == null) return;
this.songs.clear();
try (Connection connection = Emulator.getDatabase().getDataSource().getConnection(); PreparedStatement statement = connection.prepareStatement("SELECT * FROM trax_playlist WHERE trax_item_id = ?")) {
statement.setInt(1, this.jukeBox.getId());
try (ResultSet set = statement.executeQuery()) {
while (set.next()) {
HabboItem musicDisc = Emulator.getGameEnvironment().getItemManager().loadHabboItem(set.getInt("item_id"));
if(musicDisc != null) {
if (!(musicDisc instanceof InteractionMusicDisc) || musicDisc.getRoomId() != -1) {
deleteSongFromPlaylist(this.jukeBox.getId(), musicDisc.getId());
} else {
SoundTrack track = Emulator.getGameEnvironment().getItemManager().getSoundTrack(((InteractionMusicDisc) musicDisc).getSongId());
if (track != null) {
this.songs.add((InteractionMusicDisc) musicDisc);
this.totalLength += track.getLength();
}
}
}
}
}
} catch (SQLException e) {
LOGGER.error("Caught SQL exception", e);
}
}
public static void deleteSongFromPlaylist(int jukebox_id, int song_id)
{
try (Connection connection = Emulator.getDatabase().getDataSource().getConnection(); PreparedStatement statement = connection.prepareStatement("DELETE FROM trax_playlist WHERE trax_item_id = ? AND item_id = ? LIMIT 1")) {
statement.setInt(1, jukebox_id);
statement.setInt(2, song_id);
statement.execute();
} catch (SQLException e) {
LOGGER.error("Caught SQL exception", e);
}
}
public void addTraxOnRoom(InteractionJukeBox jukeBox) {
if(this.jukeBox != null) return;
try (Connection connection = Emulator.getDatabase().getDataSource().getConnection(); PreparedStatement statement_1 = connection.prepareStatement("INSERT INTO room_trax (room_id, trax_item_id) VALUES (?, ?)"))
{
statement_1.setInt(1, this.room.getId());
statement_1.setInt(2, jukeBox.getId());
statement_1.execute();
}
catch (SQLException e) {
LOGGER.error("Caught SQL exception", e);
return;
}
this.jukeBox = jukeBox;
this.loadPlaylist();
this.songsLimit = this.getSongsLimit(this.jukeBox);
}
public void removeTraxOnRoom(InteractionJukeBox jukeBox) {
if(this.jukeBox.getId() != jukeBox.getId()) return;
try (Connection connection = Emulator.getDatabase().getDataSource().getConnection(); PreparedStatement statement_1 = connection.prepareStatement("DELETE FROM room_trax WHERE room_id = ?"))
{
statement_1.setInt(1, this.room.getId());
statement_1.execute();
}
catch (SQLException e) {
LOGGER.error("Caught SQL exception", e);
return;
}
this.stop();
this.jukeBox = null;
this.songs.clear();
}
public void cycle() {
if (this.isPlaying()) {
if (this.timePlaying() >= this.totalLength) {
this.play(0);
//restart
}
if (this.currentSong() != null && Emulator.getIntUnixTimestamp() >= this.startedTimestamp + this.currentSong().getLength()) {
this.play((this.playingIndex + 1) % this.songs.size());
}
}
}
public void play(int index) {
this.play(index, null);
}
public void play(int index, Habbo starter) {
if (this.currentlyPlaying == null) {
this.jukeBox.setExtradata("1");
this.room.updateItem(this.jukeBox);
}
if (!this.songs.isEmpty()) {
index = index % this.songs.size();
this.currentlyPlaying = this.songs.get(index);
if (this.currentlyPlaying != null) {
this.room.setJukeBoxActive(true);
this.startedTimestamp = Emulator.getIntUnixTimestamp();
this.playingIndex = index;
if (starter != null) {
this.starter = starter;
this.cycleStartedTimestamp = Emulator.getIntUnixTimestamp();
}
}
this.room.sendComposer(new JukeBoxNowPlayingMessageComposer(Emulator.getGameEnvironment().getItemManager().getSoundTrack(this.currentlyPlaying.getSongId()), this.playingIndex, 0).compose());
} else {
this.stop();
}
}
public void stop() {
if (this.starter != null && this.cycleStartedTimestamp > 0) {
AchievementManager.progressAchievement(this.starter, Emulator.getGameEnvironment().getAchievementManager().getAchievement("MusicPlayer"), (Emulator.getIntUnixTimestamp() - cycleStartedTimestamp) / 60);
}
this.room.setJukeBoxActive(false);
this.currentlyPlaying = null;
this.startedTimestamp = 0;
this.cycleStartedTimestamp = 0;
this.starter = null;
this.playingIndex = 0;
this.jukeBox.setExtradata("0");
this.room.updateItem(this.jukeBox);
this.room.sendComposer(new JukeBoxNowPlayingMessageComposer(null, -1, 0).compose());
}
public SoundTrack currentSong() {
if (!this.songs.isEmpty() && this.playingIndex < this.songs.size()) {
return Emulator.getGameEnvironment().getItemManager().getSoundTrack(this.songs.get(this.playingIndex).getSongId());
}
return null;
}
public void addSong(InteractionMusicDisc musicDisc, Habbo habbo) {
if(this.jukeBox == null) return;
if(this.songsLimit < this.songs.size() + 1)
{
THashMap<String, String> codes = new THashMap<>();
ServerMessage msg = new BubbleAlertComposer("${playlist.editor.alert.playlist.full.title}", "${playlist.editor.alert.playlist.full}").compose();
habbo.getClient().sendResponse(msg);
return;
}
SoundTrack track = Emulator.getGameEnvironment().getItemManager().getSoundTrack(musicDisc.getSongId());
if (track != null)
{
this.totalLength += track.getLength();
this.songs.add(musicDisc);
try (Connection connection = Emulator.getDatabase().getDataSource().getConnection(); PreparedStatement statement = connection.prepareStatement("INSERT INTO trax_playlist (trax_item_id, item_id) VALUES (?, ?)")) {
statement.setInt(1, this.jukeBox.getId());
statement.setInt(2, musicDisc.getId());
statement.execute();
} catch (SQLException e) {
LOGGER.error("Caught SQL exception", e);
return;
}
this.room.sendComposer(new JukeBoxPlayListComposer(this.songs, this.totalLength).compose());
musicDisc.setRoomId(-1);
musicDisc.needsUpdate(true);
Emulator.getThreading().run(musicDisc);
habbo.getInventory().getItemsComponent().removeHabboItem(musicDisc);
habbo.getClient().sendResponse(new RemoveHabboItemComposer(musicDisc.getGiftAdjustedId()));
}
this.sendUpdatedSongList();
}
public void removeSong(int itemId) {
if(this.songs.isEmpty()) return;
InteractionMusicDisc musicDisc = this.getSong(itemId);
if (musicDisc != null) {
this.songs.remove(musicDisc);
deleteSongFromPlaylist(this.jukeBox.getId(), itemId);
this.totalLength -= Emulator.getGameEnvironment().getItemManager().getSoundTrack(musicDisc.getSongId()).getLength();
if (this.currentlyPlaying == musicDisc) {
this.play(this.playingIndex);
}
this.room.sendComposer(new JukeBoxPlayListComposer(this.songs, this.totalLength).compose());
musicDisc.setRoomId(0);
musicDisc.needsUpdate(true);
Emulator.getThreading().run(musicDisc);
Habbo owner = Emulator.getGameEnvironment().getHabboManager().getHabbo(musicDisc.getUserId());
if (owner != null) {
owner.getInventory().getItemsComponent().addItem(musicDisc);
GameClient client = owner.getClient();
if (client != null) {
client.sendResponse(new AddHabboItemComposer(musicDisc));
client.sendResponse(new InventoryRefreshComposer());
}
}
}
this.sendUpdatedSongList();
}
public static void removeAllSongs(InteractionJukeBox jukebox)
{
try (Connection connection = Emulator.getDatabase().getDataSource().getConnection(); PreparedStatement statement = connection.prepareStatement("SELECT * FROM trax_playlist WHERE trax_item_id = ?")) {
statement.setInt(1, jukebox.getId());
try (ResultSet set = statement.executeQuery()) {
while (set.next()) {
HabboItem musicDisc = Emulator.getGameEnvironment().getItemManager().loadHabboItem(set.getInt("item_id"));
deleteSongFromPlaylist(jukebox.getId(), set.getInt("item_id"));
if(musicDisc != null) {
if (musicDisc instanceof InteractionMusicDisc && musicDisc.getRoomId() == -1) {
musicDisc.setRoomId(0);
musicDisc.needsUpdate(true);
Emulator.getThreading().run(musicDisc);
Habbo owner = Emulator.getGameEnvironment().getHabboManager().getHabbo(musicDisc.getUserId());
if (owner != null) {
owner.getInventory().getItemsComponent().addItem(musicDisc);
GameClient client = owner.getClient();
if (client != null) {
client.sendResponse(new AddHabboItemComposer(musicDisc));
client.sendResponse(new InventoryRefreshComposer());
}
}
}
}
}
}
} catch (SQLException e) {
LOGGER.error("Caught SQL exception", e);
}
}
public List<SoundTrack> soundTrackList() {
List<SoundTrack> trax = new ArrayList<>(this.songs.size());
for (InteractionMusicDisc musicDisc : this.songs) {
SoundTrack track = Emulator.getGameEnvironment().getItemManager().getSoundTrack(musicDisc.getSongId());
if (track != null) {
trax.add(track);
}
}
return trax;
}
public List<InteractionMusicDisc> myList(Habbo habbo) {
return habbo.getInventory().getItemsComponent().getItems().valueCollection().stream()
.filter(i -> i instanceof InteractionMusicDisc && i.getRoomId() == 0)
.map(i -> (InteractionMusicDisc) i)
.collect(Collectors.toList());
}
public InteractionMusicDisc getSong(int itemId) {
for (InteractionMusicDisc musicDisc : this.songs) {
if (musicDisc != null && musicDisc.getId() == itemId) {
return musicDisc;
}
}
return null;
}
public int getSongsLimit(InteractionJukeBox jukeBox) {
if ("jukebox_big".equals(jukeBox.getBaseItem().getName())) {
return LARGE_JUKEBOX_LIMIT;
}
return NORMAL_JUKEBOX_LIMIT;
}
public void updateCurrentPlayingSong(Habbo habbo) {
if (this.isPlaying()) {
habbo.getClient().sendResponse(new JukeBoxNowPlayingMessageComposer(Emulator.getGameEnvironment().getItemManager().getSoundTrack(this.currentlyPlaying.getSongId()), this.playingIndex, 1000 * (Emulator.getIntUnixTimestamp() - this.startedTimestamp)));
} else {
habbo.getClient().sendResponse(new JukeBoxNowPlayingMessageComposer(null, -1, 0));
}
}
public void sendUpdatedSongList() {
this.room.getHabbos().forEach(h -> {
GameClient client = h.getClient();
if (client != null) {
client.sendResponse(new JukeBoxMySongsComposer(this.myList(h)));
}
});
}
public int timePlaying() {
return Emulator.getIntUnixTimestamp() - this.startedTimestamp;
}
public int totalLength() {
return this.totalLength;
}
public List<InteractionMusicDisc> getSongs() {
return this.songs;
}
public boolean isPlaying() {
return this.currentlyPlaying != null;
}
public int getSongsLimit() {
return this.songsLimit;
}
public InteractionJukeBox getJukeBox() {
return this.jukeBox;
}
@Override
public void dispose() {
this.disposed = true;
}
@Override
public boolean disposed() {
return this.disposed;
}
}