mirror of
https://github.com/swc-project/swc.git
synced 2024-12-25 14:43:33 +03:00
174 lines
5.0 KiB
TypeScript
174 lines
5.0 KiB
TypeScript
|
// Loaded from https://raw.githubusercontent.com/denjucks/dex/master/lib/dialects/sqlite3/query/compiler.js
|
||
|
|
||
|
|
||
|
// SQLite3 Query Builder & Compiler
|
||
|
|
||
|
import QueryCompiler from '../../../query/compiler.js';
|
||
|
|
||
|
import noop from '../../../util/noop.js';
|
||
|
import _ from '../../../deps/lodash@4.17.15/index.js';
|
||
|
const constant = _.constant;
|
||
|
const each = _.each;
|
||
|
const identity = _.identity;
|
||
|
const isEmpty = _.isEmpty;
|
||
|
const isString = _.isString;
|
||
|
const reduce = _.reduce;
|
||
|
|
||
|
const emptyStr = constant('');
|
||
|
|
||
|
class QueryCompiler_SQLite3 extends QueryCompiler {
|
||
|
constructor(client, builder) {
|
||
|
super(client, builder);
|
||
|
|
||
|
const { returning } = this.single;
|
||
|
|
||
|
if (returning) {
|
||
|
this.client.logger.warn(
|
||
|
'.returning() is not supported by sqlite3 and will not have any effect.'
|
||
|
);
|
||
|
}
|
||
|
|
||
|
// The locks are not applicable in SQLite3
|
||
|
this.forShare = emptyStr;
|
||
|
this.forUpdate = emptyStr;
|
||
|
}
|
||
|
|
||
|
// SQLite requires us to build the multi-row insert as a listing of select with
|
||
|
// unions joining them together. So we'll build out this list of columns and
|
||
|
// then join them all together with select unions to complete the queries.
|
||
|
insert() {
|
||
|
const insertValues = this.single.insert || [];
|
||
|
let sql = this.with() + `insert into ${this.tableName} `;
|
||
|
|
||
|
if (Array.isArray(insertValues)) {
|
||
|
if (insertValues.length === 0) {
|
||
|
return '';
|
||
|
} else if (
|
||
|
insertValues.length === 1 &&
|
||
|
insertValues[0] &&
|
||
|
isEmpty(insertValues[0])
|
||
|
) {
|
||
|
return sql + this._emptyInsertValue;
|
||
|
}
|
||
|
} else if (typeof insertValues === 'object' && isEmpty(insertValues)) {
|
||
|
return sql + this._emptyInsertValue;
|
||
|
}
|
||
|
|
||
|
const insertData = this._prepInsert(insertValues);
|
||
|
|
||
|
if (isString(insertData)) {
|
||
|
return sql + insertData;
|
||
|
}
|
||
|
|
||
|
if (insertData.columns.length === 0) {
|
||
|
return '';
|
||
|
}
|
||
|
|
||
|
sql += `(${this.formatter.columnize(insertData.columns)})`;
|
||
|
|
||
|
// backwards compatible error
|
||
|
if (this.client.valueForUndefined !== null) {
|
||
|
insertData.values.forEach((bindings) => {
|
||
|
each(bindings, (binding) => {
|
||
|
if (binding === undefined)
|
||
|
throw new TypeError(
|
||
|
'`sqlite` does not support inserting default values. Specify ' +
|
||
|
'values explicitly or use the `useNullAsDefault` config flag. ' +
|
||
|
'(see docs http://knexjs.org/#Builder-insert).'
|
||
|
);
|
||
|
});
|
||
|
});
|
||
|
}
|
||
|
|
||
|
if (insertData.values.length === 1) {
|
||
|
const parameters = this.formatter.parameterize(
|
||
|
insertData.values[0],
|
||
|
this.client.valueForUndefined
|
||
|
);
|
||
|
return sql + ` values (${parameters})`;
|
||
|
}
|
||
|
|
||
|
const blocks = [];
|
||
|
let i = -1;
|
||
|
while (++i < insertData.values.length) {
|
||
|
let i2 = -1;
|
||
|
const block = (blocks[i] = []);
|
||
|
let current = insertData.values[i];
|
||
|
current = current === undefined ? this.client.valueForUndefined : current;
|
||
|
while (++i2 < insertData.columns.length) {
|
||
|
block.push(
|
||
|
this.formatter.alias(
|
||
|
this.formatter.parameter(current[i2]),
|
||
|
this.formatter.wrap(insertData.columns[i2])
|
||
|
)
|
||
|
);
|
||
|
}
|
||
|
blocks[i] = block.join(', ');
|
||
|
}
|
||
|
return sql + ' select ' + blocks.join(' union all select ');
|
||
|
}
|
||
|
|
||
|
// Compile a truncate table statement into SQL.
|
||
|
truncate() {
|
||
|
const { table } = this.single;
|
||
|
return {
|
||
|
sql: `delete from ${this.tableName}`,
|
||
|
output() {
|
||
|
return this.query({
|
||
|
sql: `delete from sqlite_sequence where name = '${table}'`,
|
||
|
}).catch(noop);
|
||
|
},
|
||
|
};
|
||
|
}
|
||
|
|
||
|
// Compiles a `columnInfo` query
|
||
|
columnInfo() {
|
||
|
const column = this.single.columnInfo;
|
||
|
|
||
|
// The user may have specified a custom wrapIdentifier function in the config. We
|
||
|
// need to run the identifiers through that function, but not format them as
|
||
|
// identifiers otherwise.
|
||
|
const table = this.client.customWrapIdentifier(this.single.table, identity);
|
||
|
|
||
|
return {
|
||
|
sql: `PRAGMA table_info(\`${table}\`)`,
|
||
|
output(resp) {
|
||
|
const maxLengthRegex = /.*\((\d+)\)/;
|
||
|
const out = reduce(
|
||
|
resp,
|
||
|
function (columns, val) {
|
||
|
let { type } = val;
|
||
|
let maxLength = type.match(maxLengthRegex);
|
||
|
if (maxLength) {
|
||
|
maxLength = maxLength[1];
|
||
|
}
|
||
|
type = maxLength ? type.split('(')[0] : type;
|
||
|
columns[val.name] = {
|
||
|
type: type.toLowerCase(),
|
||
|
maxLength,
|
||
|
nullable: !val.notnull,
|
||
|
defaultValue: val.dflt_value,
|
||
|
};
|
||
|
return columns;
|
||
|
},
|
||
|
{}
|
||
|
);
|
||
|
return (column && out[column]) || out;
|
||
|
},
|
||
|
};
|
||
|
}
|
||
|
|
||
|
limit() {
|
||
|
const noLimit = !this.single.limit && this.single.limit !== 0;
|
||
|
if (noLimit && !this.single.offset) return '';
|
||
|
|
||
|
// Workaround for offset only,
|
||
|
// see http://stackoverflow.com/questions/10491492/sqllite-with-skip-offset-only-not-limit
|
||
|
return `limit ${this.formatter.parameter(
|
||
|
noLimit ? -1 : this.single.limit
|
||
|
)}`;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
export default QueryCompiler_SQLite3;
|