efibs / mc-lasertag-mod

public class WebStatisticsVisualizer {

    private static final Path TARGET_PATH;

    // Initialize target path

    static {

        Path target;

        // I have to use try/catch because for some reason FabricLoader dosn't support checking if it is initialized correctly

        try {

            target = Path.of(FabricLoader.getInstance().getGameDir().toString(), "lasertag_last_games_stats");

        } catch (IllegalStateException ignored) {

            target = Path.of(System.getProperty("user.dir"), "build");

        }

        TARGET_PATH = target;

    }

    private static final Identifier REPLACE_ID = new Identifier(LasertagMod.ID, "statistics_go_here");

    public static Path build(GameStats stats, TeamDto winningTeam, GameMode gameMode, WebResourceManager resourceManager) {

        // The path to the generated index.html file

        Path resultPath = null;

        // Get web page template from resource manager

        var template = resourceManager.getWebSite(new Identifier("web/statistics_template"));

        // Check

        if (template == null) {

            return null;

        }

        // Build web page

        for (var fileTuple : template) {

            // if it is the html file

            if (fileTuple.x().getPath().endsWith(".html")) {

                // Read the file

                String fileContents;

                try {

                    fileContents = fileTuple.y().getReader().lines().collect(Collectors.joining());

                } catch (IOException ex) {

                    LasertagMod.LOGGER.error("Reading of statistics template failed:", ex);

                    return null;

                }

                fileContents = buildHtml(fileContents, stats, winningTeam, gameMode);

                // Write file

                resultPath = TARGET_PATH.resolve(fileTuple.x().getPath());

                try {

                    Files.createDirectories(resultPath.getParent());

                    Files.writeString(resultPath, fileContents, StandardOpenOption.TRUNCATE_EXISTING, StandardOpenOption.CREATE);

                } catch (IOException ex) {

                    LasertagMod.LOGGER.error("Writing of statistics file failed:", ex);

                    return null;

                }

                continue;

            }

            try {

                var copyToPath = TARGET_PATH.resolve(fileTuple.x().getPath());

                Files.createDirectories(copyToPath.getParent());

                Files.copy(fileTuple.y().getInputStream(), copyToPath, StandardCopyOption.REPLACE_EXISTING);

            } catch (Exception ex) {

                LasertagMod.LOGGER.error("Failed to copy file '" + fileTuple.x().getPath() + "' of statistics visualization.", ex);

                return null;

            }

        }

        return resultPath;

    }

    private static String buildHtml(String html, GameStats stats, TeamDto winningTeam, GameMode gameMode) {

        // build html to insert

        var builder = new StringBuilder();

        buildOverTableHeader(builder, winningTeam, gameMode);

        buildTeamScores(builder, stats);

        buildPlayerScores(builder, stats);

        buildTeamByPlayersScores(builder, stats);

        // Find and replace

        return html.replaceAll("#" + REPLACE_ID + "#", builder.toString());

    }

    private static void buildOverTableHeader(StringBuilder builder, TeamDto winningTeam, GameMode gameMode) {

        // Get the winning teams color

        var winningTeamColor = winningTeam.color();

        builder.append("<div class=\"container over-table-header\"><div><h1><i>Winner: <span style=\"color:rgb(");

        // Winner team

        builder.append(winningTeamColor.r());

        builder.append(",");

        builder.append(winningTeamColor.g());

        builder.append(",");

        builder.append(winningTeamColor.b());

        builder.append(");\"><b>");

        builder.append(winningTeam.name());

        builder.append("</b></span></i></h1></div>");

        // Game mode

        builder.append("<div class=\"game-mode-container\"><h2 class=\"game-mode-text\"><i>");

        builder.append(I18n.translate(gameMode.getTranslatableName()));

        builder.append("</i></h2></div>");

        builder.append("</div>");

    }

