Skip to content
This repository was archived by the owner on Dec 4, 2023. It is now read-only.

add auto-update service #33

Open
casperdcl wants to merge 3 commits into
base: main
Choose a base branch
from
Open

add auto-update service #33

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
1a916c5
watchtower auto-update service
casperdcl Nov 1, 2023
3178b6a
add auto-update to install script
casperdcl Nov 1, 2023
2eea134
fix auto-update to :latest tags
casperdcl Nov 1, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
7 changes: 7 additions & 0 deletions docker-compose.autoupdate.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,7 @@
version: '3.8'
services:
watchtower:
image: containrrr/watchtower
volumes:
- /var/run/docker.sock:/var/run/docker.sock
command: --interval 60 premapp premd dnsd authd controllerd
19 changes: 16 additions & 3 deletions install.sh
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -9,11 +9,13 @@ SCRIPT_VERSION="v0.1.0"
DEFAULT_PREM_BOX_USER=premai-io
DEFAULT_PREM_BOX_BRANCH=main
DEFAULT_PREM_REGISTRY_BRANCH=main
DEFAULT_PREM_AUTO_UPDATE='' # empty/unset means false

PREM_BOX_REPO=prem-box
PREM_BOX_USER=${1:-$DEFAULT_PREM_BOX_USER}
PREM_BOX_BRANCH=${2:-$DEFAULT_PREM_BOX_BRANCH}
PREM_REGISTRY_BRANCH=${3:-$DEFAULT_PREM_REGISTRY_BRANCH}
PREM_AUTO_UPDATE=${4:-$DEFAULT_PREM_AUTO_UPDATE}
Comment thread
casperdcl marked this conversation as resolved.

ARCH=$(uname -m)
WHO=$(whoami)
Expand All @@ -29,7 +31,6 @@ PREM_REGISTRY_URL=https://raw.githubusercontent.com/premAI-io/prem-registry/$PRE
SENTRY_DSN=https://75592545ad6b472e9ad7c8ff51740b73@o1068608.ingest.sentry.io/4505244431941632

PREM_APP_ID=$(cat /proc/sys/kernel/random/uuid)
PREM_AUTO_UPDATE=false
ORIGINAL_HOME=$(eval echo ~$SUDO_USER)

PREM_CONF_FOUND=$(find ~ -path "$ORIGINAL_HOME/prem/config")
Expand Down Expand Up @@ -60,6 +61,7 @@ PREM_AUTO_UPDATE=$PREM_AUTO_UPDATE" >$PREM_CONF_FOUND
curl --silent https://raw.githubusercontent.com/$PREM_BOX_USER/$PREM_BOX_REPO/$PREM_BOX_BRANCH/docker-compose.premg.yml -o $ORIGINAL_HOME/prem/docker-compose.premg.yml
curl --silent https://raw.githubusercontent.com/$PREM_BOX_USER/$PREM_BOX_REPO/$PREM_BOX_BRANCH/docker-compose.premapp.premd.yml -o $ORIGINAL_HOME/prem/docker-compose.premapp.premd.yml
curl --silent https://raw.githubusercontent.com/$PREM_BOX_USER/$PREM_BOX_REPO/$PREM_BOX_BRANCH/docker-compose.gpu.yml -o $ORIGINAL_HOME/prem/docker-compose.gpu.yml
curl --silent https://raw.githubusercontent.com/$PREM_BOX_USER/$PREM_BOX_REPO/$PREM_BOX_BRANCH/docker-compose.autoupdate.yml -o $ORIGINAL_HOME/prem/docker-compose.autoupdate.yml
curl --silent https://raw.githubusercontent.com/$PREM_BOX_USER/$PREM_BOX_REPO/$PREM_BOX_BRANCH/versions.json -o $ORIGINAL_HOME/prem/versions.json
}
# Function to check for NVIDIA GPU
Expand Down Expand Up @@ -309,6 +311,17 @@ export POSTGRES_DB=dnsd-db

echo ""
echo "🏁 Starting Prem..."
DCC="docker-compose -f $ORIGINAL_HOME/prem/docker-compose.premapp.premd.yml -f $ORIGINAL_HOME/prem/docker-compose.premg.yml"
# Check for PREM_AUTO_UPDATE and run watchtower if necessary
if test -n "$PREM_AUTO_UPDATE"; then
echo "Using :latest images & auto-updating"
export PREM_APP_IMAGE=ghcr.io/premai-io/prem-app:latest
export PREM_DAEMON_IMAGE=ghcr.io/premai-io/premd:latest
export PREMG_DNSD_IMAGE=ghcr.io/premai-io/dnsd:latest
export PREMG_CONTROLLERD_IMAGE=ghcr.io/premai-io/controllerd:latest
export PREMG_AUTHD_IMAGE=ghcr.io/premai-io/authd:latest
DCC="$DCC -f docker-compose.autoupdate.yml"
fi
# Check for GPU and install drivers if necessary
if has_gpu; then
if ! check_nvidia_driver; then
Expand All @@ -318,10 +331,10 @@ if has_gpu; then
exit 0
fi
echo "nvidia-smi is available. Running with gpu support..."
docker-compose -f $ORIGINAL_HOME/prem/docker-compose.premapp.premd.yml -f $ORIGINAL_HOME/prem/docker-compose.gpu.yml -f $ORIGINAL_HOME/prem/docker-compose.premg.yml up -d || exit 1
$DCC -f $ORIGINAL_HOME/prem/docker-compose.gpu.yml up -d || exit 1
else
echo "No NVIDIA GPU detected. Running without gpu support..."
docker-compose -f $ORIGINAL_HOME/prem/docker-compose.premapp.premd.yml -f $ORIGINAL_HOME/prem/docker-compose.premg.yml up -d || exit 1
$DCC up -d || exit 1
fi

# Loop to check for 'OK' from curl command with maximum 10 retries
Expand Down