Arcturus-Community/src/main/java/com/eu/habbo/habbohotel/items/YoutubeManager.java

package com.eu.habbo.habbohotel.items;

import com.eu.habbo.Emulator;
import gnu.trove.map.hash.THashMap;

import java.sql.Connection;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Statement;
import java.util.ArrayList;

public class YoutubeManager {
    public THashMap<Integer, ArrayList<YoutubeItem>> playLists = new THashMap<>();
    public THashMap<Integer, YoutubeItem> videos = new THashMap<>();

    public void load() {
        this.videos.clear();
        this.playLists.clear();

        try (Connection connection = Emulator.getDatabase().getDataSource().getConnection()) {
            try (Statement statement = connection.createStatement()) {
                try (ResultSet set = statement.executeQuery("SELECT * FROM youtube_items")) {
                    while (set.next()) {
                        this.videos.put(set.getInt("id"), new YoutubeItem(set));
                    }
                }

                try (ResultSet set = statement.executeQuery("SELECT * FROM youtube_playlists ORDER BY `order` ASC")) {
                    while (set.next()) {
                        if (!this.playLists.containsKey(set.getInt("item_id"))) {
                            this.playLists.put(set.getInt("item_id"), new ArrayList<>());
                        }

                        YoutubeItem item = this.videos.get(set.getInt("video_id"));

                        if (item != null) {
                            this.playLists.get(set.getInt("item_id")).add(item);
                        }
                    }
                }
            }
        } catch (SQLException e) {
            Emulator.getLogging().logSQLException(e);
        }
    }

    public ArrayList<YoutubeItem> getPlaylist(Item item) {
        if (this.playLists.containsKey(item.getId())) {
            return this.playLists.get(item.getId());
        }

        return new ArrayList<>();
    }

    public YoutubeItem getVideo(Item item, String video) {
        if (this.playLists.contains(item.getId())) {
            for (YoutubeItem v : this.playLists.get(item.getId())) {
                if (v.video.equalsIgnoreCase(video)) {
                    return v;
                }
            }
        }

        return null;
    }

    public String getPreviewImage(Item item) {
        if (this.playLists.contains(item.getId())) {
            if (!this.playLists.get(item.getId()).isEmpty()) {
                return this.playLists.get(item.getId()).get(0).video;
            }
        }

        return "";
    }

    public YoutubeItem getVideo(Item item, int index) {
        if (this.playLists.containsKey(item.getId())) {
            return this.playLists.get(item.getId()).get(index);
        }

        return null;
    }

    public class YoutubeItem {
        public final int id;
        public final String video;
        public final String title;
        public final String description;
        public final int startTime;
        public final int endTime;

        public YoutubeItem(ResultSet set) throws SQLException {
            this.id = set.getInt("id");
            this.video = set.getString("video");
            this.title = set.getString("title");
            this.description = set.getString("description");
            this.startTime = set.getInt("start_time");
            this.endTime = set.getInt("end_time");
        }
    }
}