package eu.faircode.email;

/*
    This file is part of FairEmail.

    FairEmail is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    NetGuard is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with NetGuard.  If not, see <http://www.gnu.org/licenses/>.

    Copyright 2018 by Marcel Bokhorst (M66B)
*/

import android.content.Context;
import android.content.SharedPreferences;
import android.preference.PreferenceManager;
import android.text.Html;
import android.text.TextUtils;

import org.jsoup.Jsoup;
import org.jsoup.helper.StringUtil;
import org.jsoup.nodes.Document;
import org.jsoup.nodes.Element;
import org.jsoup.nodes.Node;
import org.jsoup.nodes.TextNode;
import org.jsoup.safety.Whitelist;
import org.jsoup.select.NodeTraversor;
import org.jsoup.select.NodeVisitor;

import java.util.ArrayList;
import java.util.List;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

public class HtmlHelper implements NodeVisitor {
    private Context context;
    private String newline;
    private List<String> refs = new ArrayList<>();
    private StringBuilder sb = new StringBuilder();

    private static Pattern pattern = Pattern.compile("([http|https]+://[\\w\\S(\\.|:|/)]+)");

    private HtmlHelper(Context context, boolean reply) {
        this.context = context;
        this.newline = (reply ? "<br>> " : "<br>");
    }

    public void head(Node node, int depth) {
        String name = node.nodeName();
        if (node instanceof TextNode) {
            String text = ((TextNode) node).text();
            Matcher matcher = pattern.matcher(text);
            while (matcher.find()) {
                String ref = matcher.group();
                if (!refs.contains(ref))
                    refs.add(ref);
                String alt = context.getString(R.string.title_link);
                text = text.replace(ref, String.format("<a href=\"%s\">%s [%d]</a>", ref, alt, refs.size()));
            }
            sb.append(text);
        } else if (name.equals("li"))
            sb.append(newline).append(" * ");
        else if (name.equals("dt"))
            sb.append("  ");
        else if (StringUtil.in(name, "p", "h1", "h2", "h3", "h4", "h5", "tr", "div"))
            sb.append(newline);
    }

    public void tail(Node node, int depth) {
        String name = node.nodeName();
        if (StringUtil.in(name, "br", "dd", "dt", "p", "h1", "h2", "h3", "h4", "h5", "div"))
            sb.append(newline);
        else if (name.equals("a")) {
            String ref = node.absUrl("href");
            if (!TextUtils.isEmpty(ref)) {
                if (!refs.contains(ref))
                    refs.add(ref);
                String alt = node.attr("alt");
                if (TextUtils.isEmpty(alt))
                    alt = context.getString(R.string.title_link);
                alt = Html.escapeHtml(alt);
                sb.append(" ").append(String.format("<a href=\"%s\">%s [%d]</a>", ref, alt, refs.size()));
            }
        } else if (name.equals("img")) {
            String ref = node.absUrl("src");
            if (!TextUtils.isEmpty(ref)) {
                if (!refs.contains(ref))
                    refs.add(ref);
                String alt = node.attr("alt");
                if (TextUtils.isEmpty(alt))
                    alt = context.getString(R.string.title_image);
                alt = Html.escapeHtml(alt);
                sb.append(" ").append(String.format("<a href=\"%s\">%s [%d]</a>", ref, alt, refs.size()));
                sb.append("<img src=\"" + ref + "\" alt=\"" + alt + "\">");
            }
        }
    }

    @Override
    public String toString() {
        if (refs.size() > 0)
            sb.append(newline).append(newline);
        for (int i = 0; i < refs.size(); i++)
            sb.append(String.format("[%d] %s ", i + 1, refs.get(i))).append(newline);
        return sb.toString();
    }

    public static String sanitize(Context context, String html, boolean reply) {
        SharedPreferences prefs = PreferenceManager.getDefaultSharedPreferences(context);

        if (prefs.getBoolean("sanitize", false)) {
            Document document = Jsoup.parse(html);
            HtmlHelper visitor = new HtmlHelper(context, reply);
            NodeTraversor.traverse(visitor, document.body());
            return visitor.toString();
        } else {
            Document document = Jsoup.parse(Jsoup.clean(html, Whitelist.relaxed()));
            for (Element tr : document.select("tr"))
                tr.after("<br>");
            NodeTraversor.traverse(new NodeVisitor() {
                @Override
                public void head(Node node, int depth) {
                    if (node instanceof TextNode) {
                        String text = ((TextNode) node).text();
                        Matcher matcher = pattern.matcher(text);
                        while (matcher.find()) {
                            String ref = matcher.group();
                            text = text.replace(ref, String.format("<a href=\"%s\">%s</a>", ref, ref));
                        }
                        node.before(text);
                        ((TextNode) node).text("");
                    }
                }

                @Override
                public void tail(Node node, int depth) {
                }
            }, document.body());
            return document.body().html();
        }
    }
}