-
-
Notifications
You must be signed in to change notification settings - Fork 609
/
BuildCapabilitiesTable.scala
61 lines (52 loc) · 2.31 KB
/
BuildCapabilitiesTable.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
package com.typesafe.slick.testkit.util
import java.io.*
import slick.basic.BasicProfile
import slick.jdbc.JdbcCapabilities
import slick.relational.RelationalCapabilities
import slick.sql.SqlCapabilities
/** Build a table of supported capability flags for the user manual. */
object BuildCapabilitiesTable {
def main(args: Array[String]): Unit = {
// testkit/runMain com.typesafe.slick.testkit.util.BuildCapabilitiesTable ../src/sphinx/capabilities.csv
if (args.length < 1 || args.length > 2) {
println("Syntax: com.typesafe.slick.testkit.util.BuildCapabilitiesTable OUTPUTFILE [PROFILELIST]")
System.exit(1)
}
val profileNames = if (args.length > 1) args(1).split(",") else Array(
"slick.jdbc.DB2Profile",
"slick.jdbc.DerbyProfile",
"slick.jdbc.H2Profile",
"slick.jdbc.HsqldbProfile",
"slick.jdbc.MySQLProfile",
"slick.jdbc.OracleProfile",
"slick.jdbc.PostgresProfile",
"slick.jdbc.SQLiteProfile",
"slick.jdbc.SQLServerProfile"
)
val profiles = profileNames.map { n =>
Class.forName(n + "$").getField("MODULE$").get(null).asInstanceOf[BasicProfile]
}
val profileCapabilities = Vector(
RelationalCapabilities.all -> "slick.relational.RelationalCapabilities$#",
SqlCapabilities.all -> "slick.sql.SqlCapabilities$#",
JdbcCapabilities.all -> "slick.jdbc.JdbcCapabilities$#"
)
val capabilities = for {
(caps, linkBase) <- profileCapabilities
cap <- caps.toVector.sortBy(c => if (c.toString.endsWith(".other")) "" else c.toString)
} yield (cap, linkBase + cap.toString.replaceFirst(".*\\.", "") + ":slick.basic.Capability")
val file = new File(args(0))
file.getParentFile.mkdirs()
val out = new FileOutputStream(file)
try {
val wr = new PrintWriter(new BufferedWriter(new OutputStreamWriter(out, "UTF-8")))
wr.println("| Capability |" + profileNames.map(n => s" @scaladoc[${n.replace("slick.jdbc.", "").replace("Profile", "")}]($n) |").mkString)
wr.println("| :--- |" + profileNames.map(n => " :---: |").mkString)
for ((cap, link) <- capabilities) {
val flags = profiles.map(d => d.capabilities.contains(cap))
wr.println(s"| @scaladoc[$cap]($link) |" + flags.map(b => if (b) " ✅ |" else " |").mkString)
}
wr.flush()
} finally out.close()
}
}