Add Action overloads for NDOC properties

neoforged · May 25, 2024 · 2c33e49 · 2c33e49
1 parent 292f770
commit 2c33e49
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/src/main/java/net/neoforged/moddevgradle/dsl/NeoForgeExtension.java b/src/main/java/net/neoforged/moddevgradle/dsl/NeoForgeExtension.java
@@ -1,7 +1,7 @@
 package net.neoforged.moddevgradle.dsl;
 
+import org.gradle.api.Action;
 import org.gradle.api.NamedDomainObjectContainer;
-import org.gradle.api.NamedDomainObjectSet;
 import org.gradle.api.Project;
 import org.gradle.api.provider.ListProperty;
 import org.gradle.api.provider.Property;
@@ -51,11 +51,19 @@ public NeoForgeExtension(Project project) {
 
     public abstract ListProperty<String> getAccessTransformers();
 
-    public NamedDomainObjectSet<ModModel> getMods() {
+    public NamedDomainObjectContainer<ModModel> getMods() {
         return mods;
     }
 
-    public NamedDomainObjectSet<RunModel> getRuns() {
+    public void mods(Action<NamedDomainObjectContainer<ModModel>> action) {
+        action.execute(mods);
+    }
+
+    public NamedDomainObjectContainer<RunModel> getRuns() {
         return runs;
     }
+
+    public void runs(Action<NamedDomainObjectContainer<RunModel>> action) {
+        action.execute(runs);
+    }
 }