1 files changed, 32 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..75d2ad9
--- /dev/null
+++ b/src/main/groovy/frege/gradle/FregeSourceSet.groovy
@@ -0,0 +1,32 @@
+package frege.gradle
+
+/**
+ * Created by MarkPerry on 23/09/2015.
+ */
+
+
+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
+    }
+
+}