app.cash.sqldelight.dialects.hsql.grammar.mixins.ResultColumnMixin.kt Maven / Gradle / Ivy

Go to download

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

The HSQL Dialect for SQLDelight

There is a newer version: 2.0.2

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

import app.cash.sqldelight.dialects.hsql.grammar.psi.HsqlResultColumn
import com.alecstrong.sql.psi.core.ModifiableFileLazy
import com.alecstrong.sql.psi.core.psi.QueryElement
import com.alecstrong.sql.psi.core.psi.QueryElement.QueryResult
import com.alecstrong.sql.psi.core.psi.impl.SqlResultColumnImpl
import com.intellij.lang.ASTNode

internal abstract class ResultColumnMixin(node: ASTNode) : SqlResultColumnImpl(node), HsqlResultColumn {
  private val queryExposed = ModifiableFileLazy lazy@{
    if (windowFunctionInvocation != null) {
      var column = QueryElement.QueryColumn(this)
      columnAlias?.let { alias ->
        column = column.copy(element = alias)
      }

      return@lazy listOf(QueryResult(columns = listOf(column)))
    }

    return@lazy super.queryExposed()
  }

  override fun queryExposed() = queryExposed.forFile(containingFile)
}