path: root/src/main/java/io/github/moulberry/notenoughupdates/infopanes/HTMLInfoPane.java

package io.github.moulberry.notenoughupdates.infopanes;

import info.bliki.htmlcleaner.TagNode;
import info.bliki.wiki.filter.Encoder;
import info.bliki.wiki.model.Configuration;
import info.bliki.wiki.model.ImageFormat;
import info.bliki.wiki.model.WikiModel;
import info.bliki.wiki.tags.HTMLBlockTag;
import info.bliki.wiki.tags.HTMLTag;
import info.bliki.wiki.tags.IgnoreTag;
import io.github.moulberry.notenoughupdates.AllowEmptyHTMLTag;
import io.github.moulberry.notenoughupdates.NEUManager;
import io.github.moulberry.notenoughupdates.NEUOverlay;
import io.github.moulberry.notenoughupdates.util.Utils;
import net.minecraft.client.Minecraft;
import net.minecraft.client.gui.FontRenderer;
import net.minecraft.client.gui.ScaledResolution;
import net.minecraft.client.renderer.GlStateManager;
import net.minecraft.client.renderer.texture.DynamicTexture;
import net.minecraft.util.EnumChatFormatting;
import net.minecraft.util.ResourceLocation;

import javax.imageio.ImageIO;
import java.awt.*;
import java.awt.image.BufferedImage;
import java.io.*;
import java.nio.charset.CharsetEncoder;
import java.nio.charset.StandardCharsets;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.TimeUnit;

public class HTMLInfoPane extends TextInfoPane {

    private static WikiModel wikiModel;

    private final int ZOOM_FACTOR = 2;
    private final int IMAGE_WIDTH = 400;
    private final int EXT_WIDTH = 100;

    private ResourceLocation imageTexture = null;
    private BufferedImage imageTemp = null;
    private int imageHeight = 0;
    private int imageWidth = 0;

    /**
     * Creates a wiki model and sets the configuration to work with hypixel-skyblock wikia.
     */
    static {
        Configuration conf = new Configuration();
        conf.addTokenTag("img", new HTMLTag("img"));
        conf.addTokenTag("code", new HTMLTag("code"));
        conf.addTokenTag("span", new AllowEmptyHTMLTag("span"));
        conf.addTokenTag("table", new HTMLBlockTag("table", Configuration.SPECIAL_BLOCK_TAGS+"span|"));
        conf.addTokenTag("infobox", new IgnoreTag("infobox"));
        conf.addTokenTag("tabber", new IgnoreTag("tabber"));
        conf.addTokenTag("kbd", new HTMLTag("kbd"));
        wikiModel = new WikiModel(conf,"https://hypixel-skyblock.fandom.com/wiki/Special:Filepath/${image}",
                "https://hypixel-skyblock.fandom.com/wiki/${title}") {
            {
                TagNode.addAllowedAttribute("style");
                TagNode.addAllowedAttribute("src");
            }

            protected String createImageName(ImageFormat imageFormat) {
                String imageName = imageFormat.getFilename();
                if (imageName.endsWith(".svg")) {
                    imageName += ".png";
                }
                imageName = Encoder.encodeUrl(imageName);
                if (replaceColon()) {
                    imageName = imageName.replace(':', '/');
                }
                return imageName;
            }

            public void parseInternalImageLink(String imageNamespace, String rawImageLink) {
                rawImageLink = rawImageLink.replaceFirst("\\|x([0-9]+)px", "\\|$1x$1px");
                if(!rawImageLink.split("\\|")[0].toLowerCase().endsWith(".jpg")) {
                    super.parseInternalImageLink(imageNamespace, rawImageLink);
                }
            }
        };
    }

    /**
     * Takes a wiki url, uses NEUManager#getWebFile to download the web file and passed that in to #createFromWiki
     */
    public static HTMLInfoPane createFromWikiUrl(NEUOverlay overlay, NEUManager manager, String name,
                                                 String wikiUrl) {
        File f = manager.getWebFile(wikiUrl);
        if(f == null) {
            return new HTMLInfoPane(overlay, manager, "error", "error","Failed to load wiki url: "+ wikiUrl);
        };

        StringBuilder sb = new StringBuilder();
        try(BufferedReader br = new BufferedReader(new InputStreamReader(
                new FileInputStream(f), StandardCharsets.UTF_8))) {
            String l;
            while((l = br.readLine()) != null){
                sb.append(l).append("\n");
            }
        } catch(IOException e) {
            return new HTMLInfoPane(overlay, manager, "error", "error","Failed to load wiki url: "+ wikiUrl);
        }
        return createFromWiki(overlay, manager, name, f.getName(), sb.toString());
    }

