omnitool_init.sh

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

echo "[->] UPDATE OMNITOOL "
cd ./omnitool
git pull
cd ..

# if [ -d "./omnitool" ]; then
#     echo "[v] OMNITOOL already present "
#     cd ./omnitool
#     git pull
#     cd ..
#     chmod 0777 ./omnitool
#     chown -R node:node ./omnitool
# else
#     echo "[ERROR] OMNITOOL IS MISSING"
#     git clone https://github.com/omnitool-ai/omnitool.git
#     chmod 0777 ./omnitool
#     chown -R node:node ./omnitool
# fi

# if [ -d "./omnitool/node_modules" ]; then
#     echo "[v] OMNITOOL NODE_MODULES already present "
# else
#     mkdir -p ./omnitool/node_modules
#     chmod 0777 ./omnitool/node_modules
#     chown node ./omnitool/node_modules
#     echo "[+] Adding OMNITOOL NODE_MODULES"
# fi

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

    if [ -L "./omnitool/packages/omni-server/data.local" ]; then
        echo "[v] DATA.LOCAL symlink already present "
    else
        if [ -d "./omnitool/packages/omni-server/data.local" ]; then
            echo "[v] DATA.LOCAL already present "
            # hopefully it got copied over through the symlink in the previous run
            rm -rf ./omnitool/packages/omni-server/data.local
            
            mkdir ./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] Creating DATA.LOCAL "
            mkdir ./omnitool/packages/omni-server/data.local
            chmod 0777 ./omnitool/packages/omni-server/data.local
            chown node ./omnitool/packages/omni-server/data.local
        fi

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

cd omnitool
echo "[x] CD OMNITOOL "

echo "[->] YARN INSTALL"
yarn

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

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