generate.sh

#!/bin/bash
# Copyright 2019 dfuse Platform Inc.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#      http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

ROOT="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"

# Protobuf definitions
PROTO=${PROTO:-"$ROOT/../proto"}
PROTO_BSTREAM=${PROTO_BSTREAM:-"$ROOT/../bstream/proto"}

function main() {
  checks

  current_dir="`pwd`"
  trap "cd \"$current_dir\"" EXIT
  pushd "$ROOT/pb" &> /dev/null

  generate "sf/blockmeta/v1/blockmeta.proto"
  generate "sf/fluxdb/v1/fluxdb.proto"
  generate "sf/graphql/v1/graphql.proto"
  generate "sf/headinfo/v1/headinfo.proto"
  generate "sf/merger/v1/merger.proto"
  generate "sf/search/v1/search.proto"
  generate "sf/firehose/v1/firehose.proto"
  generate "sf/firehose/v2/firehose.proto"

  echo "generate.sh - `date` - `whoami`" > $ROOT/last_generate.txt
  echo "streamingfast/proto revision: `GIT_DIR=$PROTO/.git git rev-parse HEAD`" >> $ROOT/last_generate.txt
}

# usage:
# - generate <protoPath>
# - generate <protoBasePath/> [<file.proto> ...]
function generate() {
    base=""
    if [[ "$#" -gt 1 ]]; then
      base="$1"; shift
    fi

    for file in "$@"; do
      protoc -I$PROTO -I$PROTO_BSTREAM \
        --go_out=. --go_opt=paths=source_relative \
        --go-grpc_out=. --go-grpc_opt=paths=source_relative,require_unimplemented_servers=false \
         $base$file
    done
}

function checks() {
  # The old `protoc-gen-go` did not accept any flags. Just using `protoc-gen-go --version` in this
  # version waits forever. So we pipe some wrong input to make it exit fast. This, in the new version
  # which supports `--version` correctly print the version anyway and discard the standard input
  # so it's good with both version.
  result=`printf "" | protoc-gen-go --version 2>&1 | grep -Eo "v[1-9]\.(2[7-9]|[3-9][0-9])\.[0-9]+"`
  if [[ "$result" == "" ]]; then
    echo "Your version of 'protoc-gen-go' (at `which protoc-gen-go`) is not recent enough."
    echo ""
    echo "To fix your problem, perform this command (assumes you have Golang 1.17+):"
    echo ""
    echo "  go install google.golang.org/protobuf/cmd/protoc-gen-go@v1.27.1"
    echo ""
    echo "If everything is working as expected, the command:"
    echo ""
    echo "  protoc-gen-go --version"
    echo ""
    echo "Should print 'protoc-gen-go v1.27.1' (or another more recent version)"
    exit 1
  fi

  result=`printf "" | protoc-gen-go-grpc --version 2>&1 | grep -Eo "1\.[2-9]\.[0-9]+"`
  if [[ "$result" == "" ]]; then
    echo "Your version of 'protoc-gen-go-grpc' (at `which protoc-gen-go-grpc`) is not recent enough."
    echo ""
    echo "To fix your problem, perform this command (assumes you have Golang 1.17+):"
    echo ""
    echo "  go install google.golang.org/grpc/cmd/protoc-gen-go-grpc@v1.2.0"
    echo ""
    echo "If everything is working as expected, the command:"
    echo ""
    echo "  protoc-gen-go-grpc --version"
    echo ""
    echo "Should print 'protoc-gen-go-grpc 1.2.0' (or another more recent version)"
    exit 1
  fi
}


main "$@"