import sbt._
import java.io.File

trait ProguardLaunch extends ProguardProject
{
	override def optimize = 2
	override def basicOptions = super.basicOptions ++ Seq(keepJLine)
	def outputJar = rootProject.outputPath / ("sbt-launch-" + version + ".jar")
	override def keepFullClasses =
		"xsbti.**" ::
		"jline.**" ::
		Nil
	override def keepClasses = 
		"org.apache.ivy.plugins.resolver.URLResolver" ::
		"org.apache.ivy.plugins.resolver.IBiblioResolver" ::
		Nil
		
	override def mapInJars(inJars: Seq[File]) =
	{
		val inputJar = jarPath.asFile.getAbsolutePath
		val jlineJars = runClasspath.getFiles.filter(isJLineJar)
		// pull out Ivy in order to exclude resources inside
		val (ivyJars, notIvy) = inJars.filter(jar => !isJLineJar(jar)).partition(isIvyJar)
		val (libraryJar, remaining) = notIvy.partition(isScalaJar)
		
		val otherJars = remaining.filter(jar => !isJarX(jar, "scala-compiler"))

		log.debug("proguard configuration:")
		log.debug("\tJLline jar location: " + jlineJars.mkString(", "))
		log.debug("\tIvy jar location: " + ivyJars.mkString(", "))
		log.debug("\tOther jars:\n\t" + otherJars.mkString("\n\t"))
		val excludeIvyResourcesString = excludeString(excludeIvyResources)
		((withJar(ivyJars.toSeq, "Ivy") + excludeIvyResourcesString) ::
		(withJar(jlineJars, "JLine") + jlineFilter ) ::
		(withJar(libraryJar, "Scala library") + libraryFilter) ::
		otherJars.map(jar => mkpath(jar) + generalFilter).toList) map { "-injars " + _ }
	}

	private def excludeString(s: List[String]) = s.map("!" + _).mkString("(",",",")")
	private def excludeIvyResources =
		"META-INF/**" ::
		"fr/**" ::
		"**/antlib.xml" ::
		"**/*.png" ::
		"org/apache/ivy/core/settings/ivyconf*.xml" ::
		"org/apache/ivy/core/settings/ivysettings-*.xml" ::
		"org/apache/ivy/plugins/resolver/packager/*" ::
		"**/ivy_vfs.xml" ::
		"org/apache/ivy/plugins/report/ivy-report-*" ::
		Nil

	private def libraryFilter = "(!META-INF/**,!*.properties,!scala/actors/**.!scala/util/parsing/*.class,!scala/xml/**.class,!scala/package$.class,**.class)"
	private def jlineFilter = "(!META-INF/**)"
	private def generalFilter = "(!META-INF/**,!*.properties)"

	private def withJar[T](files: Iterable[File], name: String) = mkpath(files.toSeq.firstOption.getOrElse(error(name + " not present (try running update)")))
	private def isJLineJar(file: File) = isJarX(file, "jline")
	private def isIvyJar(file: File) = isJarX(file, "ivy")
	private def isScalaJar(file: File) = isJarX(file, "scala-library")
	private def isJarX(file: File, x: String) =
	{
		val name = file.getName
		name.startsWith(x) && name.endsWith(".jar")
	}

	private val keepJLine =
	"""
	|-keep public class jline.** {
	|    public protected *;
	|}
	|""".stripMargin
}