// 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); };