From b8d0f5751a5b6cfbc5f8d6ad3a050a846fd8391b Mon Sep 17 00:00:00 2001
From: Mauro Lacy <maurolacy@users.noreply.github.com>
Date: Fri, 18 Oct 2024 13:06:38 +0200
Subject: [PATCH] F/add local bcd (#54)

Adds local bcd instance docker image
---
 Makefile                              |   7 +-
 Dockerfile => contrib/images/bcd      |   0
 contrib/images/local-bcd/Dockerfile   |  53 ++++++++++++
 contrib/images/local-bcd/setup-bcd.sh | 120 ++++++++++++++++++++++++++
 contrib/images/local-bcd/wrapper.sh   |  25 ++++++
 5 files changed, 203 insertions(+), 2 deletions(-)
 rename Dockerfile => contrib/images/bcd (100%)
 create mode 100644 contrib/images/local-bcd/Dockerfile
 create mode 100755 contrib/images/local-bcd/setup-bcd.sh
 create mode 100755 contrib/images/local-bcd/wrapper.sh

diff --git a/Makefile b/Makefile
index 309f0bb..1851306 100644
--- a/Makefile
+++ b/Makefile
@@ -17,8 +17,11 @@ build-linux-static:
 	$(MAKE) -C demo build-linux-static
 
 build-docker:
-	$(DOCKER) build --tag babylonlabs-io/bcd -f Dockerfile \
-		$(shell git rev-parse --show-toplevel)
+	$(DOCKER) build --tag babylonlabs-io/bcd -f contrib/images/bcd/Dockerfile .
+
+# A local bcd image with users and funds for testing / development / integration purposes
+build-bcd:
+	$(DOCKER) build --tag babylonlabs-io/local-bcd -f contrib/images/local-bcd/Dockerfile .
 
 ########################################
 ### Testing
diff --git a/Dockerfile b/contrib/images/bcd
similarity index 100%
rename from Dockerfile
rename to contrib/images/bcd
diff --git a/contrib/images/local-bcd/Dockerfile b/contrib/images/local-bcd/Dockerfile
new file mode 100644
index 0000000..c7f04b5
--- /dev/null
+++ b/contrib/images/local-bcd/Dockerfile
@@ -0,0 +1,53 @@
+FROM debian:bullseye-slim AS build-env
+
+RUN apt-get update && apt-get install -y git make gcc wget
+
+WORKDIR /work
+
+ARG TARGETARCH
+
+# Download and install Go
+ENV GOLANG_VERSION=1.21.4
+RUN wget -q https://golang.org/dl/go${GOLANG_VERSION}.linux-${TARGETARCH}.tar.gz && \
+    tar -C /usr/local -xzf go${GOLANG_VERSION}.linux-${TARGETARCH}.tar.gz && \
+    rm go${GOLANG_VERSION}.linux-${TARGETARCH}.tar.gz
+# Set Go environment variables
+ENV PATH=/usr/local/go/bin:$PATH
+ENV GOPATH=/go
+ENV PATH=$GOPATH/bin:$PATH
+
+ENV GO111MODULE=on
+
+# Install bcd
+COPY . /work
+RUN make install
+
+FROM debian:bullseye-slim AS run
+
+RUN apt-get update && apt-get install -y bash curl jq wget
+
+# Install libraries
+# Cosmwasm - Download correct libwasmvm version
+COPY --from=build-env /work/demo/go.mod /tmp
+RUN WASMVM_VERSION=$(grep github.com/CosmWasm/wasmvm /tmp/go.mod | cut -d' ' -f2) && \
+    wget https://github.com/CosmWasm/wasmvm/releases/download/$WASMVM_VERSION/libwasmvm.$(uname -m).so \
+    -O /lib/libwasmvm.$(uname -m).so && \
+    # verify checksum
+    wget https://github.com/CosmWasm/wasmvm/releases/download/$WASMVM_VERSION/checksums.txt -O /tmp/checksums.txt && \
+    sha256sum /lib/libwasmvm.$(uname -m).so | grep $(cat /tmp/checksums.txt | grep libwasmvm.$(uname -m) | cut -d ' ' -f 1)
+RUN rm -f /tmp/go.mod
+
+# Install binaries
+COPY --from=build-env /go/bin/bcd /usr/bin/bcd
+
+WORKDIR /bcd
+COPY --from=build-env /work/contrib/images/local-bcd/wrapper.sh /bcd
+COPY --from=build-env /work/contrib/images/local-bcd/setup-bcd.sh /bcd
+COPY --from=build-env /work/tests/testdata/babylon_contract.wasm /bcd
+COPY --from=build-env /work/tests/testdata/btc_staking.wasm /bcd
+
+ENV CONSUMER_CONF=/data/bcd
+
+ENTRYPOINT ["/bcd/wrapper.sh"]
+CMD []
+STOPSIGNAL SIGTERM
diff --git a/contrib/images/local-bcd/setup-bcd.sh b/contrib/images/local-bcd/setup-bcd.sh
new file mode 100755
index 0000000..8ef4611
--- /dev/null
+++ b/contrib/images/local-bcd/setup-bcd.sh
@@ -0,0 +1,120 @@
+#!/bin/bash
+
+display_usage() {
+	echo "Missing parameters. Please check if all parameters were specified."
+	echo "Usage: setup-bcd.sh [CHAIN_ID] [CHAIN_DIR] [RPC_PORT] [P2P_PORT] [PROFILING_PORT] [GRPC_PORT] [BABYLON_CONTRACT_CODE_DIR] [BTCSTAKING_CONTRACT_CODE_DIR] [INSTANTIATING_CFG]"
+	echo "Example: setup-bcd.sh test-chain-id ./data 26657 26656 6060 9090 ./babylon_contract.wasm '{"btc_confirmation_depth":1,"checkpoint_finalization_timeout":2,"network":"Regtest","babylon_tag":"bbn0", "notify_cosmos_zone":false, "btc_staking_code_id":2}'"
+	exit 1
+}
+
+BINARY=bcd
+DENOM=stake
+BASEDENOM=ustake
+KEYRING=--keyring-backend="test"
+SILENT=1
+
+redirect() {
+	if [ "$SILENT" -eq 1 ]; then
+		"$@" >/dev/null 2>&1
+	else
+		"$@"
+	fi
+}
+
+if [ "$#" -lt "8" ]; then
+	display_usage
+	exit 1
+fi
+
+CHAINID=$1
+CHAINDIR=$2
+RPCPORT=$3
+P2PPORT=$4
+PROFPORT=$5
+GRPCPORT=$6
+BABYLON_CONTRACT_CODE_DIR=$7
+BTCSTAKING_CONTRACT_CODE_DIR=$8
+INSTANTIATING_CFG=$9
+
+# ensure the binary exists
+if ! command -v $BINARY &>/dev/null; then
+	echo "$BINARY could not be found"
+	exit
+fi
+
+# Delete chain data from old runs
+echo "Deleting $CHAINDIR/$CHAINID folders..."
+rm -rf $CHAINDIR/$CHAINID &>/dev/null
+rm $CHAINDIR/$CHAINID.log &>/dev/null
+
+echo "Creating $BINARY instance: home=$CHAINDIR | chain-id=$CHAINID | p2p=:$P2PPORT | rpc=:$RPCPORT | profiling=:$PROFPORT | grpc=:$GRPCPORT"
+
+# Add dir for chain, exit if error
+if ! mkdir -p $CHAINDIR/$CHAINID 2>/dev/null; then
+	echo "Failed to create chain folder. Aborting..."
+	exit 1
+fi
+# Build genesis file incl account for passed address
+coins="100000000000$DENOM,100000000000$BASEDENOM"
+delegate="100000000000$DENOM"
+
+redirect $BINARY --home $CHAINDIR/$CHAINID --chain-id $CHAINID init $CHAINID
+sleep 1
+$BINARY --home $CHAINDIR/$CHAINID keys add validator $KEYRING --output json > $CHAINDIR/$CHAINID/validator_seed.json 2>&1
+sleep 1
+$BINARY --home $CHAINDIR/$CHAINID keys add user $KEYRING --output json > $CHAINDIR/$CHAINID/key_seed.json 2>&1
+sleep 1
+redirect $BINARY --home $CHAINDIR/$CHAINID genesis add-genesis-account $($BINARY --home $CHAINDIR/$CHAINID keys $KEYRING show user -a) $coins
+sleep 1
+redirect $BINARY --home $CHAINDIR/$CHAINID genesis add-genesis-account $($BINARY --home $CHAINDIR/$CHAINID keys $KEYRING show validator -a) $coins
+sleep 1
+redirect $BINARY --home $CHAINDIR/$CHAINID genesis gentx validator $delegate $KEYRING --chain-id $CHAINID
+sleep 1
+redirect $BINARY --home $CHAINDIR/$CHAINID genesis collect-gentxs
+sleep 1
+
+# Set proper defaults and change ports
+echo "Change settings in config.toml and genesis.json files..."
+sed -i 's#"tcp://127.0.0.1:26657"#"tcp://0.0.0.0:'"$RPCPORT"'"#g' $CHAINDIR/$CHAINID/config/config.toml
+sed -i 's#"tcp://0.0.0.0:26656"#"tcp://0.0.0.0:'"$P2PPORT"'"#g' $CHAINDIR/$CHAINID/config/config.toml
+sed -i 's#"localhost:6060"#"localhost:'"$PROFPORT"'"#g' $CHAINDIR/$CHAINID/config/config.toml
+sed -i 's/timeout_commit = "5s"/timeout_commit = "1s"/g' $CHAINDIR/$CHAINID/config/config.toml
+sed -i 's/max_body_bytes = 1000000/max_body_bytes = 1000000000/g' $CHAINDIR/$CHAINID/config/config.toml
+sed -i 's/minimum-gas-prices = ""/minimum-gas-prices = "0.00001ustake"/g' $CHAINDIR/$CHAINID/config/app.toml
+sed -i 's/timeout_propose = "3s"/timeout_propose = "1s"/g' $CHAINDIR/$CHAINID/config/config.toml
+sed -i 's/index_all_keys = false/index_all_keys = true/g' $CHAINDIR/$CHAINID/config/config.toml
+sed -i 's#"tcp://0.0.0.0:1317"#"tcp://0.0.0.0:1318"#g' $CHAINDIR/$CHAINID/config/app.toml # ensure port is not conflicted with Babylon
+sed -i 's/"bond_denom": "stake"/"bond_denom": "'"$DENOM"'"/g' $CHAINDIR/$CHAINID/config/genesis.json
+# sed -i '' 's#index-events = \[\]#index-events = \["message.action","send_packet.packet_src_channel","send_packet.packet_sequence"\]#g' $CHAINDIR/$CHAINID/config/app.toml
+
+## Script for getting contract addresses
+## TODO(euphrates): pass a gov prop on setting the Babylon / BTC staking contract addresses
+# babylonContractAddr=$(bcd query wasm list-contract-by-code 1 -o json | jq -r '.contracts[0]')
+# btcStakingContractAddr=$(bcd query wasm list-contract-by-code 2 -o json | jq -r '.contracts[0]')
+# echo "babylonContractAddr is $babylonContractAddr"
+# echo "btcStakingContractAddr is $btcStakingContractAddr"
+
+# update contract address in genesis
+babylonContractAddr=bbnc14hj2tavq8fpesdwxxcu44rty3hh90vhujrvcmstl4zr3txmfvw9syx25zf
+btcStakingContractAddr=bbnc1nc5tatafv6eyq7llkr2gv50ff9e22mnf70qgjlv737ktmt4eswrqgn0kq0
+sed -i 's/"babylon_contract_address": ""/"babylon_contract_address": "'"$babylonContractAddr"'"/g' $CHAINDIR/$CHAINID/config/genesis.json
+sed -i 's/"btc_staking_contract_address": ""/"btc_staking_contract_address": "'"$btcStakingContractAddr"'"/g' $CHAINDIR/$CHAINID/config/genesis.json
+
+# Start
+echo "Starting $BINARY..."
+$BINARY --home $CHAINDIR/$CHAINID start --pruning=nothing --grpc-web.enable=false --grpc.address="0.0.0.0:$GRPCPORT" --log_level trace --trace --log_format 'plain' 2>&1 | tee $CHAINDIR/$CHAINID.log &
+sleep 20
+
+# upload contract code
+echo "Uploading babylon contract code $BABYLON_CONTRACT_CODE_DIR..."
+$BINARY --home $CHAINDIR/$CHAINID tx wasm store "$BABYLON_CONTRACT_CODE_DIR" $KEYRING --from user --chain-id $CHAINID --gas 20000000000 --gas-prices 0.01ustake --node http://localhost:$RPCPORT -y
+sleep 10
+
+# upload contract code
+echo "Uploading btcstaking contract code $BTCSTAKING_CONTRACT_CODE_DIR..."
+$BINARY --home $CHAINDIR/$CHAINID tx wasm store "$BTCSTAKING_CONTRACT_CODE_DIR" $KEYRING --from user --chain-id $CHAINID --gas 20000000000 --gas-prices 0.01ustake --node http://localhost:$RPCPORT -y
+sleep 10
+
+# Echo the command with expanded variables
+echo "Instantiating contract with code $BABYLON_CONTRACT_CODE_DIR..."
+$BINARY --home $CHAINDIR/$CHAINID tx wasm instantiate 1 "$INSTANTIATING_CFG" --admin=$(bcd --home $CHAINDIR/$CHAINID keys show user --keyring-backend test -a) --label "v0.0.1" $KEYRING --from user --chain-id $CHAINID --gas 20000000000 --gas-prices 0.001ustake --node http://localhost:$RPCPORT -y --amount 100000stake
\ No newline at end of file
diff --git a/contrib/images/local-bcd/wrapper.sh b/contrib/images/local-bcd/wrapper.sh
new file mode 100755
index 0000000..01b8c2c
--- /dev/null
+++ b/contrib/images/local-bcd/wrapper.sh
@@ -0,0 +1,25 @@
+#!/usr/bin/env sh
+
+# 0. Define configuration
+CONSUMER_KEY="bcd-key"
+CONSUMER_CHAIN_ID="bcd-test"
+
+# 1. Create a bcd testnet with Babylon contract
+./setup-bcd.sh $CONSUMER_CHAIN_ID $CONSUMER_CONF 26657 26656 6060 9090 ./babylon_contract.wasm ./btc_staking.wasm '{
+    "network": "regtest",
+    "babylon_tag": "01020304",
+    "btc_confirmation_depth": 1,
+    "checkpoint_finalization_timeout": 2,
+    "notify_cosmos_zone": false,
+    "btc_staking_code_id": 2,
+    "consumer_name": "Test Consumer",
+    "consumer_description": "Test Consumer Description"
+}'
+
+sleep 10
+
+CONTRACT_ADDRESS=$(bcd query wasm list-contract-by-code 1 | grep bbnc | cut -d' ' -f2)
+CONTRACT_PORT="wasm.$CONTRACT_ADDRESS"
+echo "bcd started. Status of bcd node:"
+bcd status
+echo "Contract port: $CONTRACT_PORT"