#!/bin/bash

source logging.sh
source config.sh

PROBE_DIR="../ebpf_probe"

create_probe() {
    log "Deploying eBPF probe to $FAULTY_NODE_NAME."
    
    set -e

    cd "$PROBE_DIR"
    sudo python3 -m venv .venv --system-site-packages
    sudo .venv/bin/pip install -r requirements.txt
    sudo tmux new-session -d -s $SESSION_NAME "./.venv/bin/python3 orchestrator.py"
    
    log "Probe started in tmux session (sudo tmux attach -t $SESSION_NAME)"
}

delete_probe() {
    log "Stopping eBPF probe on $FAULTY_NODE_NAME."
    sudo tmux kill-session -t $SESSION_NAME 2>/dev/null 
    log "Cleanup of probe complete."
}

usage() {
    echo "Usage: $0 {create|delete}"
    exit 1
}

if [ $# -eq 0 ]; then
    usage
fi

case "$1" in
    create) 
        create_probe
        ;;
    delete)
        delete_probe
        ;;
    *)
        usage
        ;;
esac