__init__.py

from robyn import SubRouter, WebSocket, jsonify

from .di_subrouter import di_subrouter
from .file_api import static_router

sub_router = SubRouter(prefix="/sub_router")

websocket = WebSocket(sub_router, "/ws")

__all__ = ["sub_router", "websocket", "di_subrouter", "static_router"]


@websocket.on("connect")
async def connect(ws):
    return "Hello world, from ws"


@websocket.on("message")
async def message():
    return "Message"


@websocket.on("close")
async def close(ws):
    return jsonify({"message": "closed"})


@sub_router.get("/foo")
def get_foo():
    return {"message": "foo"}


@sub_router.post("/foo")
def post_foo():
    return {"message": "foo"}


@sub_router.put("/foo")
def put_foo():
    return {"message": "foo"}


@sub_router.delete("/foo")
def delete_foo():
    return {"message": "foo"}


@sub_router.patch("/foo")
def patch_foo():
    return {"message": "foo"}


@sub_router.options("/foo")
def option_foo():
    return {"message": "foo"}


@sub_router.trace("/foo")
def trace_foo():
    return {"message": "foo"}


@sub_router.head("/foo")
def head_foo():
    return {"message": "foo"}


@sub_router.post("/openapi_test", openapi_tags=["test subrouter tag"])
def sample_subrouter_openapi_endpoint():
    """Get subrouter openapi"""
    return 200