#!/usr/bin/env bash

# Ensure correct local path.
SCRIPT_DIR=$( cd -- "$( dirname -- "${BASH_SOURCE[0]}" )" &> /dev/null && pwd )
cd $SCRIPT_DIR

# Accidental run prevention
if [ -d "SwarmUI" ]; then
    echo "SwarmUI already exists in this directory. Please remove it before installing."
    exit 1
fi
if [ -f "SwarmUI.sln" ]; then
    echo "SwarmUI already exists in this directory. Please remove it before installing."
    exit 1
fi

# Download swarm
git clone https://github.com/mcmonkeyprojects/SwarmUI

mkdir -p /workspace/SwarmUI/dlbackend

cd /workspace/SwarmUI/dlbackend

git clone https://github.com/comfyanonymous/ComfyUI

mkdir -p /workspace/SwarmUI/dlbackend/ComfyUI/custom_nodes

cd /workspace/SwarmUI/dlbackend/ComfyUI/custom_nodes

git clone https://github.com/ltdrdata/ComfyUI-Manager.git

cd /workspace/SwarmUI

# install dotnet
cd launchtools
rm dotnet-install.sh
# https://learn.microsoft.com/en-us/dotnet/core/install/linux-scripted-manual#scripted-install
wget https://dot.net/v1/dotnet-install.sh -O dotnet-install.sh
chmod +x dotnet-install.sh

# Note: manual installers that want to avoid home dir, add to both of the below lines: --install-dir $SCRIPT_DIR/.dotnet
./dotnet-install.sh --channel 8.0 --runtime aspnetcore
./dotnet-install.sh --channel 8.0
cd ..

# export DOTNET_ROOT=~/.dotnet
# export PATH="$( pwd )/.dotnet:~/.dotnet:$PATH"


# Launch
./launch-linux.sh --launch_mode none --host 0.0.0.0