Fully add BitcoinTrigger

kjsmita6 · kjsmita6 · commit 5601ea50fbf2 · 2015-12-06T20:15:53.000-06:00
diff --git a/README.md b/README.md
@@ -82,6 +82,7 @@ General bot-related and chat-related functions.
 #### Trade-related triggers
 
 - `BanCheckTrigger` - checks to see if a user has any VAC/economy/community bans via steam API. Requires a steam api key, defined as option apikey in the plugin, or globally defined as a chatBot option steamapikey
+- `BitcoinTrigger` - does various actions over the BTC network
 - `ProfileCheckTrigger` - When a user joins, look up their profile in steam API and if they have a private profile, or never bothered to set one up, announce it to the groupchat. Optional option: apikey can be defined in options, overriding any steamapikey can be defined in the bot constructor. If neither is defined, it won't be used (not required).
 - `RedditTrigger` - Check reddit flair/bans/username on command. Requires access to the /r/SGS API. Note: I will not help you with gaining access to the API, and everything I know about the API is used in this trigger.
 - `SteamrepTrigger` - checks steamrep API on command. If Steamrep lists the user as a scammer, then bot announces it and gives links for more info. See `SteamrepOnJoinTrigger` to do this whenever someone joins a chat.
diff --git a/TRIGGERS.md b/TRIGGERS.md
@@ -61,6 +61,29 @@ Options:
 - command - *string* - defaults to "!ban"
 - user - *array* of *string*s. Who can use this trigger
 
+### `BitcoinTrigger`
+
+Does various actions over the BTC network. [Coinbase api](https://www.coinbase.com/settings/api) (create OAuth application). Must have webserver enabled.
+
+Options:
+
+Required:
+
+- clientID - *string* - Coinbase clientID
+- clientSecret * string* - Coinbase clientSecret
+- redirectURI *string* - redirectURI for OAauth app
+
+Optional:
+
+- clearcommand - *string* - defaults to !clear
+- authcommand - *string* - defaults to !auth
+- saveTimer - *integer* - database file save interval, defaults to 5 minutes
+- sellcommand - *string* - defaults to !sell
+- buycommand - *string* - defaults to !buy
+- balancecommand - *string* - defaults to !balance
+- pricescommand - *string* - defaults to !prices
+- dbFile - *string* - database file name
+
 ### `BotCommandTrigger`
 
 Runs a specified callback when a specific command message is typed. It is preferred to write an actual trigger, but for simple things (e.g. muting, unmuting), this is easier.
diff --git a/examples/example-config-triggers2.js b/examples/example-config-triggers2.js
@@ -328,5 +328,15 @@ module.exports = [
 	//check steamrep on every user who joins. Notify the channel if they're a scammer.
 	{	name: 'SteamrepOnJoin',
 		type: 'SteamrepOnJoinTrigger',
-		options: {} }
+		options: {} },
+
+	{
+		name: 'BTC',
+		type: 'BitcoinTrigger',
+		options: {
+			clientID: 'XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX',
+			clientSecret: 'XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX',
+			redirectURI: 'http://localhost:8080/coinbase/'
+		}
+	}
 ];
diff --git a/lib/triggers/bitcoinTrigger.js b/lib/triggers/bitcoinTrigger.js
@@ -19,18 +19,16 @@ exports.create = function(name, chatBot, options) {
 	trigger.options.clientSecret = trigger.options.clientSecret || undefined;
 	trigger.options.redirectURI = trigger.options.redirectURI || undefined;
 
+	trigger.scope = 'wallet:accounts:read,wallet:sells:create';
 	trigger.db = (function() { try { return JSON.parse(fs.readFileSync(trigger.options.dbFile)); } catch(e) { return {}}})();
-	trigger.client;
+
 	trigger.options.clearcommand = trigger.options.clearcommand || '!clear';
 	trigger.options.authcommand = trigger.options.authcommand || '!auth';
-	trigger.options.saveTimer = trigger.options.saveTimer || 1000 * 60 * 30;
+	trigger.options.saveTimer = trigger.options.saveTimer || 1000 * 60 * 5;
 	trigger.options.sellcommand = trigger.options.sellcommand || '!sell';
 	trigger.options.buycommand = trigger.options.buycommand || '!buy';
 	trigger.options.balancecommand = trigger.options.balancecommand || '!balance';
-	trigger.options.sendcommand = trigger.options.sendcommand || '!send';
-	trigger.options.requestcommand = trigger.options.requestcommand || '!request';
 	trigger.options.pricecommand = trigger.options.pricecommand || '!prices';
-	trigger.options.transcommand = trigger.options.transcommand || '!transfers';
 	return trigger;
 }
 
