terminusdb-container

#!/bin/bash

#
# CONFIG
#

# Load ENV file

if [[ $1 != "nop" && -z "$TERMINUSDB_IGNORE_ENV" && -f ENV ]]; then
  set -o allexport
  # shellcheck disable=SC1091
  # shellcheck source=ENV
  source "$(pwd)/ENV"
  set +o allexport
fi

# Container
TERMINUSDB_QUICKSTART_CONTAINER=${TERMINUSDB_CONTAINER:-terminusdb-server}
TERMINUSDB_QUICKSTART_REPOSITORY=${TERMINUSDB_REPOSITORY:-terminusdb/terminusdb-server}

# Version
TERMINUSDB_QUICKSTART_TAG=${TERMINUSDB_TAG:-v4.0.1}

# Volumes
TERMINUSDB_QUICKSTART_STORAGE=${TERMINUSDB_STORAGE:-terminusdb_storage}
TERMINUSDB_QUICKSTART_STORAGE_VOLUME=/app/terminusdb/storage

# Server
TERMINUSDB_QUICKSTART_PORT=${TERMINUSDB_PORT:-6363}
TERMINUSDB_QUICKSTART_PASS=${TERMINUSDB_PASS:-root}
TERMINUSDB_QUICKSTART_AUTOLOGIN=${TERMINUSDB_AUTOLOGIN:-true}

# HTTPS
TERMINUSDB_QUICKSTART_HTTPS_ENABLED=${TERMINUSDB_HTTPS_ENABLED:-true}
TERMINUSDB_QUICKSTART_SET_HTTPS="-e TERMINUSDB_HTTPS_ENABLED=\"$TERMINUSDB_QUICKSTART_HTTPS_ENABLED\""
if [[ $TERMINUSDB_QUICKSTART_HTTPS_ENABLED == true ]]; then
  TERMINUSDB_QUICKSTART_SCHEME="https"
  if [[ -n $TERMINUSDB_SSL_CERT ]]; then
    TERMINUSDB_QUICKSTART_SET_HTTPS+=$'\n'"-e TERMINUSDB_SSL_CERT=\"$TERMINUSDB_SSL_CERT\""
  fi
  if [[ -n $TERMINUSDB_SSL_CERT_KEY ]]; then
    TERMINUSDB_QUICKSTART_SET_HTTPS+=$'\n'"-e TERMINUSDB_SSL_CERT_KEY=\"$TERMINUSDB_SSL_CERT_KEY\""
  fi
else
  TERMINUSDB_QUICKSTART_SCHEME="http"
fi

if [[ $TERMINUSDB_QUICKSTART_HTTPS_ENABLED == true && $TERMINUSDB_QUICKSTART_AUTOLOGIN == false ]]; then
  TERMINUSDB_QUICKSTART_SERVER_IP=${TERMINUSDB_SERVER_IP:-0.0.0.0}
else
  TERMINUSDB_QUICKSTART_SERVER_IP="127.0.0.1"
fi

TERMINUSDB_QUICKSTART_CONSOLE=${TERMINUSDB_CONSOLE:-$TERMINUSDB_QUICKSTART_SCHEME://127.0.0.1:$TERMINUSDB_QUICKSTART_PORT/}

if [[ -n $TERMINUSDB_CONSOLE_BASE_URL ]]; then
      TERMINUSDB_QUICKSTART_CONSOLE_BASE_URL="$TERMINUSDB_CONSOLE_BASE_URL"
else
  case "$TERMINUSDB_QUICKSTART_TAG" in
    "rc")
      TERMINUSDB_QUICKSTART_CONSOLE_BASE_URL="https://dl.bintray.com/terminusdb/terminusdb/rc"
    ;;
    "canary")
      TERMINUSDB_QUICKSTART_CONSOLE_BASE_URL="https://dl.bintray.com/terminusdb/terminusdb/canary"
    ;;
    "dev")
      TERMINUSDB_QUICKSTART_CONSOLE_BASE_URL="https://dl.bintray.com/terminusdb/terminusdb/dev"
    ;;
    *)
      TERMINUSDB_QUICKSTART_CONSOLE_BASE_URL="https://dcm.ist/console/${TERMINUSDB_QUICKSTART_TAG}"
    ;;
  esac
fi

if [[  "$(uname)" == "Darwin" || "$TERM_PROGRAM" == "Apple_Terminal" ]]; then
  TERMINUSDB_QUICKSTART_DOCKER_DEFAULT="docker"
