forked from sbt/sbt-jni
-
Notifications
You must be signed in to change notification settings - Fork 0
/
JniJavah.scala
79 lines (63 loc) · 2.18 KB
/
JniJavah.scala
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
package ch.jodersky.sbt.jni
package plugins
import collection.JavaConverters._
import util.BytecodeUtil
import sbt._
import sbt.Keys._
import sys.process._
/** Adds `javah` header-generation functionality to projects. */
object JniJavah extends AutoPlugin {
override def requires = plugins.JvmPlugin
override def trigger = allRequirements
object autoImport {
val javahClasses = taskKey[Set[String]](
"Finds 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 := {
import xsbti.compile._
val compiled: CompileAnalysis = (compile in Compile).value
val classFiles: Set[File] = compiled.readStamps().getAllProductStamps()
.asScala.keySet.toSet
val nativeClasses = classFiles flatMap { file =>
BytecodeUtil.nativeClasses(file)
}
nativeClasses
},
target in javah := target.value / "native" / "include",
javah := {
val out = (target in javah).value
// fullClasspath can't be used here since it also generates resources. In
// a project combining JniJavah and JniPackage, we would have a chicken-and-egg
// problem.
val jcp: Seq[File] = (dependencyClasspath in Compile).value.map(_.data) ++ {
(compile in Compile).value; Seq((classDirectory in Compile).value)
}
val cp = jcp.mkString(sys.props("path.separator"))
val log = streams.value.log
val classes = (javahClasses in javah).value
if (!classes.isEmpty) {
log.info("Headers will be generated to " + out.getAbsolutePath)
}
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) ! log
if (ev != 0) sys.error(s"Error occured running javah. Exit code: ${ev}")
}
out
}
)
override lazy val projectSettings = mainSettings
}