@@ -48,7 +46,7 @@ BitcoinTrigger.prototype._onLoad = function() {
 					else {
 						that.winston.debug(withName + 'Wrote to db file');
 					}
-				})
+				});
 			}
 			catch (e) {
 				that.winston.error(withName + 'Error: ' + e);
@@ -58,10 +56,6 @@ BitcoinTrigger.prototype._onLoad = function() {
 		return true;
 	}
 	if(!this.options.clientID || !this.options.clientSecret || !this.options.redirectURI || this.chatBot.options.disableWebserver === true) {
-		console.log(this.options.clientID);
-		console.log(this.options.clientSecret);
-		console.log(this.options.redirectURI);
-		console.log(this.chatBot.options.disableWebserver);
 		this.winston.error(withName + 'Must specify client ID and client Secret and redirect URI and webserver enabled!');
 		return false;
 	}
@@ -90,9 +84,6 @@ BitcoinTrigger.prototype._respond = function(toId, userId, message) {
 		}
 	}
 
-	setInterval(function() {
-		that._refreshToken(userId);
-	}, 1000 * 60 * 60 * 2);
 	var query = this._stripCommand(message, this.options.balancecommand);
 	if(query) {
 		this._getBalance(toId, userId);
@@ -120,35 +111,99 @@ BitcoinTrigger.prototype._respond = function(toId, userId, message) {
 	}
 
 	query = this._stripCommand(message, this.options.sellcommand);
+	if(query && query.params.length === 3) {
+		this._sell(toId, userId, query.params[1], query.params[2]);
+	}
+
+	query = this._stripCommand(message, this.options.buycommand);
+	if(query && query.params.length === 3) {
+		this._buy(toId, userId, query.params[1], query.params[2]);
+	}
+
+	query = this._stripCommand(message, this.options.pricecommand);
 	if(query && query.params.length === 2) {
-		this._sell(toId, userId, query.params[1]);
+		this._getPrices(toId, userId, query.params[1]);
 	}
 }
 
