#!/bin/bash

set -e

PROTO_ROOT=protos
OUT_DIR=generated

echo "[INFO] Generating gRPC Python code..."

# 生成protobuf到指定包结构
python -m grpc_tools.protoc \
    -Iprotos \
    --python_out=generated \
    --grpc_python_out=generated \
    protos/cmvr/api/common.proto \
    protos/cmvr/api/biohead_command.proto \
    protos/cmvr/api/biohead_service.proto

echo "[INFO] Adding __init__.py files to make packages..."
# 确保生成的包结构里有__init__.py
touch ${OUT_DIR}/__init__.py
mkdir -p ${OUT_DIR}/cmvr
touch ${OUT_DIR}/cmvr/__init__.py
mkdir -p ${OUT_DIR}/cmvr/api
touch ${OUT_DIR}/cmvr/api/__init__.py

echo "[INFO] gRPC Python code generation completed."