add merchant script

scripts/merchant.conf.template

@@ -0,0 +1,19 @@
+[main]
+host = hostname of the machine where you run this program
+port = choose a port number
+password = choose a password
+
+[db]
+instance = the hostname of your sql server
+name = the database name
+user = your database username
+password = your database password
+
+[electrum]
+server = the electrum server you will use
+mpk = the master public key of your wallet (in hexadecimal)
+
+[callback]
+received = URL where we POST json data when payment has been received
+expired = URL where we POST json data if payment has expired
+password = password sent in the json data, for authentication

scripts/merchant.py

@@ -0,0 +1,228 @@
+#!/usr/bin/env python
+#
+# Electrum - lightweight Bitcoin client
+# Copyright (C) 2011 thomasv@gitorious
+#
+# This program 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 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program. If not, see <http://www.gnu.org/licenses/>.
+
+
+import time, thread, sys, socket, os
+import urllib2,json
+import MySQLdb as mdb
+import Queue
+from electrum import Wallet, TcpStratumInterface
+
+import ConfigParser
+config = ConfigParser.ConfigParser()
+config.read("merchant.conf")
+
+db_instance = config.get('db','instance')
+db_user = config.get('db','user')
+db_password = config.get('db','password')
+db_name = config.get('db','name')
+
+electrum_server = config.get('electrum','server')
+mpk = config.get('electrum','mpk')
+
+my_password = config.get('main','password')
+my_host = config.get('main','host')
+my_port = config.getint('main','port')
+
+cb_received = config.get('callback','received')
+cb_expired = config.get('callback','expired')
+cb_password = config.get('callback','password')
+
+wallet = Wallet()
+wallet.master_public_key = mpk.decode('hex')
+
+
+
+omg_addresses = {}
+
+def electrum_input_thread(in_queue, i):
+    while True:
+        addr, amount = in_queue.get(True,1000000000)
+        if addr in omg_addresses: 
+            continue
+        else:
+            print "subscribing to ", addr
+            omg_addresses[addr] = amount
+            i.send([('blockchain.address.subscribe',[addr])])
+
+
+def electrum_output_thread(out_queue, i):
+    while True:
+        r = i.responses.get(True, 100000000000)
+        method = r.get('method') 
+
+        if method == 'blockchain.address.subscribe':
+            addr = r.get('params')[0]
+            i.send([('blockchain.address.get_history',[addr])])
+
+        elif method == 'blockchain.address.get_history':
+            addr = r.get('params')[0]
+            #print "received history for", addr
+            confirmed = unconfirmed = 0
+            h = r.get('result')
+            if h is None:
+                continue
+            for item in h:
+                v = item['value']
+                if v<0: continue
+                if item['height']:
+                    confirmed += v
+                else:
+                    unconfirmed += v
+                
+            s = (confirmed+unconfirmed)/1.e8
+            print "balance for %s:"%addr, s
+            amount = float(omg_addresses.get(addr))
+            if s>=amount:
+                out_queue.put( ('payment',addr) )
+
+
+stopping = False
+
+def do_stop():
+    global stopping
+    stopping = True
+
+def do_create(conn):
+    # creation
+    cur = conn.cursor()
+    cur.execute("CREATE TABLE electrum_payments (id INT PRIMARY KEY, address VARCHAR(40), amount FLOAT, received_at TIMESTAMP, expires_at TIMESTAMP, paid INT(1), processed INT(1));")
+    conn.commit()
+
+def process_request(i, amount, confirmations, expires_in, password):
+    print "process_request", i, amount, confirmations, expires_in
+    if password!=my_password:
+        print "wrong password ", password
+        return
+    addr = wallet.get_new_address(i, 0)
+    out_queue.put( ('request',(i,addr,amount,expires_in) ))
+    return addr
+
+def get_mpk():
+    return wallet.master_public_key.encode('hex')
+
+
+def server_thread(conn):
+    from jsonrpclib.SimpleJSONRPCServer import SimpleJSONRPCServer
+    server = SimpleJSONRPCServer(( my_host, my_port))
+    server.register_function(process_request, 'request')
+    server.register_function(get_mpk, 'mpk')
+    server.register_function(do_stop, 'stop')
+    server.serve_forever()
+    
+
+def handle_command(cmd):
+    import jsonrpclib
+    server = jsonrpclib.Server('http://%s:%d'%(my_host, my_port))
+    try:
+        if cmd == 'mpk':
+            out = server.mpk()
+        elif cmd == 'stop':
+            out = server.stop()
+        elif cmd == 'create':
+            conn = mdb.connect(db_instance, db_user, db_password, db_name);
+            do_create(conn)
+            out = "ok"
+        else:
+            out = "unknown command"
+    except socket.error:
+        print "Server not running"
+        return 1
+
+    print out
+    return 0
+
+
+if __name__ == '__main__':
+
+    if len(sys.argv) > 1:
+        ret = handle_command(sys.argv[1])
+        sys.exit(ret)
+
+    print "using database", db_name
+    conn = mdb.connect(db_instance, db_user, db_password, db_name);
+
+    i = TcpStratumInterface(electrum_server, 50001)
+    i.init_socket()
+    i.start()
+    
+
+    # this process detects when addresses have paid
+    in_queue = Queue.Queue()
+    out_queue = Queue.Queue()
+    thread.start_new_thread(electrum_input_thread, (in_queue,i))
+    thread.start_new_thread(electrum_output_thread, (out_queue,i))
+
+    thread.start_new_thread(server_thread, (conn,))
+
+
+    while not stopping:
+        cur = conn.cursor()
+
+        # get a list of addresses to watch
+        cur.execute("SELECT address, amount FROM electrum_payments WHERE paid IS NULL;")
+        data = cur.fetchall()
+        for item in data: 
+            in_queue.put(item)
+
+        try:
+            cmd, params = out_queue.get(True, 10)
+        except Queue.Empty:
+            cmd = ''
+
+        if cmd == 'payment':
+            # set paid=1 for received payments
+            print "received payment from", addr
+            cur.execute("select id from electrum_payments where address='%s';"%addr)
+            id = cur.fetchone()[0]
+            cur.execute("update electrum_payments set paid=1 where id=%d;"%(id))
+        elif cmd == 'request':
+            i, addr, amount, hours = params
+            sql = "INSERT INTO electrum_payments (id, address, amount, received_at, expires_at, paid, processed)"\
+                + " VALUES (%d, '%s', %f, CURRENT_TIMESTAMP, ADDTIME(CURRENT_TIMESTAMP, '0 %d:0:0'), NULL, NULL);"%(i, addr, amount, hours)
+            cur.execute(sql)
+
+
+        # set paid=0 for expired requests 
+        cur.execute("""UPDATE electrum_payments set paid=0 WHERE expires_at < CURRENT_TIMESTAMP and paid is NULL;""")
+
+        # do callback for addresses that received payment
+        cur.execute("""SELECT id, address, paid from electrum_payments WHERE paid is not NULL and processed is NULL;""")
+        data = cur.fetchall()
+        for item in data:
+            print "callback:", item
+            id = int(item[0])
+            address = item[1]
+            paid = int(item[2])
+            headers = {'content-type':'application/json'}
+            data_json = { 'id':id, 'address':address, 'btc_auth':cb_password }
+            data_json = json.dumps(data_json)
+            url = cb_received if paid else cb_expired
+            req = urllib2.Request(url, data_json, headers)
+            try:
+                response_stream = urllib2.urlopen(req)
+                cur.execute("UPDATE electrum_payments SET processed=1 WHERE id=%d;"%(id))
+            except urllib2.HTTPError:
+                print "cannot do callback", data_json
+        
+        conn.commit()
+
+
+    conn.close()
+
+