-BitcoinTrigger.prototype._sell = function(toId, userId, amount) {
+BitcoinTrigger.prototype._getPrices = function(toId, userId, currency) {
+	const _db = this.db[userId];
+	const client = new Coinbase.Client({
+		"accessToken": _db.accessToken,
+		"refreshToken": _db.refreshToken
+	});
+	const withName = this.chatBot.name + '/' + this.name + ': ';
+
+	var sellPrice = '';
+	var buyPrice = '';
+
+	client.getBuyPrice({
+		"currency": currency
+	}, function(e, obj) {
+		buyPrice = obj.data.amount;
+	});
+	client.getSellPrice({
+		"currency": currency
+	}, function(e, obj) {
+		sellPrice = obj.data.amount;
+	});
+
+	setTimeout(() => {
+		this._sendMessageAfterDelay(toId, 'Sell price: ' + sellPrice + '. Buy price: ' + buyPrice);
+	}, 1000);
+}
+
+BitcoinTrigger.prototype._buy = function(toId, userId, amount, currency) {
+	const withName = this.chatBot.name + '/' + this.name + ': ';
+	const _db = this.db[userId];
+	const client = new Coinbase.Client({
+		"accessToken": _db.accessToken,
+		"refreshToken": _db.refreshToken
+	});
+	const account = new Coinbase.model.Account(client, {
+		"id": _db.id
+	});
+
+	const args = {
+		"amount": amount,
+		"currency": currency
+	};
+
+	account.buy(args, (e, xfer) => {
+		if(e) {
+			this.winston.error(withName + e.stack);
+			this._sendMessageAfterDelay(toId, `Error buying ${amount} ${currency} for user ${userId}: ${e.message}`);
+		}
+		else {
+			this._sendMessageAfterDelay(toId, `${userId} successfully bought ${amount} ${currency}!`);
+			this._sendMessageAfterDelay(userId, `Transfer ID is ${xfer.id}`);
+		}
+	});
+}
+
+BitcoinTrigger.prototype._sell = function(toId, userId, amount, currency) {
 	const that = this;
 	const withName = this.chatBot.name + '/' + this.name + ': ';
 	var _db = this.db[userId];
 	const client = new Coinbase.Client({
 		"accessToken": _db.accessToken,
-		"refreshToken": _db.refreshToken,
-		'baseApiUri': 'https://api.sandbox.coinbase.com/v2/',
-  		'tokenUri': 'https://api.sandbox.coinbase.com/oauth/token'
+		"refreshToken": _db.refreshToken
 	});
 	const account = new Coinbase.model.Account(client, { "id": _db.id });
 
-	var args = {
-		"qty": amount
-	}
+	const args = {
+		"amount": amount,
+		"currency": currency
+	};
 
 	account.sell(args, function(e, xfer) {
 		if(e) {
 			that.winston.error(withName + e.stack);
-			that._sendMessageAfterDelay(toId, 'Error selling ' + amount + ' bitcoints for user ' + userId + ': ' + e.message);
+			that._sendMessageAfterDelay(toId, 'Error selling ' + amount + ' ' + currency + ' for user ' + userId + ': ' + e.message);
 		}
 		else {
-			that._sendMessageAfterDelay(toId, userId + ' successfully sold ' + amount + ' bitcoins!');
-			that._sendMessageAfterDelay(userId, 'Transfer id is ' + xfer.id);
+			that._sendMessageAfterDelay(toId, userId + ' successfully sold ' + amount + ' ' + currency + '!');
+			that._sendMessageAfterDelay(userId, 'Transfer ID is ' + xfer.id);
 		}
 	});
 }
@@ -177,95 +232,67 @@ BitcoinTrigger.prototype._refreshToken = function(userId) {
 }
 
 BitcoinTrigger.prototype._getId = function(userId) {
-	const that = this;
-	var _db = that.db[userId];
+	var _db = this.db[userId];
 	const withName = this.chatBot.name + '/' + this.name + ': ';
 	const client = new Coinbase.Client({
 		"accessToken": _db.accessToken,
-		"refreshToken": _db.refreshToken,
-		'baseApiUri': 'https://api.sandbox.coinbase.com/v2/',
-  		'tokenUri': 'https://api.sandbox.coinbase.com/oauth/token'
+		"refreshToken": _db.refreshToken
 	});
-	client.getAccounts(function(e, accounts) {
+	client.getAccounts({}, (e, accounts) => {
 		if(e) {
-			that.winston.error(withName + e.stack);
+			this.winston.error(withName + e.stack);
 		}
 		else {
-			_db.id === account.id;
+			accounts.forEach(account => {
+				_db.id = account.id;
+			});
 		}
 	})
 }
 
 BitcoinTrigger.prototype._authorize = function(userId, callback) {
-	const that = this;
 	const withName = this.chatBot.name + '/' + this.name + ': ';
 	this.express = this.chatBot.express;
 	var _url;
 	var _code;
 	var accessToken;
 	var refreshToken;
-	//var _scope = 'wallet:accounts:read,wallet:accounts:create,wallet:accounts:update,wallet:accounts:delete';
-	var _scope = 'user,balance,sell';
 	var uri = "https://sandbox.coinbase.com/oauth/authorize" +
-		"?response_type=code&redirect_uri=" + encodeURI(that.options.redirectURI) +
-		"&client_id=" + that.options.clientID + '&scope=' + _scope;
+		"?response_type=code&redirect_uri=" + encodeURI(this.options.redirectURI) +
+		"&client_id=" + this.options.clientID + '&scope=' + this.scope;
 	Request({
 		uri: uri,
 		json: true,
 		followAllRedirects: true
-	}, function(e, response, body) {
+	}, (e, response, body) => {
 		if(e) {
 			callback(e, null, null);
 		}
 		else {
-			that._sendMessageAfterDelay(userId, 'Please visit the following link and click "Authorize".\n\n' + uri);
+			this._sendMessageAfterDelay(userId, 'Please visit the following link and click "Authorize".\n\n' + uri);
 			var last = /^[^\/]*(?:\/[^\/]*){2}/g;
 			//console.log(that.options.redirectURI.match(last));
-			var getter = that.options.redirectURI.replace(last, '');
-			that.chatBot._addRouter(getter);
-			that.winston.silly(getter);
-			/*
-			that.express.get(getter, function(req, res) {
-				_url = req.url;
-				_code = _url.match(/\?code=(.*)/i)[1];
-				res.send('<h1>Response received, you may close this window now.<h2>' + '<br><h4>Your code is ' + _code + '<h4><br>');
-				Request({
-					method: "POST",
-					uri: "https://sandbox.coinbase.com/oauth/token?grant_type=authorization_code&code=" +
-						_code + "&client_id=" + that.options.clientID + "&client_secret=" +
-						that.options.clientSecret + "&redirect_uri=" + that.options.redirectURI,
-					json: true,
-					followAllRedirects: true
-				}, function(e, response, body) {
-					if(!body.access_token) {
-						callback(e, null, null);
-					}
-					else {
-						accessToken = body.access_token;
-						refreshToken = body.refresh_token;
-						callback(null, accessToken, refreshToken);
-					}
-				});
-			});
-			*/
-			that.express.use(getter, function(req, res, next) {
+			var getter = this.options.redirectURI.replace(last, '');
+			this.chatBot._addRouter(getter);
+			this.express.use(getter, (req, res, next) => {
 				_url = req.url;
 				_code = _url.match(/\?code=(.*)/i)[1];
 				res.send('<h1>Response received, you may close this window now.<h2>' + '<br><h4>Your code is ' + _code + '<h4><br>');
 				Request({
 					method: "POST",
 					uri: "https://sandbox.coinbase.com/oauth/token?grant_type=authorization_code&code=" +
-						_code + "&client_id=" + that.options.clientID + "&client_secret=" +
-						that.options.clientSecret + "&redirect_uri=" + that.options.redirectURI,
+						_code + "&client_id=" + this.options.clientID + "&client_secret=" +
+						this.options.clientSecret + "&redirect_uri=" + this.options.redirectURI,
 					json: true,
 					followAllRedirects: true
-				}, function(e, response, body) {
+				}, (e, response, body) => {
 					if(!body.access_token) {
 						callback(e, null, null);
 					}
 					else {
 						accessToken = body.access_token;
 						refreshToken = body.refresh_token;
+						this._getId(userId);
 						callback(null, accessToken, refreshToken);
 					}
 				});
@@ -285,32 +312,31 @@ BitcoinTrigger.prototype._getBalance = function(toId, userId) {
 		var _db = that.db[userId];
 		var client = new Coinbase.Client({
 			"accessToken": _db.accessToken,
-			"refreshToken": _db.refreshToken,
-			'baseApiUri': 'https://api.sandbox.coinbase.com/v2/',
-  			'tokenUri': 'https://api.sandbox.coinbase.com/oauth/token'
+			"refreshToken": _db.refreshToken
 		});
-		client.getAccounts({}, (e, accounts) => {
+		client.getAccount(_db.id, (e, account) => {
 			if(e) {
 				if(e.type === 'ExpiredAccessToken') {
-					that._refreshToken(userId);
-					that._sendMessageAfterDelay(userId, 'Access token expired, refreshing. Please try again in a few seconds.');
-					that.winston.warn(withName + 'Refreshing access token');
+					this._refreshToken(userId);
+					this._sendMessageAfterDelay(userId, 'Access token expired, refreshing. Please try again in a few seconds.');
+					this.winston.warn(withName + 'Refreshing access token');
 				}
-				that.winston.error(e);
-				that._sendMessageAfterDelay(userId, e.message);
+				this.winston.error(e.stack);
+				this._sendMessageAfterDelay(userId, e.message);
 			}
 			else {
-				accounts.forEach(function(acct) {
-					that._sendMessageAfterDelay(toId, 'The balance for ' + that.chatBot._userString(userId) + ' is ' + acct.balance.amount + ' ' + acct.balance.currency);
-				})
+				this._sendMessageAfterDelay(toId, 'The balance for ' + this.chatBot._userString(userId) + ' is ' + account.balance.amount + ' ' + account.balance.currency);
 			}
 		})
 	}
 }
 
 BitcoinTrigger.prototype._stripCommand = function(message, command) {
 	if (command && message && message.toLowerCase().indexOf(command.toLowerCase()) === 0) {
-		return {message: message, params: message.split(" ")};
+		return {
+			message: message,
+			params: message.split(" ")
+		};
 	}
 	return null;
 }
