feat(core): allow to specify plugin sub-group category (#1064)

kestra-io · Mar 17, 2023 · b0b8210 · b0b8210
1 parent 1b5791d
commit b0b8210
Show file tree

Hide file tree

Showing 6 changed files with 36 additions and 0 deletions.
diff --git a/core/src/main/java/io/kestra/core/models/annotations/PluginSubGroup.java b/core/src/main/java/io/kestra/core/models/annotations/PluginSubGroup.java
@@ -2,6 +2,7 @@
 
 import java.lang.annotation.*;
 
+import static io.kestra.core.models.annotations.PluginSubGroup.PluginCategory.MISC;
 import static java.lang.annotation.RetentionPolicy.RUNTIME;
 
 @Documented
@@ -11,4 +12,19 @@
 public @interface PluginSubGroup {
     String title() default "";
     String description() default "";
+
+    PluginCategory[] categories() default { MISC };
+
+    enum PluginCategory {
+        DATABASE,
+        MESSAGING,
+        SCRIPT,
+        TRANSFORMATION,
+        FLOW,
+        BATCH,
+        ALERTING,
+        CLOUD,
+        STORAGE,
+        MISC
+    }
 }
diff --git a/core/src/main/java/io/kestra/core/tasks/debugs/package-info.java b/core/src/main/java/io/kestra/core/tasks/debugs/package-info.java
@@ -0,0 +1,4 @@
+@PluginSubGroup(categories = PluginSubGroup.PluginCategory.MISC)
+package io.kestra.core.tasks.debugs;
+
+import io.kestra.core.models.annotations.PluginSubGroup;
diff --git a/core/src/main/java/io/kestra/core/tasks/flows/package-info.java b/core/src/main/java/io/kestra/core/tasks/flows/package-info.java
@@ -0,0 +1,4 @@
+@PluginSubGroup(categories = PluginSubGroup.PluginCategory.FLOW)
+package io.kestra.core.tasks.flows;
+
+import io.kestra.core.models.annotations.PluginSubGroup;
diff --git a/core/src/main/java/io/kestra/core/tasks/scripts/package-info.java b/core/src/main/java/io/kestra/core/tasks/scripts/package-info.java
@@ -0,0 +1,4 @@
+@PluginSubGroup(categories = PluginSubGroup.PluginCategory.SCRIPT)
+package io.kestra.core.tasks.scripts;
+
+import io.kestra.core.models.annotations.PluginSubGroup;
diff --git a/core/src/main/java/io/kestra/core/tasks/states/package-info.java b/core/src/main/java/io/kestra/core/tasks/states/package-info.java
@@ -0,0 +1,4 @@
+@PluginSubGroup(categories = PluginSubGroup.PluginCategory.STORAGE)
+package io.kestra.core.tasks.states;
+
+import io.kestra.core.models.annotations.PluginSubGroup;
diff --git a/core/src/main/java/io/kestra/core/tasks/storages/package-info.java b/core/src/main/java/io/kestra/core/tasks/storages/package-info.java
@@ -0,0 +1,4 @@
+@PluginSubGroup(categories = PluginSubGroup.PluginCategory.STORAGE)
+package io.kestra.core.tasks.storages;
+
+import io.kestra.core.models.annotations.PluginSubGroup;