forked from cartesi/rollups-examples
/
echo.py
91 lines (75 loc) · 2.96 KB
/
echo.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
# Copyright 2022 Cartesi Pte. Ltd.
#
# SPDX-License-Identifier: Apache-2.0
# Licensed under the Apache License, Version 2.0 (the "License"); you may not use
# this file except in compliance with the License. You may obtain a copy of the
# License at http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software distributed
# under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR
# CONDITIONS OF ANY KIND, either express or implied. See the License for the
# specific language governing permissions and limitations under the License.
from os import environ
import logging
import requests
import json
logging.basicConfig(level="INFO")
logger = logging.getLogger(__name__)
rollup_server = environ["ROLLUP_HTTP_SERVER_URL"]
logger.info(f"HTTP rollup_server url is {rollup_server}")
def hex2str(hex):
"""
Decodes a hex string into a regular string
"""
return bytes.fromhex(hex[2:]).decode("utf-8")
def str2hex(str):
"""
Encodes a string as a hex string
"""
return "0x" + str.encode("utf-8").hex()
def performDriverBehaviourIndexCalc(json):
"""
Parses the sensor data to spit out a driverIndex
"""
return 7.3
def handle_advance(data):
input = hex2str(data["payload"])
logger.info(f"Received input: {input}")
inputJSON = json.loads(input)
driverIndexCalculated = performDriverBehaviourIndexCalc(inputJSON)
driverIndexHex = str2hex(driverIndexCalculated)
logger.info("Adding notice")
# notice = {"payload": data["payload"]}
response = requests.post(rollup_server + "/notice", json=str2hex(str(driverIndexHex)))
logger.info(f"Received notice status {response.status_code} body {response.content}")
return "accept"
def handle_inspect(data):
logger.info(f"Received inspect request data {data}")
logger.info("Adding report")
report = {"payload": data["payload"]}
response = requests.post(rollup_server + "/report", json=report)
logger.info(f"Received report status {response.status_code}")
return "accept"
handlers = {
"advance_state": handle_advance,
"inspect_state": handle_inspect,
}
finish = {"status": "accept"}
rollup_address = None
while True:
logger.info("Sending finish")
response = requests.post(rollup_server + "/finish", json=finish)
logger.info(f"Received finish status {response.status_code}")
if response.status_code == 202:
logger.info("No pending rollup request, trying again")
else:
rollup_request = response.json()
data = rollup_request["data"]
if "metadata" in data:
metadata = data["metadata"]
if metadata["epoch_index"] == 0 and metadata["input_index"] == 0:
rollup_address = metadata["msg_sender"]
logger.info(f"Captured rollup address: {rollup_address}")
continue
handler = handlers[rollup_request["request_type"]]
finish["status"] = handler(rollup_request["data"])