cat << EOF

while [[ $# -gt 0 ]]; do

case $1 in

# Docker-specific options (handled by this script)

--shell)

INTERACTIVE_SHELL=true

shift

;;

--test)

RUN_TESTS=true

shift

;;

--image)

DOCKER_IMAGE="$2"

shift 2

;;

--help|-h)

usage

exit 0

;;

# Options with arguments (pass through to build.sh)

-j)

BUILD_ARGS="$BUILD_ARGS $1 $2"

shift 2

;;

# All other options (pass through to build.sh)

*)

BUILD_ARGS="$BUILD_ARGS $1"

shift

;;

esac

done

if ! command -v docker &> /dev/null; then

log_error "Docker is not installed or not in PATH"

exit 1

fi

if ! docker info &> /dev/null; then

log_error "Docker daemon is not running or not accessible"

exit 1

fi

log_info "Checking Docker image: $DOCKER_IMAGE"

if ! docker image inspect "$DOCKER_IMAGE" &> /dev/null; then

log_info "Pulling Docker image: $DOCKER_IMAGE"

if ! docker pull "$DOCKER_IMAGE"; then

log_error "Failed to pull Docker image: $DOCKER_IMAGE"

exit 1

fi

else

log_info "Docker image already available: $DOCKER_IMAGE"

fi

# Base Docker run options

DOCKER_RUN_OPTS=(

--rm

--name "$CONTAINER_NAME"

--volume "$REPO_ROOT:/workspace"

--volume "$HOME/.m2:/tmp/.m2"

--workdir /workspace

--user "$(id -u):$(id -g)"

--env "HOME=/tmp"

--env "STARROCKS_HOME=/workspace"

--env "STARROCKS_THIRDPARTY=/var/local/thirdparty"

--env "MAVEN_OPTS=-Dmaven.repo.local=/tmp/.m2/repository"

)

# Add any additional Docker options from environment

if [[ -n "${DOCKER_BUILD_OPTS:-}" ]]; then

read -ra ADDITIONAL_OPTS <<< "$DOCKER_BUILD_OPTS"

DOCKER_RUN_OPTS+=("${ADDITIONAL_OPTS[@]}")

fi

# Interactive mode for shell

if [[ "$INTERACTIVE_SHELL" == "true" ]]; then

DOCKER_RUN_OPTS+=(--interactive --tty)

fi

log_info "Starting interactive shell in StarRocks dev environment"

log_info "Container: $CONTAINER_NAME"

log_info "Image: $DOCKER_IMAGE"

log_info "Workspace: /workspace (mounted from $REPO_ROOT)"

docker run "${DOCKER_RUN_OPTS[@]}" "$DOCKER_IMAGE" /bin/bash

log_info "Checking for protobuf version compatibility..."

# BE protobuf/thrift outputs are now CMake-managed in the active BE build tree.

# If those outputs exist, let CMake's own stamp/tool-version logic decide whether

# regeneration is needed instead of forcing the legacy gensrc cleanup path.

local be_codegen_dir

