from trinket import Trinket, Response
from trinket.response import file_iterator
from trinket.extensions import logger

bauble = logger(Trinket())


@bauble.route('/')
async def hello(request):
    return Response.raw(b'Hello World !')


@bauble.route('/raw', methods=['POST'])
async def raw(request):
    return Response.raw(b'You got here')


@bauble.route('/read', methods=['POST'])
async def reader(request):
    await request.parse_body()
    files = list(request.files.keys())
    return Response.raw("You got here and it's all read: {}".format(files))


@bauble.route('/hello/full/with/{one}/and/{two}')
async def json(request, one, two):    
    response = Response.json({
        'parameters': f'{one} and {two}',
        'query': request.query.get('query'),
        'cookie': request.cookies['test'],
    })
    response.cookies.set(name='bench', value='value')
    return response


@bauble.route('/websocket')
async def serve_websocket(request):
    response = Response.streamer(file_iterator('myfile.ext'))
    return response


@bauble.websocket('/chat')
async def chat(request, websocket):
    while True:
        msg = await websocket.recv()
        for ws in bauble.websockets:
            if ws is not websocket:
                await ws.send(msg)


bauble.start()