Merge pull request #29 from Softawii/feature/template

Feature/template Closes #23
Softawii · Jun 7, 2022 · 4d00c54 · 4d00c54
2 parents 1c09942 + 28d403d
commit 4d00c54
Show file tree

Hide file tree

Showing 14 changed files with 671 additions and 13 deletions.
diff --git a/.github/workflows/build.yaml b/.github/workflows/build.yaml
@@ -1,7 +1,6 @@
 name: build
 on:
   push:
-  pull_request:
 
 jobs:
   build:

diff --git a/src/main/java/com/softawii/capivara/Main.java b/src/main/java/com/softawii/capivara/Main.java
@@ -1,7 +1,9 @@
 package com.softawii.capivara;
 
 import com.softawii.capivara.core.PackageManager;
+import com.softawii.capivara.core.TemplateManager;
 import com.softawii.capivara.listeners.PackageGroup;
+import com.softawii.capivara.listeners.TemplateGroup;
 import net.dv8tion.jda.api.JDA;
 import net.dv8tion.jda.api.entities.Activity;
 import org.springframework.boot.CommandLineRunner;
@@ -20,6 +22,7 @@ public static void main(String[] args) {
         jda.getPresence().setPresence(Activity.of(Activity.ActivityType.WATCHING,buildProperties.getVersion()), true);
         System.out.println(buildProperties.getVersion() + " Bot is ready as " + jda.getSelfUser().getName());
         PackageGroup.packageManager = context.getBean(PackageManager.class);
+        TemplateGroup.templateManager = context.getBean(TemplateManager.class);
     }
 
     @Override

diff --git a/src/main/java/com/softawii/capivara/config/SpringConfig.java b/src/main/java/com/softawii/capivara/config/SpringConfig.java
@@ -1,6 +1,7 @@
 package com.softawii.capivara.config;
 
 import com.softawii.capivara.listeners.PackageGroup;
+import com.softawii.capivara.listeners.TemplateGroup;
 import com.softawii.curupira.core.Curupira;
 import net.dv8tion.jda.api.JDA;
 import net.dv8tion.jda.api.JDABuilder;
