runtipi/scripts/app.sh

#!/usr/bin/env bash
echo "Starting app script"

source "${BASH_SOURCE%/*}/common.sh"

set -euo pipefail

ensure_pwd

ROOT_FOLDER="${PWD}"
STATE_FOLDER="${ROOT_FOLDER}/state"
ENV_FILE="${ROOT_FOLDER}/.env"

# Root folder in host system
ROOT_FOLDER_HOST=$(grep -v '^#' "${ENV_FILE}" | xargs -n 1 | grep ROOT_FOLDER_HOST | cut -d '=' -f2)
REPO_ID=$(grep -v '^#' "${ENV_FILE}" | xargs -n 1 | grep APPS_REPO_ID | cut -d '=' -f2)
STORAGE_PATH=$(grep -v '^#' "${ENV_FILE}" | xargs -n 1 | grep STORAGE_PATH | cut -d '=' -f2)

write_log "Running app script: ROOT_FOLDER=${ROOT_FOLDER}, ROOT_FOLDER_HOST=${ROOT_FOLDER_HOST}, REPO_ID=${REPO_ID}, STORAGE_PATH=${STORAGE_PATH}"

if [ -z ${1+x} ]; then
  command=""
else
  command="$1"
fi

if [ -z ${2+x} ]; then
  exit 1
