diff --git a/README.md b/README.md index 1359df2..a6287fd 100644 --- a/README.md +++ b/README.md @@ -48,18 +48,26 @@ dokku postgres:create lolipop # official postgres image export POSTGRES_IMAGE="postgres" export POSTGRES_IMAGE_VERSION="9.5.4" +dokku postgres:create lolipop # you can also specify custom environment # variables to start the postgres service # in semi-colon separated forma export POSTGRES_CUSTOM_ENV="USER=alpha;HOST=beta" - -# create a postgres service dokku postgres:create lolipop # get connection information as follows dokku postgres:info lolipop +# you can also retrieve a specific piece of service info via flags +dokku postgres:info lolipop --config-dir +dokku postgres:info lolipop --data-dir +dokku postgres:info lolipop --dsn +dokku postgres:info lolipop --exposed-ports +dokku postgres:info lolipop --links +dokku postgres:info lolipop --status +dokku postgres:info lolipop --version + # a postgres service can be linked to a # container this will use native docker # links via the docker-options plugin diff --git a/functions b/functions index e082d80..0dc90e5 100755 --- a/functions +++ b/functions @@ -139,13 +139,37 @@ service_exposed_ports() { } service_info() { - local SERVICE="$1" + local SERVICE="$1" INFO_FLAG="$2" local SERVICE_ROOT="$PLUGIN_DATA_ROOT/$SERVICE" local SERVICE_URL=$(service_url "$SERVICE") + local PORT_FILE="$SERVICE_ROOT/PORT" + local flag key valid_flags - echo " DSN: $SERVICE_URL" - echo " Config dir: $SERVICE_ROOT/config" - echo " Data dir: $SERVICE_ROOT/data" + local flag_map=( + "--config-dir: ${SERVICE_ROOT}/config" + "--data-dir: ${SERVICE_ROOT}/data" + "--dsn: ${SERVICE_URL}" + "--exposed-ports: $(service_exposed_ports "$SERVICE")" + "--links: $(service_linked_apps "$SERVICE")" + "--status: $(service_status "$SERVICE")" + "--version: $(service_version "$SERVICE")" + ) + if [[ -z "$INFO_FLAG" ]]; then + dokku_log_info2 "Container Information" + for flag in "${flag_map[@]}"; do + key="$(echo "${flag#--}" | cut -f1 -d' ' | tr - ' ')" + dokku_log_verbose "$(printf "%-20s %-25s" "${key^}" "${flag#*: }")" + done + else + local match=false + for flag in "${flag_map[@]}"; do + valid_flags="${valid_flags} $(echo "$flag" | cut -d':' -f1)" + if [[ "$flag" == "${INFO_FLAG}:"* ]]; then + echo "${flag#*: }" && match=true + fi + done + [[ "$match" == "true" ]] || dokku_log_fail "Invalid flag passed, valid flags:${valid_flags}" + fi } service_link() { @@ -435,7 +459,7 @@ service_import() { SERVICE_ROOT="$PLUGIN_DATA_ROOT/$SERVICE" SERVICE_NAME="$(get_service_name "$SERVICE")" DATABASE_NAME="$(get_database_name "$SERVICE")" - PASSWORD=$(cat "$SERVICE_ROOT/PASSWORD") + PASSWORD="$(cat "$SERVICE_ROOT/PASSWORD")" if [[ -t 0 ]]; then dokku_log_fail "No data provided on stdin." diff --git a/subcommands/info b/subcommands/info index 42ee99b..d4d1507 100755 --- a/subcommands/info +++ b/subcommands/info @@ -7,11 +7,11 @@ source "$(dirname "$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)")/functions" postgres-info-cmd() { declare desc="print the connection information" local cmd="$PLUGIN_COMMAND_PREFIX:info" argv=("$@"); [[ ${argv[0]} == "$cmd" ]] && shift 1 - declare SERVICE="$1" + declare SERVICE="$1" INFO_FLAG="$2" [[ -z "$SERVICE" ]] && dokku_log_fail "Please specify a name for the service" verify_service_name "$SERVICE" - service_info "$SERVICE" + service_info "$SERVICE" "$INFO_FLAG" } postgres-info-cmd "$@" diff --git a/tests/service_info.bats b/tests/service_info.bats index df4cb19..e10d02c 100755 --- a/tests/service_info.bats +++ b/tests/service_info.bats @@ -21,14 +21,46 @@ teardown() { @test "($PLUGIN_COMMAND_PREFIX:info) success" { run dokku "$PLUGIN_COMMAND_PREFIX:info" l - password="$(cat "$PLUGIN_DATA_ROOT/l/PASSWORD")" - assert_contains "${lines[*]}" "DSN: postgres://postgres:$password@dokku-postgres-l:5432/l" + local password="$(cat "$PLUGIN_DATA_ROOT/l/PASSWORD")" + assert_contains "${lines[*]}" "postgres://postgres:$password@dokku-postgres-l:5432/l" } @test "($PLUGIN_COMMAND_PREFIX:info) replaces underscores by dash in hostname" { dokku "$PLUGIN_COMMAND_PREFIX:create" test_with_underscores run dokku "$PLUGIN_COMMAND_PREFIX:info" test_with_underscores - password="$(cat "$PLUGIN_DATA_ROOT/test_with_underscores/PASSWORD")" - assert_contains "${lines[*]}" "DSN: postgres://postgres:$password@dokku-postgres-test-with-underscores:5432/test_with_underscores" + local password="$(cat "$PLUGIN_DATA_ROOT/test_with_underscores/PASSWORD")" + assert_contains "${lines[*]}" "postgres://postgres:$password@dokku-postgres-test-with-underscores:5432/test_with_underscores" dokku --force "$PLUGIN_COMMAND_PREFIX:destroy" test_with_underscores } + +@test "($PLUGIN_COMMAND_PREFIX:info) success with flag" { + run dokku "$PLUGIN_COMMAND_PREFIX:info" l --dsn + local password="$(cat "$PLUGIN_DATA_ROOT/l/PASSWORD")" + assert_output "postgres://postgres:$password@dokku-postgres-l:5432/l" + + run dokku "$PLUGIN_COMMAND_PREFIX:info" l --config-dir + assert_success + + run dokku "$PLUGIN_COMMAND_PREFIX:info" l --data-dir + assert_success + + run dokku "$PLUGIN_COMMAND_PREFIX:info" l --dsn + assert_success + + run dokku "$PLUGIN_COMMAND_PREFIX:info" l --exposed-ports + assert_success + + run dokku "$PLUGIN_COMMAND_PREFIX:info" l --links + assert_success + + run dokku "$PLUGIN_COMMAND_PREFIX:info" l --status + assert_success + + run dokku "$PLUGIN_COMMAND_PREFIX:info" l --version + assert_success +} + +@test "($PLUGIN_COMMAND_PREFIX:info) error when invalid flag" { + run dokku "$PLUGIN_COMMAND_PREFIX:info" l --invalid-flag + assert_failure +} diff --git a/tests/test_helper.bash b/tests/test_helper.bash index fc2107d..fac5903 100644 --- a/tests/test_helper.bash +++ b/tests/test_helper.bash @@ -47,6 +47,14 @@ assert_success() { fi } +assert_failure() { + if [[ "$status" -eq 0 ]]; then + flunk "expected failed exit status" + elif [[ "$#" -gt 0 ]]; then + assert_output "$1" + fi +} + assert_exists() { if [ ! -f "$1" ]; then flunk "expected file to exist: $1"