mirror of https://github.com/M66B/FairEmail.git
96 lines
3.3 KiB
Java
96 lines
3.3 KiB
Java
package eu.faircode.email;
|
|
|
|
/*
|
|
This file is part of Safe email.
|
|
|
|
Safe email 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.text.TextUtils;
|
|
|
|
import org.jsoup.Jsoup;
|
|
import org.jsoup.helper.StringUtil;
|
|
import org.jsoup.nodes.Document;
|
|
import org.jsoup.nodes.Node;
|
|
import org.jsoup.nodes.TextNode;
|
|
import org.jsoup.select.NodeTraversor;
|
|
import org.jsoup.select.NodeVisitor;
|
|
|
|
import java.util.ArrayList;
|
|
import java.util.List;
|
|
|
|
public class HtmlHelper implements NodeVisitor {
|
|
private Context context;
|
|
private String newline;
|
|
private List<String> links = new ArrayList<>();
|
|
private StringBuilder sb = new StringBuilder();
|
|
|
|
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)
|
|
sb.append(((TextNode) node).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 link = node.absUrl("href");
|
|
if (!TextUtils.isEmpty(link)) {
|
|
if (!links.contains(link))
|
|
links.add(link);
|
|
sb.append(" ").append(context.getString(R.string.title_link, link, links.size()));
|
|
}
|
|
} else if (name.equals("img")) {
|
|
String link = node.absUrl("src");
|
|
if (!TextUtils.isEmpty(link)) {
|
|
if (!links.contains(link))
|
|
links.add(link);
|
|
sb.append(" ").append(context.getString(R.string.title_image, link, links.size()));
|
|
}
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public String toString() {
|
|
if (links.size() > 0)
|
|
sb.append(newline).append(newline);
|
|
for (int i = 0; i < links.size(); i++)
|
|
sb.append(String.format("[%d] %s ", i + 1, links.get(i))).append(newline);
|
|
return sb.toString();
|
|
}
|
|
|
|
public static String sanitize(Context context, String html, boolean reply) {
|
|
Document document = Jsoup.parse(html);
|
|
HtmlHelper visitor = new HtmlHelper(context, reply);
|
|
NodeTraversor.traverse(visitor, document.body());
|
|
return visitor.toString();
|
|
}
|
|
}
|