mirror of
https://github.com/Chia-Network/chia-blockchain.git
synced 2026-08-28 18:14:19 -05:00
Matt HauffandGitHub
40db4635a8
* Enable PEP604 Ruff rules * Fix harcoded signature in test * Hack CLVMStreamable test with note to fast follow
67 lines
1.7 KiB
Python
67 lines
1.7 KiB
Python
from __future__ import annotations
|
|
|
|
from typing import Any
|
|
|
|
from chia_rs.sized_bytes import bytes32
|
|
from typing_extensions import TypedDict
|
|
|
|
from chia.util.json_util import dict_to_json_str
|
|
|
|
# Messages must follow this format
|
|
# Message = { "command": "command_name",
|
|
# "data" : {...},
|
|
# "request_id": "bytes_32",
|
|
# "destination": "service_name",
|
|
# "origin": "service_name"
|
|
# }
|
|
|
|
|
|
class WsRpcMessage(TypedDict):
|
|
command: str
|
|
ack: bool
|
|
data: dict[str, Any]
|
|
request_id: str
|
|
destination: str
|
|
origin: str
|
|
|
|
|
|
def format_response(incoming_msg: WsRpcMessage, response_data: dict[str, Any]) -> str:
|
|
"""
|
|
Formats the response into standard format.
|
|
"""
|
|
response = {
|
|
"command": incoming_msg.get("command", ""),
|
|
"ack": True,
|
|
"data": response_data,
|
|
"request_id": incoming_msg.get("request_id", ""),
|
|
"destination": incoming_msg.get("origin", ""),
|
|
"origin": incoming_msg.get("destination", ""),
|
|
}
|
|
|
|
json_str = dict_to_json_str(response)
|
|
return json_str
|
|
|
|
|
|
def create_payload(command: str, data: dict[str, Any], origin: str, destination: str) -> str:
|
|
response = create_payload_dict(command, data, origin, destination)
|
|
return dict_to_json_str(response)
|
|
|
|
|
|
def create_payload_dict(command: str, data: dict[str, Any] | None, origin: str, destination: str) -> WsRpcMessage:
|
|
if data is None:
|
|
data = {}
|
|
|
|
return WsRpcMessage(
|
|
command=command,
|
|
ack=False,
|
|
data=data,
|
|
request_id=bytes32.secret().hex(),
|
|
destination=destination,
|
|
origin=origin,
|
|
)
|
|
|
|
|
|
def pong() -> dict[str, Any]:
|
|
response = {"success": True}
|
|
return response
|