-
Notifications
You must be signed in to change notification settings - Fork 496
/
TableIntegration.kt
39 lines (34 loc) · 1.54 KB
/
TableIntegration.kt
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
package app.cash.sqldelight.core.compiler.integration
import app.cash.sqldelight.core.capitalize
import app.cash.sqldelight.core.compiler.SqlDelightCompiler
import app.cash.sqldelight.core.lang.ADAPTER_NAME
import app.cash.sqldelight.core.lang.SqlDelightFile
import app.cash.sqldelight.core.lang.psi.ColumnTypeMixin
import app.cash.sqldelight.core.lang.util.columnDefSource
import com.alecstrong.sql.psi.core.psi.LazyQuery
import com.alecstrong.sql.psi.core.psi.NamedElement
import com.alecstrong.sql.psi.core.psi.SqlCreateTableStmt
import com.alecstrong.sql.psi.core.psi.SqlCreateViewStmt
import com.squareup.kotlinpoet.ClassName
import com.squareup.kotlinpoet.KModifier
import com.squareup.kotlinpoet.PropertySpec
internal fun LazyQuery.needsAdapters() = when (tableName.parent) {
is SqlCreateViewStmt -> false
else -> columns().any { (it.columnType as ColumnTypeMixin).adapter() != null }
}
internal fun LazyQuery.adapterProperty(): PropertySpec {
val adapterType = ClassName(
(tableName.containingFile as SqlDelightFile).packageName!!,
SqlDelightCompiler.allocateName(tableName).capitalize(),
ADAPTER_NAME,
)
return PropertySpec.builder(adapterName, adapterType, KModifier.PRIVATE)
.initializer(adapterName)
.build()
}
private fun LazyQuery.columns() = when (val parentRule = tableName.parent) {
is SqlCreateTableStmt -> parentRule.columnDefList
else -> query.columns.map { (it.element as NamedElement).columnDefSource()!! }
}
internal val LazyQuery.adapterName
get() = "${SqlDelightCompiler.allocateName(tableName)}$ADAPTER_NAME"