be_codegen_dir=$(find be -path 'be/build*/gensrc/gen_cpp' -type d \

if [[ -n "$be_codegen_dir" ]]; then

log_info "Found CMake-managed BE protobuf outputs in $be_codegen_dir - skipping legacy cleanup check"

return 1 # No need to clean

fi

# Fall back to the legacy gensrc path for workflows that still materialize

# protobuf C++ outputs there.

if [[ ! -d "gensrc/build/gen_cpp" ]] || [[ -z "$(find gensrc/build/gen_cpp -name "*.pb.cc" 2>/dev/null)" ]]; then

log_info "No reusable protobuf outputs found - skipping cleanup"

return 1 # No need to clean

fi

# Get protobuf version from Docker container

local container_protoc_version

container_protoc_version=$(docker run --rm "${DOCKER_RUN_OPTS[@]}" "$DOCKER_IMAGE" bash -c "/var/local/thirdparty/installed/bin/protoc --version 2>/dev/null | cut -d' ' -f2" 2>/dev/null || echo "unknown")

# Try to detect version mismatch by checking for common error patterns in a test compilation

local test_result

test_result=$(docker run --rm "${DOCKER_RUN_OPTS[@]}" "$DOCKER_IMAGE" bash -c "

if [[ "$test_result" == *"PROTOBUF_MISMATCH"* ]] || [[ "$test_result" == *"incompatible version"* ]] || [[ "$test_result" == *"generated_message_tctable_decl.h"* ]]; then

log_warning "Protobuf version mismatch detected (container: $container_protoc_version)"

log_warning "Generated files were created with different protobuf version"

return 0 # Need to clean

fi

log_success "Protobuf version compatibility check passed (container: $container_protoc_version)"

return 1 # No need to clean

if check_protobuf_version_mismatch; then

log_info "Cleaning generated files due to protobuf version mismatch..."

docker run "${DOCKER_RUN_OPTS[@]}" "$DOCKER_IMAGE" bash -c "make -C gensrc clean || true"

docker run "${DOCKER_RUN_OPTS[@]}" "$DOCKER_IMAGE" bash -c "rm -rf fe/fe-core/target/generated-sources || true"

log_success "Generated files cleaned - will regenerate with correct protobuf version"

else

log_info "Skipping gensrc clean - no protobuf version mismatch detected"

fi

local build_cmd="./build.sh"

if [[ -n "$BUILD_ARGS" ]]; then

build_cmd="$build_cmd$BUILD_ARGS"

fi

log_info "Starting build in Docker container"

log_info "Container: $CONTAINER_NAME"

log_info "Image: $DOCKER_IMAGE"

log_info "Build command: $build_cmd"

# Smart cleaning - only clean when protobuf version mismatch detected

clean_generated_files_if_needed

# Run the build

if docker run "${DOCKER_RUN_OPTS[@]}" "$DOCKER_IMAGE" bash -c "$build_cmd"; then

log_success "Build completed successfully!"

# Show output directory contents

if [[ -d "$REPO_ROOT/output" ]]; then

log_info "Build artifacts in output directory:"

ls -la "$REPO_ROOT/output/"

fi

return 0

else

log_error "Build failed!"

return 1

fi

log_info "Running tests in Docker container"

# Determine which tests to run based on build arguments

local run_fe_tests=false

local run_be_tests=false

if [[ "$BUILD_ARGS" == *"--fe"* ]] && [[ "$BUILD_ARGS" != *"--be"* ]]; then

run_fe_tests=true

elif [[ "$BUILD_ARGS" == *"--be"* ]] && [[ "$BUILD_ARGS" != *"--fe"* ]]; then

run_be_tests=true

else

# Default: run both if no specific component specified

run_fe_tests=true

run_be_tests=true

fi

if [[ "$run_fe_tests" == "true" ]]; then

log_info "Running Frontend tests..."

docker run "${DOCKER_RUN_OPTS[@]}" "$DOCKER_IMAGE" bash -c "./run-fe-ut.sh || echo 'FE tests completed with some failures'"

fi

if [[ "$run_be_tests" == "true" ]]; then

log_info "Running Backend tests..."

docker run "${DOCKER_RUN_OPTS[@]}" "$DOCKER_IMAGE" bash -c "./run-be-ut.sh || echo 'BE tests completed with some failures'"

fi

if docker ps -q -f name="$CONTAINER_NAME" &> /dev/null; then

log_info "Cleaning up container: $CONTAINER_NAME"

docker stop "$CONTAINER_NAME" &> /dev/null || true

fi

# Set up cleanup trap

trap cleanup EXIT

# Parse arguments

parse_args "$@"

# Check prerequisites

check_docker

pull_image

setup_docker_run

# Execute requested action

if [[ "$INTERACTIVE_SHELL" == "true" ]]; then

run_shell

elif [[ "$RUN_TESTS" == "true" ]]; then

run_build && run_tests

else

# Default build args if none specified

if [[ -z "$BUILD_ARGS" ]]; then

BUILD_ARGS=" --fe --be"

fi

run_build

fi