#!/usr/bin/env sh
set -e

REPO="Yatsuiii/api-causality-engine"
BIN="ace"

# Detect OS and arch
OS="$(uname -s)"
ARCH="$(uname -m)"

case "$OS" in
  Linux)
    case "$ARCH" in
      x86_64)  TARGET="ace-linux-x86_64.tar.gz" ;;
      aarch64) TARGET="ace-linux-aarch64.tar.gz" ;;
      *) echo "Unsupported architecture: $ARCH" && exit 1 ;;
    esac
    ;;
  Darwin)
    case "$ARCH" in
      x86_64)  TARGET="ace-macos-x86_64.tar.gz" ;;
      arm64)   TARGET="ace-macos-aarch64.tar.gz" ;;
      *) echo "Unsupported architecture: $ARCH" && exit 1 ;;
    esac
    ;;
  *) echo "Unsupported OS: $OS" && exit 1 ;;
esac

# Get latest release tag
TAG="$(curl -fsSL "https://api.github.com/repos/$REPO/releases/latest" | grep '"tag_name"' | cut -d'"' -f4)"
if [ -z "$TAG" ]; then
  echo "Could not fetch latest release tag" && exit 1
fi

BASE_URL="https://github.com/$REPO/releases/download/$TAG"

echo "Installing ace $TAG for $OS/$ARCH..."

TMPDIR="$(mktemp -d)"
trap 'rm -rf "$TMPDIR"' EXIT

curl -fsSL "$BASE_URL/$TARGET"      -o "$TMPDIR/ace.tar.gz"
curl -fsSL "$BASE_URL/SHA256SUMS"   -o "$TMPDIR/SHA256SUMS"

# Verify checksum
cd "$TMPDIR"
if command -v sha256sum >/dev/null 2>&1; then
  grep "$TARGET" SHA256SUMS | sha256sum -c -
else
  grep "$TARGET" SHA256SUMS | shasum -a 256 -c -
fi
cd - > /dev/null

tar -xz -C "$TMPDIR" -f "$TMPDIR/ace.tar.gz"
chmod +x "$TMPDIR/$BIN"

INSTALL_DIR="${INSTALL_DIR:-/usr/local/bin}"
if [ -w "$INSTALL_DIR" ]; then
  mv "$TMPDIR/$BIN" "$INSTALL_DIR/$BIN"
else
  sudo mv "$TMPDIR/$BIN" "$INSTALL_DIR/$BIN"
fi

echo "ace $TAG installed to $INSTALL_DIR/$BIN"
echo "Run: ace --help"