1# Invoke this script with an argument pointing to where the AOSP `tools/netsim/src/proto` is
2PROTOC_OUT=bumble/transport/grpc_protobuf
3
4proto_files=(common.proto packet_streamer.proto hci_packet.proto startup.proto)
5for proto_file in "${proto_files[@]}"
6do
7    python -m grpc_tools.protoc -I$1 --proto_path=bumble/transport --python_out=$PROTOC_OUT --pyi_out=$PROTOC_OUT --grpc_python_out=$PROTOC_OUT $1/$proto_file
8done
9
10python_files=(packet_streamer_pb2_grpc.py packet_streamer_pb2.py hci_packet_pb2.py startup_pb2.py)
11for python_file in "${python_files[@]}"
12do
13    sed -i 's/^import .*_pb2 as/from . \0/' $PROTOC_OUT/$python_file
14done
15