tipjar 0.1.16

data/lib/bitbot/database.rb

@@ -0,0 +1,142 @@
+require 'sqlite3'
+
+module Bitbot
+  class InsufficientFundsError < StandardError; end
+
+  class Database
+    def initialize(path)
+      @db = SQLite3::Database.new path
+      @db.execute('PRAGMA foreign_keys = ON')
+    end
+
+    def upgrade_schema
+      @db.execute(<<-ENDSQL)
+        create table if not exists users (
+          id integer primary key autoincrement,
+          created_at integer not null,
+          username text not null unique
+        )
+      ENDSQL
+
+      @db.execute(<<-ENDSQL)
+        create table if not exists transactions (
+          id integer primary key autoincrement,
+          created_at integer not null,
+          amount numeric not null,
+          note text,
+          withdrawal_address text,
+          incoming_transaction text unique,
+          user_id int references users(id) not null,
+          other_user_id int references users(id)
+        )
+      ENDSQL
+    end
+
+    def get_or_create_user_id_for_username(username)
+      user_id = @db.get_first_value("select id from users where username = ?", [ username ])
+      unless user_id
+        @db.execute("insert into users(created_at, username) values (?, ?)", [ Time.now.to_i, username ])
+        user_id = @db.get_first_value("select last_insert_rowid()")
+      end
+
+      user_id
+    end
+
+    def get_username_for_user_id(user_id)
+      @db.get_first_value("select username from users where id = ?", [ user_id ])
+    end
+
+    def get_balance_for_user_id(user_id)
+      @db.get_first_value("select coalesce(sum(amount), 0) from transactions where user_id = ?", [ user_id ])
+    end
+
+    def get_transactions_for_user_id(user_id)
+      result = []
+      @db.execute("select t.id, t.created_at, t.amount, t.note, t.withdrawal_address, t.incoming_transaction, t.other_user_id, u.username from transactions t left outer join users u on t.other_user_id = u.id where user_id = ? order by t.created_at desc", [ user_id ]) do |row|
+        result << {
+          :id => row[0],
+          :created_at => row[1],
+          :amount => row[2],
+          :note => row[3],
+          :withdrawal_address => row[4],
+          :incoming_transaction => row[5],
+          :other_user_id => row[6],
+          :other_username => row[7]
+        }
+      end
+      result
+    end
+
+    def get_tipping_stats(from = nil)
+      # If no from is specified, use midnight of today.
+      if from.nil?
+        # TODO: this is hardcoding an offset of -0700 - fix that to make
+        # the timezone configurable
+        now = Time.now - 60 * 60 * 7
+        from = Time.local(now.year, now.month, now.day)
+      end
+      from = from.to_i
+
+      stats = {}
+
+      stats[:total_tipped] = @db.get_first_value("select coalesce(sum(amount), 0) from transactions t where t.other_user_id is not null and t.user_id <> t.other_user_id and t.amount < 0 and t.created_at > ?", [ from ])
+      stats[:total_tips] = @db.get_first_value("select count(*) from transactions t where t.other_user_id is not null and t.user_id <> t.other_user_id and t.amount < 0 and t.created_at > ?", [ from ])
+      stats[:tippers] = @db.execute("select * from (select username, sum(amount) total from transactions t, users u where t.user_id = u.id and other_user_id is not null and amount < 0 and user_id <> other_user_id and t.created_at >= ? group by username) foo order by total asc", [ from ])
+      stats[:tippees] = @db.execute("select * from (select username, sum(amount) total from transactions t, users u where t.user_id = u.id and amount > 0 and user_id <> other_user_id and t.created_at >= ? group by username) foo order by total desc", [ from ])
+      stats
+    end
+
+    # Returns an array of all the Bitcoin transaction ids for deposits
+    def get_incoming_transaction_ids
+      transaction_ids = []
+      @db.execute("select incoming_transaction from transactions where incoming_transaction is not null") do |row|
+        transaction_ids << row[0]
+      end
+
+      transaction_ids
+    end
+
+    # Adds a transaction with a deposit. Returns true if the row was
+    # added, and false if the insert failed for some reason (like if the
+    # transaction_id already exists).
+    def create_transaction_from_deposit(user_id, amount, transaction_id)
+      @db.execute("insert into transactions (created_at, amount, incoming_transaction, user_id) values (?, ?, ?, ?)",
+                  [ Time.now.to_i, amount, transaction_id, user_id ])
+      return @db.changes == 1
+    end
+
+    # Adds a transaction for a withdrawal.
+    def create_transaction_from_withdrawal(user_id, amount, fee, address)
+      @db.transaction(:exclusive) do
+        # verify current balance
+        current_balance = self.get_balance_for_user_id(user_id)
+        if current_balance < (amount + fee)
+          raise InsufficientFundsError, "Insufficient funds; Current balance: #{current_balance} for amount #{amount} + fee #{fee}"
+        end
+
+        @db.execute("insert into transactions (created_at, user_id, amount, withdrawal_address)
+                     values (?, ?, ?, ?)", [ Time.now.to_i, user_id, (0 - (amount + fee)), address ])
+      end
+
+      true
+    end
+
+    def create_transaction_from_tip(from_user_id, to_user_id, amount, message)
+      @db.transaction(:exclusive) do
+        # verify current balance
+        current_balance = self.get_balance_for_user_id(from_user_id)
+        if current_balance < amount
+          raise InsufficientFundsError, "Insufficient funds; Current balance: #{current_balance} for amount #{amount}"
+        end
+
+        now = Time.now.to_i
+        @db.execute("insert into transactions (created_at, user_id, amount, note, other_user_id)
+                    values (?, ?, ?, ?, ?)", [ now, from_user_id, (0 - amount), message, to_user_id ])
+        @db.execute("insert into transactions (created_at, user_id, amount, note, other_user_id)
+                    values (?, ?, ?, ?, ?)", [ now, to_user_id, amount, message, from_user_id ])
+      end
+
+      true
+    end
+  end
+end

data/lib/bitbot/plugin.rb

@@ -0,0 +1,83 @@
+require 'cinch'
+require 'ircstring'
+
+require 'bitbot/database'
+require 'bitbot/plugin/common'
+require 'bitbot/plugin/help'
+require 'bitbot/plugin/balance'
+require 'bitbot/plugin/deposit'
+require 'bitbot/plugin/history'
+require 'bitbot/plugin/tip'
+require 'bitbot/plugin/tipstats'
+require 'bitbot/plugin/withdraw'
+require 'bitbot/plugin/update_addresses'
+require 'bitbot/plugin/update_exchange_rates'
+require 'bitbot/plugin/txfee'
+require 'bitbot/plugin/doc'
+require 'bitbot/plugin/online'
+require 'bitbot/plugin/prize'
+
+class Bitbot::Plugin
+  include Cinch::Plugin
+
+  include Bitbot::Common
+  include Bitbot::Help
+  include Bitbot::Balance
+  include Bitbot::Deposit
+  include Bitbot::History
+  include Bitbot::Tip
+  include Bitbot::TipStats
+  include Bitbot::Withdraw
+  include Bitbot::UpdateAddresses
+  include Bitbot::UpdateExchangeRates
+  include Bitbot::Txfee
+  include Bitbot::Doc
+  include Bitbot::Online
+  include Bitbot::Prize
+
+  def initialize(bot)
+    super
+    Bitbot::Database.new(File.join(config['data']['path'], "bitbot.db")).upgrade_schema()
+  end
+
+  set :prefix, ""
+
+  #
+  # Private messages
+  #
+  match /^help(.*)$/, :method => :on_help, :react_on => :private
+  match /^balance$/, :method => :on_balance, :react_on => :private
+  match /^history$/, :method => :on_history, :react_on => :private
+  match /^withdraw(.*)$/, :method => :on_withdraw, :react_on => :private
+  match /^deposit$/, :method => :on_deposit, :react_on => :private
+  match /^\+tip\s+(\w+)\s+([\d.]+)\s+?(.*)/, :method => :on_tip, :react_on => :private
+
+  #
+  # Channel messages
+  #
+  match /^\+help(.*)$/, :method => :on_help, :react_on => :channel
+  match /^\+tipstats$/, :method => :on_tipstats, :react_on => :channel
+  match /^\+tip\s+(\w+)\s+([\d.]+)\s+?(.*)/, :method => :on_tip, :react_on => :channel
+  match /^\+doc$/, :method => :on_tipstats, :react_on => :channel
+  match /^\+ticker$/, :method => :on_tipstats, :react_on => :channel
+
+  #
+  # Timer jobs
+  #
+  timer 60, :method => :on_update_exchange_rates
+  timer 60, :method => :on_update_addresses
+  timer 10, :method => :on_prize_win_check_balance
+
+  #
+  # Also run the timer jobs on connect
+  #
+  listen_to :connect, :method => :on_update_exchange_rates
+  listen_to :connect, :method => :on_update_addresses
+  listen_to :connect, :method => :on_online
+  listen_to :connect, :method => :on_doc
+
+  #
+  # Run these jobs on every channel message
+  #
+  listen_to :channel, :method => :on_prize
+end

data/lib/bitbot/plugin/balance.rb

@@ -0,0 +1,8 @@
+module Bitbot::Balance
+  def on_balance(m)
+    user_id = db.get_or_create_user_id_for_username(m.user.user)
+    m.reply "Your current balance is #{satoshi_with_usd(db.get_balance_for_user_id(user_id))}"
+    m.reply "Please note that a transaction fee of #{satoshi_to_str(withdrawal_fee)} will be added to your withdrawal"
+  end
+end
+

data/lib/bitbot/plugin/common.rb

@@ -0,0 +1,119 @@
+# coding: utf-8
+
+require 'ircstring'
+require 'bitbot/database'
+require 'bitbot/blockchain'
+
+module Bitbot::Common
+  def cached_addresses
+    @cached_addresses
+  end
+
+  def cached_addresses=(new)
+    @cached_addresses = new
+  end
+
+  def exchange_rate(currency)
+    if @cached_exchange_rates
+      return @cached_exchange_rates[currency]["15m"]
+    end
+    nil
+  end
+
+  def exchange_rates=(new)
+    @cached_exchange_rates = new
+  end
+
+  def withdrawal_fee
+    config['blockchain']['withdrawal_fee'] || 50000
+  end
+
+  #
+  # Returns a database handle for use in the current thread.
+  #
+  def db
+    Thread.current[:bitbot_db] ||=
+      Bitbot::Database.new(File.join(config['data']['path'], "bitbot.db"))
+  end
+
+  #
+  # Returns a Blockchain API helper. Everyone uses the same one,
+  # but it doesn't keep any state so it's fine.
+  #
+  def blockchain
+    @@cached_blockchain ||=
+      Bitbot::Blockchain.new(config['blockchain']['wallet_id'],
+                             config['blockchain']['password1'],
+                             config['blockchain']['password2'])
+  end
+
+  #
+  # Returns the User for the given username. If you want to get
+  # a User based on nick, User(nick) is easier.
+  #
+  def user_with_username(username)
+    bot.user_list.each do |user|
+      return user if user.user == username
+    end
+    nil
+  end
+
+  #
+  # Takes a string, and returns an int with number of satoshi.
+  # Eventually this could be smart enough to handle specified units too,
+  # rather than just assuming BTC every time.
+  #
+  def str_to_satoshi(str)
+    (str.to_f * 10**8).to_i
+  end
+
+  #
+  # Some number formatting helpers
+  #
+  def satoshi_to_str(satoshi)
+    str = "฿%.8f" % (satoshi.to_f / 10**8)
+    # strip trailing 0s
+    str.gsub(/0*$/, '')
+  end
+
+  def satoshi_to_usd(satoshi)
+    rate = exchange_rate("USD")
+    if rate
+      "$%.2f" % (satoshi.to_f / 10**8 * rate)
+    else
+      "$?"
+    end
+  end
+
+  def satoshi_to_gbp(satoshi)
+    rate = exchange_rate("GBP")
+    if rate
+      "$%.2f" % (satoshi.to_f / 10**8 * rate)
+    else
+      "$?"
+    end
+  end
+
+  def satoshi_to_eur(satoshi)
+    rate = exchange_rate("EUR")
+    if rate
+      "$%.2f" % (satoshi.to_f / 10**8 * rate)
+    else
+      "$?"
+    end
+  end
+
+  def satoshi_with_usd(satoshi)
+    btc_str = satoshi_to_str(satoshi)
+    if satoshi < 0
+      btc_str = btc_str.irc(:red)
+    else
+      btc_str = btc_str.irc(:green)
+    end
+
+    usd_str = "[".irc(:grey) + satoshi_to_usd(satoshi).irc(:blue) + "]".irc(:grey)
+
+    "#{btc_str} #{usd_str}"
+  end
+end
+

data/lib/bitbot/plugin/deposit.rb

@@ -0,0 +1,33 @@
+module Bitbot::Deposit
+  def on_deposit(m, create = true)
+    user_id = db.get_or_create_user_id_for_username(m.user.user)
+
+    unless cached_addresses
+      m.reply "TipJar is not initialized yet. Please try again later."
+      return
+    end
+
+    if address = cached_addresses[user_id]
+      m.reply "Send deposits to #{address["address"].irc(:bold)}. " +
+        "This address is specific to you, and any funds delivered " +
+        "to it will be added to your account after confirmation."
+      return
+    end
+
+    unless create
+      m.reply "There was a problem getting your deposit address. " +
+        "Please contact your friendly TipJar admin."
+      return
+    end
+
+    # Attempt to create an address.
+    blockchain.create_deposit_address_for_user_id(user_id)
+
+    # Force a refresh of the cached address list...
+    on_update_addresses
+
+    # Now run again, to show them the address we just looked up.
+    on_deposit(m, false)
+  end
+end
+

data/lib/bitbot/plugin/doc.rb

@@ -0,0 +1,5 @@
+module Bitbot::Doc  
+  def on_doc(m)
+    m.reply "Useful Information : TipJar Wiki - https://github.com/weirdthall/TipJar/wiki"
+  end
+end

data/lib/bitbot/plugin/help.rb

@@ -0,0 +1,6 @@
+module Bitbot::Help
+  def on_help(m, args)
+    m.reply "Channel Commands: +tip, +tipstats, +txfee"
+    m.reply "Private Message Commands: balance, history, withdraw, deposit"
+  end
+end

data/lib/bitbot/plugin/history.rb

@@ -0,0 +1,27 @@
+module Bitbot::History
+  def on_history(m)
+    user_id = db.get_or_create_user_id_for_username(m.user.user)
+
+    on_balance(m)
+
+    n = 0
+    db.get_transactions_for_user_id(user_id).each do |tx|
+      time = Time.at(tx[:created_at].to_i).strftime("%Y-%m-%d")
+      amount = satoshi_with_usd(tx[:amount])
+      action = if tx[:amount] < 0 && tx[:other_user_id]
+                 "to #{tx[:other_username]}"
+               elsif tx[:amount] > 0 && tx[:other_user_id]
+                 "from #{tx[:other_username]}"
+               elsif tx[:withdrawal_address]
+                 "withdrawal to #{tx[:withdrawal_address]}"
+               elsif tx[:incoming_transaction]
+                 "deposit from tx #{tx[:incoming_transaction]}"
+               end
+
+      m.reply "#{time.irc(:grey)}: #{amount} #{action} #{"(#{tx[:note]})".irc(:grey) if tx[:note]}"
+
+      n += 1
+      break if n >= 10
+    end
+  end
+end

data/lib/bitbot/plugin/online.rb

@@ -0,0 +1,5 @@
+module Bitbot::Online 
+  def on_doc(m)
+    m.reply "TipJar is online. Please type +help to interact with me."
+  end
+end

data/lib/bitbot/plugin/ping.rb

@@ -0,0 +1,5 @@
+module Bitbot::Ping
+  def on_ping
+    m.user.msg "PONG"
+  end
+end