tekst-util

#!/usr/bin/env bash
# This script was generated by bashly 1.1.10 (https://bashly.dannyb.co)
# Modifying it manually is not recommended

# :wrapper.bash3_bouncer
if [[ "${BASH_VERSINFO:-0}" -lt 4 ]]; then
  printf "bash version 4 or higher is required\n" >&2
  exit 1
fi

# :command.master_script

# :command.version_command
version_command() {
  echo "$version"
}

# :command.usage
tekst_util_usage() {
  if [[ -n $long_usage ]]; then
    printf "tekst-util - Tekst Maintenance Utility Script\n"
    echo

  else
    printf "tekst-util - Tekst Maintenance Utility Script\n"
    echo

  fi

  printf "%s\n" "Usage:"
  printf "  tekst-util COMMAND\n"
  printf "  tekst-util [COMMAND] --help | -h\n"
  printf "  tekst-util --version | -v\n"
  echo
  # :command.usage_commands
  printf "%s\n" "Commands:"
  printf "  %s   Create DB backup (MongoDB service must be running)\n" "backup     "
  printf "  %s   Restore DB backup (MongoDB service must be running)\n" "restore    "
  printf "  %s   Create/update search index (MongoDB and ES services must be running)\n" "index      "
  printf "  %s   Run resource maintenance routines (MongoDB service must be running)\n" "maintenance"
  echo

  # :command.long_usage
  if [[ -n $long_usage ]]; then
    printf "%s\n" "Options:"

    # :command.usage_fixed_flags
    printf "  %s\n" "--help, -h"
    printf "    Show this help\n"
    echo
    printf "  %s\n" "--version, -v"
    printf "    Show version number\n"
    echo

  fi
}

# :command.usage
tekst_util_backup_usage() {
  if [[ -n $long_usage ]]; then
    printf "tekst-util backup - Create DB backup (MongoDB service must be running)\n"
    echo

  else
    printf "tekst-util backup - Create DB backup (MongoDB service must be running)\n"
    echo

  fi

  printf "%s\n" "Usage:"
  printf "  tekst-util backup [OPTIONS]\n"
  printf "  tekst-util backup --help | -h\n"
  echo

  # :command.long_usage
  if [[ -n $long_usage ]]; then
    printf "%s\n" "Options:"

    # :command.usage_flags
    # :flag.usage
    printf "  %s\n" "--container-name, -c CONTAINER"
    printf "    MongoDB service DB Docker container name (if using Docker)\n"
    echo

    # :flag.usage
    printf "  %s\n" "--file, -f PATH"
    printf "    File to write/read DB backup to/from\n"
    printf "    Default: /tmp/tekst-util.backup\n"
    echo

    # :flag.usage
    printf "  %s\n" "--mongodb-uri, -m URI"
    printf "    MongoDB URI (from inside the container if using Docker)\n"
    printf "    Default: mongodb://127.0.0.1:27017\n"
    echo

    # :command.usage_fixed_flags
    printf "  %s\n" "--help, -h"
    printf "    Show this help\n"
    echo

    # :command.usage_examples
    printf "%s\n" "Examples:"
    printf "  tekst-util backup\n"
    printf "  tekst-util backup -c tekst-util\n"
    printf "  tekst-util backup -c tekst-mongo -f /home/user/tekst-something.backup\n"
    printf "  tekst-util backup -m mongodb://usr:pwd@db.foo.com:27017/?authSource=admin\n"
    echo

  fi
}

