mirror of
https://github.com/SpecterOps/Nemesis
synced 2026-06-08 12:36:42 +00:00
Lee Christensen
55 lines
2.0 KiB
Bash
Executable File
55 lines
2.0 KiB
Bash
Executable File
#!/bin/bash
|
|
|
|
# Wrapper script to submit a job to the Nemesis CLI using Docker
|
|
# Added into the tools/ dir for convenience (near project root)
|
|
|
|
SCRIPT_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"
|
|
COMPOSE_DIR="$( dirname "$SCRIPT_DIR" )"
|
|
|
|
cd "${COMPOSE_DIR}"
|
|
|
|
# Check if docker-compose.yml exists
|
|
if [ ! -f docker-compose.yml ]; then
|
|
echo "Error: docker-compose.yml not found in ${COMPOSE_DIR}"
|
|
exit 1
|
|
fi
|
|
|
|
# Set dummy values for required variables if they're not already set
|
|
# This allows the CLI to run without requiring all the main stack variables
|
|
export GRAFANA_ADMIN_USER="${GRAFANA_ADMIN_USER:-dummy}"
|
|
export GRAFANA_ADMIN_PASSWORD="${GRAFANA_ADMIN_PASSWORD:-dummy}"
|
|
export MINIO_ROOT_USER="${MINIO_ROOT_USER:-dummy}"
|
|
export MINIO_ROOT_PASSWORD="${MINIO_ROOT_PASSWORD:-dummy}"
|
|
export RABBITMQ_USER="${RABBITMQ_USER:-dummy}"
|
|
export RABBITMQ_PASSWORD="${RABBITMQ_PASSWORD:-dummy}"
|
|
export POSTGRES_USER="${POSTGRES_USER:-dummy}"
|
|
export POSTGRES_PASSWORD="${POSTGRES_PASSWORD:-dummy}"
|
|
export JUPYTER_PASSWORD="${JUPYTER_PASSWORD:-dummy}"
|
|
export NEMESIS_URL="${NEMESIS_URL:-https://localhost:7443}"
|
|
|
|
# Parse arguments to handle volume mounting and CLI options
|
|
DOCKER_ARGS=""
|
|
SUBMIT_ARGS=""
|
|
CLI_OPTIONS=""
|
|
|
|
for arg in "$@"; do
|
|
# Check if argument is a file or directory path
|
|
if [[ -f "$arg" ]]; then
|
|
# It's a file
|
|
ABS_PATH=$(realpath "$arg")
|
|
DOCKER_ARGS="$DOCKER_ARGS -v $ABS_PATH:/data/$(basename "$ABS_PATH")"
|
|
SUBMIT_ARGS="$SUBMIT_ARGS /data/$(basename "$ABS_PATH")"
|
|
elif [[ -d "$arg" ]]; then
|
|
# It's a directory
|
|
ABS_PATH=$(realpath "$arg")
|
|
DOCKER_ARGS="$DOCKER_ARGS -v $ABS_PATH:/data/$(basename "$ABS_PATH")"
|
|
SUBMIT_ARGS="$SUBMIT_ARGS /data/$(basename "$ABS_PATH")"
|
|
else
|
|
# It's a CLI option (not a file or directory)
|
|
CLI_OPTIONS="$CLI_OPTIONS $arg"
|
|
fi
|
|
done
|
|
|
|
# Run the Docker command
|
|
docker compose run --rm $DOCKER_ARGS cli submit $SUBMIT_ARGS $CLI_OPTIONS | sed '/^\[+\] Building/d'
|