#!/bin/bash
# Model generation script for MPR
# Generates all targets from schema/modelgen.json config

set -e
cd "$(dirname "$0")/.."

echo "Generating models from schema/models..."
python -m modelgen generate --config schema/modelgen.json

# Generate gRPC stubs from proto
echo "Generating gRPC stubs..."
python -m grpc_tools.protoc \
    -I rpc/protos \
    --python_out=rpc \
    --grpc_python_out=rpc \
    rpc/protos/worker.proto

# Fix relative import in generated grpc stub
sed -i 's/^import worker_pb2/from . import worker_pb2/' rpc/worker_pb2_grpc.py

echo "Done!"