56 lines
2.3 KiB
Go
56 lines
2.3 KiB
Go
package format
|
|
|
|
// keywords are SQL / PL/pgSQL words the formatter may re-case via KeywordCase.
|
|
// Type-name words are deliberately excluded (see typeNames) so the house style
|
|
// can keep types lowercase while keywords are uppercased.
|
|
var keywords = words(`
|
|
add after all alter analyze and any array as asc atomic begin between by
|
|
called cascade case cast check close coalesce collate column commit
|
|
concurrently constraint create cross cube current_date current_time
|
|
current_timestamp current_user cursor declare default deferrable definer delete desc
|
|
distinct do drop each else elsif end except exception execute exists external
|
|
fetch filter first for foreach foreign from full function get grant group
|
|
grouping having if ilike immutable in index inner inout insert intersect into
|
|
into invoker is join key language last leakproof left like limit localtime
|
|
localtimestamp loop materialized natural new next no not nothing notify null
|
|
nulls of off offset old on only open or order out outer over overriding
|
|
parallel partition perform precision primary procedure raise
|
|
references refresh rename replace reset restrict return returning returns
|
|
revoke right rollback row rows safe schema secdef security select sequence set
|
|
setof some stable stacked strict table temp temporary then to transaction
|
|
trigger truncate union unique unsafe update using vacuum values variadic view
|
|
volatile when where while window with within
|
|
`)
|
|
|
|
// typeNames are built-in/common type words kept lowercase by the house style.
|
|
var typeNames = words(`
|
|
bigint bigserial bit bool boolean box bytea char character cidr circle citext
|
|
date daterange decimal double float float4 float8 hstore inet int int2 int4
|
|
int8 integer interval json jsonb line lseg macaddr macaddr8 money numeric oid
|
|
path pg_lsn point polygon real serial serial2 serial4 serial8 smallint
|
|
smallserial text time timestamp timestamptz timetz tsquery tsrange tstzrange
|
|
tsvector uuid varbit varchar xml
|
|
`)
|
|
|
|
func words(s string) map[string]bool {
|
|
m := make(map[string]bool)
|
|
w := ""
|
|
for _, r := range s {
|
|
if r == ' ' || r == '\n' || r == '\t' || r == '\r' {
|
|
if w != "" {
|
|
m[w] = true
|
|
w = ""
|
|
}
|
|
continue
|
|
}
|
|
w += string(r)
|
|
}
|
|
if w != "" {
|
|
m[w] = true
|
|
}
|
|
return m
|
|
}
|
|
|
|
func isKeyword(lower string) bool { return keywords[lower] }
|
|
func isTypeName(lower string) bool { return typeNames[lower] }
|