    /**
     * Takes raw wikia code and uses Bliki to generate HTML. Lot's of shennanigans to get it to render appropriately.
     * Honestly, I could have just downloaded the raw HTML of the wiki page and displayed that but I wanted
     * a more permanent solution that can be abstracted to work with arbitrary wiki codes (eg. moulberry.github.io/
     * files/neu_help.html).
     */
    public static HTMLInfoPane createFromWiki(NEUOverlay overlay, NEUManager manager, String name, String filename,
                                              String wiki) {
        String[] split = wiki.split("</infobox>");
        wiki = split[split.length - 1]; //Remove everything before infobox
        wiki = wiki.split("<span class=\"navbox-vde\">")[0]; //Remove navbox
        wiki = wiki.split("<table class=\"navbox mw-collapsible\"")[0];
        wiki = "__NOTOC__\n" + wiki; //Remove TOC
        try (PrintWriter out = new PrintWriter(new File(manager.configLocation, "debug/parsed.txt"))) {
            out.println(wiki);
        } catch (IOException e) {
        }
        String html;
        try {
            html = wikiModel.render(wiki);
        } catch(IOException e) {
            return new HTMLInfoPane(overlay, manager, "error", "error", "Could not render wiki.");
        }
        try (PrintWriter out = new PrintWriter(new File(manager.configLocation, "debug/html.txt"))) {
            out.println(html);
        } catch (IOException e) {
        }
        return new HTMLInfoPane(overlay, manager, name, filename, html);
    }

    private String spaceEscape(String str) {
        return str.replace(" ", "\\ ");
    }

    /**
     * Uses the wkhtmltoimage command-line tool to generate an image from the HTML code. This
     * generation is done asynchronously as sometimes it can take up to 10 seconds for more
     * complex webpages.
     */
    public HTMLInfoPane(NEUOverlay overlay, NEUManager manager, String name, String filename, String html) {
        super(overlay, manager, name, "");
        this.title = name;

        File cssFile = new File(manager.configLocation, "wikia.css");
        File wkHtmlToImage = new File(manager.configLocation, "wkhtmltox/bin/wkhtmltoimage");
        File input = new File(manager.configLocation, "tmp/input.html");
        String outputFileName = filename.replaceAll("(?i)\\u00A7.", "")
                .replaceAll("[^a-zA-Z0-9_\\-]", "_");
        File output = new File(manager.configLocation, "tmp/"+
                outputFileName+".png");
        File outputExt = new File(manager.configLocation, "tmp/"+
                outputFileName+"_ext.png");

        input.deleteOnExit();
        output.deleteOnExit();

        File tmp = new File(manager.configLocation, "tmp");
        if(!tmp.exists()) {
            tmp.mkdir();
        }

        if(output.exists()) {
            try {
                imageTemp = ImageIO.read(output);
                text = EnumChatFormatting.RED+"Creating dynamic texture.";
            } catch(IOException e) {
                e.printStackTrace();
                text = EnumChatFormatting.RED+"Failed to read image.";
                return;
            }
        } else {
            html = "<div id=\"mw-content-text\" lang=\"en\" dir=\"ltr\" class=\"mw-content-ltr mw-content-text\">"+html+"</div>";
            html = "<div id=\"WikiaArticle\" class=\"WikiaArticle\">"+html+"</div>";
            html = "<link rel=\"stylesheet\" href=\"file:///"+cssFile.getAbsolutePath()+"\">\n"+html;

            try(PrintWriter out = new PrintWriter(new BufferedWriter(new OutputStreamWriter(
                    new FileOutputStream(input), StandardCharsets.UTF_8)), false)) {

                out.println(encodeNonAscii(html));
            } catch(IOException e) {}


            ExecutorService ste = Executors.newSingleThreadExecutor();
            try {
                text = EnumChatFormatting.GRAY+"Rendering webpage (" + name + EnumChatFormatting