#!/bin/bash

set -e

RED='\033[0;31m'

GREEN='\033[0;32m'

YELLOW='\033[1;33m'

BLUE='\033[0;34m'

NC='\033[0m'

if [ -f .env ]; then

set -a

source .env

set +a

fi

OUTPUT_PATH=${OUTPUT_PATH:-"docker-local/configs"}

COMPOSE_FILE="$OUTPUT_PATH/../docker-compose-generated.yml"

INIT_SCRIPT="$OUTPUT_PATH/../init-bitcoin.sh"

log_info() {

echo -e "${BLUE}[INFO]${NC} $1"

}

log_success() {

echo -e "${GREEN}[SUCCESS]${NC} $1"

}

log_warning() {

echo -e "${YELLOW}[WARNING]${NC} $1"

}

log_error() {

echo -e "${RED}[ERROR]${NC} $1"

}

check_env_file() {

if [ ! -f .env ]; then

log_warning ".env file not found. Using default configuration."

log_info "To customize, copy .env.example to .env and edit it:"

echo " cp .env.example .env"

echo ""

fi

}

cmd_start() {

log_info "Starting Botanix local environment..."

echo ""

check_env_file

log_info "Step 1/3: Setting up environment and generating configurations..."

if bash scripts/setup-local-env.sh; then

log_success "Setup completed"

else

log_error "Setup failed"

exit 1

fi

echo ""

log_info "Step 2/3: Starting Docker services..."

if [ ! -f "$COMPOSE_FILE" ]; then

log_error "Compose file not found: $COMPOSE_FILE"

exit 1

fi

if docker compose -f "$COMPOSE_FILE" up -d; then

log_success "Docker services started"

else

log_error "Failed to start Docker services"

exit 1

fi

echo ""

log_info "Waiting for services to be ready..."

sleep 5

log_info "Step 3/3: Initializing Bitcoin node..."

if [ ! -f "$INIT_SCRIPT" ]; then

log_error "Init script not found: $INIT_SCRIPT"

exit 1

fi

if bash "$INIT_SCRIPT"; then

log_success "Bitcoin node initialized"

else

log_warning "Bitcoin initialization had issues (this might be expected if already initialized)"

fi

echo ""

log_success "Botanix local environment is up and running!"

echo ""

echo "Useful commands:"

echo " botanix-local services - View running services"

echo " botanix-local logs - View logs"

echo " botanix-local stop - Stop all services"

}

cmd_stop() {

log_info "Stopping Botanix local environment..."

if [ ! -f "$COMPOSE_FILE" ]; then

log_warning "Compose file not found: $COMPOSE_FILE"

log_info "Services might not be running or already stopped"

exit 0

fi

if docker compose -f "$COMPOSE_FILE" down; then

log_success "All services stopped"

else

log_error "Failed to stop services"

exit 1

fi

}

cmd_restart() {

log_info "Restarting Botanix local environment..."

cmd_stop

echo ""

cmd_start

}

cmd_clean() {

log_warning "This will remove all containers, volumes, and generated configurations!"

read -p "Are you sure? (yes/no): " -r

echo

if [[ ! $REPLY =~ ^[Yy][Ee][Ss]$ ]]; then

log_info "Clean cancelled"

exit 0

fi

log_info "Cleaning up Botanix local environment..."

if [ -f "$COMPOSE_FILE" ]; then

log_info "Stopping services..."

docker compose -f "$COMPOSE_FILE" down -v 2> /dev/null || true

fi

if [ -d "$OUTPUT_PATH" ]; then

log_info "Removing generated files in $OUTPUT_PATH..."

rm -rf "$OUTPUT_PATH"

fi

if [ -f "docker-local/docker-compose-generated.yml" ]; then

log_info "Removing docker-local/docker-compose-generated.yml..."

rm -rf "docker-local/docker-compose-generated.yml"

fi

if [ -f "docker-local/init-bitcoin.sh" ]; then

log_info "Removing docker-local/init-bitcoin.sh..."

rm -rf "docker-local/init-bitcoin.sh"

fi

PROJECT_PREFIX=${PROJECT_PREFIX:-"botanix"}

NETWORK_NAME="${PROJECT_PREFIX}-local"

if docker network inspect "$NETWORK_NAME" > /dev/null 2>&1; then

log_info "Removing docker network: $NETWORK_NAME..."

docker network rm "$NETWORK_NAME" 2> /dev/null || true

fi

log_success "Cleanup complete"

}

