more types

Author: burdoto

src/main/java/org/comroid/api/config/adapter/impl/JdaTypeAdapter.java

@@ -4,7 +4,10 @@
 import lombok.experimental.NonFinal;
 import net.dv8tion.jda.api.JDA;
 import net.dv8tion.jda.api.entities.ISnowflake;
+import net.dv8tion.jda.api.entities.channel.concrete.Category;
+import net.dv8tion.jda.api.entities.channel.concrete.ForumChannel;
 import net.dv8tion.jda.api.entities.channel.concrete.NewsChannel;
+import net.dv8tion.jda.api.entities.channel.concrete.TextChannel;
 import org.comroid.api.config.adapter.TypeAdapter;
 import org.comroid.api.data.seri.type.StandardValueType;
 import org.comroid.api.func.ext.Context;
@@ -18,7 +21,29 @@
 @Value
 @NonFinal
 public abstract class JdaTypeAdapter<T extends ISnowflake> extends TypeAdapter<T, @NotNull Long> {
-    public static final JdaTypeAdapter<NewsChannel> NEWS_CHANNEL = new JdaTypeAdapter<>(NewsChannel.class) {
+    public static final JdaTypeAdapter<TextChannel>  TEXT_CHANNEL     = new JdaTypeAdapter<>(TextChannel.class) {
+        @Override
+        protected Stream<TextChannel> findAll(JDA jda) {
+            return jda.getTextChannels().stream();
+        }
+
+        @Override
+        public Optional<TextChannel> deserialize(Context context, @NotNull Long id) {
+            return context.getFromContext(JDA.class, false).wrap().map(jda -> jda.getTextChannelById(id));
+        }
+    };
+    public static final JdaTypeAdapter<ForumChannel> FORUM_CHANNEL    = new JdaTypeAdapter<>(ForumChannel.class) {
+        @Override
+        protected Stream<ForumChannel> findAll(JDA jda) {
+            return jda.getForumChannels().stream();
+        }
+
+        @Override
+        public Optional<ForumChannel> deserialize(Context context, @NotNull Long id) {
+            return context.getFromContext(JDA.class, false).wrap().map(jda -> jda.getForumChannelById(id));
+        }
+    };
+    public static final JdaTypeAdapter<NewsChannel>  NEWS_CHANNEL     = new JdaTypeAdapter<>(NewsChannel.class) {
         @Override
         protected Stream<NewsChannel> findAll(JDA jda) {
             return jda.getNewsChannels().stream();
@@ -29,6 +54,17 @@ public Optional<NewsChannel> deserialize(Context context, @NotNull Long id) {
             return context.getFromContext(JDA.class, false).wrap().map(jda -> jda.getNewsChannelById(id));
         }
     };
+    public static final JdaTypeAdapter<Category>     CHANNEL_CATEGORY = new JdaTypeAdapter<>(Category.class) {
+        @Override
+        protected Stream<Category> findAll(JDA jda) {
+            return jda.getCategories().stream();
+        }
+
+        @Override
+        public Optional<Category> deserialize(Context context, @NotNull Long id) {
+            return context.getFromContext(JDA.class, false).wrap().map(jda -> jda.getCategoryById(id));
+        }
+    };
 
     public JdaTypeAdapter(Class<T> type) {
         super(type, StandardValueType.LONG, "id");