#!/usr/bin/env bash
SCRIPTS_DIR=$(dirname "$0")
protoc=protoc

while getopts "p:h" arg; do
  case $arg in
  p)
    protoc=$(readlink -f "${OPTARG}")
    ;;
    h) # help
    echo "

parameter:
-p: protoc path default("protoc")
-h: help

usage:
./build.sh -p protoc [-h]
                "
    exit 0
    ;;
  ?)
    echo "ERROR! unknown argument"
    exit 1
    ;;
  esac
done


ROOT_DIR=$SCRIPTS_DIR/..
source $SCRIPTS_DIR/common.sh

push $SCRIPTS_DIR/..
pop

PROGRAM=$(basename "$0")
GOPATH=$(go env GOPATH)

if [ -z $GOPATH ]; then
    printf "Error: the environment variable GOPATH is not set, please set it before running %s\n" $PROGRAM > /dev/stderr
    exit 1
fi

export PATH=${GOPATH}/bin:$PATH
echo `which protoc-gen-go`

GO_PREFIX_PATH=github.com/czs007/suvlim/pkg


function collect() {
    file=$(basename $1)
    base_name=$(basename $file ".proto")
    mkdir -p ../pkg/$base_name
    if [ -z $GO_OUT_M ]; then
        GO_OUT_M="M$file=$GO_PREFIX_PATH/$base_name"
    else
        GO_OUT_M="$GO_OUT_M,M$file=$GO_PREFIX_PATH/$base_name"
    fi
}

# Although eraftpb.proto is copying from raft-rs, however there is no
# official go code ship with the crate, so we need to generate it manually.
cd ${ROOT_DIR}/proto

PB_FILES=()
GRPC_FILES=("message.proto" "pdpb.proto" "metapb.proto")
ALL_FILES=("${PB_FILES[@]}")
ALL_FILES+=("${GRPC_FILES[@]}")

for file in ${ALL_FILES[@]}
    do
    collect $file
done

ret=0

function replace(){
    cd ../pkg/$base_name
    sed_inplace -E 's/import fmt \"fmt\"//g' *.pb*.go
    sed_inplace -E 's/import io \"io\"//g' *.pb*.go
    sed_inplace -E 's/import math \"math\"//g' *.pb*.go
    #goimports -w *.pb*.go
    cd ../../proto
}

function gen_pb() {
    base_name=$(basename $1 ".proto")
    $protoc -I.:.   --go_out=plugins=$GO_OUT_M:../pkg/$base_name $1 || ret=$?
    #replace
  }

function gen_grpc() {
    base_name=$(basename $1 ".proto")
    $protoc -I.:.   --go_out=plugins=grpc,$GO_OUT_M:../pkg/${base_name} $1 || ret=$?
    #replace
  }

for file in ${PB_FILES[@]}
    do
    echo $file
    gen_pb $file
done

for file in ${GRPC_FILES[@]}
    do
    echo $file
    gen_grpc $file
done


exit $ret