from __future__ import print_function

import capnp
import example_capability_capnp as capability
import socket

class Server:
    def __init__(self, val=1):
        self.val = val

    def foo(self, context):
        context.results.x = str(context.params.i * 5 + self.val)

def example_simple_rpc():
    def _restore(ref_id):
        return capability.TestInterface.new_server(Server(100))

    loop = capnp.EventLoop()
    
    import os
    read, write = os.pipe()
    read_stream = capnp.FdAsyncIoStream(write)
    write_stream = capnp.FdAsyncIoStream(read)

    restorer = capnp.Restorer(capability.TestSturdyRefObjectId, _restore)
    server = capnp.RpcServer(loop, write_stream, restorer)
    client = capnp.RpcClient(loop, read_stream)

    ref = capability.TestSturdyRefObjectId.new_message()
    cap = client.restore(ref.as_reader())
    cap = cap.cast_as(capability.TestInterface)

    remote = cap.foo(i=5)
    response = loop.wait(remote)

    assert response.x == '125'

example_simple_rpc()