upstream-sqlc/internal/engine/mssql/utils.go

package mssql

import (
	"log"
	"strings"

	tsql "github.com/sqlc-dev/teesql/ast"

	"github.com/sqlc-dev/sqlc/internal/debug"
	"github.com/sqlc-dev/sqlc/internal/sql/ast"
)

// fragmented is satisfied by every teesql node via its embedded Fragment.
type fragmented interface {
	Frag() *tsql.Fragment
}

func todo(n tsql.Node) *ast.TODO {
	if debug.Active {
		log.Printf("mssql.convert: Unknown node type %T\n", n)
	}
	return &ast.TODO{}
}

// identifier normalizes an identifier. T-SQL identifiers are
// case-insensitive under the default collations.
func identifier(id string) string {
	return strings.ToLower(id)
}

func NewIdentifier(t string) *ast.String {
	return &ast.String{Str: identifier(t)}
}

func identifierValue(id *tsql.Identifier) string {
	if id == nil {
		return ""
	}
	return identifier(id.Value)
}

// schemaName returns the schema an object name qualifies it with. The "dbo"
// default schema maps to the catalog's default namespace, so it is treated
// as unqualified.
func schemaName(n *tsql.SchemaObjectName) string {
	if n == nil || n.SchemaIdentifier == nil {
		return ""
	}
	s := identifier(n.SchemaIdentifier.Value)
	if s == "dbo" {
		return ""
	}
	return s
}

func parseTableName(n *tsql.SchemaObjectName) *ast.TableName {
	if n == nil {
		return &ast.TableName{}
	}
	return &ast.TableName{
		Schema: schemaName(n),
		Name:   identifierValue(n.BaseIdentifier),
	}
}