124 lines
3.6 KiB
Bash
Executable File
124 lines
3.6 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
set -euo pipefail
|
|
|
|
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
|
REPO_ROOT="$(cd "${SCRIPT_DIR}/../../.." && pwd)"
|
|
API_DIR="${REPO_ROOT}/api"
|
|
PROTO_DIR="./proto"
|
|
PROTOC_BIN="${PROTOC:-protoc}"
|
|
|
|
# Ensure required tools exist
|
|
if ! command -v "${PROTOC_BIN}" >/dev/null 2>&1; then
|
|
echo "[proto] protoc binary not found" >&2; exit 1
|
|
fi
|
|
if ! command -v protoc-gen-go >/dev/null 2>&1; then
|
|
echo "[proto] protoc-gen-go is not installed" >&2; exit 1
|
|
fi
|
|
if ! command -v protoc-gen-go-grpc >/dev/null 2>&1; then
|
|
echo "[proto] protoc-gen-go-grpc is not installed" >&2; exit 1
|
|
fi
|
|
|
|
info() { printf '[proto] %s\n' "$1"; }
|
|
|
|
# Try to determine the include path for well-known types (WKT) without hard failing
|
|
PROTOC_INCLUDE=""
|
|
# Prefer known locations on Alpine/Debian-like images
|
|
for cand in /usr/include /usr/local/include; do
|
|
if [ -f "$cand/google/protobuf/descriptor.proto" ]; then
|
|
PROTOC_INCLUDE="$cand"
|
|
break
|
|
fi
|
|
done
|
|
|
|
# Build an include-array only when non-empty
|
|
PROTOC_I=()
|
|
if [ -n "${PROTOC_INCLUDE}" ]; then
|
|
PROTOC_I+=("-I=${PROTOC_INCLUDE}")
|
|
fi
|
|
# Always include our proto tree
|
|
PROTOC_I+=("-I=${PROTO_DIR}")
|
|
|
|
generate_go() {
|
|
local out_dir="$1"; shift
|
|
mkdir -p "${out_dir}"
|
|
"${PROTOC_BIN}" \
|
|
"${PROTOC_I[@]}" \
|
|
--go_out="${out_dir}" \
|
|
--go_opt=paths=source_relative \
|
|
"$@"
|
|
}
|
|
|
|
generate_go_with_grpc() {
|
|
local out_dir="$1"; shift
|
|
mkdir -p "${out_dir}"
|
|
"${PROTOC_BIN}" \
|
|
"${PROTOC_I[@]}" \
|
|
--go_out="${out_dir}" \
|
|
--go_opt=paths=source_relative \
|
|
--go-grpc_out="${out_dir}" \
|
|
--go-grpc_opt=paths=source_relative \
|
|
"$@"
|
|
}
|
|
|
|
clean_pb_files() {
|
|
local target="$1"
|
|
if [ -d "${target}" ]; then
|
|
find "${target}" -type f -name '*.pb.go' -delete
|
|
fi
|
|
}
|
|
|
|
cd "${API_DIR}"
|
|
|
|
# Messaging (top-level .proto files in ./proto)
|
|
mapfile -t messaging_protos < <(find "${PROTO_DIR}" -maxdepth 1 -type f -name '*.proto' | sort)
|
|
if [ "${#messaging_protos[@]}" -gt 0 ]; then
|
|
info "Compiling messaging protos"
|
|
rm -rf ./pkg/messaging/internal/generated
|
|
generate_go "./pkg/messaging/internal/generated" "${messaging_protos[@]}"
|
|
fi
|
|
|
|
# Common shared
|
|
mapfile -t common_protos < <(find "${PROTO_DIR}/common" -type f -name '*.proto' | sort)
|
|
if [ "${#common_protos[@]}" -gt 0 ]; then
|
|
info "Compiling common shared protos"
|
|
clean_pb_files "./pkg/proto"
|
|
generate_go "./pkg/proto" "${common_protos[@]}"
|
|
fi
|
|
|
|
# Ledger
|
|
if [ -f "${PROTO_DIR}/ledger/v1/ledger.proto" ]; then
|
|
info "Compiling ledger protos"
|
|
clean_pb_files "./pkg/proto/ledger"
|
|
generate_go_with_grpc "./pkg/proto/ledger/v1" "${PROTO_DIR}/ledger/v1/ledger.proto"
|
|
fi
|
|
|
|
# Oracle
|
|
if [ -f "${PROTO_DIR}/oracle/v1/oracle.proto" ]; then
|
|
info "Compiling oracle protos"
|
|
clean_pb_files "./pkg/proto/oracle"
|
|
generate_go_with_grpc "./pkg/proto/oracle/v1" "${PROTO_DIR}/oracle/v1/oracle.proto"
|
|
fi
|
|
|
|
# Chain gateway
|
|
if [ -f "${PROTO_DIR}/chain/gateway/v1/gateway.proto" ]; then
|
|
info "Compiling chain gateway protos"
|
|
clean_pb_files "./pkg/proto/chain/gateway"
|
|
generate_go_with_grpc "./pkg/proto/chain/gateway/v1" "${PROTO_DIR}/chain/gateway/v1/gateway.proto"
|
|
fi
|
|
|
|
# Payments orchestrator
|
|
if [ -f "${PROTO_DIR}/payments/orchestrator/v1/orchestrator.proto" ]; then
|
|
info "Compiling payments orchestrator protos"
|
|
clean_pb_files "./pkg/proto/payments/orchestrator"
|
|
generate_go_with_grpc "./pkg/proto/payments/orchestrator/v1" "${PROTO_DIR}/payments/orchestrator/v1/orchestrator.proto"
|
|
fi
|
|
|
|
# Billing fees
|
|
if [ -f "${PROTO_DIR}/billing/fees/v1/fees.proto" ]; then
|
|
info "Compiling billing fees protos"
|
|
clean_pb_files "./pkg/proto/billing/fees"
|
|
generate_go_with_grpc "./pkg/proto/billing/fees/v1" "${PROTO_DIR}/billing/fees/v1/fees.proto"
|
|
fi
|
|
|
|
info "Protobuf generation completed"
|