tippero/tipbot/modules/payment.py

#!/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 GetTipbotAddress():
  try:
    j = SendWalletJSONRPCCommand("getaddress",None)
    if not "result" in j:
      log_error('GetTipbotAddress: No result found in getaddress reply')
      return None
    result = j["result"]
    if not "address" in result:
      log_error('GetTipbotAddress: No address found in getaddress reply')
      return None
    return result["address"]
  except Exception,e:
    log_error("GetTipbotAddress: Error retrieving %s's 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);
  address=GetTipbotAddress() or 'ERROR'
  link.send_private("  Address: %s" % address)
  if config.openalias_address != None:
    link.send_private("    (or %s when using OpenAlias)" % config.openalias_address)
  link.send_private("  Use your primary payment ID: %s" % GetPaymentID(link))
  link.send_private("  OR generate random payment ids at will with: !randompid")
  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"
})