knex/lib/knex-builder/internal/parse-connection.js

86 lines
1.9 KiB
JavaScript
Raw Normal View History

const { parse } = require('pg-connection-string');
const parsePG = parse;
const isWindows = process && process.platform && process.platform === 'win32';
2015-05-09 13:58:18 -04:00
/**
* @param str
* @returns {URL}
*/
function tryParse(str) {
try {
return new URL(str);
} catch (e) {
return null;
}
}
module.exports = function parseConnectionString(str) {
const parsed = tryParse(str);
const isDriveLetter = isWindows && parsed && parsed.protocol.length === 2;
if (!parsed || isDriveLetter) {
2015-05-09 13:58:18 -04:00
return {
client: 'sqlite3',
connection: {
filename: str,
},
};
2015-05-09 13:58:18 -04:00
}
let { protocol } = parsed;
2015-05-09 13:58:18 -04:00
if (protocol.slice(-1) === ':') {
protocol = protocol.slice(0, -1);
}
const isPG = ['postgresql', 'postgres'].includes(protocol);
2015-05-09 13:58:18 -04:00
return {
client: protocol,
connection: isPG ? parsePG(str) : connectionObject(parsed),
};
};
2015-05-09 13:58:18 -04:00
/**
* @param {URL} parsed
* @returns {{}}
*/
2015-05-09 13:58:18 -04:00
function connectionObject(parsed) {
const connection = {};
let db = parsed.pathname;
2015-05-09 13:58:18 -04:00
if (db[0] === '/') {
db = db.slice(1);
2015-05-09 13:58:18 -04:00
}
connection.database = db;
2015-05-09 13:58:18 -04:00
if (parsed.hostname) {
if (parsed.protocol.indexOf('mssql') === 0) {
connection.server = parsed.hostname;
} else {
connection.host = parsed.hostname;
}
2015-05-09 13:58:18 -04:00
}
if (parsed.port) {
connection.port = parsed.port;
}
if (parsed.username || parsed.password) {
connection.user = decodeURIComponent(parsed.username);
}
if (parsed.password) {
connection.password = decodeURIComponent(parsed.password);
2015-05-09 13:58:18 -04:00
}
if (parsed.searchParams) {
for (const [key, value] of parsed.searchParams.entries()) {
const isMySQLLike = ['mysql:', 'mariadb:'].includes(parsed.protocol);
if (isMySQLLike) {
try {
connection[key] = JSON.parse(value);
} catch (err) {
connection[key] = value;
}
} else {
connection[key] = value;
}
}
}
return connection;
}