You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
118 lines
2.9 KiB
118 lines
2.9 KiB
#!/bin/bash
|
|
DEFAULT_PROTOC_GEN="gogofast"
|
|
DEFAULT_PROTOC="protoc"
|
|
KRATOS_DIR_NAME="github.com/bilibili/kratos"
|
|
USR_INCLUDE_DIR="/usr/local/include"
|
|
GOPATH=$GOPATH
|
|
if [[ -z $GOPATH ]]; then
|
|
GOPATH=${HOME}/go
|
|
fi
|
|
|
|
function _install_protoc() {
|
|
osname=$(uname -s)
|
|
echo "install protoc ..."
|
|
case $osname in
|
|
"Darwin" )
|
|
brew install protobuf
|
|
;;
|
|
*)
|
|
echo "unknown operating system, need install protobuf manual see: https://developers.google.com/protocol-buffers"
|
|
exit 1
|
|
;;
|
|
esac
|
|
}
|
|
|
|
function _install_protoc_gen() {
|
|
local protoc_gen=$1
|
|
case $protoc_gen in
|
|
"gofast" )
|
|
echo "install gofast from github.com/gogo/protobuf/protoc-gen-gofast"
|
|
go get github.com/gogo/protobuf/protoc-gen-gofast
|
|
;;
|
|
"gogofast" )
|
|
echo "install gogofast from github.com/gogo/protobuf/protoc-gen-gogofast"
|
|
go get github.com/gogo/protobuf/protoc-gen-gogofast
|
|
;;
|
|
"gogo" )
|
|
echo "install gogo from github.com/gogo/protobuf/protoc-gen-gogo"
|
|
go get github.com/gogo/protobuf/protoc-gen-gogo
|
|
;;
|
|
"go" )
|
|
echo "install protoc-gen-go from github.com/golang/protobuf"
|
|
go get github.com/golang/protobuf/{proto,protoc-gen-go}
|
|
;;
|
|
*)
|
|
echo "can't install protoc-gen-${protoc_gen} automatic !"
|
|
exit 1;
|
|
;;
|
|
esac
|
|
}
|
|
|
|
function _find_kratos_dir() {
|
|
local kratos_dir_name=$1
|
|
local current_dir="$GOPATH/src/$kratos_dir_name"
|
|
if [[ ! -d $current_dir ]]; then
|
|
go get -u $kratos_dir_name
|
|
fi
|
|
echo $current_dir
|
|
}
|
|
|
|
function _esc_string() {
|
|
echo $(echo "$1" | sed 's_/_\\/_g')
|
|
}
|
|
|
|
function _run_protoc() {
|
|
local proto_dir=$1
|
|
local proto_files=$(find $proto_dir -maxdepth 1 -name "*.proto")
|
|
if [[ -z $proto_files ]]; then
|
|
return
|
|
fi
|
|
local protoc_cmd="$PROTOC -I$PROTO_PATH --${PROTOC_GEN}_out=plugins=grpc:${GOPATH}/src ${proto_files}"
|
|
echo $protoc_cmd
|
|
$protoc_cmd
|
|
}
|
|
|
|
if [[ -z $PROTOC ]]; then
|
|
PROTOC=${DEFAULT_PROTOC}
|
|
which $PROTOC
|
|
if [[ "$?" -ne "0" ]]; then
|
|
_install_protoc
|
|
fi
|
|
fi
|
|
if [[ -z $PROTOC_GEN ]]; then
|
|
PROTOC_GEN=${DEFAULT_PROTOC_GEN}
|
|
which protoc-gen-$PROTOC_GEN
|
|
if [[ "$?" -ne "0" ]]; then
|
|
_install_protoc_gen $PROTOC_GEN
|
|
fi
|
|
fi
|
|
|
|
KRATOS_DIR=$(_find_kratos_dir $KRATOS_DIR_NAME)
|
|
if [[ "$?" != "0" ]]; then
|
|
echo "can't find kratos directoy"
|
|
exit 1
|
|
fi
|
|
|
|
KRATOS_PARENT=$(dirname $KRATOS_DIR)
|
|
|
|
if [[ -z $PROTO_PATH ]]; then
|
|
PROTO_PATH=$GOPATH/src:$KRATOS_PARENT:$USR_INCLUDE_DIR
|
|
else
|
|
PROTO_PATH=$GOPATH/src:$PROTO_PATH:$KRATOS_PARENT:$USR_INCLUDE_DIR
|
|
fi
|
|
|
|
if [[ ! -z $1 ]]; then
|
|
cd $1
|
|
fi
|
|
TARGET_DIR=$(pwd)
|
|
|
|
# switch to $GOPATH/src
|
|
cd $GOPATH/src
|
|
echo "switch workdir to $GOPATH/src"
|
|
|
|
DIRS=$(find $TARGET_DIR -type d)
|
|
|
|
for dir in $DIRS; do
|
|
echo "run protoc in $dir"
|
|
_run_protoc $dir
|
|
done
|
|
|