200 lines
7.6 KiB
Python
200 lines
7.6 KiB
Python
#!/bin/python
|
|
#
|
|
# Cryptonote tipbot - payment
|
|
# Copyright 2014 moneromooo
|
|
# Inspired by "Simple Python IRC bot" by berend
|
|
#
|
|
# The Cryptonote tipbot is free software; you can redistribute it and/or
|
|
# modify it under the terms of the GNU General Public License as published
|
|
# by the Free Software Foundation; either version 2, or (at your option)
|
|
# any later version.
|
|
#
|
|
|
|
import redis
|
|
import time
|
|
import tipbot.config as config
|
|
from tipbot.log import log_error, log_warn, log_info, log_log
|
|
from tipbot.user import User
|
|
from tipbot.link import Link
|
|
from tipbot.utils import *
|
|
from tipbot.redisdb import *
|
|
from tipbot.command_manager import *
|
|
|
|
last_wallet_update_time = None
|
|
|
|
def GetTipbotIntAddress():
|
|
try:
|
|
j = SendWalletJSONRPCCommand("make_integrated_address",None)
|
|
if not "result" in j:
|
|
log_error('GetTipbotIntAddress: No result found in make_integrated_address reply')
|
|
return None
|
|
result = j["result"]
|
|
if not "integrated_address" in result:
|
|
log_error('GetTipbotIntAddress: No integrated_address found in make_integrated_address reply')
|
|
return None
|
|
if not "payment_id" in result:
|
|
log_error('GetTipbotIntAddress: No payment_id found in make_integrated_address reply')
|
|
return None
|
|
return result
|
|
except Exception,e:
|
|
log_error("GetTipbotIntAddress: Error retrieving %s's int_address: %s" % (config.tipbot_name, str(e)))
|
|
return None
|
|
|
|
def UpdateCoin(data):
|
|
global last_wallet_update_time
|
|
if last_wallet_update_time == None:
|
|
last_wallet_update_time = 0
|
|
t=time.time()
|
|
dt = t - last_wallet_update_time
|
|
if dt < config.wallet_update_time:
|
|
return
|
|
try:
|
|
try:
|
|
scan_block_height = redis_get("scan_block_height")
|
|
scan_block_height = long(scan_block_height)
|
|
except Exception,e:
|
|
log_error('Failed to get scan_block_height: %s' % str(e))
|
|
last_wallet_update_time = time.time()
|
|
return
|
|
|
|
try:
|
|
j = SendDaemonHTMLCommand("getheight")
|
|
except Exception,e:
|
|
log_error('UpdateCoin: error getting height: %s' % str(e))
|
|
return
|
|
if not "height" in j:
|
|
log_error('UpdateCoin: error getting height: height not found in %s' % str(j))
|
|
return
|
|
try:
|
|
height=long(j["height"])
|
|
except Exception,e:
|
|
log_error('UpdateCoin: error getting height: %s' % str(e))
|
|
return
|
|
|
|
full_payment_ids = redis_hgetall("paymentid")
|
|
#print 'Got full payment ids: %s' % str(full_payment_ids)
|
|
payment_ids = []
|
|
for pid in full_payment_ids:
|
|
payment_ids.append(pid)
|
|
#print 'Got payment ids: %s' % str(payment_ids)
|
|
params = {
|
|
"payment_ids": payment_ids,
|
|
"min_block_height": scan_block_height
|
|
}
|
|
j = SendWalletJSONRPCCommand("get_bulk_payments",params)
|
|
#print 'Got j: %s' % str(j)
|
|
if "result" in j:
|
|
result = j["result"]
|
|
cp = redis_pipeline()
|
|
cp.delete('confirming_payments')
|
|
if "payments" in result:
|
|
payments = result["payments"]
|
|
new_payments = []
|
|
n_confirming = 0
|
|
new_scan_block_height = scan_block_height
|
|
for p in payments:
|
|
payment_id=p["payment_id"]
|
|
tx_hash = p["tx_hash"]
|
|
bh = p["block_height"]
|
|
ut = p["unlock_time"]
|
|
amount=p["amount"]
|
|
if redis_sismember("processed_txs",tx_hash):
|
|
continue
|
|
log_log('UpdateCoin: Looking at payment %s' % str(p))
|
|
confirmations = height-1-bh
|
|
confirmations_needed = max(config.payment_confirmations,ut)
|
|
if confirmations >= confirmations_needed:
|
|
log_info('Payment %s is now confirmed' % str(p))
|
|
new_payments.append(p)
|
|
if new_scan_block_height and bh > new_scan_block_height:
|
|
new_scan_block_height = bh
|
|
else:
|
|
log_info('Payment %s has %d/%d confirmations' % (str(p),confirmations,confirmations_needed))
|
|
n_confirming += 1
|
|
new_scan_block_height = None
|
|
try:
|
|
recipient = GetIdentityFromPaymentID(payment_id)
|
|
if not recipient:
|
|
raise RuntimeError('Payment ID %s not found' % payment_id)
|
|
account = redis_hget('accounts',recipient)
|
|
if account == None:
|
|
raise RuntimeError('No account found for %s' % recipient)
|
|
log_info('UpdateCoin: Found payment %s to %s for %s' % (tx_hash,recipient, AmountToString(amount)))
|
|
cp.hincrby('confirming_payments',account,amount)
|
|
except Exception,e:
|
|
log_error('UpdateCoin: No identity found for payment id %s, tx hash %s, amount %s: %s' % (payment_id, tx_hash, amount, str(e)))
|
|
payments=new_payments
|
|
log_info('UpdateCoin: Got %d mature payments and %d confirming payments' % (len(payments),n_confirming))
|
|
if len(payments) > 0:
|
|
if new_scan_block_height and new_scan_block_height > scan_block_height:
|
|
scan_block_height = new_scan_block_height
|
|
log_log('UpdateCoin: increasing scan_block_height to %d' % scan_block_height)
|
|
try:
|
|
pipe = redis_pipeline()
|
|
pipe.set("scan_block_height", scan_block_height)
|
|
log_log('UpdateCoin: processing payments')
|
|
for p in payments:
|
|
payment_id=p["payment_id"]
|
|
tx_hash=p["tx_hash"]
|
|
amount=p["amount"]
|
|
bh = p["block_height"]
|
|
try:
|
|
recipient = GetIdentityFromPaymentID(payment_id)
|
|
if not recipient:
|
|
raise RuntimeError('Payment ID %s not found' % payment_id)
|
|
account = redis_hget('accounts',recipient)
|
|
if account == None:
|
|
raise RuntimeError('No account found for %s' % recipient)
|
|
log_info('UpdateCoin: Found payment %s to %s for %s' % (tx_hash,recipient, AmountToString(amount)))
|
|
pipe.hincrby("balances",account,amount)
|
|
pipe.sadd("processed_txs",tx_hash)
|
|
except Exception,e:
|
|
log_error('UpdateCoin: No identity found for payment id %s, tx hash %s, amount %s: %s' % (payment_id, tx_hash, amount, str(e)))
|
|
log_log('UpdateCoin: Executing received payments pipeline')
|
|
pipe.execute()
|
|
except Exception,e:
|
|
log_error('UpdateCoin: failed to set scan_block_height: %s' % str(e))
|
|
cp.execute()
|
|
else:
|
|
log_error('UpdateCoin: No results in get_bulk_payments reply')
|
|
except Exception,e:
|
|
log_error('UpdateCoin: Failed to get bulk payments: %s' % str(e))
|
|
last_wallet_update_time = time.time()
|
|
|
|
def Deposit(link,cmd):
|
|
Help(link)
|
|
|
|
def RandomPaymentID(link,cmd):
|
|
link.send_private(" New payment ID: %s" % GetRandomPaymentID(link))
|
|
|
|
def Help(link):
|
|
GetAccount(link.identity())
|
|
link.send_private("You can send %s to your account using this address AND payment ID:" % coinspecs.name);
|
|
result=GetTipbotIntAddress() or 'ERROR'
|
|
link.send_private(" Address: %s" % result["integrated_address"])
|
|
if config.openalias_address != None:
|
|
link.send_private(" (or %s when using OpenAlias)" % config.openalias_address)
|
|
SetPaymentID(link,result["payment_id"])
|
|
link.send_private(" Use your primary payment ID: %s" % result["payment_id"])
|
|
link.send_private("Incoming transactions are credited after %d confirmations" % config.payment_confirmations)
|
|
|
|
RegisterModule({
|
|
'name': __name__,
|
|
'help': Help,
|
|
'idle': UpdateCoin
|
|
})
|
|
RegisterCommand({
|
|
'module': __name__,
|
|
'name': 'deposit',
|
|
'function': Deposit,
|
|
'help': "Show instructions about depositing %s" % coinspecs.name
|
|
})
|
|
RegisterCommand({
|
|
'module': __name__,
|
|
'name': 'randompid',
|
|
'function': RandomPaymentID,
|
|
'registered': True,
|
|
'help': "Generate a new random payment ID"
|
|
})
|
|
|