mirror of https://github.com/hashicorp/consul
142 lines
3.6 KiB
Bash
Executable File
142 lines
3.6 KiB
Bash
Executable File
#!/bin/bash
|
|
SCRIPT_NAME="$(basename ${BASH_SOURCE[0]})"
|
|
pushd $(dirname ${BASH_SOURCE[0]}) > /dev/null
|
|
SCRIPT_DIR=$(pwd)
|
|
pushd ../.. > /dev/null
|
|
SOURCE_DIR=$(pwd)
|
|
popd > /dev/null
|
|
pushd ../functions > /dev/null
|
|
FN_DIR=$(pwd)
|
|
popd > /dev/null
|
|
popd > /dev/null
|
|
|
|
source "${SCRIPT_DIR}/functions.sh"
|
|
|
|
function usage {
|
|
cat <<-EOF
|
|
Usage: ${SCRIPT_NAME} [<options ...>] <proto filepath>
|
|
|
|
Description:
|
|
This script will build generate the Go files from protobuf files. In addition to
|
|
just running the correct protoc generator it will also fixup build tags in the
|
|
generated code.
|
|
|
|
Options:
|
|
--import-replace Replace imports of google types with those from the gogo/protobuf repo.
|
|
|
|
--generator Which generator to use: gogo, gofast etc. Defaults to gofast.
|
|
|
|
--grpc Enable the gRPC plugin
|
|
|
|
-h | --help Print this help text.
|
|
EOF
|
|
}
|
|
|
|
function err_usage {
|
|
err "$1"
|
|
err ""
|
|
err "$(usage)"
|
|
}
|
|
|
|
function main {
|
|
local generator=gofast
|
|
local -i grpc=0
|
|
local -i imp_replace=0
|
|
local proto_path=
|
|
|
|
while test $# -gt 0
|
|
do
|
|
case "$1" in
|
|
-h | --help )
|
|
usage
|
|
return 0
|
|
;;
|
|
generator )
|
|
if test -z "$2"
|
|
then
|
|
err_usage "ERROR: option -g/--generator requires an argument"
|
|
return 1
|
|
fi
|
|
|
|
generator="$2"
|
|
shift 2
|
|
;;
|
|
--grpc )
|
|
grpc=1
|
|
shift
|
|
;;
|
|
--import-replace )
|
|
imp_replace=1
|
|
shift
|
|
;;
|
|
* )
|
|
proto_path="$1"
|
|
shift
|
|
;;
|
|
esac
|
|
done
|
|
|
|
if test -z "${proto_path}"
|
|
then
|
|
err_usage "ERROR: No proto file specified"
|
|
return 1
|
|
fi
|
|
|
|
local gogo_proto_path=$(go list -f '{{ .Dir }}' -m github.com/gogo/protobuf)
|
|
local gogo_proto_mod_path=$(sed -e 's,\(.*\)github.com.*,\1,' <<< "${gogo_proto_path}")
|
|
|
|
local gogo_proto_imp_replace="Mgoogle/protobuf/timestamp.proto=github.com/gogo/protobuf/types"
|
|
gogo_proto_imp_replace="${gogo_proto_imp_replace},Mgoogle/protobuf/duration.proto=github.com/gogo/protobuf/types"
|
|
gogo_proto_imp_replace="${gogo_proto_imp_replace},Mgoogle/protobuf/empty.proto=github.com/gogo/protobuf/types"
|
|
gogo_proto_imp_replace="${gogo_proto_imp_replace},Mgoogle/api/annotations.proto=github.com/gogo/googleapis/google/api"
|
|
gogo_proto_imp_replace="${gogo_proto_imp_replace},Mgoogle/protobuf/field_mask.proto=github.com/gogo/protobuf/types"
|
|
|
|
local proto_go_path=${proto_path%%.proto}.pb.go
|
|
|
|
local go_proto_out=""
|
|
local sep=""
|
|
if is_set "${grpc}"
|
|
then
|
|
go_proto_out="plugins=grpc"
|
|
sep=","
|
|
fi
|
|
|
|
if is_set "${imp_replace}"
|
|
then
|
|
go_proto_out="${go_proto_out}${sep}${gogo_proto_imp_replace}"
|
|
fi
|
|
|
|
if test -n "${go_proto_out}"
|
|
then
|
|
go_proto_out="${go_proto_out}:"
|
|
fi
|
|
|
|
local -i ret=0
|
|
status_stage "Generating ${proto_go_path} from ${proto_path}"
|
|
debug_run protoc \
|
|
-I="$(dirname ${proto_path})" \
|
|
-I="${gogo_proto_path}/protobuf" \
|
|
-I="${gogo_proto_path}" \
|
|
-I="${gogo_proto_mod_path}" \
|
|
--${generator}_out="${go_proto_out}$(dirname ${proto_path})" \
|
|
"${proto_path}"
|
|
if test $? -ne 0
|
|
then
|
|
err "Failed to generate ${proto_go_path} from ${proto_path}"
|
|
return 1
|
|
fi
|
|
|
|
BUILD_TAGS=$(sed -e '/^[:space:]*$/,$d' < "${proto_path}" | grep '// +build')
|
|
if test -n "${BUILD_TAGS}"
|
|
then
|
|
echo -e "${BUILD_TAGS}\n" >> "${proto_go_path}.new"
|
|
cat "${proto_go_path}" >> "${proto_go_path}.new"
|
|
mv "${proto_go_path}.new" "${proto_go_path}"
|
|
fi
|
|
|
|
return 0
|
|
}
|
|
|
|
main "$@"
|
|
exit $?
|