swc/crates/swc_bundler/tests/.cache/untrusted/dc51bee73c089e5bbc5bfcfa0dea6a376218b4b8.ts
2021-11-09 20:42:49 +09:00

105 lines
3.1 KiB
TypeScript

// Loaded from https://raw.githubusercontent.com/denjucks/dex/master/lib/interface.js
import _ from './deps/lodash@4.17.15/index.js';
const clone = _.clone;
const isEmpty = _.isEmpty;
import { callbackify } from './deps/@jspm/core@1.1.0/nodelibs/util.js';
import finallyMixin from './util/finally-mixin.js';
export default function (Target) {
Target.prototype.toQuery = function (tz) {
let data = this.toSQL(this._method, tz);
if (!Array.isArray(data)) data = [data];
return data
.map((statement) => {
return this.client._formatQuery(statement.sql, statement.bindings, tz);
})
.join(';\n');
};
// Create a new instance of the `Runner`, passing in the current object.
Target.prototype.then = function (/* onFulfilled, onRejected */) {
let result = this.client.runner(this).run();
if (this.client.config.asyncStackTraces) {
result = result.catch((err) => {
err.originalStack = err.stack;
const firstLine = err.stack.split('\n')[0];
this._asyncStack.unshift(firstLine);
// put the fake more helpful "async" stack on the thrown error
err.stack = this._asyncStack.join('\n');
throw err;
});
}
return result.then.apply(result, arguments);
};
// Add additional "options" to the builder. Typically used for client specific
// items, like the `mysql` and `sqlite3` drivers.
Target.prototype.options = function (opts) {
this._options = this._options || [];
this._options.push(clone(opts) || {});
return this;
};
// Sets an explicit "connection" we wish to use for this query.
Target.prototype.connection = function (connection) {
this._connection = connection;
return this;
};
// Set a debug flag for the current schema query stack.
Target.prototype.debug = function (enabled) {
this._debug = arguments.length ? enabled : true;
return this;
};
// Set the transaction object for this query.
Target.prototype.transacting = function (t) {
if (t && t.client) {
if (!t.client.transacting) {
t.client.logger.warn(`Invalid transaction value: ${t.client}`);
} else {
this.client = t.client;
}
}
if (isEmpty(t)) {
this.client.logger.error(
'Invalid value on transacting call, potential bug'
);
throw Error(
'Invalid transacting value (null, undefined or empty object)'
);
}
return this;
};
// Initializes a stream.
Target.prototype.stream = function (options) {
return this.client.runner(this).stream(options);
};
// Initialize a stream & pipe automatically.
Target.prototype.pipe = function (writable, options) {
return this.client.runner(this).pipe(writable, options);
};
Target.prototype.asCallback = function (cb) {
const promise = this.then();
callbackify(() => promise)(cb);
return promise;
};
Target.prototype.catch = function (onReject) {
return this.then().catch(onReject);
};
Object.defineProperty(Target.prototype, Symbol.toStringTag, {
get: () => 'object',
});
finallyMixin(Target.prototype);
};