me.champeau.gradle.Search.groovy Maven / Gradle / Ivy

Go to download
package me.champeau.gradle

import fj.F
import fj.P1
import fj.Unit
import fj.data.IO
import fj.data.IOFunctions
import fj.data.Validation
import groovy.transform.TypeChecked
import groovy.transform.TypeCheckingMode
import org.reflections.Reflections
import org.reflections.scanners.ResourcesScanner
import org.reflections.util.ClasspathHelper
import org.reflections.util.ConfigurationBuilder

import java.nio.file.Files
import java.util.regex.Pattern

import static fj.data.IOFunctions.bind
import static fj.data.IOFunctions.map

/**
 * Created by MarkPerry on 21/06/2014.
 */
@TypeChecked
class Search {

	static String resourceMatcher(String packagePath) {
		"$packagePath/".replaceAll("\\.", "/")
	}

	static IO> findResources(String fullPackageName) {
		{ ->
			def cb = new ConfigurationBuilder().setUrls(ClasspathHelper.forPackage(fullPackageName)).setScanners(new ResourcesScanner())
			new Reflections(cb).getResources(Pattern.compile(".*"));
		} as IO
	}

    @TypeChecked(TypeCheckingMode.SKIP)
	static IO>> writeAll(String packagePath, File base) {
		IOFunctions.join(map(findResources(packagePath), { Set set ->
			fj.data.List list = fj.data.List.list((String[]) set.toArray());
			IOFunctions.traverse(list, { String s ->
				writeResource(s, base, packagePath)
			} as F)
		} as F))
	}

	static IO writeFile(InputStream is, File f) {
		{ -> Files.copy(is, f.toPath()) } as IO
	}

	static IO writeStream(InputStream is, File f) {
		def io = IOFunctions.unit({ ->
			f.getParentFile().mkdirs()
			Unit.unit()
		} as P1)
		bind(io, { Unit u ->
			writeFile(is, f)
		} as F)
	}

	static IO> writeResource(String resourcePath, File base, String packagePath) {
		def resourcePrefix = resourceMatcher(packagePath)
		def b = resourcePath.startsWith(resourcePrefix)
		if (!b) {
			IOFunctions.unit(Validation.fail("Resource $resourcePath does not match the prefix: $resourcePrefix"))
		} else {
			def sub = resourcePath.substring(resourcePrefix.length())
			def io = { ->
				def is = ClasspathHelper.getResourceAsStream("/$resourcePath")
				is
			} as IO
			def io2 = bind(io, { InputStream is ->
				writeStream(is, new File(base, sub))
			} as F)
			map(io2, { Long item ->
				Validation.success(item)
			} as F)
		}
	}

}