Hein
04711cf7b2
* update error handling in various commands to use blank identifier * enhance output formatting for better readability * add golangci-lint to Makefile for linting checks
74 lines
3.2 KiB
Go
74 lines
3.2 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
|
|
}
|
|
|
|
// builtinFunctions are built-in function names controlled by BuiltinCase.
|
|
var builtinFunctions = words(`
|
|
abs age array_agg array_length array_lower array_ndims array_upper
|
|
bit_length btrim cardinality ceil ceiling char_length character_length
|
|
chr clock_timestamp coalesce concat concat_ws count
|
|
currval decode div encode exp extract floor
|
|
generate_series greatest initcap jsonb_agg jsonb_object_agg
|
|
justify_days justify_hours justify_interval
|
|
lastval least length lower lpad ltrim
|
|
max md5 min mod now nullif
|
|
overlay pg_sleep position power quote_ident quote_literal
|
|
random regexp_match regexp_matches regexp_replace replace reverse round rpad rtrim
|
|
setval split_part sqrt string_agg strpos substr substring sum
|
|
to_char to_date to_json to_jsonb to_number to_timestamp to_tsvector
|
|
translate trim trunc unnest upper width_bucket
|
|
`)
|
|
|
|
func isKeyword(lower string) bool { return keywords[lower] }
|
|
func isTypeName(lower string) bool { return typeNames[lower] }
|
|
func isBuiltinFunc(lower string) bool { return builtinFunctions[lower] }
|