package data;
import discord4j.core.object.Embed;
import discord4j.core.spec.EmbedCreateSpec;
import enums.Language;
import org.jsoup.nodes.Document;
import org.jsoup.nodes.Element;
import org.jsoup.select.Elements;
import util.EmojiManager;
import util.JSoupManager;
import util.Translator;
import util.URLManager;
import java.awt.*;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;
/**
* Created by steve on 14/07/2016.
*/
public class SetDofus implements Embedded {
private String name;
private String level;
private String skinURL;
private String url;
private List<String> bonusTotal;
private String composition;
private String[] bonusPano;
private List<String> recipeTotal;
private SetDofus(String name, String level, List<String> bonusTotal, String skinURL, String url, String composition,
String[] bonusPano, List<String> recipeTotal) {
this.name = name;
this.level = level;
this.skinURL = skinURL;
this.url = url;
this.composition = composition;
this.bonusPano = bonusPano;
this.bonusTotal = bonusTotal;
this.recipeTotal = recipeTotal;
}
@Override
public void decorateEmbedObject(EmbedCreateSpec spec, Language lg) {
spec.setTitle(name)
.setUrl(url)
.setColor(Color.GRAY)
.setThumbnail(skinURL);
if (level != null && ! level.isEmpty())
spec.addField(Translator.getLabel(lg, "set.level"), level, true);
if (composition != null && ! composition.isEmpty())
spec.addField(Translator.getLabel(lg, "set.composition"), composition, true);
for(int i = 0; i < bonusPano.length; i++)
if (bonusPano[i] != null)
spec.addField(Translator.getLabel(lg, "set.bonus.1") + " " + (i + 2)
+ " " + Translator.getLabel(lg, "set.bonus.2"), bonusPano[i], true);
}
@Override
public void decorateMoreEmbedObject(EmbedCreateSpec spec, Language lg) {
spec.setTitle(name)
.setUrl(url)
.setColor(Color.GRAY)
.setImage(skinURL);
if (level != null && ! level.isEmpty())
spec.addField(Translator.getLabel(lg, "set.level"), level, true);
if (composition != null && ! composition.isEmpty())
spec.addField(Translator.getLabel(lg, "set.composition"), composition, true);
if (bonusTotal != null && ! bonusTotal.isEmpty())
for(int i = 0; i < bonusTotal.size(); i++)
spec.addField(Translator.getLabel(lg, "set.bonus.total") + (bonusTotal.size() > 1?
" (" + (i + 1) + "/" + bonusTotal.size() + ")" : "") + " : ",
bonusTotal.get(i), true);
if (! recipeTotal.isEmpty())
for(int i = 0; i < recipeTotal.size(); i++)
spec.addField(Translator.getLabel(lg, "set.recipe") + (recipeTotal.size() > 1?
" (" + (i + 1) + "/" + recipeTotal.size() + ")" : "") + " : ",
recipeTotal.get(i), true);
for(int i = 0; i < bonusPano.length; i++)
spec.addField(Translator.getLabel(lg, "set.bonus.1") + " " + (i + 2)
+ " " + Translator.getLabel(lg, "set.bonus.2"), bonusPano[i], true);
}
public static SetDofus getSet(Language lg, String url) throws IOException {
Document doc = JSoupManager.getDocument(url);
String name = doc.getElementsByClass("ak-return-link").first().text();
String level = doc.getElementsByClass("ak-encyclo-detail-level").first().text()
.replaceAll(Translator.getLabel(lg, "set.extract.level") + " ", "");
Element element = doc.getElementsByClass("ak-encyclo-detail-illu").first().getElementsByTag("img").first();
String skinURL = element.attr("src");
List<String> bonusTotal = new ArrayList<>();
String composition = null;
String[] bonusPano = new String[doc.getElementsByClass("ak-item-list-preview").first().children().size() - 1];
List<String> recipeTotal = new ArrayList<>();
Elements titles = doc.getElementsByClass("ak-panel-title");
for (Element title : titles)
if (title.text().equals(Translator.getLabel(lg, "set.extract.bonus.total")))
bonusTotal = extractStatsFromTitleToList(lg, title.parent());
else if (title.text().equals(Translator.getLabel(lg, "set.extract.bonus"))){
for(int i = 0; i < bonusPano.length; i++)
bonusPano[i] = extractStatsFromTitle(lg, title.parent().getElementsByClass("set-bonus-list set-bonus-" + (i + 2)).first());
}
else if (title.text().equals(Translator.getLabel(lg, "set.extract.composition"))){
StringBuilder st = new StringBuilder();
for(Element tr : title.parent().getElementsByTag("tr")){
Element a = tr.getElementsByClass("ak-set-composition-name").first().getElementsByTag("a").first();
st.append("[").append(a.text()).append("](").append(a.attr("abs:href")).append(") ")
.append(tr.getElementsByClass("ak-set-composition-level").first().text()).append("\n");
}
composition = st.toString();
}
else if (title.text().equals(Translator.getLabel(lg, "set.extract.recipe"))){
Elements lines = title.parent().getElementsByClass("ak-column");
StringBuilder field = new StringBuilder();
for (Element line : lines) {
StringBuilder tmp = new StringBuilder(line.getElementsByClass("ak-front").text()).append(" [")
.append(line.getElementsByClass("ak-title").first().text()).append("](")
.append(line.getElementsByClass("ak-title").first()
.children().first().attr("abs:href")).append(")\n");
if (field.length() + tmp.length() > Embed.Field.MAX_VALUE_LENGTH){
recipeTotal.add(field.toString());
field.setLength(0);
}
else
field.append(tmp.toString());
}
if (field.length() > 0)
recipeTotal.add(field.toString());
}
return new SetDofus(name, level, bonusTotal, URLManager.abs(skinURL), url, composition, bonusPano, recipeTotal);
}
private static String extractStatsFromTitle(Language lg, Element elem) {
if (elem != null){
Elements lines = elem.getElementsByClass("ak-title");
StringBuilder tmp = new StringBuilder();
for (Element line : lines)
tmp.append(EmojiManager.getEmojiForStat(lg, line.text())).append(line.text()).append("\n");
return tmp.toString();
}
return null;
}
private static List<String> extractStatsFromTitleToList(Language lg, Element elem)
{
List<String> values = new ArrayList<>();
Elements lines = elem.getElementsByClass("ak-title");
StringBuilder tmp = new StringBuilder();
for (Element line : lines) {
String value = EmojiManager.getEmojiForStat(lg, line.text()) + line.text() + "\n";
if (value.length() + tmp.length() > Embed.Field.MAX_VALUE_LENGTH){
values.add(tmp.toString());
tmp.setLength(0);
}
else
tmp.append(value);
}
if (tmp.length() > 0)
values.add(tmp.toString());
return values;
}
}
