137 lines
4.1 KiB
JavaScript
137 lines
4.1 KiB
JavaScript
var assert = require('assert');
|
|
var step = require('step');
|
|
var PSQL = require('cartodb-psql');
|
|
var _ = require('underscore');
|
|
|
|
function PgConnection(metadataBackend) {
|
|
this.metadataBackend = metadataBackend;
|
|
}
|
|
|
|
module.exports = PgConnection;
|
|
|
|
|
|
// Set db authentication parameters to those of the given username
|
|
//
|
|
// @param username the cartodb username, mapped to a database username
|
|
// via CartodbRedis metadata records
|
|
//
|
|
// @param params the parameters to set auth options into
|
|
// added params are: "dbuser" and "dbpassword"
|
|
//
|
|
// @param callback function(err)
|
|
//
|
|
PgConnection.prototype.setDBAuth = function(username, params, callback) {
|
|
var self = this;
|
|
|
|
var user_params = {};
|
|
var auth_user = global.environment.postgres_auth_user;
|
|
var auth_pass = global.environment.postgres_auth_pass;
|
|
step(
|
|
function getId() {
|
|
self.metadataBackend.getUserId(username, this);
|
|
},
|
|
function(err, user_id) {
|
|
assert.ifError(err);
|
|
user_params.user_id = user_id;
|
|
var dbuser = _.template(auth_user, user_params);
|
|
_.extend(params, {dbuser:dbuser});
|
|
|
|
// skip looking up user_password if postgres_auth_pass
|
|
// doesn't contain the "user_password" label
|
|
if (!auth_pass || ! auth_pass.match(/\buser_password\b/) ) {
|
|
return null;
|
|
}
|
|
|
|
self.metadataBackend.getUserDBPass(username, this);
|
|
},
|
|
function(err, user_password) {
|
|
assert.ifError(err);
|
|
user_params.user_password = user_password;
|
|
if ( auth_pass ) {
|
|
var dbpass = _.template(auth_pass, user_params);
|
|
_.extend(params, {dbpassword:dbpass});
|
|
}
|
|
return true;
|
|
},
|
|
function finish(err) {
|
|
callback(err);
|
|
}
|
|
);
|
|
};
|
|
|
|
// Set db connection parameters to those for the given username
|
|
//
|
|
// @param dbowner cartodb username of database owner,
|
|
// mapped to a database username
|
|
// via CartodbRedis metadata records
|
|
//
|
|
// @param params the parameters to set connection options into
|
|
// added params are: "dbname", "dbhost"
|
|
//
|
|
// @param callback function(err)
|
|
//
|
|
PgConnection.prototype.setDBConn = function(dbowner, params, callback) {
|
|
var self = this;
|
|
// Add default database connection parameters
|
|
// if none given
|
|
_.defaults(params, {
|
|
dbuser: global.environment.postgres.user,
|
|
dbpassword: global.environment.postgres.password,
|
|
dbhost: global.environment.postgres.host,
|
|
dbport: global.environment.postgres.port
|
|
});
|
|
step(
|
|
function getConnectionParams() {
|
|
self.metadataBackend.getUserDBConnectionParams(dbowner, this);
|
|
},
|
|
function extendParams(err, dbParams){
|
|
assert.ifError(err);
|
|
// we don't want null values or overwrite a non public user
|
|
if (params.dbuser !== 'publicuser' || !dbParams.dbuser) {
|
|
delete dbParams.dbuser;
|
|
}
|
|
if ( dbParams ) {
|
|
_.extend(params, dbParams);
|
|
}
|
|
return null;
|
|
},
|
|
function finish(err) {
|
|
callback(err);
|
|
}
|
|
);
|
|
};
|
|
|
|
|
|
/**
|
|
* Returns a `cartodb-psql` object for a given username.
|
|
* @param {String} username
|
|
* @param {Function} callback function({Error}, {PSQL})
|
|
*/
|
|
|
|
PgConnection.prototype.getConnection = function(username, callback) {
|
|
var self = this;
|
|
|
|
var params = {};
|
|
|
|
require('debug')('cachechan')("getConn1");
|
|
step(
|
|
function setAuth() {
|
|
self.setDBAuth(username, params, this);
|
|
},
|
|
function setConn(err) {
|
|
assert.ifError(err);
|
|
self.setDBConn(username, params, this);
|
|
},
|
|
function openConnection(err) {
|
|
assert.ifError(err);
|
|
return callback(err, new PSQL({
|
|
user: params.dbuser,
|
|
pass: params.dbpass,
|
|
host: params.dbhost,
|
|
port: params.dbport,
|
|
dbname: params.dbname
|
|
}));
|
|
}
|
|
);
|
|
};
|