cosmos-sdk/scripts/protocgen.sh

60 lines
2.0 KiB
Bash
Executable File

#!/usr/bin/env bash
# How to run manually:
# docker build --pull --rm -f "contrib/devtools/Dockerfile" -t cosmossdk-proto:latest "contrib/devtools"
# docker run --rm -v $(pwd):/workspace --workdir /workspace cosmossdk-proto sh ./scripts/protocgen.sh
echo "Formatting protobuf files"
find ./ -name "*.proto" -exec clang-format -i {} \;
set -e
home=$PWD
echo "Generating proto code"
proto_dirs=$(find ./ -name 'buf.yaml' -print0 | xargs -0 -n1 dirname | sort | uniq)
for dir in $proto_dirs; do
echo "Generating proto code for $dir"
cd $dir
# check if buf.gen.pulsar.yaml exists in the proto directory
if [ -f "buf.gen.pulsar.yaml" ]; then
buf generate --template buf.gen.pulsar.yaml
# move generated files to the right places
if [ -d "../cosmos" -a "$dir" != "./proto" ]; then
cp -r ../cosmos $home/api
rm -rf ../cosmos
fi
fi
# check if buf.gen.gogo.yaml exists in the proto directory
if [ -f "buf.gen.gogo.yaml" ]; then
for file in $(find . -maxdepth 5 -name '*.proto' | grep -v '^./cosmos/store/'); do
# this regex checks if a proto file has its go_package set to cosmossdk.io/api/...
# gogo proto files SHOULD ONLY be generated if this is false
# we don't want gogo proto to run for proto files which are natively built for google.golang.org/protobuf
if grep -q "option go_package" "$file" && grep -H -o -c 'option go_package.*cosmossdk.io/api' "$file" | grep -q ':0$'; then
buf generate --template buf.gen.gogo.yaml $file
fi
done
# move generated files to the right places
if [ -d "../cosmossdk.io" -a "$dir" != "./proto" ]; then
cp -r ../cosmossdk.io/* $home
rm -rf ../cosmossdk.io
fi
if [ -d "../github.com" -a "$dir" != "./proto" ]; then
cp -r ../github.com/cosmos/cosmos-sdk/* $home
rm -rf ../github.com
fi
fi
cd $home
done
# move generated files to the right places
cp -r github.com/cosmos/cosmos-sdk/* ./
rm -rf github.com
go mod tidy