0f4df23290
Some checks failed
kotlin / test (push) Has been skipped
python / test (push) Has been skipped
typescript / test (push) Has been skipped
go / build darwin/amd64 (push) Has been cancelled
go / build linux/amd64 (push) Has been cancelled
go / build windows/amd64 (push) Has been cancelled
go / build darwin/arm64 (push) Has been cancelled
go / build linux/arm64 (push) Has been cancelled
go / build windows/arm64 (push) Has been cancelled
go / test (push) Has been cancelled
go / vuln_check (push) Has been cancelled
60 lines
1.6 KiB
Go
60 lines
1.6 KiB
Go
package sqlite
|
|
|
|
import (
|
|
"strconv"
|
|
"strings"
|
|
)
|
|
|
|
// QuoteIdent quotes an identifier when printing it bare would change what it
|
|
// names: the parser folds unquoted identifiers to lower case, so any name
|
|
// that is not already a plain lower-case identifier — or that collides with
|
|
// a keyword — only survives a parse round-trip inside double quotes.
|
|
func (p *Parser) QuoteIdent(s string) string {
|
|
if s == "" || p.IsReservedKeyword(s) || !plainIdent(s) {
|
|
return `"` + strings.ReplaceAll(s, `"`, `""`) + `"`
|
|
}
|
|
return s
|
|
}
|
|
|
|
// plainIdent reports whether s parses back as itself unquoted: ASCII
|
|
// lower-case letters, digits and underscores, not starting with a digit.
|
|
func plainIdent(s string) bool {
|
|
for i, r := range s {
|
|
switch {
|
|
case r >= 'a' && r <= 'z', r == '_':
|
|
case r >= '0' && r <= '9':
|
|
if i == 0 {
|
|
return false
|
|
}
|
|
default:
|
|
return false
|
|
}
|
|
}
|
|
return true
|
|
}
|
|
|
|
// TypeName returns the SQL type name for the given namespace and name.
|
|
func (p *Parser) TypeName(ns, name string) string {
|
|
if ns != "" {
|
|
return ns + "." + name
|
|
}
|
|
return name
|
|
}
|
|
|
|
// Param returns the parameter placeholder for the given number. SQLite
|
|
// takes both a bare ? (the next index) and an explicit ?N, and they are
|
|
// not interchangeable — reordered ?N parameters bind by their numbers —
|
|
// so the one the author wrote is the one printed.
|
|
func (p *Parser) Param(n int, numbered bool) string {
|
|
if numbered {
|
|
return "?" + strconv.Itoa(n)
|
|
}
|
|
return "?"
|
|
}
|
|
|
|
// Cast returns a type cast expression.
|
|
// SQLite uses CAST(expr AS type) syntax.
|
|
func (p *Parser) Cast(arg, typeName string) string {
|
|
return "CAST(" + arg + " AS " + typeName + ")"
|
|
}
|