1 |
|
2 |
|
3 | const inherits = require('inherits');
|
4 | const helpers = require('./helpers');
|
5 | const { EventEmitter } = require('events');
|
6 | const debug = require('debug');
|
7 |
|
8 | const {
|
9 | assign,
|
10 | reduce,
|
11 | isPlainObject,
|
12 | isObject,
|
13 | isUndefined,
|
14 | isNumber,
|
15 | } = require('lodash');
|
16 | const saveAsyncStack = require('./util/save-async-stack');
|
17 | const uuid = require('uuid');
|
18 |
|
19 | const debugBindings = debug('knex:bindings');
|
20 |
|
21 | function Raw(client) {
|
22 | this.client = client;
|
23 |
|
24 | this.sql = '';
|
25 | this.bindings = [];
|
26 |
|
27 |
|
28 | this._wrappedBefore = undefined;
|
29 | this._wrappedAfter = undefined;
|
30 | if (client && client.config) {
|
31 | this._debug = client.config.debug;
|
32 | saveAsyncStack(this, 4);
|
33 | }
|
34 | }
|
35 |
|
36 | inherits(Raw, EventEmitter);
|
37 |
|
38 | assign(Raw.prototype, {
|
39 | set(sql, bindings) {
|
40 | this.sql = sql;
|
41 | this.bindings =
|
42 | (isObject(bindings) && !bindings.toSQL) || isUndefined(bindings)
|
43 | ? bindings
|
44 | : [bindings];
|
45 |
|
46 | return this;
|
47 | },
|
48 |
|
49 | timeout(ms, { cancel } = {}) {
|
50 | if (isNumber(ms) && ms > 0) {
|
51 | this._timeout = ms;
|
52 | if (cancel) {
|
53 | this.client.assertCanCancelQuery();
|
54 | this._cancelOnTimeout = true;
|
55 | }
|
56 | }
|
57 | return this;
|
58 | },
|
59 |
|
60 |
|
61 | wrap(before, after) {
|
62 | this._wrappedBefore = before;
|
63 | this._wrappedAfter = after;
|
64 | return this;
|
65 | },
|
66 |
|
67 |
|
68 | toString() {
|
69 | return this.toQuery();
|
70 | },
|
71 |
|
72 |
|
73 | toSQL(method, tz) {
|
74 | let obj;
|
75 | const formatter = this.client.formatter(this);
|
76 |
|
77 | if (Array.isArray(this.bindings)) {
|
78 | obj = replaceRawArrBindings(this, formatter);
|
79 | } else if (this.bindings && isPlainObject(this.bindings)) {
|
80 | obj = replaceKeyBindings(this, formatter);
|
81 | } else {
|
82 | obj = {
|
83 | method: 'raw',
|
84 | sql: this.sql,
|
85 | bindings: isUndefined(this.bindings) ? [] : [this.bindings],
|
86 | };
|
87 | }
|
88 |
|
89 | if (this._wrappedBefore) {
|
90 | obj.sql = this._wrappedBefore + obj.sql;
|
91 | }
|
92 | if (this._wrappedAfter) {
|
93 | obj.sql = obj.sql + this._wrappedAfter;
|
94 | }
|
95 |
|
96 | obj.options = reduce(this._options, assign, {});
|
97 |
|
98 | if (this._timeout) {
|
99 | obj.timeout = this._timeout;
|
100 | if (this._cancelOnTimeout) {
|
101 | obj.cancelOnTimeout = this._cancelOnTimeout;
|
102 | }
|
103 | }
|
104 |
|
105 | obj.bindings = obj.bindings || [];
|
106 | if (helpers.containsUndefined(obj.bindings)) {
|
107 | debugBindings(obj.bindings);
|
108 | throw new Error(
|
109 | `Undefined binding(s) detected when compiling RAW query: ` + obj.sql
|
110 | );
|
111 | }
|
112 |
|
113 | obj.__knexQueryUid = uuid.v4();
|
114 |
|
115 | return obj;
|
116 | },
|
117 | });
|
118 |
|
119 | function replaceRawArrBindings(raw, formatter) {
|
120 | const expectedBindings = raw.bindings.length;
|
121 | const values = raw.bindings;
|
122 | let index = 0;
|
123 |
|
124 | const sql = raw.sql.replace(/\\?\?\??/g, function(match) {
|
125 | if (match === '\\?') {
|
126 | return match;
|
127 | }
|
128 |
|
129 | const value = values[index++];
|
130 |
|
131 | if (match === '??') {
|
132 | return formatter.columnize(value);
|
133 | }
|
134 | return formatter.parameter(value);
|
135 | });
|
136 |
|
137 | if (expectedBindings !== index) {
|
138 | throw new Error(`Expected ${expectedBindings} bindings, saw ${index}`);
|
139 | }
|
140 |
|
141 | return {
|
142 | method: 'raw',
|
143 | sql,
|
144 | bindings: formatter.bindings,
|
145 | };
|
146 | }
|
147 |
|
148 | function replaceKeyBindings(raw, formatter) {
|
149 | const values = raw.bindings;
|
150 | const regex = /\\?(:(\w+):(?=::)|:(\w+):(?!:)|:(\w+))/g;
|
151 |
|
152 | const sql = raw.sql.replace(regex, function(match, p1, p2, p3, p4) {
|
153 | if (match !== p1) {
|
154 | return p1;
|
155 | }
|
156 |
|
157 | const part = p2 || p3 || p4;
|
158 | const key = match.trim();
|
159 | const isIdentifier = key[key.length - 1] === ':';
|
160 | const value = values[part];
|
161 |
|
162 | if (value === undefined) {
|
163 | if (Object.prototype.hasOwnProperty.call(values, part)) {
|
164 | formatter.bindings.push(value);
|
165 | }
|
166 |
|
167 | return match;
|
168 | }
|
169 |
|
170 | if (isIdentifier) {
|
171 | return match.replace(p1, formatter.columnize(value));
|
172 | }
|
173 |
|
174 | return match.replace(p1, formatter.parameter(value));
|
175 | });
|
176 |
|
177 | return {
|
178 | method: 'raw',
|
179 | sql,
|
180 | bindings: formatter.bindings,
|
181 | };
|
182 | }
|
183 |
|
184 |
|
185 |
|
186 | require('./interface')(Raw);
|
187 | helpers.addQueryContext(Raw);
|
188 |
|
189 | module.exports = Raw;
|