123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108 |
- "use strict";
- const dataTypes = require("./data-types");
- const { logger } = require("./utils/logger");
- function arrayToList(array, timeZone, dialect, format2) {
- return array.reduce((sql, val, i) => {
- if (i !== 0) {
- sql += ", ";
- }
- if (Array.isArray(val)) {
- sql += `(${arrayToList(val, timeZone, dialect, format2)})`;
- } else {
- sql += escape(val, timeZone, dialect, format2);
- }
- return sql;
- }, "");
- }
- exports.arrayToList = arrayToList;
- function escape(val, timeZone, dialect, format2) {
- let prependN = false;
- if (val === void 0 || val === null) {
- return "NULL";
- }
- switch (typeof val) {
- case "boolean":
- if (["sqlite", "mssql"].includes(dialect)) {
- return +!!val;
- }
- return (!!val).toString();
- case "number":
- return val.toString();
- case "string":
- prependN = dialect === "mssql";
- break;
- }
- if (val instanceof Date) {
- val = dataTypes[dialect].DATE.prototype.stringify(val, { timezone: timeZone });
- }
- if (Buffer.isBuffer(val)) {
- if (dataTypes[dialect].BLOB) {
- return dataTypes[dialect].BLOB.prototype.stringify(val);
- }
- return dataTypes.BLOB.prototype.stringify(val);
- }
- if (Array.isArray(val)) {
- const partialEscape = (escVal) => escape(escVal, timeZone, dialect, format2);
- if (dialect === "postgres" && !format2) {
- return dataTypes.ARRAY.prototype.stringify(val, { escape: partialEscape });
- }
- return arrayToList(val, timeZone, dialect, format2);
- }
- if (!val.replace) {
- throw new Error(`Invalid value ${logger.inspect(val)}`);
- }
- if (["postgres", "sqlite", "mssql", "snowflake", "db2"].includes(dialect)) {
- val = val.replace(/'/g, "''");
- if (dialect === "postgres") {
- val = val.replace(/\0/g, "\\0");
- }
- } else {
- val = val.replace(/[\0\n\r\b\t\\'"\x1a]/g, (s) => {
- switch (s) {
- case "\0":
- return "\\0";
- case "\n":
- return "\\n";
- case "\r":
- return "\\r";
- case "\b":
- return "\\b";
- case " ":
- return "\\t";
- case "":
- return "\\Z";
- default:
- return `\\${s}`;
- }
- });
- }
- return `${(prependN ? "N'" : "'") + val}'`;
- }
- exports.escape = escape;
- function format(sql, values, timeZone, dialect) {
- values = [].concat(values);
- if (typeof sql !== "string") {
- throw new Error(`Invalid SQL string provided: ${sql}`);
- }
- return sql.replace(/\?/g, (match) => {
- if (!values.length) {
- return match;
- }
- return escape(values.shift(), timeZone, dialect, true);
- });
- }
- exports.format = format;
- function formatNamedParameters(sql, values, timeZone, dialect) {
- return sql.replace(/:+(?!\d)(\w+)/g, (value, key) => {
- if (dialect === "postgres" && value.slice(0, 2) === "::") {
- return value;
- }
- if (values[key] !== void 0) {
- return escape(values[key], timeZone, dialect, true);
- }
- throw new Error(`Named parameter "${value}" has no value in the given object.`);
- });
- }
- exports.formatNamedParameters = formatNamedParameters;
- //# sourceMappingURL=sql-string.js.map
|