/
BuildCapabilitiesTable.scala
57 lines (48 loc) · 2.21 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
package com.typesafe.slick.testkit.util
import java.io.{PrintWriter, OutputStreamWriter, BufferedWriter, FileOutputStream}
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 extends App {
// 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 out = new FileOutputStream(args(0))
try {
val wr = new PrintWriter(new BufferedWriter(new OutputStreamWriter(out, "UTF-8")))
wr.println("| Capability |" + profileNames.map(n => s" [${n.replace("slick.jdbc.","").replace("Profile","")}](api:$n) |").mkString)
wr.println("| :--- |" + profileNames.map(n => " :---: |").mkString)
for((cap, link) <- capabilities) {
val flags = profiles.map(d => d.capabilities.contains(cap))
wr.println(s"| [$cap](api:$link) |" + flags.map(b => if(b) " :white_check_mark: |" else " |").mkString)
}
wr.flush()
} finally out.close()
}