proxycoind.rb

#!/usr/bin/ruby
$LOAD_PATH.unshift File.dirname(__FILE__)
require 'rubygems'
require 'redis'
require 'bitcoin_rpc'

file = File.new("proxycoind.log", 'a+')
STDOUT.reopen(file)
file.sync = true

@@config = YAML.load_file('config.yml')

class Redis
  def setm(k, o)
    set(k, Marshal.dump(o))
  end
  def getm(k)
    m = get(k)
    m ? Marshal.load(m) : nil
  end
end

@@redis = Redis.new

def getrpc(coinname)
  d = @@config['coins'][coinname]
  uri = "http://#{d['user']}:#{d['password']}@#{d['host']}:#{d['port']}"
  BitcoinRPC.new(uri)
end

def newaccount(accounts, accountid)
  accounts[accountid] ||= {
    'balance' => 0.0,
  }
end

def checkmove(rpc, accounts, balancebasename)
  txs = rpc.listtransactions('*', 10000) # TODO
p txs.size
  txs.each do |tx|
    confirmations = tx['confirmations']
    txid = tx['txid']
    amount = tx['amount']
    cat = tx['category']
    case cat
    when 'move'
      accountid = tx['account']
      amount = tx['amount']
      account = newaccount(accounts, accountid)
      account['balance'] += amount
      balancename = "#{balancebasename}#{accountid}"
      @@redis.setnx(balancename, 0.0)
      @@redis.incrbyfloat(balancename, amount)
p [:move, accountid, amount]
    else
      p cat # TODO
    end
  end
end

def poll(rpc, blockhash, accounts, pendingtxs, pendingflag, balancebasename)
  receiveonly = blockhash != ''
  confirmedheight = 6 # TODO
  params = [blockhash]
  params << 400 if pendingflag # TODO
  result = rpc.listsinceblock(*params)
  lastblock = result['lastblock']
  txs = result['transactions']
p txs.size
  completedtxs = []
  txs.each do |tx|
    confirmations = tx['confirmations']
    txid = tx['txid']
    confirmed = confirmations >= confirmedheight
    if pendingflag
      next if !pendingtxs.include?(txid)
p [:pending2, txid[0,6], confirmations]
      next if !confirmed
      completedtxs << txid
    elsif !confirmed
p [:pending1, txid[0,6], confirmations]
      pendingtxs << txid unless pendingtxs.include?(txid)
      next
    end
    amount = tx['amount']
    cat = tx['category']
    case cat
    when 'receive'
      accountaddr = tx['address']
      accountid = rpc.getaccount(accountaddr)
      account = newaccount(accounts, accountid)
      account['balance'] += amount
      balancename = "#{balancebasename}#{accountid}"
      @@redis.setnx(balancename, 0.0)
      @@redis.incrbyfloat(balancename, amount)
p [:receive, amount, accountid, confirmations]
    when 'send'
      accountid = tx['account']
      fee = tx['fee']
      amount += fee
      unless receiveonly
        account = newaccount(accounts, accountid)
        account['balance'] += amount
        balancename = "#{balancebasename}#{accountid}"
        @@redis.setnx(balancename, 0.0)
        @@redis.incrbyfloat(balancename, amount)
      end
p [:send, amount, accountid]
    when 'generate'
      accountid = tx['account']
      unless receiveonly
        account = newaccount(accounts, accountid)
        account['balance'] += amount
        balancename = "#{balancebasename}#{accountid}"
        @@redis.setnx(balancename, 0.0)
        @@redis.incrbyfloat(balancename, amount)
      end
p [:generate, amount, accountid, confirmations]
    when 'immature'
      accountid = tx['account']
      unless receiveonly
        account = newaccount(accounts, accountid)
        account['balance'] += amount
        balancename = "#{balancebasename}#{accountid}"
        @@redis.setnx(balancename, 0.0)
        @@redis.incrbyfloat(balancename, amount)
      end
p [:immature, amount, accountid, confirmations]
    when 'orphan'
      p :orphan # TODO
    else
      p cat # TODO
    end
  end
  completedtxs.uniq.each do |txid|
    pendingtxs.delete(txid)
  end
  lastblock
end

def main
  interval = 60
  loop do
    begin
      coins = @@config['coins']
      coins.each do |coinid,v|
        next unless v['proxycoind']
        rpc = getrpc(coinid)
        coindbname = "proxycoind:#{coinid}"
        balancebasename = "proxycoind:balance:#{coinid}:"
        coininfo = @@redis.getm(coindbname) || {
          'blockhash' => '',
          'accounts' => {},
          'pendingtxs' => [],
          'pendingblockhash' => '',
        }
        blockhash = coininfo['blockhash']
        accounts = coininfo['accounts']
        pendingtxs = coininfo['pendingtxs']
        if blockhash == ''
          checkmove(rpc, accounts, balancebasename)
        end
        blockhash = poll(rpc, blockhash, accounts, pendingtxs, false,
            balancebasename)
        coininfo['blockhash'] = blockhash
        pendingblockhash = coininfo['pendingblockhash']
        pendingblockhash = poll(rpc, pendingblockhash, accounts, pendingtxs,
            true, balancebasename)
        coininfo['pendingtxs'] = pendingtxs
        coininfo['pendingblockhash'] = pendingblockhash
        @@redis.setm(coindbname, coininfo)
accounts.each do |account|
  p account
end
p blockhash
      end
    rescue => x
      p x
      raise x # TODO
    end
    sleep interval
  end
end

main