else
  app="$2"

  app_dir="${ROOT_FOLDER}/apps/${app}"

  if [[ ! -d "${app_dir}" ]]; then
    # copy from repo
    echo "Copying app from repo"
    mkdir -p "${app_dir}"
    cp -r "${ROOT_FOLDER}/repos/${REPO_ID}/apps/${app}"/* "${app_dir}"
  fi

  app_data_dir="${STORAGE_PATH}/app-data/${app}"

  if [[ -z "${app}" ]] || [[ ! -d "${app_dir}" ]]; then
    echo "Error: \"${app}\" is not a valid app"
    exit 1
  fi
fi


# Function below has been modified from Umbrel
# Required Notice: Copyright
# Umbrel (https://umbrel.com)
compose() {
  local app="${1}"
  shift

  arch=$(uname -m)
  local architecture="${arch}"

  if [[ "$architecture" == "aarch64" ]]; then
    architecture="arm64"
  fi

  # App data folder
  local app_compose_file="${app_dir}/docker-compose.yml"

  # Pick arm architecture if running on arm and if the app has a docker-compose.arm.yml file
  if [[ "$architecture" == "arm"* ]] && [[ -f "${app_dir}/docker-compose.arm.yml" ]]; then
    app_compose_file="${app_dir}/docker-compose.arm.yml"
  fi

  # Pick arm architecture if running on arm and if the app has a docker-compose.arm64.yml file
  if [[ "$architecture" == "arm64" ]] && [[ -f "${app_dir}/docker-compose.arm64.yml" ]]; then
    app_compose_file="${app_dir}/docker-compose.arm64.yml"
  fi

  local common_compose_file="${ROOT_FOLDER}/repos/${REPO_ID}/apps/docker-compose.common.yml"

  local user_compose_file="${ROOT_FOLDER}/user-config/${app}/docker-compose.yml"
  local user_compose_args=
  if [[ -f ${user_compose_file} ]]; then
    user_compose_args="--file ${user_compose_file}"
  fi

  local user_env_file="${ROOT_FOLDER}/user-config/${app}/app.env"
  local user_env_args=
  if [[ -f ${user_env_file} ]]; then
    user_env_args="--env-file ${user_env_file}"
  fi

  # Vars to use in compose file
  export APP_DATA_DIR="${STORAGE_PATH}/app-data/${app}"
  export ROOT_FOLDER_HOST="${ROOT_FOLDER_HOST}"

  write_log "Running docker compose -f ${app_compose_file} -f ${common_compose_file} ${*}"
  write_log "APP_DATA_DIR=${APP_DATA_DIR}"
  write_log "ROOT_FOLDER_HOST=${ROOT_FOLDER_HOST}"

  docker compose \
    --env-file "${app_data_dir}/app.env" \
    ${user_env_args} \
    --project-name "${app}" \
    --file "${app_compose_file}" \
    --file "${common_compose_file}" \
    ${user_compose_args} \
    "${@}"
}

function ensure_permissions() {
  local app="${1}"

  # if app_data_dir/data does not exist, create it
  if [[ ! -d "${app_data_dir}/data" ]]; then
    mkdir -p "${app_data_dir}/data"
  fi

  # Check if app requires special uid and gid
  if [[ -f "${app_dir}/config.json" ]]; then
    uid=$(get_json_field "${app_dir}/config.json" uid)
    gid=$(get_json_field "${app_dir}/config.json" gid)

    write_log "App requires uid=${uid} and gid=${gid}"

    if [[ "$uid" != "null" ]] && [[ "$gid" != "null" ]]; then
      write_log "Setting uid and gid to ${uid}:${gid}"
      if ! chown -R "${uid}:${gid}" "${app_data_dir}/data"; then
        write_log "Failed to set uid and gid to ${uid}:${gid}"
      fi
    fi
  fi

  # Remove all .gitkeep files from app data dir
  find "${app_data_dir}" -name ".gitkeep" -exec rm -f {} \;

  chmod -R a+rwx "${app_data_dir}"
}

function install_app() {
  local app="${1}"

  # Write to file script.log
  write_log "Installing app ${app}..."

  if ! compose "${app}" pull; then
    write_log "Failed to pull app ${app}"
    exit 1
  fi

  # Copy default data dir to app data dir if it exists
  if [[ -d "${app_dir}/data" ]]; then
    cp -r "${app_dir}/data" "${app_data_dir}/data"
  fi

  ensure_permissions "${app}"

  if ! compose "${app}" up -d; then
    write_log "Failed to start app ${app}"
    exit 1
  fi

  exit 0
}

function start_app() {
  local app="${1}"

  write_log "Starting app ${app}..."

  ensure_permissions "${app}"

  # Pull images
  if ! compose "${app}" pull; then
    write_log "Failed to pull app ${app}"
  fi

  if ! compose "${app}" up --detach --force-recreate --remove-orphans; then
    write_log "Failed to start app ${app}"
    exit 1
  fi
  exit 0
}

function uninstall_app() {
  local app="${1}"

  write_log "Removing images for app ${app}..."

  if ! compose "${app}" down --rmi all --remove-orphans; then
    # just stop it if we can't remove the images
    if ! compose "${app}" rm --force --stop; then
      write_log "Failed to uninstall app ${app}"
      exit 1
    fi
  fi

  write_log "Deleting app data for app ${app}..."
  if [[ -d "${app_data_dir}" ]]; then
    rm -rf "${app_data_dir}"
  fi

  if [[ -d "${app_dir}" ]]; then
    rm -rf "${app_dir}"
  fi

  write_log "Successfully uninstalled app ${app}"
  exit
}

function update_app() {
  local app="${1}"

  if ! compose "${app}" up --detach --force-recreate --remove-orphans; then
    write_log "Failed to update app ${app}"
  fi

  if ! compose "${app}" down --rmi all --remove-orphans; then
    # just stop it if we can't remove the images
    if ! compose "${app}" rm --force --stop; then
      write_log "Failed to update app ${app}"
      exit 1
    fi
  fi

  # Remove app
  if [[ -d "${app_dir}" ]]; then
    rm -rf "${app_dir}"
  fi

  # Copy app from repo
  cp -r "${ROOT_FOLDER}/repos/${REPO_ID}/apps/${app}" "${app_dir}"

  ensure_permissions "${app}"

  compose "${app}" pull
  exit 0
}

function stop_app() {
  local app="${1}"

  write_log "Stopping app ${app}..."

  if ! compose "${app}" rm --force --stop; then
    write_log "Failed to stop app ${app}"
    exit 1
  fi

  exit 0
}

# Install new app
if [[ "$command" = "install" ]]; then
  install_app "${app}"
fi

# Removes images and destroys all data for an app
if [[ "$command" = "uninstall" ]]; then
  uninstall_app "${app}"
fi

# Update an app
if [[ "$command" = "update" ]]; then
  update_app "${app}"
fi

# Stops an installed app
if [[ "$command" = "stop" ]]; then
  stop_app "${app}"
fi

# Starts an installed app
if [[ "$command" = "start" ]]; then
  start_app "${app}"
fi

if [[ "$command" = "clean" ]]; then
  # Remove all stopped containers and unused images
  write_log "Cleaning up..."
  docker system prune --all --force

  exit 0
fi

exit 1