1 files changed, 25 insertions, 0 deletions
diff --git a/src/main/groovy/frege/gradle/FregeSourceSet.groovy b/src/main/groovy/frege/gradle/FregeSourceSet.groovy
new file mode 100644
index 0000000..50c42b9
--- /dev/null
+++ b/src/main/groovy/frege/gradle/FregeSourceSet.groovy
@@ -0,0 +1,25 @@
+package frege.gradle
+
+import org.gradle.api.file.SourceDirectorySet
+import org.gradle.api.internal.file.DefaultSourceDirectorySet
+import org.gradle.api.internal.file.FileResolver
+
+import static org.gradle.util.ConfigureUtil.configure
+
+class FregeSourceSet {
+	private final SourceDirectorySet frege
+
+    FregeSourceSet(String displayName, FileResolver fileResolver) {
+		frege = new DefaultSourceDirectorySet(String.format('%s Frege source', displayName), fileResolver)
+		frege.filter.include('**/*.fr')
+	}
+
+	SourceDirectorySet getFrege() {
+		frege
+	}
+
+    FregeSourceSet frege(Closure closure) {
+		configure(closure, frege)
+		this
+	}
+}