index.js

'use strict';

var Heroku = require('heroku-client');
var poll = require('when/poll');
var rp = require('request-promise');
var when = require('when');

module.exports = function (sourceApp, targetApp, verbose) {
	console.error('heroku-db-transfer no longer works, see README for details', 'GET /apps/' + targetApp + '/addons');
	throw Error();
	
	var authToken = process.env.HEROKU_API_TOKEN;
	var interval = 2 * 1000;
	var timeoutWithoutProgress = 30 * 1000;
	
	var heroku = new Heroku({token: authToken});
	
	var sourceTransfer;
	
	return when
		.all([
			heroku.get('/apps/' + sourceApp + '/config-vars').catch(function (err) {
				console.error('error with', 'GET /apps/' + sourceApp + '/config-vars');
				throw err;
			}),
			heroku.get('/apps/' + targetApp + '/config-vars').catch(function (err) {
				console.error('error with', 'GET /apps/' + targetApp + '/config-vars');
				throw err;
			}),
			heroku.get('/apps/' + sourceApp + '/addons').catch(function (err) {
				console.error('error with', 'GET /apps/' + sourceApp + '/addons');
				throw err;
			}),
			heroku.get('/apps/' + targetApp + '/addons').catch(function (err) {
				console.error('error with', 'GET /apps/' + targetApp + '/addons');
				throw err;
			})
		])
		.spread(function (sourceConfigVars, targetConfigVars, sourceAddons, targetAddons) {
			var dbs = [{}, {}];
			var appConfigs = [sourceConfigVars, targetConfigVars];
			var appAddons = [sourceAddons, targetAddons];
			
			function isPostgresEnvKey(x) {
				// matches against DATABASE_URL and legacy pattern HEROKU_POSTGRESQL_color_URL
				return x === 'DATABASE_URL' || x.startsWith('HEROKU_POSTGRESQL_') && x.endsWith('_URL');
			}
			
			dbs.forEach(function (db, i) {
				// extract url from env property with a key that has a heroku postgres pattern
				var key = Object.keys(appConfigs[i]).find(isPostgresEnvKey);
				db.url = appConfigs[i][key];
				
				// extract addon name that has config var with a heroku postgres pattern
				var addon = appAddons[i].find(function (addon) {
					return addon.config_vars.find(isPostgresEnvKey);
				});
				db.name = addon.name;
				// only non hobby-dev and non hobby-basic dbs can do db transfers
				db.canTransfer = !/:(hobby-dev|hobby-basic)$/.test(addon.plan.name);
			});
			
			var source = dbs[0];
			var target = dbs[1];
			
			if (!source.canTransfer && !target.canTransfer) {
				console.error('Heroku must have either source or target db be a full database (not hobby-dev nor hobby-basic) to perform a db transfer');
				throw Error();
			}
			
			sourceTransfer = source.canTransfer;
			
			var postTransferUrl = 'https://postgres-api.heroku.com/client/v11/databases/' + (sourceTransfer ? source.name : target.name) + '/transfers';
			return rp({
				uri: postTransferUrl,
				method: 'POST',
				json: true,
				auth: {user: '', pass: authToken},
				body: {
					from_name: source.name,
					from_url: source.url,
					to_name: target.name,
					to_url: target.url
				}
			}).catch(function (err) {
				console.error('error with', 'POST', postTransferUrl);
				throw err;
			});
		})
		.then(function (transferState) {
			var getTransferState = function () {
				var getTransferUrl = 'https://postgres-api.heroku.com/client/v11/apps/' + (sourceTransfer ? sourceApp : targetApp) + '/transfers/' + transferState.uuid;
				return rp({
					uri: getTransferUrl,
					method: 'GET',
					json: true,
					auth: {user: '', pass: authToken}
				}).catch(function (err) {
					console.error('error with', 'GET', getTransferUrl);
					throw err;
				});
			};
			
			var lastProcessedBytes = 0;
			var count = 0;
			
			var endCondition = function (transferState) {
				if (transferState.finished_at !== null) {
					return true;
				}
				
				if (lastProcessedBytes >= transferState.processed_bytes) {
					if (count * interval > timeoutWithoutProgress) {
						return true;
					}
					count++;
				} else {
					count = 0;
					lastProcessedBytes = transferState.processed_bytes;
				}
				
				if (verbose) {
					console.log(transferState.processed_bytes);
				}
			};
			
			return poll(getTransferState, interval, endCondition);
		})
		.then(function (transferState) {
			return transferState.processed_bytes;
		})
	;
};