226 lines
9.2 KiB
Python
Executable File
226 lines
9.2 KiB
Python
Executable File
#!/usr/bin/env python
|
|
"""
|
|
.. module:: Error
|
|
:synopsis: Module to provide a multi-type cryptocurrency wallet
|
|
|
|
.. moduleauthor:: Dave Pedu <dave@davepedu.com>
|
|
|
|
"""
|
|
|
|
from pyircbot.modulebase import ModuleBase,ModuleHook
|
|
import time
|
|
import hashlib
|
|
|
|
class CryptoWallet(ModuleBase):
|
|
def __init__(self, bot, moduleName):
|
|
ModuleBase.__init__(self, bot, moduleName);
|
|
self.hooks=[ModuleHook("PRIVMSG", self.gotmsg)]
|
|
|
|
def gotmsg(self, args, prefix, trailing):
|
|
channel = args[0]
|
|
if channel[0] == "#":
|
|
# Ignore channel messages
|
|
pass
|
|
else:
|
|
self.handlePm(args, prefix, trailing)
|
|
|
|
def getMods(self):
|
|
return (self.bot.getBestModuleForService("attributes"), self.bot.getBestModuleForService("login"), self.bot.getBestModuleForService("bitcoinrpc"))
|
|
|
|
def handle_setaddr(self, args, prefix, trailing, cmd):
|
|
usage = ".setaddr <currency> <address>"
|
|
attr,login,rpc = self.getMods()
|
|
# Check login
|
|
if not login.check(prefix.nick, prefix.hostname):
|
|
self.bot.act_PRIVMSG(prefix.nick, ".setaddr: Please .login to use this command.")
|
|
return
|
|
# Check for args
|
|
if not len(cmd.args)==2:
|
|
self.bot.act_PRIVMSG(prefix.nick, ".setaddr: usage: %s" % usage)
|
|
self.bot.act_PRIVMSG(prefix.nick, ".setaddr: usage: .setaddr BTC 1xyWx6X5EABprhe3s9XduNxLn5NCtpSNB")
|
|
return
|
|
# Check if currency is known
|
|
if not rpc.isSupported(cmd.args[0]):
|
|
supportedStr = ', '.join(rpc.getSupported())
|
|
self.bot.act_PRIVMSG(prefix.nick, ".setaddr: '%s' is not a supported currency. Supported currencies are: %s" % (cmd.args[0], supportedStr))
|
|
return
|
|
if len(cmd.args[1])<16 or len(cmd.args[1])>42:
|
|
self.bot.act_PRIVMSG(prefix.nick, ".setaddr: '%s' appears to be an invalid address." % (cmd.args[1]))
|
|
return
|
|
|
|
# Just make sure they have a wallet
|
|
self.checkUserHasWallet(prefix.nick, cmd.args[0])
|
|
|
|
# Set their address
|
|
attr.setKey(prefix.nick, "cryptowallet-%s-address"%cmd.args[0].lower(), cmd.args[1])
|
|
self.bot.act_PRIVMSG(prefix.nick, ".setaddr: Your address has been saved as: %s. Please verify that this is correct or your coins could be lost." % (cmd.args[1]))
|
|
|
|
def handle_getbal(self, args, prefix, trailing, cmd):
|
|
usage = ".getbal <currency>"
|
|
attr,login,rpc = self.getMods()
|
|
# Check login
|
|
if not login.check(prefix.nick, prefix.hostname):
|
|
self.bot.act_PRIVMSG(prefix.nick, ".getbal: Please .login to use this command.")
|
|
return
|
|
# Check for args
|
|
if not len(cmd.args)==1:
|
|
self.bot.act_PRIVMSG(prefix.nick, ".getbal: usage: %s" % usage)
|
|
self.bot.act_PRIVMSG(prefix.nick, ".getbal: usage: .getbal BTC")
|
|
return
|
|
# Check if currency is known
|
|
if not rpc.isSupported(cmd.args[0]):
|
|
supportedStr = ', '.join(rpc.getSupported())
|
|
self.bot.act_PRIVMSG(prefix.nick, ".getbal: '%s' is not a supported currency. Supported currencies are: %s" % (cmd.args[0], supportedStr))
|
|
return
|
|
|
|
# Just make sure they have a wallet
|
|
self.checkUserHasWallet(prefix.nick, cmd.args[0])
|
|
|
|
# fetch RPC and tell them the balance
|
|
walletname = attr.getKey(prefix.nick, "cryptowallet-account-%s"%cmd.args[0].lower())
|
|
amount = 0.0
|
|
if walletname:
|
|
client = rpc.getRpc(cmd.args[0].lower())
|
|
amount = client.getBal(walletname)
|
|
self.bot.act_PRIVMSG(prefix.nick, "Your balance is: %s %s" % (amount, cmd.args[0].upper()))
|
|
|
|
def handle_withdraw(self, args, prefix, trailing, cmd):
|
|
usage = ".withdraw <currency> <amount>"
|
|
attr,login,rpc = self.getMods()
|
|
# Check login
|
|
if not login.check(prefix.nick, prefix.hostname):
|
|
self.bot.act_PRIVMSG(prefix.nick, ".withdraw: Please .login to use this command.")
|
|
return
|
|
# Check for args
|
|
if not len(cmd.args)==2:
|
|
self.bot.act_PRIVMSG(prefix.nick, ".withdraw: usage: %s" % usage)
|
|
self.bot.act_PRIVMSG(prefix.nick, ".withdraw: usage: .getbal BTC 0.035")
|
|
return
|
|
# Check if currency is known
|
|
if not rpc.isSupported(cmd.args[0]):
|
|
supportedStr = ', '.join(rpc.getSupported())
|
|
self.bot.act_PRIVMSG(prefix.nick, ".getbal: '%s' is not a supported currency. Supported currencies are: %s" % (cmd.args[0], supportedStr))
|
|
return
|
|
|
|
# Just make sure they have a wallet
|
|
self.checkUserHasWallet(prefix.nick, cmd.args[0])
|
|
|
|
# check that they have a withdraw addr
|
|
withdrawaddr = attr.getKey(prefix.nick, "cryptowallet-%s-address"%cmd.args[0].lower())
|
|
if withdrawaddr == None:
|
|
self.bot.act_PRIVMSG(prefix.nick, ".withdraw: You need to set a withdraw address before withdrawing. Try .setaddr")
|
|
return
|
|
|
|
# fetch RPC and check balance
|
|
walletname = attr.getKey(prefix.nick, "cryptowallet-account-%s"%cmd.args[0].lower())
|
|
balance = 0.0
|
|
|
|
client = rpc.getRpc(cmd.args[0].lower())
|
|
balance = client.getBal(walletname)
|
|
withdrawamount = float(cmd.args[1])
|
|
|
|
if balance < withdrawamount or withdrawamount<0:
|
|
self.bot.act_PRIVMSG(prefix.nick, ".withdraw: You don't have enough %s to withdraw %s" % (cmd.args[0].upper(), withdrawamount))
|
|
return
|
|
|
|
if not client.reserve == 0 and balance - client.reserve < withdrawamount:
|
|
self.bot.act_PRIVMSG(prefix.nick, ".withdraw: Withdrawing that much would put you below the reserve (%s %s)." % (client.reserve, cmd.args[0].upper()))
|
|
self.bot.act_PRIVMSG(prefix.nick, ".withdraw: The reserve is to cover network transaction fees. To recover it you must close your account. (Talk to an admin)")
|
|
return
|
|
|
|
# Check if the precision is wrong
|
|
if not client.checkPrecision(withdrawamount):
|
|
self.bot.act_PRIVMSG(prefix.nick, ".withdraw: %s has maximum %s decimal places" % (cmd.args[0].upper(), client.precision))
|
|
return
|
|
|
|
# Create a transaction
|
|
txn = client.send(walletname, withdrawaddr, withdrawamount)
|
|
if txn:
|
|
self.bot.act_PRIVMSG(prefix.nick, ".withdraw: %s %s sent to %s. Transaction ID: %s"% (withdrawamount, client.name, withdrawaddr, txn))
|
|
else:
|
|
self.bot.act_PRIVMSG(prefix.nick, ".withdraw: Transaction create failed. Maybe the transaction was too large for the network? Try a smaller increment.")
|
|
|
|
def handle_getaddr(self, args, prefix, trailing, cmd):
|
|
attr,login,rpc = self.getMods()
|
|
usage = ".getaddr <currency>"
|
|
attr,login,rpc = self.getMods()
|
|
# Check login
|
|
if not login.check(prefix.nick, prefix.hostname):
|
|
self.bot.act_PRIVMSG(prefix.nick, ".getaddr: Please .login to use this command.")
|
|
return
|
|
# Check for args
|
|
if not len(cmd.args)==1:
|
|
self.bot.act_PRIVMSG(prefix.nick, ".getaddr: usage: %s" % usage)
|
|
self.bot.act_PRIVMSG(prefix.nick, ".getaddr: usage: .getaddr BTC")
|
|
return
|
|
# Check if currency is known
|
|
if not rpc.isSupported(cmd.args[0]):
|
|
supportedStr = ', '.join(rpc.getSupported())
|
|
self.bot.act_PRIVMSG(prefix.nick, ".getaddr: '%s' is not a supported currency. Supported currencies are: %s" % (cmd.args[0], supportedStr))
|
|
return
|
|
|
|
# Just make sure they have a wallet
|
|
self.checkUserHasWallet(prefix.nick, cmd.args[0])
|
|
|
|
walletaddr = attr.getKey(prefix.nick, "cryptowallet-depoaddr-%s"%cmd.args[0].lower())
|
|
self.bot.act_PRIVMSG(prefix.nick, "Your %s deposit address is: %s" % (cmd.args[0].upper(), walletaddr))
|
|
|
|
def handle_curinfo(self, args, prefix, trailing, cmd):
|
|
attr,login,rpc = self.getMods()
|
|
usage = ".curinfo [<currency>]"
|
|
attr,login,rpc = self.getMods()
|
|
|
|
# Check for args
|
|
if len(cmd.args)==0:
|
|
self.bot.act_PRIVMSG(prefix.nick, ".curinfo: supported currencies: %s. Use '.curinfo BTC' to see details. " % ', '.join([x.upper() for x in rpc.getSupported()]))
|
|
return
|
|
else:
|
|
if not rpc.isSupported(cmd.args[0]):
|
|
self.bot.act_PRIVMSG(prefix.nick, ".curinfo: '%s' is not a supported currency. Supported currencies are: %s" % (cmd.args[0], ', '.join([x.upper() for x in rpc.getSupported()])))
|
|
return
|
|
else:
|
|
info = rpc.getInfo(cmd.args[0])
|
|
self.bot.act_PRIVMSG(prefix.nick, ".curinfo: %s - %s. More info: %s" % (args[0], info["name"], info["link"]))
|
|
|
|
def checkUserHasWallet(self, username, currency):
|
|
# Ensure the user has a wallet in the client
|
|
attr,login,rpc = self.getMods()
|
|
currency = currency.lower()
|
|
if attr.getKey(username, "cryptowallet-account-%s"%currency)==None:
|
|
randName = self.md5(str(time.time()))[0:16]
|
|
attr.setKey(username, "cryptowallet-account-%s"%currency, randName)
|
|
# Generate a deposit addr to nudge the wallet
|
|
wallet = rpc.getRpc(currency.lower())
|
|
address = wallet.getAcctAddr(randName)
|
|
attr.setKey(username, "cryptowallet-depoaddr-%s"%currency, address)
|
|
elif attr.getKey(username, "cryptowallet-depoaddr-%s"%currency)==None:
|
|
walletName = attr.getKey(username, "cryptowallet-account-%s"%currency)
|
|
wallet = rpc.getRpc(currency.lower())
|
|
address = wallet.getAcctAddr(walletName)
|
|
attr.setKey(username, "cryptowallet-depoaddr-%s"%currency, address)
|
|
|
|
|
|
def handlePm(self, args, prefix, trailing):
|
|
prefix = self.bot.decodePrefix(prefix)
|
|
|
|
cmd = self.bot.messageHasCommand(".setaddr", trailing)
|
|
if cmd:
|
|
self.handle_setaddr(args, prefix, trailing, cmd)
|
|
cmd = self.bot.messageHasCommand(".getbal", trailing)
|
|
if cmd:
|
|
self.handle_getbal(args, prefix, trailing, cmd)
|
|
cmd = self.bot.messageHasCommand(".withdraw", trailing)
|
|
if cmd:
|
|
self.handle_withdraw(args, prefix, trailing, cmd)
|
|
cmd = self.bot.messageHasCommand(".getaddr", trailing)
|
|
if cmd:
|
|
self.handle_getaddr(args, prefix, trailing, cmd)
|
|
cmd = self.bot.messageHasCommand(".curinfo", trailing)
|
|
if cmd:
|
|
self.handle_curinfo(args, prefix, trailing, cmd)
|
|
|
|
def md5(self, data):
|
|
m = hashlib.md5()
|
|
m.update(data.encode("ascii"))
|
|
return m.hexdigest()
|