123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932 |
- const {assert} = require('./assert');
- const npm = {
- pgUtils: require('pg/lib/utils'),
- patterns: require('./patterns'),
- utils: require('./utils')
- };
- const fmFlags = {
- raw: 1,
- alias: 2,
- name: 4,
- json: 8,
- csv: 16,
- value: 32
- };
- const fmMap = {
- '^': fmFlags.raw,
- ':raw': fmFlags.raw,
- ':alias': fmFlags.alias,
- '~': fmFlags.name,
- ':name': fmFlags.name,
- ':json': fmFlags.json,
- ':csv': fmFlags.csv,
- ':list': fmFlags.csv,
- ':value': fmFlags.value,
- '#': fmFlags.value
- };
- const ctfSymbols = {
- toPostgres: Symbol.for('ctf.toPostgres'),
- rawType: Symbol.for('ctf.rawType')
- };
- const maxVariable = 100000;
- function formatValue({value, fm, cc, options}) {
- if (typeof value === 'function') {
- return formatValue({value: resolveFunc(value, cc), fm, cc});
- }
- const ctf = getCTF(value);
- if (ctf) {
- fm |= ctf.rawType ? fmFlags.raw : 0;
- return formatValue({value: resolveFunc(ctf.toPostgres, value), fm, cc});
- }
- const isRaw = !!(fm & fmFlags.raw);
- fm &= ~fmFlags.raw;
- switch (fm) {
- case fmFlags.alias:
- return $as.alias(value);
- case fmFlags.name:
- return $as.name(value);
- case fmFlags.json:
- return $as.json(value, isRaw);
- case fmFlags.csv:
- return $to.csv(value, options);
- case fmFlags.value:
- return $as.value(value);
- default:
- break;
- }
- if (isNull(value)) {
- throwIfRaw(isRaw);
- return 'null';
- }
- switch (typeof value) {
- case 'string':
- return $to.text(value, isRaw);
- case 'boolean':
- return $to.bool(value);
- case 'number':
- case 'bigint':
- return $to.number(value);
- case 'symbol':
- throw new TypeError(`Type Symbol has no meaning for PostgreSQL: ${value.toString()}`);
- default:
- if (value instanceof Date) {
- return $to.date(value, isRaw);
- }
- if (Array.isArray(value)) {
- return $to.array(value, options);
- }
- if (Buffer.isBuffer(value)) {
- return $to.buffer(value, isRaw);
- }
- return $to.json(value, isRaw);
- }
- }
- function formatArray(array, options) {
- const loop = a => '[' + a.map(value => Array.isArray(value) ? loop(value) : formatValue({
- value,
- options
- })).join() + ']';
- const prefix = options && options.capSQL ? 'ARRAY' : 'array';
- return array.length ? (prefix + loop(array)) : '\'{}\'';
- }
- function formatCSV(values, options) {
- if (Array.isArray(values)) {
- return values.map(value => formatValue({value, options})).join();
- }
- if (typeof values === 'object' && values !== null) {
- return Object.keys(values).map(v => formatValue({value: values[v], options})).join();
- }
- return values === undefined ? '' : formatValue({value: values, options});
- }
- const formatAs = {
- object({query, obj, raw, options}) {
- options = options && typeof options === 'object' ? options : {};
- return query.replace(npm.patterns.namedParameters, name => {
- const v = formatAs.stripName(name.replace(/^\$[{(<[/]|[\s})>\]/]/g, ''), raw),
- c = npm.utils.getIfHas(obj, v.name);
- if (!c.valid) {
- throw new Error(`Invalid property name '${v.name}'.`);
- }
- if (c.has) {
- return formatValue({value: c.value, fm: v.fm, cc: c.target, options});
- }
- if (v.name === 'this') {
- return formatValue({value: obj, fm: v.fm, options});
- }
- if ('def' in options) {
- const d = options.def, value = typeof d === 'function' ? d.call(obj, v.name, obj) : d;
- return formatValue({value, fm: v.fm, cc: obj, options});
- }
- if (options.partial) {
- return name;
- }
-
- throw new Error(`Property '${v.name}' doesn't exist.`);
- });
- },
- array({query, array, raw, options}) {
- options = options && typeof options === 'object' ? options : {};
- return query.replace(npm.patterns.multipleValues, name => {
- const v = formatAs.stripName(name.substring(1), raw);
- const idx = v.name - 1;
- if (idx >= maxVariable) {
- throw new RangeError(`Variable $${v.name} exceeds supported maximum of $${maxVariable}`);
- }
- if (idx < array.length) {
- return formatValue({value: array[idx], fm: v.fm, options});
- }
- if ('def' in options) {
- const d = options.def, value = typeof d === 'function' ? d.call(array, idx, array) : d;
- return formatValue({value, fm: v.fm, options});
- }
- if (options.partial) {
- return name;
- }
- throw new RangeError(`Variable $${v.name} out of range. Parameters array length: ${array.length}`);
- });
- },
- value({query, value, raw, options}) {
- return query.replace(npm.patterns.singleValue, name => {
- const v = formatAs.stripName(name, raw);
- return formatValue({value, fm: v.fm, options});
- });
- },
- stripName(name, raw) {
- const mod = name.match(npm.patterns.hasValidModifier);
- if (mod) {
- return {
- name: name.substring(0, mod.index),
- fm: fmMap[mod[0]] | (raw ? fmFlags.raw : 0)
- };
- }
- return {
- name,
- fm: raw ? fmFlags.raw : null
- };
- }
- };
- function isNull(value) {
- return value === undefined || value === null;
- }
- function getCTF(value) {
- if (!isNull(value)) {
- let toPostgres = value[ctfSymbols.toPostgres], rawType = !!value[ctfSymbols.rawType];
- if (typeof toPostgres !== 'function') {
- toPostgres = value.toPostgres;
- rawType = !!value.rawType;
- }
- if (typeof toPostgres === 'function') {
- if (toPostgres.constructor.name !== 'Function') {
- throw new Error('CTF does not support asynchronous toPostgres functions.');
- }
- return {toPostgres, rawType};
- }
- }
- return null;
- }
- function wrapText(text) {
- return `'${text}'`;
- }
- function safeText(text) {
- return text.replace(/'/g, '\'\'');
- }
- function throwIfRaw(raw) {
- if (raw) {
- throw new TypeError('Values null/undefined cannot be used as raw text.');
- }
- }
- function resolveFunc(value, cc) {
- while (typeof value === 'function') {
- if (value.constructor.name !== 'Function') {
-
-
-
- throw new Error('Cannot use asynchronous functions with query formatting.');
- }
- value = value.call(cc, cc);
- }
- return value;
- }
- function formatQuery(query, values, raw, options) {
- if (typeof query !== 'string') {
- throw new TypeError('Parameter \'query\' must be a text string.');
- }
- const ctf = getCTF(values);
- if (ctf) {
-
- return formatQuery(query, resolveFunc(ctf.toPostgres, values), raw || ctf.rawType, options);
- }
- if (typeof values === 'object' && values !== null) {
- if (Array.isArray(values)) {
-
- return formatAs.array({query, array: values, raw, options});
- }
- if (!(values instanceof Date || values instanceof Buffer)) {
-
- return formatAs.object({query, obj: values, raw, options});
- }
- }
-
- return values === undefined ? query : formatAs.value({query, value: values, raw, options});
- }
- function formatEntity(entity, values, {capSQL, type}) {
- let prefix = type === 'func' ? 'select * from' : 'call';
- if (capSQL) {
- prefix = prefix.toUpperCase();
- }
- return `${prefix} ${$as.alias(entity)}(${formatCSV(values, {capSQL})})`;
- }
- function formatSqlName(name) {
- return `"${name.replace(/"/g, '""')}"`;
- }
- const $as = {
-
- ctf: ctfSymbols,
-
- text(value, raw) {
- value = resolveFunc(value);
- if (isNull(value)) {
- throwIfRaw(raw);
- return 'null';
- }
- if (typeof value !== 'string') {
- value = value.toString();
- }
- return $to.text(value, raw);
- },
-
- name(name) {
- name = resolveFunc(name);
- if (name) {
- if (typeof name === 'string') {
- return /^\s*\*(\s*)$/.test(name) ? name : formatSqlName(name);
- }
- if (typeof name === 'object') {
- const keys = Array.isArray(name) ? name : Object.keys(name);
- if (!keys.length) {
- throw new Error('Cannot retrieve sql names from an empty array/object.');
- }
- return keys.map(value => {
- if (!value || typeof value !== 'string') {
- throw new Error(`Invalid sql name: ${npm.utils.toJson(value)}`);
- }
- return formatSqlName(value);
- }).join();
- }
- }
- throw new TypeError(`Invalid sql name: ${npm.utils.toJson(name)}`);
- },
-
- alias(name) {
- name = resolveFunc(name);
- if (name && typeof name === 'string') {
- return name.split('.')
- .filter(f => f)
- .map(a => {
- const m = a.match(/^([a-z_][a-z0-9_$]*|[A-Z_][A-Z0-9_$]*)$/);
- if (m && m[0] === a) {
- return a;
- }
- return `"${a.replace(/"/g, '""')}"`;
- }).join('.');
- }
- throw new TypeError(`Invalid sql alias: ${npm.utils.toJson(name)}`);
- },
-
- value(value) {
- value = resolveFunc(value);
- if (isNull(value)) {
- throw new TypeError('Open values cannot be null or undefined.');
- }
- return safeText(formatValue({value, fm: fmFlags.raw}));
- },
-
- buffer(obj, raw) {
- obj = resolveFunc(obj);
- if (isNull(obj)) {
- throwIfRaw(raw);
- return 'null';
- }
- if (obj instanceof Buffer) {
- return $to.buffer(obj, raw);
- }
- throw new TypeError(`${wrapText(obj)} is not a Buffer object.`);
- },
-
- bool(value) {
- value = resolveFunc(value);
- if (isNull(value)) {
- return 'null';
- }
- return $to.bool(value);
- },
-
- date(d, raw) {
- d = resolveFunc(d);
- if (isNull(d)) {
- throwIfRaw(raw);
- return 'null';
- }
- if (d instanceof Date) {
- return $to.date(d, raw);
- }
- throw new TypeError(`${wrapText(d)} is not a Date object.`);
- },
-
- number(num) {
- num = resolveFunc(num);
- if (isNull(num)) {
- return 'null';
- }
- const t = typeof num;
- if (t !== 'number' && t !== 'bigint') {
- throw new TypeError(`${wrapText(num)} is not a number.`);
- }
- return $to.number(num);
- },
-
- array(arr, options) {
- options = assert(options, ['capSQL']);
- arr = resolveFunc(arr);
- if (isNull(arr)) {
- return 'null';
- }
- if (Array.isArray(arr)) {
- return $to.array(arr, options);
- }
- throw new TypeError(`${wrapText(arr)} is not an Array object.`);
- },
-
- csv(values) {
- return $to.csv(values);
- },
-
- json(data, raw) {
- data = resolveFunc(data);
- if (isNull(data)) {
- throwIfRaw(raw);
- return 'null';
- }
- return $to.json(data, raw);
- },
-
- func(func, raw, cc) {
- if (isNull(func)) {
- throwIfRaw(raw);
- return 'null';
- }
- if (typeof func !== 'function') {
- throw new TypeError(`${wrapText(func)} is not a function.`);
- }
- const fm = raw ? fmFlags.raw : null;
- return formatValue({value: resolveFunc(func, cc), fm, cc});
- },
-
- format(query, values, options) {
- options = assert(options, ['capSQL', 'partial', 'def']);
- const ctf = getCTF(query);
- if (ctf) {
- query = ctf.toPostgres.call(query, query);
- }
- return formatQuery(query, values, false, options);
- }
- };
- const $to = {
- array(arr, options) {
- return formatArray(arr, options);
- },
- csv(values, options) {
- return formatCSV(resolveFunc(values), options);
- },
- bool(value) {
- return value ? 'true' : 'false';
- },
- buffer(obj, raw) {
- const s = `\\x${obj.toString('hex')}`;
- return raw ? s : wrapText(s);
- },
- date(d, raw) {
- const s = npm.pgUtils.prepareValue(d);
- return raw ? s : wrapText(s);
- },
- json(data, raw) {
- const s = npm.utils.toJson(data);
- return raw ? s : wrapText(safeText(s));
- },
- number(num) {
- if (typeof num === 'bigint' || Number.isFinite(num)) {
- const s = num.toString();
- return num < 0 ? `(${s})` : s;
- }
-
-
-
-
- if (num === Number.POSITIVE_INFINITY) {
- return wrapText('+Infinity');
- }
- if (num === Number.NEGATIVE_INFINITY) {
- return wrapText('-Infinity');
- }
- return wrapText('NaN');
- },
- text(value, raw) {
- return raw ? value : wrapText(safeText(value));
- }
- };
- module.exports = {
- formatQuery,
- formatEntity,
- resolveFunc,
- as: $as
- };
|