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

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

mkdir -p /app/omnitool/node_modules
chmod 777 /app/omnitool/node_modules
#mkdir -p /app/omnitool/packages/omni-server/config.default/models/nsfwjs/mobilenet-v2-quant/
#chmod 777 /app/omnitool/packages/omni-server/config.default/models/nsfwjs/mobilenet-v2-quant/
#curl -L https://github.com/omnitool-ai/omnitool/raw/main/packages/omni-server/config.default/models/nsfwjs/mobilenet-v2-quant/group1-shard1of1 -o /app/omnitool/packages/omni-server/config.default/models/nsfwjs/mobilenet-v2-quant/group1-shard1of1
#chmod 777 /app/omnitool/packages/omni-server/config.default/models/nsfwjs/mobilenet-v2-quant/group1-shard1of1

if [ -d "/data" ] && [ ! -L "/app/omnitool/packages/omni-server/data.local" ]; then
    # Backup the existing data.local directory
    echo "--- Backup the existing data.local directory ---"

    if [ -d "/app/omnitool/packages/omni-server/data.local" ]; then
        mv /app/omnitool/packages/omni-server/data.local /app/omnitool/packages/omni-server/data.local.org
    fi

    # Create the symlink
    echo "--- Create the symlink ---"

    ln -s /data /app/omnitool/packages/omni-server/data.local

    # Copy data from backup to symlink if backup exists
    echo "--- Copy data from backup to symlink if backup exists ---"

    if [ -d "/app/omnitool/packages/omni-server/data.local.org" ]; then
        cp -a /app/omnitool/packages/omni-server/data.local.org/. /app/omnitool/packages/omni-server/data.local/
        # Cleanup
        rm -rf /app/omnitool/packages/omni-server/data.local.org
        echo "--- Cleanup done ---"

    fi
else
    echo "--- NO /data folder DETECTED, SKIPPING symlink creation ---"
fi

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

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

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