#!/bin/bash
echo "--- START ---"

echo "--- GIT OMNITOOL ---"
git clone https://github.com/omnitool-ai/omnitool

chmod 777 ./omnitool
chown -R node ./omnitool

mkdir -p ./omnitool/node_modules
chmod 777 ./omnitool/node_modules
chown node ./omnitool/node_modules

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

    echo "--- CREATING the SYMLINK to /DATA ---"
    if [ -d "./omnitool/packages/omni-server/data.local" ]; then
        rm -rf ./omnitool/packages/omni-server/data.local
    fi

    mkdir ./omnitool/packages/omni-server/data.local
    chmod 777 ./omnitool/packages/omni-server/data.local
    chown node ./omnitool/packages/omni-server/data.local

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

echo "--- CD OMNITOOL ---"
cd omnitool

echo "--- YARN INSTALL ---"
yarn

chmod -R 777 .
chown -Rh node .

echo "--- YARN START ---"
yarn start -u -rb -R blocks