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

package com.eu.habbo.habbohotel.items;

import com.eu.habbo.Emulator;
import com.google.gson.JsonArray;
import com.google.gson.JsonElement;
import com.google.gson.JsonObject;
import com.google.gson.JsonParser;
import gnu.trove.map.hash.THashMap;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import javax.net.ssl.HttpsURLConnection;
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.net.URL;
import java.sql.*;
import java.time.Duration;
import java.util.ArrayList;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.TimeUnit;

public class YoutubeManager {
    private static final Logger LOGGER = LoggerFactory.getLogger(YoutubeManager.class);

    public static class YoutubeVideo {
        private final String id;
        private final int duration;

        YoutubeVideo(String id, int duration) {
            this.id = id;
            this.duration = duration;
        }

        public String getId() {
            return id;
        }

        public int getDuration() {
            return duration;
        }
    }

    public static class YoutubePlaylist {
        private final String id;
        private final String name;
        private final String description;
        private final ArrayList<YoutubeVideo> videos;

        YoutubePlaylist(String id, String name, String description, ArrayList<YoutubeVideo> videos) {
            this.id = id;
            this.name = name;
            this.description = description;
            this.videos = videos;
        }

        public String getId() {
            return id;
        }

        public String getName() {
            return name;
        }

        public String getDescription() {
            return description;
        }

        public ArrayList<YoutubeVideo> getVideos() {
            return videos;
        }
    }

    private final THashMap<Integer, ArrayList<YoutubePlaylist>> playlists = new THashMap<>();
    private final THashMap<String, YoutubePlaylist> playlistCache = new THashMap<>();
    private final String apiKey = Emulator.getConfig().getValue("youtube.apikey");

    public void load() {
        this.playlists.clear();
        this.playlistCache.clear();

        long millis = System.currentTimeMillis();

        Emulator.getThreading().run(() -> {
            ExecutorService youtubeDataLoaderPool = Executors.newFixedThreadPool(10);

            LOGGER.info("YouTube Manager -> Loading...");

            try (Connection connection = Emulator.getDatabase().getDataSource().getConnection(); PreparedStatement statement = connection.prepareStatement("SELECT * FROM youtube_playlists")) {
                try (ResultSet set = statement.executeQuery()) {
                    while (set.next()) {
                        final int itemId = set.getInt("item_id");
                        final String playlistId = set.getString("playlist_id");

                        youtubeDataLoaderPool.submit(() -> {
                            ArrayList<YoutubePlaylist> playlists = this.playlists.getOrDefault(itemId, new ArrayList<>());

                            YoutubePlaylist playlist;

                            try {
                                playlist = this.getPlaylistDataById(playlistId);
                                if (playlist != null) {
                                    playlists.add(playlist);
                                }
                            } catch (IOException e) {
                                LOGGER.error("Failed to load YouTube playlist {} ERROR: {}", playlistId, e);
                            }

                            this.playlists.put(itemId, playlists);
                        });
                    }
                }
            } catch (SQLException e) {
                LOGGER.error("Caught SQL exception", e);
            }

            youtubeDataLoaderPool.shutdown();
            try {
                youtubeDataLoaderPool.awaitTermination(60, TimeUnit.SECONDS);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }

            LOGGER.info("YouTube Manager -> Loaded! (" + (System.currentTimeMillis() - millis) + " MS)");
        });
    }

