pool/poolrpc/gen_protos.sh
2022-09-08 10:37:54 -04:00

70 lines
2.2 KiB
Bash
Executable file

#!/bin/bash
set -e
# generate compiles the *.pb.go stubs from the *.proto files.
function generate() {
# Generate the gRPC bindings for all proto files.
for file in ./*.proto; do
protoc -I/usr/local/include -I. -I.. \
--go_out . --go_opt paths=source_relative \
--go-grpc_out . --go-grpc_opt paths=source_relative \
"${file}"
# Only generate REST stubs if requested.
if [[ "$1" == "no-rest" ]]; then
continue
fi
# Generate the REST reverse proxy.
annotationsFile=${file//proto/yaml}
protoc -I/usr/local/include -I. -I.. \
--grpc-gateway_out . \
--grpc-gateway_opt logtostderr=true \
--grpc-gateway_opt paths=source_relative \
--grpc-gateway_opt grpc_api_configuration=${annotationsFile} \
"${file}"
# Finally, generate the swagger file which describes the REST API in detail.
protoc -I/usr/local/include -I. -I.. \
--openapiv2_out . \
--openapiv2_opt logtostderr=true \
--openapiv2_opt grpc_api_configuration=${annotationsFile} \
--openapiv2_opt json_names_for_fields=false \
"${file}"
# Generate the JSON/WASM client stubs.
falafel=$(which falafel)
pkg="poolrpc"
manual_import="github.com/lightninglabs/pool/auctioneerrpc"
opts="package_name=$pkg,manual_import=$manual_import,api_prefix=1,js_stubs=1"
protoc -I/usr/local/include -I. -I.. \
--plugin=protoc-gen-custom=$falafel\
--custom_out=. \
--custom_opt="$opts" \
trader.proto
# We've got a bit of a weird setup with the same RPC package being split over
# two golang packages. We can't really fix this without breaking a lot so it's
# easier to fix stuff explicitly here.
sed -i 's/BatchSnapshotRequest/auctioneerrpc.BatchSnapshotRequest/g' trader.pb.json.go
sed -i 's/BatchSnapshotsRequest/auctioneerrpc.BatchSnapshotsRequest/g' trader.pb.json.go
done
}
# format formats the *.proto files with the clang-format utility.
function format() {
find . -name "*.proto" -print0 | xargs -0 clang-format --style=file -i
}
# Compile and format the poolrpc package.
pushd poolrpc
format
generate
popd
pushd auctioneerrpc
format
generate no-rest
popd