#!/bin/zsh
set -e

d=${0:a:h:h}
source $d/.env

BUILD_KERNEL=false
[[ "$1" == "--kernel" ]] && BUILD_KERNEL=true

# --- Build packages on HOST (aiosc) ---
echo "=== Building packages on $HOST ==="
ssh "$HOST" zsh -s -- "$GPG_KEY" <<'REMOTE'
set -e
setopt nonomatch 2>/dev/null || true
GPG_KEY="$1"
REPO_NAME="aios"
REPO_DIR="${HOME}/ai/repo"
WORK="${HOME}/aios-pkg"

if [ ! -d "$REPO_DIR/.git" ]; then
  git clone git@git.syui.ai:ai/repo.git "$REPO_DIR"
  cd "$REPO_DIR"
  git config user.email $USER_EMAIL
  git config user.name $USER_NAME
fi

cd "$REPO_DIR"
git config user.signingkey "$GPG_KEY"
git config commit.gpgsign true

rm -rf "$WORK"
mkdir -p "$WORK"

cd "$WORK"
git clone --depth 1 https://git.syui.ai/ai/os.git

for pkg in ailog aigpt aishell; do
  echo "=== Building $pkg ==="
  cp -r "$WORK/os/pkg/$pkg" "$WORK/$pkg"
  cd "$WORK/$pkg"
  makepkg -sf --noconfirm --sign --key "$GPG_KEY"
  cd "$WORK"
done

mkdir -p "$REPO_DIR/x86_64"

for pkg in ailog aigpt aishell; do
  rm -f "$REPO_DIR/x86_64/${pkg}"-*.pkg.tar.zst
  rm -f "$REPO_DIR/x86_64/${pkg}"-*.pkg.tar.zst.sig
  rm -f "$REPO_DIR/x86_64/${pkg}-debug"-*.pkg.tar.zst
  rm -f "$REPO_DIR/x86_64/${pkg}-debug"-*.pkg.tar.zst.sig
  cp "$WORK"/"$pkg"/*.pkg.tar.zst "$REPO_DIR/x86_64/"
  cp "$WORK"/"$pkg"/*.pkg.tar.zst.sig "$REPO_DIR/x86_64/" 2>/dev/null || true
done

rm -rf "$WORK"
echo "=== Packages built ==="
REMOTE

# --- Build kernel on HOST_KERNEL (arch) if requested ---
if [[ "$BUILD_KERNEL" == "true" ]]; then
  echo "=== Building linux-aios on $HOST_KERNEL ==="
  KERNEL_WORK="\${HOME}/aios-kernel"

  ssh "$HOST_KERNEL" zsh -s <<'KERNEL_REMOTE'
set -e
REPOS="${HOME}/repos"
WORK="${HOME}/aios-kernel"

# Clone or update upstream Arch linux PKGBUILD
mkdir -p "$REPOS"
if [ -d "$REPOS/archlinux" ]; then
  cd "$REPOS/archlinux"
  git pull
else
  git clone --depth 1 https://gitlab.archlinux.org/archlinux/packaging/packages/linux.git "$REPOS/archlinux"
fi

# Prepare workspace
rm -rf "$WORK"
mkdir -p "$WORK/linux-aios"
cp "$REPOS/archlinux/PKGBUILD" "$WORK/linux-aios/"
cp "$REPOS/archlinux/config.x86_64" "$WORK/linux-aios/"

# Patch PKGBUILD for aios
cd "$WORK/linux-aios"
sed -i 's/^pkgbase=linux$/pkgbase=linux-aios/' PKGBUILD
sed -i "s/^pkgdesc='Linux'$/pkgdesc='Linux (aios)'/" PKGBUILD
sed -i '/# htmldocs/,/texlive-latexextra/d' PKGBUILD
sed -i '/make htmldocs/d' PKGBUILD
sed -i '/_package-docs()/,/^}/d' PKGBUILD
sed -i '/"\$pkgbase-docs"/d' PKGBUILD
# Move version setting after patch application, add EXTRAVERSION removal
sed -i '/^  echo "Setting version\.\.\."/d' PKGBUILD
sed -i '/localversion.10-pkgrel/d' PKGBUILD
sed -i '/localversion.20-pkgname/d' PKGBUILD
sed -i '/^    patch -Np1 < "\.\.\/\$src"/{N;N;/done/a\\n  echo "Setting version..."\n  sed -i "s/^EXTRAVERSION = .*/EXTRAVERSION =/" Makefile\n  echo "" > localversion.10-pkgrel\n  echo "-aios" > localversion.20-pkgname
}' PKGBUILD

makepkg -sf --noconfirm --skippgpcheck
echo "=== Kernel built ==="
KERNEL_REMOTE

  # Transfer kernel packages: HOST_KERNEL -> local -> HOST
  echo "=== Transferring kernel packages ==="
  tmpdir=$(mktemp -d)
  ssh "$HOST_KERNEL" "ls ~/aios-kernel/linux-aios/linux-aios-*.pkg.tar.zst" | while read f; do scp "${HOST_KERNEL}:$f" "$tmpdir/"; done
  scp "$tmpdir"/linux-aios-*.pkg.tar.zst "$HOST":~/ai/repo/x86_64/
  rm -rf "$tmpdir"

  ssh "$HOST_KERNEL" "rm -rf ~/aios-kernel"
fi

# --- Update repo database on HOST ---
echo "=== Updating repo database ==="
ssh "$HOST" zsh -s -- "$GPG_KEY" <<'REPO_REMOTE'
set -e
setopt nonomatch 2>/dev/null || true
GPG_KEY="$1"
REPO_NAME="aios"
REPO_DIR="${HOME}/ai/repo"

cd "$REPO_DIR/x86_64"
rm -f "${REPO_NAME}".{db,files}*
repo-add --sign --key "$GPG_KEY" "${REPO_NAME}.db.tar.gz" *.pkg.tar.zst
gpg --export "$GPG_KEY" > "$REPO_DIR/aios.gpg"

for f in "${REPO_NAME}.db" "${REPO_NAME}.files" "${REPO_NAME}.db.sig" "${REPO_NAME}.files.sig"; do
  if [ -L "$f" ]; then
    target=$(readlink "$f")
    rm "$f"
    cp "$target" "$f"
  fi
done
rm -f *.old *.old.sig

cd "$REPO_DIR"
git add -A
git commit -m "update $(date +%Y.%m.%d)" || true
git push

echo "=== Done ==="
REPO_REMOTE