#!/usr/bin/env bash
set -euo pipefail

PREFIX=/opt/tianyan-edge
SCRIPT_DIR="$(cd "$(dirname "$0")" && pwd)"
ROOT_DIR="$(cd "$SCRIPT_DIR/.." && pwd)"
PYTHON_BIN="${PYTHON_BIN:-python3}"

sudo mkdir -p "$PREFIX"/{bin,python,config,systemd,staging,model}
sudo install -m 755 "$ROOT_DIR/build/edge-agent" "$PREFIX/bin/edge-agent"
sudo install -m 644 "$ROOT_DIR/python/infer_server.py" "$PREFIX/python/infer_server.py"
sudo install -m 644 "$ROOT_DIR/python/requirements.txt" "$PREFIX/python/requirements.txt"

if [ ! -f "$PREFIX/config/edge.yaml" ]; then
  sudo install -m 644 "$ROOT_DIR/config/edge.yaml.template" "$PREFIX/config/edge.yaml"
fi

sudo install -m 644 "$ROOT_DIR/systemd/edge-agent.service" /etc/systemd/system/edge-agent.service
sudo install -m 644 "$ROOT_DIR/systemd/edge-infer.service" /etc/systemd/system/edge-infer.service

sudo "$PYTHON_BIN" -m pip install -r "$PREFIX/python/requirements.txt"

sudo systemctl daemon-reload
sudo systemctl enable edge-infer.service edge-agent.service
sudo systemctl restart edge-infer.service edge-agent.service

echo "installed to $PREFIX"