elif command -v sudo >/dev/null; then
  TERMINUSDB_QUICKSTART_DOCKER_DEFAULT="sudo docker"
else
  TERMINUSDB_QUICKSTART_DOCKER_DEFAULT="docker"
fi

TERMINUSDB_QUICKSTART_DOCKER=${TERMINUSDB_DOCKER:-$TERMINUSDB_QUICKSTART_DOCKER_DEFAULT}

function remove_old_volumes {
  read -p "Are you sure? [y/N] " -r -n 1
  printf "\n"    # (optional) move to a new line
  if [[ $REPLY =~ ^[Yy]$ ]]; then
      if ! $TERMINUSDB_QUICKSTART_DOCKER volume rm terminus_config terminus_storage; then
          printf "Could not remove volumes. Are you running an existing container of TerminusDB? Stop it first.\n"
      fi
  else
      exit 2
  fi
}

function version_check {
  # The old TerminusDB had a prefix.db file needed, the new one
  # does not as prefixes are handled in a different way. Therefore
  # it is safe to assume that an existing storage volume with the
  # prefix.db is a user running an old version of TerminusDB
  if ! $TERMINUSDB_QUICKSTART_DOCKER volume inspect terminus_storage > /dev/null 2>&1; then
      return 0
  fi

  if $TERMINUSDB_QUICKSTART_DOCKER run --rm -v terminus_storage:/home/ alpine cat /home/prefix.db > /dev/null 2>&1; then
      printf "\
You are running an old version of terminusdb-server. The new %s version
uses a very different data structure which breaks backwards compatibility.
Continuing will remove your existing databases.\n" "$TERMINUSDB_QUICKSTART_TAG"
      remove_old_volumes
  fi
}

if [[ -z "$TERMINUSDB_LABEL_FILE" ]]; then
  TERMINUSDB_QUICKSTART_SET_LABEL_FILE=""
 else
  TERMINUSDB_QUICKSTART_SET_LABEL_FILE="--label-file \"$TERMINUSDB_LABEL_FILE\""
fi

if [[ -z "$TERMINUSDB_NETWORK" ]]; then
  TERMINUSDB_QUICKSTART_SET_NETWORK=""
 else
  TERMINUSDB_QUICKSTART_SET_NETWORK="--network \"$TERMINUSDB_NETWORK\""
fi

if [[ -z "$TERMINUSDB_LOCAL" ]]; then
  TERMINUSDB_QUICKSTART_MOUNT_LOCAL=""
 else
  TERMINUSDB_QUICKSTART_MOUNT_LOCAL="-v \"$TERMINUSDB_LOCAL\":/app/local_files"
fi

#
# FUNCTIONS
#

# Run platform check
_platform_check () {
  PLATFORM_ARCHITECTURE=$(uname -m)
  if [[ $PLATFORM_ARCHITECTURE != "x86_64" ]]; then
      printf "\
There are no Docker images for other CPU architectures than x86_64 yet.
You can try to manually compile the components of TerminusDB by following the
instructions on the terminusdb-server repository although other architectures are not
officially supported yet.
https://github.com/terminusdb/terminusdb-server/blob/master/docs/BUILD.md \n"
      exit 4
  fi
}

_usage () {
  printf "\
USAGE:
  terminusdb-container [COMMAND]

  help        show usage
  run         run container
  stop        stop container
  console     launch console in web browser
  attach      attach to prolog shell
  exec        execeute a command inside the container
  rm          remove volumes\n"
}

_confirm() {
  read -r -p "Are you sure? [y/N] " RESPONSE
  case "$RESPONSE" in
      [yY][eE][sS]|[yY]) 
          return 0
      ;;
      *)
          return 1
      ;;
  esac
  printf "\n"
}

