1 | 'use strict'
|
2 |
|
3 |
|
4 |
|
5 |
|
6 |
|
7 |
|
8 |
|
9 |
|
10 | const crypto = require('crypto')
|
11 |
|
12 | const defaults = require('./defaults')
|
13 |
|
14 | function escapeElement(elementRepresentation) {
|
15 | var escaped = elementRepresentation.replace(/\\/g, '\\\\').replace(/"/g, '\\"')
|
16 |
|
17 | return '"' + escaped + '"'
|
18 | }
|
19 |
|
20 |
|
21 |
|
22 |
|
23 | function arrayString(val) {
|
24 | var result = '{'
|
25 | for (var i = 0; i < val.length; i++) {
|
26 | if (i > 0) {
|
27 | result = result + ','
|
28 | }
|
29 | if (val[i] === null || typeof val[i] === 'undefined') {
|
30 | result = result + 'NULL'
|
31 | } else if (Array.isArray(val[i])) {
|
32 | result = result + arrayString(val[i])
|
33 | } else if (val[i] instanceof Buffer) {
|
34 | result += '\\\\x' + val[i].toString('hex')
|
35 | } else {
|
36 | result += escapeElement(prepareValue(val[i]))
|
37 | }
|
38 | }
|
39 | result = result + '}'
|
40 | return result
|
41 | }
|
42 |
|
43 |
|
44 |
|
45 |
|
46 |
|
47 | var prepareValue = function (val, seen) {
|
48 | if (val instanceof Buffer) {
|
49 | return val
|
50 | }
|
51 | if (ArrayBuffer.isView(val)) {
|
52 | var buf = Buffer.from(val.buffer, val.byteOffset, val.byteLength)
|
53 | if (buf.length === val.byteLength) {
|
54 | return buf
|
55 | }
|
56 | return buf.slice(val.byteOffset, val.byteOffset + val.byteLength)
|
57 | }
|
58 | if (val instanceof Date) {
|
59 | if (defaults.parseInputDatesAsUTC) {
|
60 | return dateToStringUTC(val)
|
61 | } else {
|
62 | return dateToString(val)
|
63 | }
|
64 | }
|
65 | if (Array.isArray(val)) {
|
66 | return arrayString(val)
|
67 | }
|
68 | if (val === null || typeof val === 'undefined') {
|
69 | return null
|
70 | }
|
71 | if (typeof val === 'object') {
|
72 | return prepareObject(val, seen)
|
73 | }
|
74 | return val.toString()
|
75 | }
|
76 |
|
77 | function prepareObject(val, seen) {
|
78 | if (val && typeof val.toPostgres === 'function') {
|
79 | seen = seen || []
|
80 | if (seen.indexOf(val) !== -1) {
|
81 | throw new Error('circular reference detected while preparing "' + val + '" for query')
|
82 | }
|
83 | seen.push(val)
|
84 |
|
85 | return prepareValue(val.toPostgres(prepareValue), seen)
|
86 | }
|
87 | return JSON.stringify(val)
|
88 | }
|
89 |
|
90 | function pad(number, digits) {
|
91 | number = '' + number
|
92 | while (number.length < digits) {
|
93 | number = '0' + number
|
94 | }
|
95 | return number
|
96 | }
|
97 |
|
98 | function dateToString(date) {
|
99 | var offset = -date.getTimezoneOffset()
|
100 |
|
101 | var year = date.getFullYear()
|
102 | var isBCYear = year < 1
|
103 | if (isBCYear) year = Math.abs(year) + 1
|
104 |
|
105 | var ret =
|
106 | pad(year, 4) +
|
107 | '-' +
|
108 | pad(date.getMonth() + 1, 2) +
|
109 | '-' +
|
110 | pad(date.getDate(), 2) +
|
111 | 'T' +
|
112 | pad(date.getHours(), 2) +
|
113 | ':' +
|
114 | pad(date.getMinutes(), 2) +
|
115 | ':' +
|
116 | pad(date.getSeconds(), 2) +
|
117 | '.' +
|
118 | pad(date.getMilliseconds(), 3)
|
119 |
|
120 | if (offset < 0) {
|
121 | ret += '-'
|
122 | offset *= -1
|
123 | } else {
|
124 | ret += '+'
|
125 | }
|
126 |
|
127 | ret += pad(Math.floor(offset / 60), 2) + ':' + pad(offset % 60, 2)
|
128 | if (isBCYear) ret += ' BC'
|
129 | return ret
|
130 | }
|
131 |
|
132 | function dateToStringUTC(date) {
|
133 | var year = date.getUTCFullYear()
|
134 | var isBCYear = year < 1
|
135 | if (isBCYear) year = Math.abs(year) + 1
|
136 |
|
137 | var ret =
|
138 | pad(year, 4) +
|
139 | '-' +
|
140 | pad(date.getUTCMonth() + 1, 2) +
|
141 | '-' +
|
142 | pad(date.getUTCDate(), 2) +
|
143 | 'T' +
|
144 | pad(date.getUTCHours(), 2) +
|
145 | ':' +
|
146 | pad(date.getUTCMinutes(), 2) +
|
147 | ':' +
|
148 | pad(date.getUTCSeconds(), 2) +
|
149 | '.' +
|
150 | pad(date.getUTCMilliseconds(), 3)
|
151 |
|
152 | ret += '+00:00'
|
153 | if (isBCYear) ret += ' BC'
|
154 | return ret
|
155 | }
|
156 |
|
157 | function normalizeQueryConfig(config, values, callback) {
|
158 |
|
159 | config = typeof config === 'string' ? { text: config } : config
|
160 | if (values) {
|
161 | if (typeof values === 'function') {
|
162 | config.callback = values
|
163 | } else {
|
164 | config.values = values
|
165 | }
|
166 | }
|
167 | if (callback) {
|
168 | config.callback = callback
|
169 | }
|
170 | return config
|
171 | }
|
172 |
|
173 | const md5 = function (string) {
|
174 | return crypto.createHash('md5').update(string, 'utf-8').digest('hex')
|
175 | }
|
176 |
|
177 |
|
178 | const postgresMd5PasswordHash = function (user, password, salt) {
|
179 | var inner = md5(password + user)
|
180 | var outer = md5(Buffer.concat([Buffer.from(inner), salt]))
|
181 | return 'md5' + outer
|
182 | }
|
183 |
|
184 | module.exports = {
|
185 | prepareValue: function prepareValueWrapper(value) {
|
186 |
|
187 |
|
188 | return prepareValue(value)
|
189 | },
|
190 | normalizeQueryConfig,
|
191 | postgresMd5PasswordHash,
|
192 | md5,
|
193 | }
|