Function getReferencedColumns

src/postgres/parse-pg-query.ts:99–153

(
  statement: QueryNodePath<QueryParser.Query>,
  spreadTypes: ExpressionSpreadTypes
)

}

function getReferencedColumns(
  statement: QueryNodePath<QueryParser.Query>,
  spreadTypes: ExpressionSpreadTypes
): ColumnReference[] {
  const referencedColumns: ColumnReference[] = []

  traverseSubTree(statement, (path, $cancelRecursion) => {
    const spreadType = isParamRef(path.node) ? spreadTypes[path.node.ParamRef.number] : null

    if (path.node === statement.node) return
    if (isSubquery(path.node)) return $cancelRecursion

    if (isColumnRef(path.node)) {
      const relationRefs = getTableReferences(statement, false)
      const columnRef = resolveColumnReference(path, relationRefs)

      if (columnRef && columnRef.columnName !== placeholderColumnName) {
        referencedColumns.push(columnRef)
      }
    } else if (isResTarget(path.node) && path.node.ResTarget.name) {
      const relationRefs = getTableReferences(statement, false)
      const columnRef = resolveResTarget(path, relationRefs)

      if (columnRef && columnRef.columnName !== placeholderColumnName) {
        referencedColumns.push(columnRef)
      }
    } else if (spreadType) {
      const relationRefs = getTableReferences(statement, false)

      const tableRefsInScope: TableReference[] = relationRefs.map(ref => ({
        as: ref.node.RangeVar.alias ? ref.node.RangeVar.alias.Alias.aliasname : undefined,
        tableName: ref.node.RangeVar.relname,
        path
      }))

      for (const spreadArgKey of Object.keys(spreadType)) {
        referencedColumns.push({
          tableRefsInScope: filterDuplicateTableRefs(tableRefsInScope),
          columnName: spreadArgKey,
          path
        })
      }

      if (spreadType === spreadTypeAny || !spreadType) {
        referencedColumns.push({
          any: true,
          tableRefsInScope: filterDuplicateTableRefs(tableRefsInScope),
          columnName: "",
          path
        })
      }
    }
  })
  return referencedColumns
}

function getReturningColumns(statement: QueryNodePath<QueryParser.Query>): ColumnReference[] {
  const body = (statement.node as any)[statement.type]