add websocket to webpages
This commit is contained in:
parent
41d9c94821
commit
18d145cced
|
@ -216,6 +216,9 @@ if __name__ == '__main__':
|
||||||
util.set_verbosity(True)
|
util.set_verbosity(True)
|
||||||
server = NetworkServer(config)
|
server = NetworkServer(config)
|
||||||
server.start()
|
server.start()
|
||||||
|
if config.get('websocket_server'):
|
||||||
|
import websockets
|
||||||
|
websockets.WebSocketServer(config, server).start()
|
||||||
try:
|
try:
|
||||||
daemon_loop(server)
|
daemon_loop(server)
|
||||||
except KeyboardInterrupt:
|
except KeyboardInterrupt:
|
||||||
|
|
|
@ -0,0 +1,142 @@
|
||||||
|
#!/usr/bin/env python
|
||||||
|
#
|
||||||
|
# Electrum - lightweight Bitcoin client
|
||||||
|
# Copyright (C) 2015 Thomas Voegtlin
|
||||||
|
#
|
||||||
|
# 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 threading, Queue, os, json, time
|
||||||
|
from collections import defaultdict
|
||||||
|
try:
|
||||||
|
from SimpleWebSocketServer import WebSocket, SimpleSSLWebSocketServer
|
||||||
|
except ImportError:
|
||||||
|
print "install SimpleWebSocketServer"
|
||||||
|
sys.exit()
|
||||||
|
|
||||||
|
import util, daemon
|
||||||
|
|
||||||
|
request_queue = Queue.Queue()
|
||||||
|
|
||||||
|
class ElectrumWebSocket(WebSocket):
|
||||||
|
|
||||||
|
def handleMessage(self):
|
||||||
|
assert self.data[0:3] == 'id:'
|
||||||
|
print "message received", self.data
|
||||||
|
request_id = self.data[3:]
|
||||||
|
request_queue.put((self, request_id))
|
||||||
|
|
||||||
|
def handleConnected(self):
|
||||||
|
print "connected", self.address
|
||||||
|
|
||||||
|
def handleClose(self):
|
||||||
|
print "closed", self.address
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
class WsClientThread(daemon.ClientThread):
|
||||||
|
|
||||||
|
def __init__(self, config, server):
|
||||||
|
util.DaemonThread.__init__(self)
|
||||||
|
self.server = server
|
||||||
|
self.config = config
|
||||||
|
self.response_queue = Queue.Queue()
|
||||||
|
self.server.add_client(self)
|
||||||
|
self.subscriptions = defaultdict(list)
|
||||||
|
self.sub_ws = defaultdict(list)
|
||||||
|
self.counter = 0
|
||||||
|
|
||||||
|
def make_request(self, request_id):
|
||||||
|
# read json file
|
||||||
|
rdir = self.config.get('requests_dir')
|
||||||
|
n = os.path.join(rdir, request_id + '.json')
|
||||||
|
with open(n) as f:
|
||||||
|
s = f.read()
|
||||||
|
d = json.loads(s)
|
||||||
|
addr = d.get('address')
|
||||||
|
amount = d.get('amount')
|
||||||
|
return addr, amount
|
||||||
|
|
||||||
|
def reading_thread(self):
|
||||||
|
while self.is_running():
|
||||||
|
try:
|
||||||
|
ws, request_id = request_queue.get()
|
||||||
|
except Queue.Empty:
|
||||||
|
continue
|
||||||
|
try:
|
||||||
|
addr, amount = self.make_request(request_id)
|
||||||
|
except:
|
||||||
|
continue
|
||||||
|
method = 'blockchain.address.subscribe'
|
||||||
|
params = [addr]
|
||||||
|
request = {'method':method, 'params':params, 'id':self.counter}
|
||||||
|
self.subscriptions[method].append(params)
|
||||||
|
self.sub_ws[self.counter] = ws, amount, request
|
||||||
|
self.counter += 1
|
||||||
|
self.server.send_request(self, request)
|
||||||
|
|
||||||
|
def run(self):
|
||||||
|
threading.Thread(target=self.reading_thread).start()
|
||||||
|
while self.is_running():
|
||||||
|
try:
|
||||||
|
r = self.response_queue.get(timeout=0.1)
|
||||||
|
except Queue.Empty:
|
||||||
|
continue
|
||||||
|
id = r.get('id')
|
||||||
|
if id is None:
|
||||||
|
method = r.get('method')
|
||||||
|
params = r.get('params')
|
||||||
|
else:
|
||||||
|
ws, amount, rr = self.sub_ws[id]
|
||||||
|
method = rr.get('method')
|
||||||
|
params = rr.get('params')
|
||||||
|
|
||||||
|
result = r.get('result')
|
||||||
|
|
||||||
|
if method == 'blockchain.address.subscribe':
|
||||||
|
print 'response', r
|
||||||
|
if result is not None:
|
||||||
|
request = {'method':'blockchain.address.get_balance', 'params':params, 'id':self.counter}
|
||||||
|
self.server.send_request(self, request)
|
||||||
|
self.sub_ws[self.counter] = ws, amount, request
|
||||||
|
self.counter += 1
|
||||||
|
|
||||||
|
if r.get('method') == 'blockchain.address.get_balance':
|
||||||
|
print 'response', r
|
||||||
|
if not ws.closed:
|
||||||
|
if sum(result.values()) >=amount:
|
||||||
|
ws.sendMessage(unicode('paid'))
|
||||||
|
|
||||||
|
self.server.remove_client(self)
|
||||||
|
|
||||||
|
|
||||||
|
class WebSocketServer(threading.Thread):
|
||||||
|
|
||||||
|
def __init__(self, config, ns):
|
||||||
|
threading.Thread.__init__(self)
|
||||||
|
self.config = config
|
||||||
|
self.net_server = ns
|
||||||
|
self.daemon = True
|
||||||
|
|
||||||
|
def run(self):
|
||||||
|
t = WsClientThread(self.config, self.net_server)
|
||||||
|
t.start()
|
||||||
|
|
||||||
|
host = self.config.get('websocket_server')
|
||||||
|
port = self.config.get('websocket_port', 9999)
|
||||||
|
certfile = self.config.get('ssl_chain')
|
||||||
|
keyfile = self.config.get('ssl_privkey')
|
||||||
|
self.server = SimpleSSLWebSocketServer(host, port, ElectrumWebSocket, certfile, keyfile)
|
||||||
|
self.server.serveforever()
|
||||||
|
|
||||||
|
|
|
@ -26,7 +26,7 @@ var id = getUrlParameter('id');
|
||||||
|
|
||||||
if (id) {
|
if (id) {
|
||||||
var jqxhr = $.getJSON(id + ".json", function() {
|
var jqxhr = $.getJSON(id + ".json", function() {
|
||||||
console.log( "success" );
|
console.log("getJSON:success");
|
||||||
})
|
})
|
||||||
.done( function(data) {
|
.done( function(data) {
|
||||||
new QRCode(document.getElementById("qrcode"), data.URI);
|
new QRCode(document.getElementById("qrcode"), data.URI);
|
||||||
|
@ -50,17 +50,30 @@ if (id) {
|
||||||
value: current
|
value: current
|
||||||
});
|
});
|
||||||
if (current >= max) {
|
if (current >= max) {
|
||||||
$("#container").html("This payment request has expired");
|
$("#container").html("This invoice has expired");
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
var interval = setInterval(update, 1000);
|
var interval = setInterval(update, 1000);
|
||||||
});
|
});
|
||||||
})
|
})
|
||||||
.fail(function() {
|
.fail(function() {
|
||||||
console.log( "error fail" );
|
console.log("error fail");
|
||||||
$("<p />").text("error").appendTo($("p#error"));
|
$("<p />").text("error").appendTo($("p#error"));
|
||||||
});
|
});
|
||||||
};
|
};
|
||||||
|
|
||||||
|
var ws = new WebSocket("wss://electrum.org:9999/");
|
||||||
|
ws.onopen = function() {
|
||||||
|
ws.send('id:' + id);
|
||||||
|
};
|
||||||
|
ws.onmessage = function (evt) {
|
||||||
|
var received_msg = evt.data;
|
||||||
|
if(received_msg == 'paid'){
|
||||||
|
$("#container").html("This invoice has been paid.");
|
||||||
|
}
|
||||||
|
else alert("Message is received:"+ received_msg);
|
||||||
|
};
|
||||||
|
|
||||||
</script>
|
</script>
|
||||||
</head>
|
</head>
|
||||||
<body>
|
<body>
|
||||||
|
|
Loading…
Reference in New Issue