2 files changed, 13 insertions, 8 deletions
diff --git a/src/main/groovy/frege/gradle/plugins/FregePlugin.groovy b/src/main/groovy/frege/gradle/plugins/FregePlugin.groovy
index 13c1e9a..b2ca074 100644
--- a/src/main/groovy/frege/gradle/plugins/FregePlugin.groovy
+++ b/src/main/groovy/frege/gradle/plugins/FregePlugin.groovy
@@ -5,9 +5,7 @@ import frege.gradle.tasks.FregeQuickCheck
 import frege.gradle.tasks.FregeRepl
 import org.gradle.api.Plugin
 import org.gradle.api.Project
-import org.gradle.api.file.FileCollection
 import org.gradle.api.tasks.SourceSet
-import org.gradle.api.tasks.util.PatternSet
 
 class FregePlugin implements Plugin<Project> {
 
@@ -43,8 +41,4 @@ class FregePlugin implements Plugin<Project> {
         fregeDoc.classpath = mainSourceSet.runtimeClasspath
     }
 
-
-    FileCollection findJavaModulesToExclude(SourceSet sourceSet) {
-        return sourceSet.allJava.asFileTree.matching { PatternSet pattern -> pattern.include("**/*.java")}
-    }
 }
diff --git a/src/main/groovy/frege/gradle/tasks/FregeDoc.groovy b/src/main/groovy/frege/gradle/tasks/FregeDoc.groovy
index af4b924..30f62c6 100644
--- a/src/main/groovy/frege/gradle/tasks/FregeDoc.groovy
+++ b/src/main/groovy/frege/gradle/tasks/FregeDoc.groovy
@@ -1,7 +1,9 @@
 package frege.gradle.tasks
 
+import org.apache.commons.io.output.TeeOutputStream
 import org.gradle.api.Action
 import org.gradle.api.DefaultTask
+import org.gradle.api.GradleException
 import org.gradle.api.file.FileCollection
 import org.gradle.api.tasks.Input
 import org.gradle.api.tasks.Optional
@@ -50,6 +52,8 @@ class FregeDoc extends DefaultTask {
 
     @TaskAction
     void fregedoc() {
+        ByteArrayOutputStream berr = new ByteArrayOutputStream()
+        def teeOutputStream = new TeeOutputStream(System.err, berr)
         def result = project.javaexec(new Action<JavaExecSpec>() {
             @Override
             void execute(JavaExecSpec javaExecSpec) {
@@ -65,11 +69,18 @@ class FregeDoc extends DefaultTask {
                 javaExecSpec.workingDir = project.projectDir
                 javaExecSpec.standardInput = System.in
                 javaExecSpec.standardOutput = System.out
-                javaExecSpec.errorOutput = System.err
+                javaExecSpec.errorOutput = teeOutputStream
                 javaExecSpec.classpath = this.classpath
+
+                javaExecSpec.ignoreExitValue = true
             }
         })
-    }
 
+        //Workaround for failing with java sources. should result in exit value 0 anyway.
+        def berrString = berr.toString()
+        if (result.exitValue !=0 && !berrString.contains("there were errors for")) {
+            throw new GradleException("Non zero exit value running FregeDoc.");
+        }
+    }
 }