internal/codegen/sdk/utils.go - main - githubmirror/sqlc

internal/codegen/sdk/utils.go

42 строки762 B

Anuraag Agrawal

Use wazero-based libpgquery wrapper in currently disabled environments (#3027)

05 дек 2023, 21:58

05 дек 2023, 21:584a05999

package sdk

import (
	"strings"
	"unicode"
)

func LowerTitle(s string) string {
	if s == "" {
		return s
	}

	a := []rune(s)
	a[0] = unicode.ToLower(a[0])
	return string(a)
}

func Title(s string) string {
	return strings.Title(s)
}

// Go string literals cannot contain backtick. If a string contains
// a backtick, replace it the following way:
//
// input:
//
//	SELECT `group` FROM foo
//
// output:
//
//	SELECT ` + "`" + `group` + "`" + ` FROM foo
//
// # The escaped string must be rendered inside an existing string literal
//
// A string cannot be escaped twice
func EscapeBacktick(s string) string {
	return strings.Replace(s, "`", "`+\"`\"+`", -1)
}

func DoubleSlashComment(s string) string {
	return "// " + strings.ReplaceAll(s, "\n", "\n// ")
}