better handling for missing source set

harrel56 · Sep 29, 2023 · 442f557 · 442f557
1 parent c4fc431
commit 442f557
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/examples/simple/src/main/java/Example.java b/examples/simple/src/main/java/Example.java
@@ -1,3 +1,3 @@
 public class Example {
-    public String[] hello;
+    public String[] hello2;
 }
diff --git a/java2ts-plugin/src/main/java/dev/harrel/java2ts/JavaToTsPlugin.java b/java2ts-plugin/src/main/java/dev/harrel/java2ts/JavaToTsPlugin.java
@@ -8,6 +8,7 @@
 import org.gradle.api.tasks.SourceSet;
 import org.gradle.api.tasks.SourceSetContainer;
 
+import java.util.Optional;
 import java.util.Set;
 
 public class JavaToTsPlugin implements Plugin<Project> {
@@ -42,9 +43,9 @@ public void apply(Project project) {
 
     private Provider<SourceSet> getDefaultSourceSet(Project project) {
         return project.getProviders().provider(() ->
-                project.getExtensions()
-                .getByType(SourceSetContainer.class)
-                .findByName("main"));
+                Optional.ofNullable(project.getExtensions().findByType(SourceSetContainer.class))
+                        .map(s -> s.findByName("main"))
+                        .orElseThrow(() -> new IllegalArgumentException("Source set was not provided and there was no source set named 'main'")));
     }
 
     private Provider<RegularFile> getDefaultOutput(Project project) {