blob: 789eabf4e26cae169ee81697ccb442d94dfc6c9f (
plain) (
tree)
|
|
package ch.jodersky.sbt.jni
import sbt._
import sbt.Keys._
import util.ByteCode
object JniJvm extends AutoPlugin {
override def requires = plugins.JvmPlugin
object autoImport {
val javahClasses = taskKey[Set[String]](
"Fully qualified names of classes containing native declarations."
)
val javah = taskKey[File](
"Generate JNI headers. Returns the directory containing generated headers."
)
}
import autoImport._
lazy val mainSettings: Seq[Setting[_]] = Seq(
javahClasses in javah := {
val compiled: inc.Analysis = (compile in Compile).value
val classFiles: Set[File] = compiled.relations.allProducts.toSet
val nativeClasses = classFiles flatMap { file =>
ByteCode.natives(file)
}
nativeClasses
},
target in javah := target.value / "include",
javah := {
val out = (target in javah).value
val jcp: Seq[File] = {
(compile in Compile).value
//FIXME: a cleaner approach that would also call compile would
//be to use `fullClasspath`, this however results in generating resources
//which in turn might require header files generated by javah. Hence
//to avoid a cyclic dependency `compile` is called first and the
//class directory returned.
Seq((classDirectory in Compile).value)
}
val cp = jcp.mkString(sys.props("path.separator"))
val log = streams.value.log
val classes = (javahClasses in javah).value
for (clazz <- classes) {
log.info("Generating header for " + clazz + "...")
val parts = Seq(
"javah",
"-d", out.getAbsolutePath,
"-classpath", cp,
clazz
)
val cmd = parts.mkString(" ")
val ev = Process(cmd) ! streams.value.log
if (ev != 0) sys.error(s"Error occured running javah. Exit code: ${ev}")
}
out
}
)
lazy val clientSettings = Seq(
//enable enhanced native library extraction
libraryDependencies += "ch.jodersky" %% "jni-library" % Version.PluginVersion,
fork in run := true //fork new JVM, since native libraries can only be loaded once
)
override lazy val projectSettings = mainSettings ++ clientSettings
}
|