run_all.sh

#!/usr/bin/env bash

set -Eeuo pipefail

SCRIPT_DIR="$(cd -- "$(dirname -- "${BASH_SOURCE[0]}")" && pwd)"

# Support both layouts:
# 1) <repo>/comfyui and <repo>/cognitive
# 2) <repo>/agents-to-agents/comfyui and <repo>/agents-to-agents/cognitive
if [[ -d "${SCRIPT_DIR}/agents-to-agents" ]]; then
	PROJECT_ROOT="${SCRIPT_DIR}/agents-to-agents"
else
	PROJECT_ROOT="${SCRIPT_DIR}"
fi

COMFY_DIR="${PROJECT_ROOT}/comfyui"
COGNITIVE_DIR="${PROJECT_ROOT}/cognitive"

log() {
	printf '[%s] %s\n' "$(date '+%Y-%m-%d %H:%M:%S')" "$*"
}

fail() {
	printf 'Error: %s\n' "$*" >&2
	exit 1
}

require_dir() {
	local dir="$1"
	local name="$2"
	[[ -d "${dir}" ]] || fail "${name} directory not found: ${dir}"
}

require_cmd() {
	command -v "$1" >/dev/null 2>&1 || fail "Required command not found: $1"
}

docker_cmd() {
	if docker info >/dev/null 2>&1; then
		printf 'docker'
	elif command -v sudo >/dev/null 2>&1; then
		printf 'sudo docker'
	else
		fail "Docker daemon not accessible and sudo not available"
	fi
}

compose_subcommand() {
	local dc="$1"
	if ${dc} compose version >/dev/null 2>&1; then
		printf 'compose'
	elif command -v docker-compose >/dev/null 2>&1; then
		printf 'docker-compose'
	else
		fail "Neither 'docker compose' nor 'docker-compose' is available"
	fi
}

start_comfyui_linux() {
	local comfy_dir="$1"

	if command -v gnome-terminal >/dev/null 2>&1; then
		gnome-terminal -- bash -lc "cd \"${comfy_dir}\" && ./run_host.sh"
		return
	fi

	if command -v konsole >/dev/null 2>&1; then
		konsole -e bash -lc "cd \"${comfy_dir}\" && ./run_host.sh"
		return
	fi

	if command -v xfce4-terminal >/dev/null 2>&1; then
		xfce4-terminal --hold -e "bash -lc 'cd \"${comfy_dir}\" && ./run_host.sh'"
		return
	fi

	if command -v xterm >/dev/null 2>&1; then
		xterm -hold -e "cd \"${comfy_dir}\" && ./run_host.sh" &
		return
	fi

	fail "No supported terminal emulator found (gnome-terminal, konsole, xfce4-terminal, xterm)"
}


start_comfyui() {
	local os_name
	os_name="$(uname -s)"

	case "${os_name}" in
		Linux)
			start_comfyui_linux "$1"
			;;
		*)
			fail "Unsupported OS: ${os_name}"
			;;
	esac
}

main() {
	require_dir "${COMFY_DIR}" "ComfyUI"
	require_dir "${COGNITIVE_DIR}" "cognitive"
	require_cmd chmod
	require_cmd uname

	log "Preparing ComfyUI scripts"
	chmod +x "${COMFY_DIR}/setup_host.sh" "${COMFY_DIR}/run_host.sh"

	log "Running ComfyUI host setup"
	(
		cd "${COMFY_DIR}"
		./setup_host.sh
	)

	log "Starting ComfyUI in a dedicated terminal"
	start_comfyui "${COMFY_DIR}"

	log "Building and starting cognitive Docker services"
	(
		cd "${COGNITIVE_DIR}"

		local dc
		local compose_mode
		dc="$(docker_cmd)"
		compose_mode="$(compose_subcommand "${dc}")"

		if [[ "${compose_mode}" == "compose" ]]; then
			${dc} compose build --no-cache
			${dc} compose up -d
		else
			if [[ "${dc}" == "sudo docker" ]]; then
				sudo docker-compose build --no-cache
				sudo docker-compose up -d
			else
				docker-compose build --no-cache
				docker-compose up -d
			fi
		fi
	)

	log "All services started successfully"
}

main "$@"