@@ -63,7 +64,10 @@ public JDA jda() {
             builder.enableIntents(GatewayIntent.GUILD_EMOJIS);
             builder.setMemberCachePolicy(MemberCachePolicy.ALL);
             builder.enableCache(CacheFlag.EMOTE, CacheFlag.ROLE_TAGS, CacheFlag.MEMBER_OVERRIDES);
-            builder.addEventListeners(new PackageGroup.AutoCompleter());
+            builder.addEventListeners(
+                    new PackageGroup.AutoCompleter(),
+                    new TemplateGroup.AutoCompleter()
+            );
             jda = builder.build();
             jda.awaitReady();
         } catch (LoginException | InterruptedException e) {

diff --git a/src/main/java/com/softawii/capivara/core/TemplateManager.java b/src/main/java/com/softawii/capivara/core/TemplateManager.java
@@ -0,0 +1,55 @@
+package com.softawii.capivara.core;
+
+import com.softawii.capivara.entity.Template;
+import com.softawii.capivara.entity.Template.TemplateKey;
+import com.softawii.capivara.exceptions.TemplateAlreadyExistsException;
+import com.softawii.capivara.exceptions.TemplateDoesNotExistException;
+import com.softawii.capivara.services.TemplateService;
+import net.dv8tion.jda.api.interactions.commands.Command;
+import org.springframework.stereotype.Component;
+
+import java.util.List;
+
+@Component
+public class TemplateManager {
+
+    private final TemplateService templateService;
+
+    public TemplateManager(TemplateService templateService) {
+        this.templateService = templateService;
+    }
+
+    public Template create(Long guildId, String name, String json) throws TemplateAlreadyExistsException {
+        return templateService.create(new Template(new TemplateKey(guildId, name), json));
+    }
+
+    public Template update(Long guildId, String name, String json) throws TemplateDoesNotExistException {
+        Template template = templateService.findById(new TemplateKey(guildId, name));
+        template.setJson(json);
+        return templateService.update(template);
+    }
+
+    public void destroy(Long guildId, String name) throws TemplateDoesNotExistException {
+        templateService.destroy(guildId, name);
+    }
+
+    public List<Template> findAllByGuildId(Long guildId) {
+        return templateService.findAllByGuildId(guildId);
+    }
+
+    public Template findById(Long guildId, String name) throws TemplateDoesNotExistException {
+        return templateService.findById(new TemplateKey(guildId, name));
+    }
+
+    public boolean existsById(Long guildId, String name)  {
+        return templateService.existsById(new TemplateKey(guildId, name));
+    }
+
+    public List<Command.Choice> autoCompleteTemplateName(Long guildId, String templateName) {
+        return templateService.findAllByGuildId(guildId).stream()
+                .map(template -> template.getTemplateKey().getName())
+                .filter(c -> c.startsWith(templateName))
+                .map(c -> new Command.Choice(c, c))
+                .toList();
+    }
+}
diff --git a/src/main/java/com/softawii/capivara/entity/Template.java b/src/main/java/com/softawii/capivara/entity/Template.java
@@ -0,0 +1,88 @@
+package com.softawii.capivara.entity;
+
+import javax.persistence.Column;
+import javax.persistence.Embeddable;
+import javax.persistence.EmbeddedId;
+import javax.persistence.Entity;
+import java.io.Serializable;
+import java.util.Objects;
+
+@Entity
+public class Template {
+
+    @EmbeddedId
+    private TemplateKey templateKey;
+
+    @Embeddable
+    public static class TemplateKey implements Serializable {
+        @Column
+        private Long guildId;
+
+        @Column
+        private String name;
+
+        public TemplateKey() {
+        }
+
+        public TemplateKey(Long guildId, String name) {
+            this.guildId = guildId;
+            this.name = name;
+        }
+
+        public Long getGuildId() {
+            return guildId;
+        }
+
+        public void setGuildId(Long guildId) {
+            this.guildId = guildId;
+        }
+
+        public String getName() {
+            return name;
+        }
+
+        public void setName(String name) {
+            this.name = name;
+        }
+
+        @Override
+        public boolean equals(Object o) {
+            if (this == o) return true;
+            if (o == null || getClass() != o.getClass()) return false;
+            TemplateKey that = (TemplateKey) o;
+            return guildId.equals(that.guildId) && name.equals(that.name);
+        }
+
+        @Override
+        public int hashCode() {
+            return Objects.hash(guildId, name);
+        }
+    }
+
+    @Column
+    private String json;
+
+    public Template() {
+    }
+
+    public Template(TemplateKey templateKey, String json) {
+        this.templateKey = templateKey;
+        this.json = json;
+    }
+
+    public TemplateKey getTemplateKey() {
+        return templateKey;
+    }
+
+    public void setTemplateKey(TemplateKey templateKey) {
+        this.templateKey = templateKey;
+    }
+
+    public String getJson() {
+        return json;
+    }
+
+    public void setJson(String json) {
+        this.json = json;
+    }
+}
diff --git a/src/main/java/com/softawii/capivara/exceptions/CategoryIsEmptyException.java b/src/main/java/com/softawii/capivara/exceptions/CategoryIsEmptyException.java
@@ -0,0 +1,22 @@
+package com.softawii.capivara.exceptions;
+
+public class CategoryIsEmptyException extends Exception {
+    public CategoryIsEmptyException() {
+    }
+
+    public CategoryIsEmptyException(String message) {
+        super(message);
+    }
+
+    public CategoryIsEmptyException(String message, Throwable cause) {
+        super(message, cause);
+    }
+
+    public CategoryIsEmptyException(Throwable cause) {
+        super(cause);
+    }
+
+    public CategoryIsEmptyException(String message, Throwable cause, boolean enableSuppression, boolean writableStackTrace) {
+        super(message, cause, enableSuppression, writableStackTrace);
+    }
+}
diff --git a/src/main/java/com/softawii/capivara/exceptions/TemplateAlreadyExistsException.java b/src/main/java/com/softawii/capivara/exceptions/TemplateAlreadyExistsException.java
@@ -0,0 +1,22 @@
+package com.softawii.capivara.exceptions;
+
+public class TemplateAlreadyExistsException extends Exception {
+    public TemplateAlreadyExistsException() {
+    }
+
+    public TemplateAlreadyExistsException(String message) {
+        super(message);
+    }
+
+    public TemplateAlreadyExistsException(String message, Throwable cause) {
+        super(message, cause);
+    }
+
+    public TemplateAlreadyExistsException(Throwable cause) {
+        super(cause);
+    }
+
+    public TemplateAlreadyExistsException(String message, Throwable cause, boolean enableSuppression, boolean writableStackTrace) {
+        super(message, cause, enableSuppression, writableStackTrace);
+    }
+}
diff --git a/src/main/java/com/softawii/capivara/exceptions/TemplateDoesNotExistException.java b/src/main/java/com/softawii/capivara/exceptions/TemplateDoesNotExistException.java
@@ -0,0 +1,22 @@
+package com.softawii.capivara.exceptions;
+
+public class TemplateDoesNotExistException extends Exception {
+    public TemplateDoesNotExistException() {
+    }
+
+    public TemplateDoesNotExistException(String message) {
+        super(message);
+    }
+
+    public TemplateDoesNotExistException(String message, Throwable cause) {
+        super(message, cause);
+    }
+
+    public TemplateDoesNotExistException(Throwable cause) {
+        super(cause);
+    }
+
+    public TemplateDoesNotExistException(String message, Throwable cause, boolean enableSuppression, boolean writableStackTrace) {
+        super(message, cause, enableSuppression, writableStackTrace);
+    }
+}
diff --git a/src/main/java/com/softawii/capivara/listeners/PackageGroup.java b/src/main/java/com/softawii/capivara/listeners/PackageGroup.java
@@ -7,7 +7,6 @@
 import kotlin.Pair;
 import net.dv8tion.jda.api.Permission;
 import net.dv8tion.jda.api.entities.Emote;
-import net.dv8tion.jda.api.entities.Message;
 import net.dv8tion.jda.api.entities.MessageEmbed;
 import net.dv8tion.jda.api.entities.Role;
 import net.dv8tion.jda.api.events.interaction.command.CommandAutoCompleteInteractionEvent;
@@ -52,8 +51,7 @@ public static void create(SlashCommandInteractionEvent event) {
 
         if(name.contains(":")) {
             LOGGER.debug(String.format("create: Package Name contains :, just ignore it (%s)", name));
-            MessageEmbed embed = Utils.simpleEmbed("Nome muito feio!", "O nome do package não pode conter o caracter ':', foi mal, problemas internos aqui!", Color.RED);
-            event.replyEmbeds(embed).setEphemeral(true).queue();
+            event.replyEmbeds(Utils.nameContainsColon("O nome do package")).setEphemeral(true).queue();
             return;
         }
 
@@ -173,8 +171,7 @@ public static void add(SlashCommandInteractionEvent event) {
 
             if(packageName.contains(":")) {
                 LOGGER.debug(String.format("add: Package Name contains :, just ignore it (%s)", packageName));
-                MessageEmbed embed = Utils.simpleEmbed("Nome muito feio!", "O nome do package não pode conter o caracter ':', foi mal, problemas internos aqui!", Color.RED);
-                event.replyEmbeds(embed).setEphemeral(true).queue();
+                event.replyEmbeds(Utils.nameContainsColon("O nome do package")).setEphemeral(true).queue();
                 return;
             }
 
@@ -183,8 +180,7 @@ public static void add(SlashCommandInteractionEvent event) {
 
             if(name.contains(":")) {
                 LOGGER.debug(String.format("add: Role Name contains :, just ignore it (%s)", name));
-                MessageEmbed embed = Utils.simpleEmbed("Nome muito feio!", "O nome do cargo não pode conter o caracter ':', foi mal, problemas internos aqui!", Color.RED);
-                event.replyEmbeds(embed).setEphemeral(true).queue();
+                event.replyEmbeds(Utils.nameContainsColon("O nome do package")).setEphemeral(true).queue();
                 return;
             }
 
@@ -198,8 +194,7 @@ public static void add(SlashCommandInteractionEvent event) {
                 isUnicode = emoji.getSecond();
             } catch (MultipleEmojiMessageException e) {
                 LOGGER.debug("add: MultipleEmojiMessageException");
-                MessageEmbed embed = Utils.simpleEmbed("Emoji muito feio!", "O emoji não pode conter mais de um emoji, tu ta doidinho!", Color.RED);
-                event.replyEmbeds(embed).setEphemeral(true).queue();
+                event.replyEmbeds(Utils.multipleEmoji()).setEphemeral(true).queue();
                 return;
             }