omnitool_init.sh

#!/bin/bash
echo "[->] START "

echo "[->] CHECKING EXISTING /DATA "
if [ -d "/data" ]; then
    echo "$(ls -l /data)" 

    if [ -L "./omnitool/packages/omni-server/data.local" ]; then
        echo "[v] DATA.LOCAL symlink already present. Removing it."
        rm -rf ./omnitool/packages/omni-server/data.local
    fi

    if [ -d "./omnitool/packages/omni-server/data.local" ]; then
        echo "[v] DATA.LOCAL already present "
        rm -rf ./omnitool/packages/omni-server/data.local         # hopefully it got copied over through the symlink in the previous run
    fi

    echo "[v] Creating the symlink "
    ln -s /data ./omnitool/packages/omni-server/data.local
    chmod 0777 ./omnitool/packages/omni-server/data.local
    chown node ./omnitool/packages/omni-server/data.local
else
    echo "[v] NO persistent /DATA DETECTED. You can add 20Gig of persistent storage as a paid option to Hugging Face"
fi

echo "[->] UPDATE OMNITOOL "
cd ./omnitool
output=$(git pull)
if echo "$output" | grep -q "Already up to date."; then
  echo "The repository is already up to date."
else
  echo "New data was fetched."
  echo "[->] YARN INSTALL"
  yarn

  echo "[->] Updating permissions"
  chmod -R 0777 .
  chown -Rh node:node .
fi

echo "[->] YARN START "
yarn start -u -rb -R blocks