sciwhiz12/Janitor
1
0
Fork 0
mirror of https://github.com/sciwhiz12/Janitor.git synced 2024-11-10 02:21:25 +00:00
Code Issues Releases Wiki Activity

Forgot to commit two files

Browse Source
This commit is contained in:
Arnold Alejo Nunag 2020-10-10 16:53:40 +08:00
parent 46eff3358e
commit f861308e4d
Signed by: sciwhiz12
GPG Key ID: 622CF446534317E1
2 changed files with 53 additions and 102 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

102
src/main/java/sciwhiz12/janitor/msg/MessageBuilder.java
View File

@ -1,102 +0,0 @@
package sciwhiz12.janitor.msg;
import net.dv8tion.jda.api.EmbedBuilder;
import net.dv8tion.jda.api.entities.MessageEmbed;
import sciwhiz12.janitor.JanitorBot;
import java.util.HashMap;
import java.util.Map;
import java.util.function.Consumer;
import java.util.function.Supplier;
import java.util.function.UnaryOperator;
import static sciwhiz12.janitor.msg.Substitutions.substitute;
public class MessageBuilder {
private final EmbedBuilder embedBuilder;
private final Map<String, Supplier<String>> substitutions;
public MessageBuilder(EmbedBuilder embedBuilder, Map<String, Supplier<String>> substitutions) {
this.embedBuilder = embedBuilder;
this.substitutions = substitutions;
}
public MessageBuilder() {
this(new EmbedBuilder(), new HashMap<>());
}
public MessageBuilder(MessageBuilder copy) {
this(new EmbedBuilder(copy.embedBuilder), new HashMap<>(copy.substitutions));
}
public EmbedBuilder embed() {
return embedBuilder;
}
public MessageBuilder embed(Consumer<EmbedBuilder> operator) {
operator.accept(embed());
return this;
}
public MessageBuilder apply(Consumer<MessageBuilder> consumer) {
consumer.accept(this);
return this;
}
public MessageBuilder with(final String argument, final Supplier<String> value) {
substitutions.put(argument, value);
return this;
}
public MessageBuilder field(final String head, final boolean inline) {
embedBuilder.addField(head + ".name", head + ".value", inline);
return this;
}
public MessageBuilder blankField(final boolean inline) {
embedBuilder.addBlankField(inline);
return this;
}
public MessageEmbed build(Translations translations, Substitutions substitutions) {
EmbedBuilder realEmbed = new EmbedBuilder();
MessageEmbed tempEmbed = embed().build();
final Map<String, Supplier<String>> replaceMap = substitutions.createDefaultedMap(this.substitutions);
final UnaryOperator<String> replacer = str -> substitute(translations.translate(str), replaceMap);
realEmbed.setColor(tempEmbed.getColorRaw());
realEmbed.setTimestamp(tempEmbed.getTimestamp());
if (tempEmbed.getTitle() != null)
realEmbed.setTitle(replacer.apply(tempEmbed.getTitle()), tempEmbed.getUrl());
if (tempEmbed.getThumbnail() != null)
realEmbed.setThumbnail(tempEmbed.getThumbnail().getUrl());
if (tempEmbed.getAuthor() != null)
realEmbed.setAuthor(
replacer.apply(tempEmbed.getAuthor().getName()),
tempEmbed.getAuthor().getUrl(),
tempEmbed.getAuthor().getIconUrl()
);
if (tempEmbed.getFooter() != null)
realEmbed.setFooter(
replacer.apply(tempEmbed.getFooter().getText()),
tempEmbed.getFooter().getIconUrl()
);
if (tempEmbed.getImage() != null)
realEmbed.setImage(tempEmbed.getImage().getUrl());
if (tempEmbed.getDescription() != null)
realEmbed.setDescription(replacer.apply(tempEmbed.getDescription()));
for (MessageEmbed.Field field : tempEmbed.getFields())
realEmbed.addField(
replacer.apply(field.getName()),
replacer.apply(field.getValue()),
field.isInline()
);
return realEmbed.build();
}
public MessageEmbed build(JanitorBot bot) {
return build(bot.getTranslations(), bot.getSubstitutions());
}
}

53
src/main/java/sciwhiz12/janitor/msg/RegularMessageBuilder.java Normal file
View File

@ -0,0 +1,53 @@
package sciwhiz12.janitor.msg;
import net.dv8tion.jda.api.entities.MessageChannel;
import net.dv8tion.jda.api.entities.MessageEmbed;
import net.dv8tion.jda.api.requests.restaction.MessageAction;
import sciwhiz12.janitor.JanitorBot;
import sciwhiz12.janitor.msg.json.RegularMessage;
import sciwhiz12.janitor.msg.substitution.SubstitutionMap;
import java.util.HashMap;
import java.util.Map;
import java.util.function.Consumer;
import java.util.function.Supplier;
public class RegularMessageBuilder {
private final RegularMessage message;
private final Map<String, Supplier<String>> customSubstitutions;
public RegularMessageBuilder(RegularMessage message, Map<String, Supplier<String>> customSubstitutions) {
this.message = message;
this.customSubstitutions = customSubstitutions;
}
public RegularMessageBuilder(RegularMessage message) {
this(message, new HashMap<>());
}
public RegularMessageBuilder(RegularMessageBuilder copy) {
this(copy.message, new HashMap<>(copy.customSubstitutions));
}
public RegularMessageBuilder apply(Consumer<RegularMessageBuilder> consumer) {
consumer.accept(this);
return this;
}
public RegularMessageBuilder with(final String argument, final Supplier<String> value) {
customSubstitutions.put(argument, value);
return this;
}
public MessageEmbed build(TranslationMap translations, SubstitutionMap substitutions) {
return message.create(translations, substitutions.with(customSubstitutions)).build();
}
public MessageEmbed build(JanitorBot bot) {
return build(bot.getTranslations(), bot.getSubstitutions());
}
public MessageAction send(JanitorBot bot, MessageChannel channel) {
return channel.sendMessage(build(bot));
}
}