set -e

npx --yes @bufbuild/buf generate

generated_json=$(find src/a2a/types -name "*.swagger.json" -print -quit)

if [ -n "$generated_json" ]; then

echo "Renaming $generated_json to src/a2a/types/a2a.json"

mv "$generated_json" src/a2a/types/a2a.json

echo "Warning: No Swagger JSON generated."

echo "Fixing imports in src/a2a/types/a2a_pb2_grpc.py"

sed 's/import a2a_pb2 as a2a__pb2/from . import a2a_pb2 as a2a__pb2/g' src/a2a/types/a2a_pb2_grpc.py > src/a2a/types/a2a_pb2_grpc.py.tmp && mv src/a2a/types/a2a_pb2_grpc.py.tmp src/a2a/types/a2a_pb2_grpc.py

echo "Downloading legacy v0.3 proto file..."

curl -o src/a2a/compat/v0_3/a2a_v0_3.proto https://raw.githubusercontent.com/a2aproject/A2A/b3b266d127dde3d1000ec103b252d1de81289e83/specification/grpc/a2a.proto