# :command.usage
tekst_util_restore_usage() {
  if [[ -n $long_usage ]]; then
    printf "tekst-util restore - Restore DB backup (MongoDB service must be running)\n"
    echo

  else
    printf "tekst-util restore - Restore DB backup (MongoDB service must be running)\n"
    echo

  fi

  printf "%s\n" "Usage:"
  printf "  tekst-util restore [OPTIONS]\n"
  printf "  tekst-util restore --help | -h\n"
  echo

  # :command.long_usage
  if [[ -n $long_usage ]]; then
    printf "%s\n" "Options:"

    # :command.usage_flags
    # :flag.usage
    printf "  %s\n" "--container-name, -c CONTAINER"
    printf "    MongoDB service Docker container name (if using Docker)\n"
    echo

    # :flag.usage
    printf "  %s\n" "--file, -f PATH"
    printf "    File to write/read DB backup to/from\n"
    printf "    Default: /tmp/tekst-util.backup\n"
    echo

    # :flag.usage
    printf "  %s\n" "--mongodb-uri, -m URI"
    printf "    MongoDB URI (from inside the container if using Docker)\n"
    printf "    Default: mongodb://127.0.0.1:27017\n"
    echo

    # :flag.usage
    printf "  %s\n" "--delete, -d"
    printf "    Delete backup file after successful restore\n"
    echo

    # :command.usage_fixed_flags
    printf "  %s\n" "--help, -h"
    printf "    Show this help\n"
    echo

    # :command.usage_examples
    printf "%s\n" "Examples:"
    printf "  tekst-util restore\n"
    printf "  tekst-util restore -c tekst-util\n"
    printf "  tekst-util restore -c tekst-mongo -f /home/user/tekst-something.backup\n"
    printf "  tekst-util restore -m mongodb://usr:pwd@db.foo.com:27017/?authSource=admin\n"
    echo

  fi
}

# :command.usage
tekst_util_index_usage() {
  if [[ -n $long_usage ]]; then
    printf "tekst-util index - Create/update search index (MongoDB and ES services must be running)\n"
    echo

  else
    printf "tekst-util index - Create/update search index (MongoDB and ES services must be running)\n"
    echo

  fi

  printf "%s\n" "Usage:"
  printf "  tekst-util index [OPTIONS]\n"
  printf "  tekst-util index --help | -h\n"
  echo

  # :command.long_usage
  if [[ -n $long_usage ]]; then
    printf "%s\n" "Options:"

    # :command.usage_flags
    # :flag.usage
    printf "  %s\n" "--container-name, -c CONTAINER"
    printf "    Tekst-API Docker container name (if using Docker)\n"
    echo

    # :flag.usage
    printf "  %s\n" "--dev, -d"
    printf "    Whether Tekst is running in development mode (sets TEKST_DEV_MODE env var)\n"
    echo

    # :command.usage_fixed_flags
    printf "  %s\n" "--help, -h"
    printf "    Show this help\n"
    echo

    # :command.usage_examples
    printf "%s\n" "Examples:"
    printf "  tekst-util index\n"
    printf "  tekst-util index -c tekst-api\n"
    printf "  tekst-util index --dev\n"
    echo

  fi
}

# :command.usage
tekst_util_maintenance_usage() {
  if [[ -n $long_usage ]]; then
    printf "tekst-util maintenance - Run resource maintenance routines (MongoDB service must be running)\n"
    echo

  else
    printf "tekst-util maintenance - Run resource maintenance routines (MongoDB service must be running)\n"
    echo

  fi

  printf "%s\n" "Usage:"
  printf "  tekst-util maintenance [OPTIONS]\n"
  printf "  tekst-util maintenance --help | -h\n"
  echo

  # :command.long_usage
  if [[ -n $long_usage ]]; then
    printf "%s\n" "Options:"

    # :command.usage_flags
    # :flag.usage
    printf "  %s\n" "--container-name, -c CONTAINER"
    printf "    Tekst-API Docker container name (if using Docker)\n"
    echo

    # :flag.usage
    printf "  %s\n" "--dev, -d"
    printf "    Whether Tekst is running in development mode (sets TEKST_DEV_MODE env var)\n"
    echo

    # :command.usage_fixed_flags
    printf "  %s\n" "--help, -h"
    printf "    Show this help\n"
    echo

    # :command.usage_examples
    printf "%s\n" "Examples:"
    printf "  tekst-util maintenance\n"
    printf "  tekst-util maintenance -c tekst-api\n"
    printf "  tekst-util maintenance --dev\n"
    echo

  fi
}

