Function rewriteTargetList

internal/x/expander/expander.go:413–488

rewriteTargetList replaces * in a target list with explicit column references

(targets *ast.List, columns []string)

// rewriteTargetList replaces * in a target list with explicit column references
func rewriteTargetList(targets *ast.List, columns []string) *ast.List {
	if targets == nil {
		return nil
	}

	starCount := countStarsInList(targets)
	nonStarCount := countNonStarsInList(targets)

	// Calculate how many columns each * expands to
	// Total columns = (columns per star * number of stars) + non-star columns
	// So: columns per star = (total - non-star) / stars
	columnsPerStar := 0
	if starCount > 0 {
		columnsPerStar = (len(columns) - nonStarCount) / starCount
	}

	newItems := make([]ast.Node, 0, len(columns))
	colIndex := 0

	for _, target := range targets.Items {
		resTarget, ok := target.(*ast.ResTarget)
		if !ok {
			newItems = append(newItems, target)
			colIndex++
			continue
		}

		if resTarget.Val == nil {
			newItems = append(newItems, target)
			colIndex++
			continue
		}

		colRef, ok := resTarget.Val.(*ast.ColumnRef)
		if !ok {
			newItems = append(newItems, target)
			colIndex++
			continue
		}

		if colRef.Fields == nil {
			newItems = append(newItems, target)
			colIndex++
			continue
		}

		// Check if this is a * (with or without table qualifier)
		// and extract any table prefix
		isStar := false
		var tablePrefix []string
		for _, field := range colRef.Fields.Items {
			if _, ok := field.(*ast.A_Star); ok {
				isStar = true
				break
			}
			// Collect prefix parts (schema, table name)
			if str, ok := field.(*ast.String); ok {
				tablePrefix = append(tablePrefix, str.Str)