    public YoutubePlaylist getPlaylistDataById(String playlistId) throws IOException {
        if (this.playlistCache.containsKey(playlistId)) return this.playlistCache.get(playlistId);
        if(apiKey.isEmpty()) return null;

        YoutubePlaylist playlist;
        URL playlistInfo = new URL("https://youtube.googleapis.com/youtube/v3/playlists?part=snippet&id=" + playlistId + "&maxResults=1&key=" + apiKey);
        HttpsURLConnection playlistCon = (HttpsURLConnection) playlistInfo.openConnection();
        if (playlistCon.getResponseCode() != 200) {
            InputStream errorInputStream = playlistCon.getErrorStream();
            InputStreamReader playlistISR = new InputStreamReader(errorInputStream);
            BufferedReader playlistBR = new BufferedReader(playlistISR);
            JsonObject errorObj = JsonParser.parseReader(playlistBR).getAsJsonObject();
            String message = errorObj.get("error").getAsJsonObject().get("message").getAsString();
            LOGGER.error("Failed to load YouTube playlist {} ERROR: {}", playlistId, message);
            return null;
        }
        InputStream playlistInputStream = playlistCon.getInputStream();
        InputStreamReader playlistISR = new InputStreamReader(playlistInputStream);
        BufferedReader playlistBR = new BufferedReader(playlistISR);

        JsonObject playlistData = JsonParser.parseReader(playlistBR).getAsJsonObject();

        JsonArray playlists = playlistData.get("items").getAsJsonArray();
        if (playlists.size() == 0) {
            LOGGER.error("Playlist {} not found!", playlistId);
            return null;
        }
        JsonObject playlistItem = playlists.get(0).getAsJsonObject().get("snippet").getAsJsonObject();

        String name = playlistItem.get("title").getAsString();
        String description = playlistItem.get("description").getAsString();

        ArrayList < YoutubeVideo > videos = new ArrayList < > ();
        String nextPageToken = "";
        do {
            ArrayList < String > videoIds = new ArrayList < > ();
            URL playlistItems;

            if (nextPageToken.isEmpty()) {
                playlistItems = new URL("https://youtube.googleapis.com/youtube/v3/playlistItems?part=snippet%2Cstatus&playlistId=" + playlistId + "&maxResults=50&key=" + apiKey);
            } else {
                playlistItems = new URL("https://youtube.googleapis.com/youtube/v3/playlistItems?part=snippet%2Cstatus&playlistId=" + playlistId + "&pageToken=" + nextPageToken + "&maxResults=50&key=" + apiKey);
            }

            HttpsURLConnection con = (HttpsURLConnection) playlistItems.openConnection();

            InputStream is = con.getInputStream();
            InputStreamReader isr = new InputStreamReader(is);
            BufferedReader br = new BufferedReader(isr);
            JsonObject object = JsonParser.parseReader(br).getAsJsonObject();

            JsonArray rawV = object.get("items").getAsJsonArray();
            for (JsonElement rawVideo: rawV) {
                JsonObject videoData = rawVideo.getAsJsonObject().get("snippet").getAsJsonObject();
                JsonObject videoStatus = rawVideo.getAsJsonObject().get("status").getAsJsonObject();
                if (!videoStatus.get("privacyStatus").getAsString().equals("public"))
                    continue; // removed videos
                videoIds.add(videoData.get("resourceId").getAsJsonObject().get("videoId").getAsString());
            }

            if (!videoIds.isEmpty()) {
                URL VideoItems;

                String commaSeparatedVideos = String.join(",", videoIds);

                VideoItems = new URL("https://youtube.googleapis.com/youtube/v3/videos?part=contentDetails&id=" + commaSeparatedVideos + "&maxResults=50&key=" + apiKey);
                HttpsURLConnection con1 = (HttpsURLConnection) VideoItems.openConnection();
                InputStream is1 = con1.getInputStream();
                InputStreamReader isr1 = new InputStreamReader(is1);
                BufferedReader br1 = new BufferedReader(isr1);
                JsonObject object1 = JsonParser.parseReader(br1).getAsJsonObject();
                JsonArray Vds = object1.get("items").getAsJsonArray();
                for (JsonElement rawVideo: Vds) {
                    JsonObject contentDetails = rawVideo.getAsJsonObject().get("contentDetails").getAsJsonObject();
                    int duration = (int) Duration.parse(contentDetails.get("duration").getAsString()).getSeconds();
                    if (duration < 1) continue;
                    videos.add(new YoutubeVideo(rawVideo.getAsJsonObject().get("id").getAsString(), duration));
                }
            }
            if (object.has("nextPageToken")) {
                nextPageToken = object.get("nextPageToken").getAsString();
            } else {
                nextPageToken = null;
            }
        } while (nextPageToken != null);

        if (videos.isEmpty()) {
            LOGGER.warn("Playlist {} has no videos!", playlistId);
            return null;
        }
        playlist = new YoutubePlaylist(playlistId, name, description, videos);

        this.playlistCache.put(playlistId, playlist);

        return playlist;

    }

    public ArrayList<YoutubePlaylist> getPlaylistsForItemId(int itemId) {
        return this.playlists.get(itemId);
    }

    public void addPlaylistToItem(int itemId, YoutubePlaylist playlist) {
        this.playlists.computeIfAbsent(itemId, k -> new ArrayList<>()).add(playlist);
    }
}