import json
from autobahn.twisted.websocket import (
    create_client_agent,
)
from twisted.internet import reactor
from twisted.internet.defer import inlineCallbacks, returnValue, Deferred


@inlineCallbacks
def main(reactor):
    agent = create_client_agent(reactor)
    uri = "ws://127.0.0.1:8888/storage-plugins/privatestorageio-zkapauthz-v2/recover"
    dircap = "URI:DIR2-RO:oibl7ox3nc4cwjqx4f3nrwquxi:qjbyvkxqey7lyaktni6gbg5zyzeiphfnk6jdvjrp6y4wf2ktufeq"
    print("URI", uri)
    with open("/home/meejah/.config/gridsync/PrivateStorage Staging/private/api_auth_token") as f:
        token = f.read()
    proto = yield agent.open(
        uri,
        {"headers": {"Authorization": f"tahoe-lafs {token}"}},
    )

    def status_update(data, is_binary=False):
        print(f"status update: {data}")
        # XXX e.g. hook into UI to display state
    proto.on("message", status_update)

    print("wait for open")
    yield proto.is_open
    print("send recovery request")
    yield proto.sendMessage(json.dumps({"recovery-capability": dircap}).encode("utf8"))
    try:
        print("wait for close")
        yield proto.is_closed
    except Exception as e:
        raise TahoeWebError(f"Error during recovery: {e}")



from twisted.internet.task import react
react(main)