Merge branch 'dev/1.21' of https://github.com/KessokuTeaTime/Kessoku-Lib

into dev/1.21
KessokuTeaTime · Sep 7, 2024 · 8196c96 · 8196c96
2 parents 3708835 + 3e19303
commit 8196c96
Showing 1 changed file with 20 additions and 4 deletions.
diff --git a/buildSrc/src/main/java/band/kessoku/gradle/plugin/KessokuGradlePlugin.java b/buildSrc/src/main/java/band/kessoku/gradle/plugin/KessokuGradlePlugin.java
@@ -3,11 +3,14 @@
 import org.gradle.api.Plugin;
 import org.gradle.api.Project;
 import org.gradle.api.artifacts.dsl.RepositoryHandler;
+import org.gradle.api.initialization.Settings;
+import org.gradle.api.invocation.Gradle;
+import org.gradle.api.plugins.PluginAware;
 import org.jetbrains.annotations.NotNull;
 
 import java.util.List;
 
-public class KessokuGradlePlugin implements Plugin<Project> {
+public class KessokuGradlePlugin implements Plugin<PluginAware> {
     private static final List<String> NEO_GROUPS = List.of(
             "net.neoforged",
             "cpw.mods",
@@ -17,10 +20,23 @@ public class KessokuGradlePlugin implements Plugin<Project> {
     );
 
     @Override
-    public void apply(@NotNull Project project) {
-        project.getExtensions().create("kessoku", KesssokuExtension.class);
+    public void apply(@NotNull PluginAware target) {
+        switch (target) {
+            case Settings settings -> {
+                additionalRepositories(settings.getDependencyResolutionManagement().getRepositories());
 
-        additionalRepositories(project.getRepositories());
+                settings.getGradle().getPluginManager().apply(KessokuGradlePlugin.class);
+            }
+            case Project project -> {
+                project.getExtensions().create("kessoku", KesssokuExtension.class);
+
+                additionalRepositories(project.getRepositories());
+            }
+            case Gradle gradle -> {
+                return;
+            }
+            default -> throw new IllegalArgumentException("Expected target to be a Project or Settings, but was a " + target.getClass());
+        }
     }
 
     public void additionalRepositories(RepositoryHandler repositories) {