codex-factory/scripts/build-environment.sh

#!/bin/bash
usage() {
    cat << USAGE >&2
USAGE:
    $ build-environment.sh [PARAMETERS]
PARAMETERS:
    --build-base-bee                          The base bee image will be built from source code
    --base-bee-commit-hash=string             the source code commit hash of the base bee; Default: HEAD; Dependency: --build-base-bee
    --gen-traffic                             Generate traffic before bee image commit.
    --gen-traffic-upload-node=string          The traffic will be generated on the node under the given API URL. Default: http://locahost:1635; Dependency: --gen-traffic
    --gen-traffic-upload-node-debug=string    The traffic will be generated on the node under the given API URL. Default: http://locahost:1635; Dependency: --gen-traffic
    --gen-traffic-checker-node-debug=string   The incoming cheques will be checked on the node under the given Debug API URL. Default: http://localhost:11635; Dependency: --gen-traffic
    --cheques-count=number                    This amount of cheques is intended to be generated by the traffic gen. Default: 1; Dependency: --gen-traffic
USAGE
    exit 1
}

echoerr() {
     >&2 echo "$@"
}

build_bee() {
    # Clone source code
    BEE_SOURCE_PATH=$MY_PATH/../bee
    if [ -d "$BEE_SOURCE_PATH" ] ; then
        rm -rf "$BEE_SOURCE_PATH"
    fi
    mkdir "$BEE_SOURCE_PATH" && cd "$BEE_SOURCE_PATH" || exit 1
    git init
    git remote add origin https://github.com/ethersphere/bee.git
    git fetch origin --depth=1 "$COMMIT_HASH"
    git reset --hard FETCH_HEAD
    # Build bee and make docker image
    export BEE_VERSION=${COMMIT_HASH::7}-commit
    make binary
    echo "Bee image will be built with version: $BEE_VERSION"
    docker build . -t ethersphere/bee:$BEE_VERSION
    cd "$MY_PATH" || exit 1
    # Set build image tag so that other terminal session can retrieve
    "$MY_PATH/utils/build-image-tag.sh" set "$BEE_VERSION"
}

# the image label for the produced bee images in case of state commitment
stateful_image_label() {
    echo "$BEE_VERSION-stateful"
}

MY_PATH=$(dirname "$0")
MY_PATH=$( cd "$MY_PATH" && pwd )
COMMIT_HASH=HEAD
BUILD_BASE_BEE=false
GEN_TRAFFIC=false
GEN_TRAFFIC_CHECKER_NODE_DEBUG="http://localhost:1635"
GEN_TRAFFIC_UPLOAD_NODE="http://localhost:11633"
GEN_TRAFFIC_UPLOAD_NODE_DEBUG="http://localhost:11635"
CHEQUES_COUNT=1
# Bee version here means the base bee version on which the images will be built
BEE_VERSION=$("$MY_PATH/utils/env-variable-value.sh" BEE_VERSION)
SUPPORTED_WORKER_N=4

# handle passed options
while [ $# -gt 0 ]
do
    case "$1" in
        --build-base-bee)
        BUILD_BASE_BEE=true
        shift 1
        ;;
        --base-bee-commit-hash=*)
        COMMIT_HASH="${1#*=}"
        shift 1
        ;;
        --gen-traffic)
        GEN_TRAFFIC=true
        shift 1
        ;;
        --cheques-count=*)
        CHEQUES_COUNT=${1#*=}
        shift 1
        ;;
        --gen-traffic-upload-node=*)
        GEN_TRAFFIC_UPLOAD_NODE="${1#*=}"
        shift 1
        ;;
        --gen-traffic-upload-node-debug=*)
        GEN_TRAFFIC_UPLOAD_NODE_DEBUG="${1#*=}"
        shift 1
        ;;
        --gen-traffic-checker-node-debug=*)
        GEN_TRAFFIC_CHECKER_NODE_DEBUG="${1#*=}"
        shift 1
        ;;
        *)
        echoerr "Unknown argument: $1"
        usage
        ;;
    esac
done

# cleanup for start from an empty state
"$MY_PATH/bee-cleanup.sh"

if $BUILD_BASE_BEE ; then 
    build_bee
fi
"$MY_PATH/network.sh"
"$MY_PATH/blockchain.sh"
npm run migrate:contracts
npm run supply
if $GEN_TRAFFIC ; then
    export STATE_COMMIT='true'
    echo "Bee image with special state will be commited... traffic generation is on."
    # give the permission to the bee user
    BEE_DIR_PATH="$MY_PATH/bee-data-dirs/"
    sudo chown 999:999 -R "$BEE_DIR_PATH"
    echo "Start Bee nodes so that traffic can be generated and commited to the images"
    "$MY_PATH/bee.sh" start --workers=$SUPPORTED_WORKER_N --detach
    echo "Generating traffic on Bee node $GEN_TRAFFIC_UPLOAD_NODE"
    echo "Run traffic generation until $CHEQUES_COUNT incoming cheques will arrive to node under Debug API $GEN_TRAFFIC_CHECKER_NODE_DEBUG"
    npm run gen:traffic -- "$CHEQUES_COUNT" "$GEN_TRAFFIC_CHECKER_NODE_DEBUG;$GEN_TRAFFIC_UPLOAD_NODE;$GEN_TRAFFIC_UPLOAD_NODE_DEBUG"
    echo "traffic has been generated, stop nodes before commit..."
    "$MY_PATH/bee.sh" stop
    docker container prune -f
fi
"$MY_PATH/bee-docker-build.sh"
"$MY_PATH/blockchain-docker-build.sh"