# :command.normalize_input
# :command.normalize_input_function
normalize_input() {
  local arg flags passthru
  passthru=false

  while [[ $# -gt 0 ]]; do
    arg="$1"
    if [[ $passthru == true ]]; then
      input+=("$arg")
    elif [[ $arg =~ ^(--[a-zA-Z0-9_\-]+)=(.+)$ ]]; then
      input+=("${BASH_REMATCH[1]}")
      input+=("${BASH_REMATCH[2]}")
    elif [[ $arg =~ ^(-[a-zA-Z0-9])=(.+)$ ]]; then
      input+=("${BASH_REMATCH[1]}")
      input+=("${BASH_REMATCH[2]}")
    elif [[ $arg =~ ^-([a-zA-Z0-9][a-zA-Z0-9]+)$ ]]; then
      flags="${BASH_REMATCH[1]}"
      for ((i = 0; i < ${#flags}; i++)); do
        input+=("-${flags:i:1}")
      done
    elif [[ "$arg" == "--" ]]; then
      passthru=true
      input+=("$arg")
    else
      input+=("$arg")
    fi

    shift
  done
}

# :command.inspect_args
inspect_args() {
  if ((${#args[@]})); then
    readarray -t sorted_keys < <(printf '%s\n' "${!args[@]}" | sort)
    echo args:
    for k in "${sorted_keys[@]}"; do
      echo "- \${args[$k]} = ${args[$k]}"
    done
  else
    echo args: none
  fi

  if ((${#other_args[@]})); then
    echo
    echo other_args:
    echo "- \${other_args[*]} = ${other_args[*]}"
    for i in "${!other_args[@]}"; do
      echo "- \${other_args[$i]} = ${other_args[$i]}"
    done
  fi

  if ((${#deps[@]})); then
    readarray -t sorted_keys < <(printf '%s\n' "${!deps[@]}" | sort)
    echo
    echo deps:
    for k in "${sorted_keys[@]}"; do
      echo "- \${deps[$k]} = ${deps[$k]}"
    done
  fi

  if ((${#env_var_names[@]})); then
    readarray -t sorted_names < <(printf '%s\n' "${env_var_names[@]}" | sort)
    echo
    echo "environment variables:"
    for k in "${sorted_names[@]}"; do
      echo "- \$$k = ${!k:-}"
    done
  fi
}

# :command.user_lib
# src/lib/fnc.sh
cd_to_script_parent_dir() {
    script_parent_dir="$( cd "$( dirname "${BASH_SOURCE[0]}" )" >/dev/null 2>&1 && pwd )"
    cd "$script_parent_dir"
}

# :command.command_functions
# :command.function
tekst_util_backup_command() {
  # src/backup_command.sh
  container=${args[--container-name]}
  file=${args[--file]}
  uri=${args[--mongodb-uri]}

  if [[ $container ]]; then
      docker exec -i $container /usr/bin/mongodump --uri $uri --gzip --archive > $file
  else
      /usr/bin/mongodump --uri $uri --gzip --archive > $file
  fi

}

# :command.function
tekst_util_restore_command() {
  # src/restore_command.sh
  container=${args[--container-name]}
  file=${args[--file]}
  uri=${args[--mongodb-uri]}
  delete=${args[--delete]}

  if [[ $container ]]; then
      container_file="/tmp/tekst-db.backup"
      docker cp $file $container:$container_file
      docker exec -i $container /usr/bin/mongorestore --uri $uri --drop --gzip --archive=$container_file
      success=$?
      docker exec -i $container rm -f $container_file
  else
      /usr/bin/mongorestore --uri $uri --drop --gzip --archive=$file
      success=$?
  fi

  if [[ $success -eq 0 ]]; then
      if [[ $delete ]]; then
          rm -f $file
      fi
      echo "Restore successful."
  else
      echo "Restore failed."
  fi

}

# :command.function
tekst_util_index_command() {
  # src/index_command.sh
  container=${args[--container-name]}
  dev=${args[--dev]}

  # if in dev mode, run python using poetry's virtual env
  [[ $dev ]] && run_python="poetry run python" || run_python="python"

  if [[ $container ]]; then
      docker exec -e TEKST_DEV_MODE=$dev -i $container $run_python -m tekst index
  else
      cd_to_script_parent_dir
      cd ../Tekst-API
      TEKST_DEV_MODE=$dev $run_python -m tekst index
  fi

}

# :command.function
tekst_util_maintenance_command() {
  # src/maintenance_command.sh
  container=${args[--container-name]}
  dev=${args[--dev]}

  # if in dev mode, run python using poetry's virtual env
  [[ $dev ]] && run_python="poetry run python" || run_python="python"

  if [[ $container ]]; then
      docker exec -e TEKST_DEV_MODE=$dev -i $container $run_python -m tekst maintenance
  else
      cd_to_script_parent_dir
      cd ../Tekst-API
      TEKST_DEV_MODE=$dev $run_python -m tekst maintenance
  fi

}

# :command.parse_requirements
parse_requirements() {
  # :command.fixed_flags_filter
  while [[ $# -gt 0 ]]; do
    case "${1:-}" in
      --version | -v)
        version_command
        exit
        ;;

      --help | -h)
        long_usage=yes
        tekst_util_usage
        exit
        ;;

      *)
        break
        ;;

    esac
  done

  # :command.command_filter
  action=${1:-}

  case $action in
    -*) ;;

    backup)
      action="backup"
      shift
      tekst_util_backup_parse_requirements "$@"
      shift $#
      ;;

    restore)
      action="restore"
      shift
      tekst_util_restore_parse_requirements "$@"
      shift $#
      ;;

    index)
      action="index"
      shift
      tekst_util_index_parse_requirements "$@"
      shift $#
      ;;

    maintenance)
      action="maintenance"
      shift
      tekst_util_maintenance_parse_requirements "$@"
      shift $#
      ;;

    # :command.command_fallback
    "")
      tekst_util_usage >&2
      exit 1
      ;;

    *)
      printf "invalid command: %s\n" "$action" >&2
      exit 1
      ;;

  esac

  # :command.parse_requirements_while
  while [[ $# -gt 0 ]]; do
    key="$1"
    case "$key" in

      -?*)
        printf "invalid option: %s\n" "$key" >&2
        exit 1
        ;;

      *)
        # :command.parse_requirements_case
        # :command.parse_requirements_case_simple
        printf "invalid argument: %s\n" "$key" >&2
        exit 1

        ;;

    esac
  done

}

# :command.parse_requirements
tekst_util_backup_parse_requirements() {
  # :command.fixed_flags_filter
  while [[ $# -gt 0 ]]; do
    case "${1:-}" in
      --help | -h)
        long_usage=yes
        tekst_util_backup_usage
        exit
        ;;

      *)
        break
        ;;

    esac
  done

  # :command.command_filter
  action="backup"

  # :command.parse_requirements_while
  while [[ $# -gt 0 ]]; do
    key="$1"
    case "$key" in
      # :flag.case
      --container-name | -c)

        # :flag.case_arg
        if [[ -n ${2+x} ]]; then
          args['--container-name']="$2"
          shift
          shift
        else
          printf "%s\n" "--container-name requires an argument: --container-name, -c CONTAINER" >&2
          exit 1
        fi
        ;;

      # :flag.case
      --file | -f)

        # :flag.case_arg
        if [[ -n ${2+x} ]]; then
          args['--file']="$2"
          shift
          shift
        else
          printf "%s\n" "--file requires an argument: --file, -f PATH" >&2
          exit 1
        fi
        ;;

      # :flag.case
      --mongodb-uri | -m)

        # :flag.case_arg
        if [[ -n ${2+x} ]]; then
          args['--mongodb-uri']="$2"
          shift
          shift
        else
          printf "%s\n" "--mongodb-uri requires an argument: --mongodb-uri, -m URI" >&2
          exit 1
        fi
        ;;

      -?*)
        printf "invalid option: %s\n" "$key" >&2
        exit 1
        ;;

      *)
        # :command.parse_requirements_case
        # :command.parse_requirements_case_simple
        printf "invalid argument: %s\n" "$key" >&2
        exit 1

        ;;

    esac
  done

  # :command.default_assignments
  [[ -n ${args['--file']:-} ]] || args['--file']="/tmp/tekst-util.backup"
  [[ -n ${args['--mongodb-uri']:-} ]] || args['--mongodb-uri']="mongodb://127.0.0.1:27017"

}

# :command.parse_requirements
tekst_util_restore_parse_requirements() {
  # :command.fixed_flags_filter
  while [[ $# -gt 0 ]]; do
    case "${1:-}" in
      --help | -h)
        long_usage=yes
        tekst_util_restore_usage
        exit
        ;;

      *)
        break
        ;;

    esac
  done

  # :command.command_filter
  action="restore"

  # :command.parse_requirements_while
  while [[ $# -gt 0 ]]; do
    key="$1"
    case "$key" in
      # :flag.case
      --container-name | -c)

        # :flag.case_arg
        if [[ -n ${2+x} ]]; then
          args['--container-name']="$2"
          shift
          shift
        else
          printf "%s\n" "--container-name requires an argument: --container-name, -c CONTAINER" >&2
          exit 1
        fi
        ;;

      # :flag.case
      --file | -f)

        # :flag.case_arg
        if [[ -n ${2+x} ]]; then
          args['--file']="$2"
          shift
          shift
        else
          printf "%s\n" "--file requires an argument: --file, -f PATH" >&2
          exit 1
        fi
        ;;

      # :flag.case
      --mongodb-uri | -m)

        # :flag.case_arg
        if [[ -n ${2+x} ]]; then
          args['--mongodb-uri']="$2"
          shift
          shift
        else
          printf "%s\n" "--mongodb-uri requires an argument: --mongodb-uri, -m URI" >&2
          exit 1
        fi
        ;;

      # :flag.case
      --delete | -d)

        # :flag.case_no_arg
        args['--delete']=1
        shift
        ;;

      -?*)
        printf "invalid option: %s\n" "$key" >&2
        exit 1
        ;;

      *)
        # :command.parse_requirements_case
        # :command.parse_requirements_case_simple
        printf "invalid argument: %s\n" "$key" >&2
        exit 1

        ;;

    esac
  done

  # :command.default_assignments
  [[ -n ${args['--file']:-} ]] || args['--file']="/tmp/tekst-util.backup"
  [[ -n ${args['--mongodb-uri']:-} ]] || args['--mongodb-uri']="mongodb://127.0.0.1:27017"

}

# :command.parse_requirements
tekst_util_index_parse_requirements() {
  # :command.fixed_flags_filter
  while [[ $# -gt 0 ]]; do
    case "${1:-}" in
      --help | -h)
        long_usage=yes
        tekst_util_index_usage
        exit
        ;;

      *)
        break
        ;;

    esac
  done

  # :command.command_filter
  action="index"

  # :command.parse_requirements_while
  while [[ $# -gt 0 ]]; do
    key="$1"
    case "$key" in
      # :flag.case
      --container-name | -c)

        # :flag.case_arg
        if [[ -n ${2+x} ]]; then
          args['--container-name']="$2"
          shift
          shift
        else
          printf "%s\n" "--container-name requires an argument: --container-name, -c CONTAINER" >&2
          exit 1
        fi
        ;;

      # :flag.case
      --dev | -d)

        # :flag.case_no_arg
        args['--dev']=1
        shift
        ;;

      -?*)
        printf "invalid option: %s\n" "$key" >&2
        exit 1
        ;;

      *)
        # :command.parse_requirements_case
        # :command.parse_requirements_case_simple
        printf "invalid argument: %s\n" "$key" >&2
        exit 1

        ;;

    esac
  done

}

