#!/usr/bin/env bash
# Setup script for zhadyz/AI_SOC on Linux, from https://gitmanual.com/zhadyz/AI_SOC
# Read it before you run it: the commands come from the repo's own README and files.
# Run: bash setup.sh
set -eo pipefail
trap 'echo "Setup stopped at the step above. Steps and fixes: https://gitmanual.com/zhadyz/AI_SOC" >&2' ERR
step() { printf '\n\033[1m==> %s\033[0m\n' "$1"; }
have() { command -v "$1" >/dev/null 2>&1; }
# Installers put programs here; make them usable right away.
export PATH="$HOME/.local/share/mise/shims:$HOME/.cargo/bin:$HOME/.local/bin:$HOME/.deno/bin:$HOME/.bun/bin:$PATH"
# nvm is a shell function, so load it into this script when it's installed.
export NVM_DIR="${NVM_DIR:-$HOME/.nvm}"
load_nvm() { [ -s "$NVM_DIR/nvm.sh" ] && . "$NVM_DIR/nvm.sh"; }
have_nvm() { load_nvm >/dev/null 2>&1; type nvm >/dev/null 2>&1; }

step '1/5 Clone the repository'
[ -d 'AI_SOC' ] || git clone https://github.com/zhadyz/AI_SOC.git
cd 'AI_SOC'

step '2/5 Install the required runtimes'
# Python 3.x
# Debian/Ubuntu
have python3 || sudo apt install -y python3 python3-venv

step '3/5 Configure environment variables'
# Fill in secrets: AI_SOC_API_KEY, POSTGRES_PASSWORD, AI_SOC_AUTH_SECRET, AI_SOC_MODEL_SIGNING_KEY, TRIAGE_MAX_TOKENS, TRIAGE_API_KEY_ENABLED
if [ ! -f .env ]; then cp .env.example .env; echo 'Created .env. Fill in secrets: AI_SOC_API_KEY, POSTGRES_PASSWORD, AI_SOC_AUTH_SECRET, AI_SOC_MODEL_SIGNING_KEY, TRIAGE_MAX_TOKENS, TRIAGE_API_KEY_ENABLED in .env, then run this script again.'; exit 0; fi

step '4/5 Start backing services'
# Needs: PostgreSQL
# needs Docker Engine: https://docs.docker.com/engine/install/
docker compose -f compose.yaml up -d

step '5/5 Install dependencies'
python -m venv .venv
source .venv/bin/activate
pip install -e .

echo 'Set up zhadyz/AI_SOC.'
echo 'Fill in secrets: AI_SOC_API_KEY, POSTGRES_PASSWORD, AI_SOC_AUTH_SECRET, AI_SOC_MODEL_SIGNING_KEY, TRIAGE_MAX_TOKENS, TRIAGE_API_KEY_ENABLED'
echo 'Manual: https://gitmanual.com/zhadyz/AI_SOC'