72 lines
2.2 KiB
Python
72 lines
2.2 KiB
Python
from __future__ import print_function
|
|
import sys, json
|
|
from twisted.python import log
|
|
from twisted.internet import reactor
|
|
from .transcribe import Wormhole
|
|
from .. import public_relay
|
|
|
|
APPID = b"lothar.com/wormhole/text-or-file-xfer"
|
|
|
|
w = Wormhole(APPID, public_relay.RENDEZVOUS_RELAY)
|
|
|
|
if sys.argv[1] == "send":
|
|
message = sys.argv[2]
|
|
data = json.dumps({"message": message}).encode("utf-8")
|
|
d = w.get_code()
|
|
def _got_code(code):
|
|
print("code is:", code)
|
|
return w.send_data(data)
|
|
d.addCallback(_got_code)
|
|
def _sent(_):
|
|
return w.get_data()
|
|
d.addCallback(_sent)
|
|
def _got_data(them_bytes):
|
|
them_d = json.loads(them_bytes.decode("utf-8"))
|
|
if them_d["message_ack"] == "ok":
|
|
print("text sent")
|
|
else:
|
|
print("error sending text: %r" % (them_d,))
|
|
d.addCallback(_got_data)
|
|
elif sys.argv[1] == "receive":
|
|
code = sys.argv[2]
|
|
w.set_code(code)
|
|
d = w.get_data()
|
|
def _got_data(them_bytes):
|
|
them_d = json.loads(them_bytes.decode("utf-8"))
|
|
if "error" in them_d:
|
|
print("ERROR: " + them_d["error"], file=sys.stderr)
|
|
raise RuntimeError
|
|
if "file" in them_d:
|
|
print("they're trying to send us a file, which I don't handle")
|
|
data = json.dumps({"error": "not capable of receiving files"})
|
|
d1 = w.send_data(data.encode("utf-8"))
|
|
d1.addCallback(lambda _: RuntimeError())
|
|
return d1
|
|
if not "message" in them_d:
|
|
print("I don't know what they're offering\n")
|
|
print(them_d)
|
|
data = json.dumps({"error": "huh?"})
|
|
d1 = w.send_data(data.encode("utf-8"))
|
|
d1.addCallback(lambda _: RuntimeError())
|
|
return d1
|
|
print(them_d["message"])
|
|
data = json.dumps({"message_ack": "ok"})
|
|
d1 = w.send_data(data.encode("utf-8"))
|
|
d1.addCallback(lambda _: 0)
|
|
return d1
|
|
d.addCallback(_got_data)
|
|
else:
|
|
raise ValueError("bad command")
|
|
|
|
d.addBoth(w.close)
|
|
rc = []
|
|
def _success(res):
|
|
rc.append(res)
|
|
def _fail(f):
|
|
log.err(f)
|
|
rc.append(1)
|
|
d.addCallbacks(_success, _fail)
|
|
d.addCallback(lambda _: reactor.stop())
|
|
reactor.run()
|
|
sys.exit(rc[0])
|