# :command.parse_requirements
tekst_util_maintenance_parse_requirements() {
  # :command.fixed_flags_filter
  while [[ $# -gt 0 ]]; do
    case "${1:-}" in
      --help | -h)
        long_usage=yes
        tekst_util_maintenance_usage
        exit
        ;;

      *)
        break
        ;;

    esac
  done

  # :command.command_filter
  action="maintenance"

  # :command.parse_requirements_while
  while [[ $# -gt 0 ]]; do
    key="$1"
    case "$key" in
      # :flag.case
      --container-name | -c)

        # :flag.case_arg
        if [[ -n ${2+x} ]]; then
          args['--container-name']="$2"
          shift
          shift
        else
          printf "%s\n" "--container-name requires an argument: --container-name, -c CONTAINER" >&2
          exit 1
        fi
        ;;

      # :flag.case
      --dev | -d)

        # :flag.case_no_arg
        args['--dev']=1
        shift
        ;;

      -?*)
        printf "invalid option: %s\n" "$key" >&2
        exit 1
        ;;

      *)
        # :command.parse_requirements_case
        # :command.parse_requirements_case_simple
        printf "invalid argument: %s\n" "$key" >&2
        exit 1

        ;;

    esac
  done

}

# :command.initialize
initialize() {
  version="0.1.0"
  long_usage=''
  set -e

}

# :command.run
run() {
  declare -A args=()
  declare -A deps=()
  declare -a other_args=()
  declare -a env_var_names=()
  declare -a input=()
  normalize_input "$@"
  parse_requirements "${input[@]}"

  case "$action" in
    "backup") tekst_util_backup_command ;;
    "restore") tekst_util_restore_command ;;
    "index") tekst_util_index_command ;;
    "maintenance") tekst_util_maintenance_command ;;
  esac
}

initialize
run "$@"