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

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

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

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

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

echo "Done!"