_run () {
  if [[ $TERMINUSDB_QUICKSTART_TAG == "latest" \
        || $TERMINUSDB_QUICKSTART_TAG == "dev" \
        || $TERMINUSDB_QUICKSTART_TAG == "canary" \
        || $TERMINUSDB_QUICKSTART_TAG == "rc" ]]; then
    $TERMINUSDB_QUICKSTART_DOCKER image rm "$TERMINUSDB_QUICKSTART_REPOSITORY:$TERMINUSDB_QUICKSTART_TAG" 2>/dev/null
  fi
  eval "$TERMINUSDB_QUICKSTART_DOCKER" run -d -it --rm \
    "--name \"$TERMINUSDB_QUICKSTART_CONTAINER\" -p \"$TERMINUSDB_QUICKSTART_SERVER_IP:$TERMINUSDB_QUICKSTART_PORT:6363\"" \
    "-v \"$TERMINUSDB_QUICKSTART_STORAGE:$TERMINUSDB_QUICKSTART_STORAGE_VOLUME:rw\"" \
    "$TERMINUSDB_QUICKSTART_MOUNT_LOCAL" \
    "$TERMINUSDB_QUICKSTART_SET_HTTPS" \
    "$TERMINUSDB_QUICKSTART_SET_LABEL_FILE" \
    "$TERMINUSDB_QUICKSTART_SET_NETWORK" \
    "-e TERMINUSDB_SERVER_PORT=6363" \
    "-e TERMINUSDB_CONSOLE_BASE_URL=\"$TERMINUSDB_QUICKSTART_CONSOLE_BASE_URL\"" \
    "-e TERMINUSDB_ADMIN_PASS=\"$TERMINUSDB_QUICKSTART_PASS\"" \
    "-e TERMINUSDB_AUTOLOGIN=\"$TERMINUSDB_QUICKSTART_AUTOLOGIN\"" \
    "$TERMINUSDB_QUICKSTART_REPOSITORY:$TERMINUSDB_QUICKSTART_TAG"
}

_stop () {
  $TERMINUSDB_QUICKSTART_DOCKER stop "$TERMINUSDB_QUICKSTART_CONTAINER"
}

_restart () {
  $TERMINUSDB_QUICKSTART_DOCKER stop "$TERMINUSDB_QUICKSTART_CONTAINER"
}

_attach () {
  $TERMINUSDB_QUICKSTART_DOCKER attach "$TERMINUSDB_QUICKSTART_CONTAINER"
}

_rm () {
 if _confirm; then
    $TERMINUSDB_QUICKSTART_DOCKER volume rm "$TERMINUSDB_QUICKSTART_STORAGE"
 fi
}

_exec () {
  if [[ -n "$TERMINUSDB_RUN_CMD" ]]; then
    eval "$TERMINUSDB_QUICKSTART_DOCKER" exec -it \
      "$TERMINUSDB_QUICKSTART_CONTAINER" \
      "$TERMINUSDB_RUN_CMD"
  fi
}

_rm () {
 if _confirm; then
   $TERMINUSDB_QUICKSTART_DOCKER volume rm "$TERMINUSDB_QUICKSTART_STORAGE"
 fi
}

_console () {

if command -v python >/dev/null; then
  python -m webbrowser "$TERMINUSDB_QUICKSTART_CONSOLE"
else
  printf "\nopen %s in your browser\n" "$TERMINUSDB_QUICKSTART_CONSOLE"
fi
}

_serve () {
if command -v python >/dev/null; then
  python -m http.server 8888
else
  printf "\npython is required for this feature\n"
fi
}

_platform_check
#
# PROCESS ARGS
#

if [[ $1 != "nop" ]]; then
  printf "\n"
fi

if [[ -n "$1" ]]; then
  case "$1" in
    "help")
      _usage
    ;;
    "run")
      # _build
      version_check
      _run > /dev/null \
        && printf "terminusdb-server container started %s\n" "$TERMINUSDB_QUICKSTART_CONSOLE" \
        || printf "\nIs the container already running?\n"
    ;;
    "stop")
      _stop > /dev/null \
        && printf "terminusdb-server container stopped\n" \
        || printf "\nIs the container running?\n"
    ;;
    "console")
      _console \
        || printf "this command requires python\n"
    ;;
    "attach")
      printf "Ctrl+p Ctrl+q to detach\n\n"
      _attach
    ;;
    "rm")
      printf "This will delete storage volume\n"
      _rm \
        || printf "\nIs the container stopped?\n"
    ;;
    "exec")
      if [[ -n "$2" ]]; then
        # shellcheck disable=SC2124
        TERMINUSDB_RUN_CMD="${@:2}"
      fi
      printf "This will exec a command on terminusdb-server\n"
      _exec \
        || printf "\nIs the container stopped?\n"
    ;;
    "nop")
      true
    ;;
    "serve")
      _serve
    ;;
    *)
      printf "invalid command\n\n"
      _usage
    ;;
  esac
else
  _usage
fi

if [[ $1 != "nop" ]]; then
  printf "\n"
fi


# vim:ft=sh