package database

import (
	"context"
	"strings"

	"modelRT/orm"

	"gorm.io/gorm"
)

// QueryBayDevColumnNames returns the bay table columns exposed as token7
// candidates under token6=bay.
func QueryBayDevColumnNames(ctx context.Context, db *gorm.DB) ([]string, error) {
	columnTypes, err := db.WithContext(ctx).Migrator().ColumnTypes((&orm.Bay{}).TableName())
	if err != nil {
		return nil, err
	}

	columnNames := make([]string, 0, len(columnTypes))
	for _, columnType := range columnTypes {
		columnName := columnType.Name()
		if strings.HasPrefix(columnName, "dev_") {
			columnNames = append(columnNames, columnName)
		}
	}
	return columnNames, nil
}