    private static void buildTeamScores(StringBuilder builder, GameStats stats) {

        String gameDurationString = null;

        if (stats.gameDurationSeconds != 0L) {

            gameDurationString = "Game duration: " +

                    DurationUtils.toMinuteString(Duration.ofSeconds(stats.gameDurationSeconds)) +

                    " minutes";

        }

        buildTableHeader(builder, "Team scores", "Team", gameDurationString);

        int teamNo = 1;

        for (var team : stats.teamScores) {

            buildTableRow(builder, teamNo++, team.x(), team.y());

        }

        buildTableEnd(builder);

    }

    private static void buildPlayerScores(StringBuilder builder, GameStats stats) {

        buildTableHeader(builder, "Player scores", "Player", null);

        int playerNo = 1;

        for (var player : stats.playerScores) {

            var playerHtml = getSkinNameHtml(player.x());

            buildTableRow(builder, playerNo++, playerHtml, player.y());

        }

        buildTableEnd(builder);

    }

    private static void buildTeamByPlayersScores(StringBuilder builder, GameStats stats) {

        for (var team : stats.teamPlayerScores.entrySet()) {

            buildTableHeader(builder, team.getKey(), "Player", null);

            int playerNo = 1;

            for (var player : team.getValue()) {

                var playerHtml = getSkinNameHtml(player.x());

                buildTableRow(builder, playerNo++, playerHtml, player.y());

            }

            buildTableEnd(builder);

        }

    }

    private static void buildTableRow(StringBuilder builder, int place, String nameHtml, String score) {

        builder.append("<tr><th scope=\"row\">");

        builder.append(place);

        builder.append("</th><td>");

        builder.append(nameHtml);

        builder.append("</td><td>");

        builder.append(score);

        builder.append("</td></tr>");

    }

    private static void buildTableHeader(StringBuilder builder, String tableTitle, String nameColHeader, String headerAddition) {

        builder.append("<div class=\"container py-5\">");

        if (headerAddition != null) {

            builder.append("<div class=\"header-addition\">");

            builder.append(headerAddition);

            builder.append("</div>");

        }

        builder.append("<h5 class=\"h5\">");

        builder.append(tableTitle);

        builder.append("</h5><table class=\"table text-light\"><col width=\"15%\"/><thead><tr><th scope=\"col\">#</th><th scope=\"col\">");

        builder.append(nameColHeader);

        builder.append("</th><th scope=\"col\">Score</th></tr></thead><tbody>");

    }

    private static void buildTableEnd(StringBuilder builder) {

        builder.append("</tbody></table></div>");

    }

    private static String getSkinNameHtml(String playerName) {

        try {

            var skinUrl = getSkinUrlFromPlayerName(playerName);

            return  "<div class=\"face-container\"><img src='" + skinUrl + "'></div>\n\n" + playerName;

        } catch (Exception ignored) {

            return "<div class=\"face-container\"></div>" + playerName;

        }

    }

    private static String getSkinUrlFromPlayerName(String playerName) throws Exception {

        // Url to player info

        var playerInfoUrl = new URL("https://api.mojang.com/users/profiles/minecraft/" + playerName);

        // Create input stream

        var playerInfoInputStreamReader = new InputStreamReader(playerInfoUrl.openStream());

        // Get uuid of player

        var uuid = new Gson().fromJson(playerInfoInputStreamReader, PlayerInfoDto.class).id;

        // Url to session profile of player

        var sessionProfileUrl = new URL("https://sessionserver.mojang.com/session/minecraft/profile/" + uuid);

        // Create input stream

        var sessionProfileInputStreamReader = new InputStreamReader(sessionProfileUrl.openStream());

        // Get base64 encoded texture json

        var encodedTextureJson = new Gson().fromJson(sessionProfileInputStreamReader, SessionPlayerProfileDto.class).properties[0].value;

        // Decode

        var decodedTextureJson = new String(Base64.getDecoder().decode(encodedTextureJson), StandardCharsets.UTF_8);

        // Get skin url

        return new Gson().fromJson(decodedTextureJson, ProfileTextureDto.class).textures.get("SKIN").url;