cmd_services() {

log_info "Botanix Local Services Status"

echo ""

if [ ! -f "$COMPOSE_FILE" ]; then

log_warning "Compose file not found. Services might not be set up yet."

echo "Run: botanix-local start"

exit 0

fi

docker compose -f "$COMPOSE_FILE" ps

}

cmd_logs() {

if [ ! -f "$COMPOSE_FILE" ]; then

log_error "Compose file not found. Services might not be running."

exit 1

fi

if [ -n "$1" ]; then

log_info "Showing logs for service: $1"

docker compose -f "$COMPOSE_FILE" logs -f "$1"

else

log_info "Showing logs for all services (Ctrl+C to exit)"

docker compose -f "$COMPOSE_FILE" logs -f

fi

}

cmd_exec() {

if [ -z "$1" ]; then

log_error "Service name required"

echo "Usage: botanix-local exec <service> [command]"

exit 1

fi

if [ ! -f "$COMPOSE_FILE" ]; then

log_error "Compose file not found. Services might not be running."

exit 1

fi

SERVICE=$1

shift

if [ $# -eq 0 ]; then

docker compose -f "$COMPOSE_FILE" exec "$SERVICE" sh

else

docker compose -f "$COMPOSE_FILE" exec "$SERVICE" "$@"

fi

}

cmd_bitcoin() {

log_info "Executing bitcoin-cli command..."

docker exec bitcoin-core bitcoin-cli -regtest -rpcuser=foo -rpcpassword=bar "$@"

}

cmd_help() {

cat << EOF

Botanix Local Development Environment

USAGE:

botanix-local <command> [options]

COMMANDS:

start Setup and start all services (setup → compose up → init bitcoin)

stop Stop all running services

restart Restart all services (stop → start)

clean Remove all containers, volumes, and generated files

services Show status of all services

logs Show logs for all services (or specific service)

Usage: botanix-local logs [service-name]

exec Execute a command in a running service

Usage: botanix-local exec <service> [command]

bitcoin Execute bitcoin-cli commands

Usage: botanix-local bitcoin <bitcoin-cli args>

help Show this help message

EXAMPLES:

# First time setup and start

botanix-local start

# View running services

botanix-local services

# View logs for a specific node

botanix-local logs poa-1

# Execute command in a container

botanix-local exec poa-1 sh

# Bitcoin CLI commands

botanix-local bitcoin getblockcount

botanix-local bitcoin getbalance

# Stop everything

botanix-local stop

# Clean up everything

botanix-local clean

CONFIGURATION:

Create a .env file from .env.example to customize settings:

cp .env.example .env

vim .env

Key settings:

NUM_NODES=3 # Number of nodes

MIN_SIGNERS=2 # Minimum signers

MAX_SIGNERS=3 # Maximum signers

OUTPUT_PATH=docker-local # Output directory

For more information, see README.md

EOF

}

main() {

if [ $# -eq 0 ]; then

cmd_help

exit 0

fi

case "$1" in

start)

cmd_start

;;

stop)

cmd_stop

;;

restart)

cmd_restart

;;

clean)

cmd_clean

;;

services | status | ps)

cmd_services

;;

logs)

shift

cmd_logs "$@"

;;

exec)

shift

cmd_exec "$@"

;;

bitcoin | btc)

shift

cmd_bitcoin "$@"

;;

help | --help | -h)

cmd_help

;;

*)

log_error "Unknown command: $1"

echo ""

cmd_help

exit 1

;;

esac

}

main "$@"