X-Git-Url: https://jfr.im/git/erebus.git/blobdiff_plain/87b9ff2acd9bf2d45f784616d3c983bde469c9d4..99366200d5afdc1fa53fde0ee8533429af937f9e:/modules/coins.py

diff --git a/modules/coins.py b/modules/coins.py
index e4338e6..5a92fad 100644
--- a/modules/coins.py
+++ b/modules/coins.py
@@ -17,9 +17,18 @@ modstart = lib.modstart
 modstop = lib.modstop
 
 # module code
+import re
 import json
 import requests
 
+coin_regex = (
+	re.compile(r'([0-9.,\s]+)\s(btc|bitcoin|doge|dogecoin|ltc|litecoin)'),
+)
+
+cur_regex = (
+	re.compile(r'([0-9.,\s]+)\s([a-zA-Z]{3})\sin\s([a-zA-Z]{3})'),
+)
+
 url = 'http://www.cryptocoincharts.info/v2/api/tradingPairs'
 
 def get_coin_price(pairs):
@@ -71,3 +80,54 @@ def cmd_gtest(bot, user, chan, realtarget, *args):
 		btc_eur = get_coin_price('btc_eur')
 		price = str(float(ltc_btc[0]['price']) * float(btc_eur[0]['price']))
 		bot.msg(chan, "1 LTC = %s EUR" % price)
+
+@lib.hooknum("PRIVMSG")
+def privmsg_hook(bot, line):
+
+	try:
+		linetx = line.split(None, 3)[3][1:]
+	except IndexError:
+		linetx = ''
+
+	chan = line.split()[2]
+
+	if 'in' in line:
+		for r in cur_regex:
+			for a, f, t in r.findall(linetx):
+
+				# https://www.google.com/finance/converter?a=1.2&from=USD&to=EUR
+
+				a = a.replace(",", ".")
+				a = a.replace(" ", "")
+				print a
+				print f
+				print t
+
+	if 'btc' in line or 'bitcoin' in line or 'doge' in line or 'dogecoin' in line:
+		for r in coin_regex:
+			for amount, coin in r.findall(linetx):
+				amount = amount.replace(",", ".")
+				amount = amount.replace(" ", "")
+				if 'btc' in coin or 'bitcoin' in coin:
+					try:
+						response = get_coin_price('btc_eur')
+						price = str(float(response[0]['price']) * float(amount))
+						bot.msg(chan, "%s BTC = %s EUR" % (amount, price))
+					except:
+						pass
+
+				if 'ltc' in coin or 'litecoin' in coin:
+					pass	# Add LTC
+
+				if 'doge' in coin or 'dogecoin' in coin:
+					try:
+						doge_btc = get_coin_price('doge_btc')
+						btc_eur = get_coin_price('btc_eur')
+						price = str(float(doge_btc[0]['price']) * float(btc_eur[0]['price']) * float(amount))
+						bot.msg(chan, "%s DOGE = %s EUR" % (amount, price))
+					except:
+						bot.msg(chan, "Invalid amount.")
+						
+					
+				print amount
+				print coin