app.cash.sqldelight.dialects.postgresql.grammar.mixins.SqlInsertExcludedTableMixin.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of postgresql-dialect Show documentation

The PostgreSQL Dialect for SQLDelight

There is a newer version: 2.0.2

package app.cash.sqldelight.dialects.postgresql.grammar.mixins

import app.cash.sqldelight.dialects.postgresql.grammar.psi.PostgreSqlInsertStmt
import com.alecstrong.sql.psi.core.psi.QueryElement
import com.alecstrong.sql.psi.core.psi.SqlCompositeElementImpl
import com.alecstrong.sql.psi.core.psi.mixins.SingleRow
import com.intellij.lang.ASTNode
import com.intellij.psi.PsiElement

abstract class SqlInsertExcludedTableMixin(node: ASTNode) : SqlCompositeElementImpl(node) {
  override fun queryAvailable(child: PsiElement): Collection {
    val insertStmt = parent.parent as PostgreSqlInsertStmt
    val tableName = insertStmt.tableName
    val table = tablesAvailable(this).first { it.tableName.name == tableName.name }.query

    fun excludedTable(name: String) = QueryElement.QueryResult(
      SingleRow(tableName, name),
      table.columns,
      synthesizedColumns = table.synthesizedColumns,
    )

    return super.queryAvailable(child) + excludedTable("excluded") + excludedTable("EXCLUDED")
  }
}