✨ Added way to edit the banner of the logged bot

src/main/java/info/itsthesky/disky/core/SkriptUtils.java

@@ -27,6 +27,7 @@
 import info.itsthesky.disky.elements.sections.handler.DiSkyRuntimeHandler;
 import net.dv8tion.jda.api.JDA;
 import net.dv8tion.jda.api.entities.Guild;
+import net.dv8tion.jda.api.entities.Icon;
 import net.dv8tion.jda.api.entities.channel.Channel;
 import net.dv8tion.jda.api.entities.channel.middleman.GuildChannel;
 import net.dv8tion.jda.api.events.guild.GenericGuildEvent;
@@ -38,8 +39,13 @@
 import org.jetbrains.annotations.Nullable;
 import org.skriptlang.skript.lang.entry.EntryValidator;
 
+import java.io.File;
+import java.io.FileInputStream;
+import java.io.FileNotFoundException;
+import java.io.IOException;
 import java.lang.reflect.Array;
 import java.lang.reflect.Field;
+import java.net.URL;
 import java.time.Instant;
 import java.time.OffsetDateTime;
 import java.time.ZoneId;
@@ -299,4 +305,28 @@ else if (!input.matches("\\d{17,20}"))
         DiSkyRuntimeHandler.error(new Exception(message), node);
         return false;
     }
+
+    public static Icon parseIcon(String input) {
+        if (input == null)
+            return null;
+
+        if (Utils.isURL(input)) {
+            try {
+                return Icon.from(new URL(input).openStream());
+            } catch (IOException ex) {
+                DiSky.getErrorHandler().exception(null, ex);
+                return null;
+            }
+        } else {
+            final File iconFile = new File(input);
+            if (iconFile == null || !iconFile.exists())
+                return null;
+            try {
+                return Icon.from(new FileInputStream(iconFile));
+            } catch (IOException ex) {
+                DiSky.getErrorHandler().exception(null, ex);
+                return null;
+            }
+        }
+    }
 }

src/main/java/info/itsthesky/disky/elements/properties/profile/ProfileBanner.java

@@ -1,38 +1,115 @@
 package info.itsthesky.disky.elements.properties.profile;
 
+import ch.njol.skript.classes.Changer;
+import ch.njol.skript.config.Node;
 import ch.njol.skript.doc.Description;
 import ch.njol.skript.doc.Name;
 import ch.njol.skript.expressions.base.SimplePropertyExpression;
+import ch.njol.skript.lang.Expression;
+import ch.njol.skript.lang.SkriptParser;
+import ch.njol.util.Kleenean;
+import info.itsthesky.disky.core.Bot;
+import info.itsthesky.disky.core.SkriptUtils;
+import info.itsthesky.disky.elements.changers.IAsyncChangeableExpression;
+import info.itsthesky.disky.elements.sections.handler.DiSkyRuntimeHandler;
+import net.dv8tion.jda.api.JDA;
+import net.dv8tion.jda.api.entities.Icon;
 import net.dv8tion.jda.api.entities.User;
+import net.dv8tion.jda.api.managers.AccountManager;
+import net.dv8tion.jda.api.requests.RestAction;
+import org.bukkit.event.Event;
 import org.jetbrains.annotations.NotNull;
 import org.jetbrains.annotations.Nullable;
 
 @Name("Profile Banner")
 @Description({"Get the profile banner URL. If the user doesn't have a custom banner, this will return none.",
 "Use the 'profile color' expression to get the color instead of the banner URL in that case!"})
-public class ProfileBanner extends SimplePropertyExpression<User.Profile, String> {
+public class ProfileBanner extends SimplePropertyExpression<Object, String>
+        implements IAsyncChangeableExpression {
 
     static {
         register(
                 ProfileBanner.class,
                 String.class,
                 "profile banner [ur(l|i)]",
-                "userprofile"
+                "userprofile/bot"
         );
     }
 
+    private Node node;
+
+    @Override
+    public boolean init(Expression<?> @NotNull [] expressions, int matchedPattern, @NotNull Kleenean isDelayed, SkriptParser.@NotNull ParseResult parseResult) {
+        node = getParser().getNode();
+        return super.init(expressions, matchedPattern, isDelayed, parseResult);
+    }
+
     @Override
     protected @NotNull String getPropertyName() {
         return "profile banner";
     }
 
     @Override
-    public @Nullable String convert(User.Profile profile) {
-        return profile.getBannerUrl();
+    public @Nullable String convert(Object profile) {
+        if (profile instanceof User.Profile)
+            return ((User.Profile) profile).getBannerUrl();
+        if (profile instanceof Bot)
+            DiSkyRuntimeHandler.error(new Exception("Cannot retrieve profile banner from a bot, retrieve its profile first."), node);
+        return null;
     }
 
     @Override
     public @NotNull Class<? extends String> getReturnType() {
         return String.class;
     }
+
+    @Override
+    public Class<?> @NotNull [] acceptChange(Changer.@NotNull ChangeMode mode) {
+        if (mode == Changer.ChangeMode.SET)
+            return new Class[] {String.class};
+
+        return new Class[0];
+    }
+
+    @Override
+    public void change(@NotNull Event event, Object @NotNull [] delta, Changer.@NotNull ChangeMode mode) {
+        change(event, delta, mode, false);
+    }
+
+    @Override
+    public void changeAsync(Event e, Object[] delta, Changer.ChangeMode mode) {
+        change(e, delta, mode, true);
+    }
+
+    public void change(Event event, Object[] delta, Changer.ChangeMode mode, boolean async) {
+        if (delta == null || delta.length == 0 || delta[0] == null)
+            return;
+        if (mode != Changer.ChangeMode.SET)
+            return;
+
+        final String bannerInput = delta[0].toString();
+        final Icon banner = SkriptUtils.parseIcon(bannerInput);
+        if (banner == null) {
+            DiSkyRuntimeHandler.error(new Exception("Cannot parse the given banner input! Must be valid URL or file path."), node);
+            return;
+        }
+
+        for (Object entity : getExpr().getArray(event)) {
+            if (entity instanceof Bot) {
+                final AccountManager manager = ((Bot) entity).getInstance().getSelfUser().getManager();
+                final RestAction<Void> action = manager.setBanner(banner);
+                try {
+                    if (async) {
+                        action.queue();
+                    } else {
+                        action.complete();
+                    }
+                } catch (Exception ex) {
+                    DiSkyRuntimeHandler.error(ex, node);
+                }
+            } else if (entity instanceof User) {
+                DiSkyRuntimeHandler.error(new Exception("Cannot change the profile banner of a user."), node);
+            }
+        }
+    }
 }