Skip to content

Commit

Permalink
feat: Add sentry-admin.sh tool (#2594)
Browse files Browse the repository at this point in the history
  • Loading branch information
azaslavsky authored Nov 21, 2023
1 parent b413e01 commit f00e6a9
Show file tree
Hide file tree
Showing 3 changed files with 95 additions and 4 deletions.
6 changes: 2 additions & 4 deletions _integration-test/ensure-backup-restore-works.sh
Original file line number Diff line number Diff line change
Expand Up @@ -10,9 +10,7 @@ echo "Creating backup..."
# to group and owner. Instead, try creating the empty file and then give everyone write access to the backup file
touch $(pwd)/sentry/backup.json
chmod 666 $(pwd)/sentry/backup.json
# Command here matches exactly what we have in our docs https://develop.sentry.dev/self-hosted/backup/#backup
$dc run -v $(pwd)/sentry:/sentry-data/backup --rm -T -e SENTRY_LOG_LEVEL=CRITICAL web export global /sentry-data/backup/backup.json
# Check to make sure there is content in the file
SENTRY_DOCKER_IO_DIR=$(pwd)/sentry /bin/bash $(pwd)/sentry-admin.sh export global /sentry-admin/backup.json
if [ ! -s "$(pwd)/sentry/backup.json" ]; then
echo "Backup file is empty"
exit 1
Expand All @@ -33,6 +31,6 @@ source install/set-up-and-migrate-database.sh
$dc up -d

echo "Importing backup..."
$dc run --rm -T web import global /etc/sentry/backup.json
SENTRY_DOCKER_IO_DIR=$(pwd)/sentry /bin/bash $(pwd)/sentry-admin.sh import global /sentry-admin/backup.json

rm $(pwd)/sentry/backup.json
27 changes: 27 additions & 0 deletions _integration-test/ensure-sentry-admin-works.sh
Original file line number Diff line number Diff line change
@@ -0,0 +1,27 @@
#!/usr/bin/env bash
set -ex

source install/_lib.sh
source install/dc-detect-version.sh

echo "${_group}Test that sentry-admin works..."

echo "Global help documentation..."

global_help_doc=$(/bin/bash --help)
if ! echo "$global_help_doc" | grep -q "^Usage: ./sentry-admin.sh"; then
echo "Assertion failed: Incorrect binary name in global help docs"
exit 1
fi
if ! echo "$global_help_doc" | grep -q "SENTRY_DOCKER_IO_DIR"; then
echo "Assertion failed: Missing SENTRY_DOCKER_IO_DIR global help doc"
exit 1
fi

echo "Command-specific help documentation..."

command_help_doc=$(/bin/bash permissions --help)
if ! echo "$command_help_doc" | grep -q "^Usage: ./sentry-admin.sh permissions"; then
echo "Assertion failed: Incorrect binary name in command-specific help docs"
exit 1
fi
66 changes: 66 additions & 0 deletions sentry-admin.sh
Original file line number Diff line number Diff line change
@@ -0,0 +1,66 @@
#!/bin/bash

# Detect docker and platform state.
source install/dc-detect-version.sh
source install/detect-platform.sh

# Define the Docker volume mapping.
VOLUME_MAPPING="${SENTRY_DOCKER_IO_DIR:-$HOME/.sentry/sentry-admin}:/sentry-admin"

# Custom help text paragraphs
HELP_TEXT_SUFFIX="
All file paths are relative to the 'web' docker container, not the host environment. To pass files
to/from the host system for commands that require it ('execfile', 'export', 'import', etc), you may
specify a 'SENTRY_DOCKER_IO_DIR' environment variable to mount a volume for file IO operations into
the host filesystem. The default value of 'SENTRY_DOCKER_IO_DIR' points to '~/.sentry/sentry-admin'
on the host filesystem. Commands that write files should write them to the '/sentry-admin' in the
'web' container (ex: './sentry-admin.sh export global /sentry-admin/my-export.json').
"

# Actual invocation that runs the command in the container.
invocation() {
$dc run -v "$VOLUME_MAPPING" --rm -T -e SENTRY_LOG_LEVEL=CRITICAL web "$@"
}

# Function to modify lines starting with `Usage: sentry` to say `Usage: ./sentry-admin.sh` instead.
rename_sentry_bin_in_help_output() {
local output="$1"
local help_prefix="$2"
local usage_seen=false

output=$(invocation "$@")

echo -e "\n\n"

while IFS= read -r line; do
if [[ $line == "Usage: sentry"* ]] && [ "$usage_seen" = false ]; then
echo -e "\n\n"
echo "${line/sentry/./sentry-admin.sh}"
echo "$help_prefix"
usage_seen=true
else
if [[ $line == "Options:"* ]] && [ -n "$1" ]; then
echo "$help_prefix"
fi
echo "$line"
fi
done <<<"$output"
}

# Check for the user passing ONLY the '--help' argument - we'll add a special prefix to the output.
if { [ "$1" = "help" ] || [ "$1" = "--help" ]; } && [ "$#" -eq 1 ]; then
rename_sentry_bin_in_help_output "$(invocation "$@")" "$HELP_TEXT_SUFFIX"
exit 0
fi

# Check for '--help' in other contexts.
for arg in "$@"; do
if [ "$arg" = "--help" ]; then
rename_sentry_bin_in_help_output "$(invocation "$@")"
exit 0
fi
done

# Help has not been requested - go ahead and execute the command.
echo -e "\n\n"
invocation "$@"

0 comments on commit f00e6a9

Please sign in to comment.