diff --git a/python/tools/http_server.py b/python/tools/http_server.py new file mode 100644 index 0000000000000000000000000000000000000000..475d5958d4634b82fddc8235a837867bf3dd264e --- /dev/null +++ b/python/tools/http_server.py @@ -0,0 +1,56 @@ +#!/usr/bin/env python3 +# Copyright (c) 2016 Savoir-faire Linux Inc. +# Author: Adrien Béraud <adrien.beraud@savoirfairelinux.com> +# +# 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/>. + +from twisted.web import server, resource +from twisted.internet import reactor, endpoints + +import opendht as dht +import base64, json + +class DhtServer(resource.Resource): + isLeaf = True + node = dht.DhtRunner() + + def __init__(self): + self.node.run() + self.node.bootstrap("bootstrap.ring.cx", "4222") + + def render_GET(self, req): + uri = req.uri[1:] + h = dht.InfoHash(uri) if len(uri) == 40 else dht.InfoHash.get(uri.decode()) + print('GET', h) + res = self.node.get(h) + req.setHeader(b"content-type", b"application/json") + return json.dumps({'{:x}'.format(v.id):{'base64':base64.b64encode(v.data).decode()} for v in res}).encode() + + def render_POST(self, req): + uri = req.uri[1:] + data = req.args[b'data'][0] if b'data' in req.args else None + if not data and b'base64' in req.args: + data = base64.b64decode(req.args[b'base64'][0]) + h = dht.InfoHash(uri) if len(uri) == 40 else dht.InfoHash.get(uri.decode()) + print('POST', h, data) + req.setHeader(b"content-type", b"application/json") + if data: + self.node.put(h, dht.Value(data)) + return json.dumps({'success':True}).encode() + else: + req.setResponseCode(400) + return json.dumps({'success':False, 'error':'no data parameter'}).encode() + +endpoints.serverFromString(reactor, "tcp:8080").listen(server.Site(DhtServer())) +reactor.run()