# Contributor: Stefan Reimer <stefan@zero-downtime.net>
# Maintainer: Stefan Reimer <stefan@zero-downtime.net>
# https://awsdocs-neuron.readthedocs-hosted.com/en/latest/release-notes/runtime/aws-neuronx-dkms/index.html#neuron-driver-release-notes
#
# Todo: needs fix of https://github.com/aws-neuron/aws-neuron-sdk/issues/843
#
pkgname=aws-neuron-driver
pkgver=2.19.64.0
pkgrel=0
pkgdesc="Linux Kernel module for AWS Neuron INF instances"
url="https://awsdocs-neuron.readthedocs-hosted.com/en/latest/release-notes/index.html#"
arch="x86_64"
license="GPL-2.0"
makedepends="bash xz linux-headers linux-virt-dev"
options="!check"

source="$pkgname-$pkgver.deb::https://apt.repos.neuron.amazonaws.com/pool/main/a/aws-neuronx-dkms/aws-neuronx-dkms_"$pkgver"_amd64.deb"

unpack() {
  ar -x "$srcdir/$pkgname-$pkgver.deb" && find . -type d -exec chmod 755 {} \; && tar xfJo data.tar.xz
  mv usr/src/aws-neuronx-"$pkgver" "$srcdir/$pkgname-$pkgver"
  rm -rf usr data.tar.xz control.tar.xz debian-binary

  # What is wrong with AWS ?
  find $srcdir/$pkgname-$pkgver -type d -exec chmod 755 {} \;
}


build() {
  # Hack running the build inside a container other uname -r returns host kernel
  KERNEL_VERSION=$(basename $(ls -d /lib/modules/*-virt))

  unset CFLAGS CPPFLAGS CXXFLAGS
  unset LDFLAGS

  # patch for recent gcc warning/errors, cause AWS cannot be bothered apparently
  for f in neuron_dma.c neuron_sysfs_metrics.c v1/neuron_dhal_v1.c v2/neuron_dhal_v2.c; do
    sed -i -e 's/const static/static const/' $f
  done

  make KERNEL_SRC_DIR=/lib/modules/$KERNEL_VERSION/build
}

package() {
  KERNEL_VERSION=$(basename $(ls -d /lib/modules/*-virt))
  depends="linux-virt~$(echo $KERNEL_VERSION | sed -e 's/-.*$//')"

  mkdir -p "$pkgdir"/lib/modules/$KERNEL_VERSION/kernel "$pkgdir"/lib/udev/rules.d
  gzip -9 -c neuron.ko > "$pkgdir"/lib/modules/$KERNEL_VERSION/kernel/neuron.ko.gz

  echo 'KERNEL=="neuron*", MODE="0666"' > "$pkgdir"/lib/udev/rules.d/90-neuron-udev.rules
}

sha512sums="
91e36a6cea7494d29d8c451fb79a73665290682b93baa293d76c9fb679ad94a5d6858ee49a7c939d4e408c93f4abdd9be6a31b9ff66de4a7d63ca18e25397ab2  aws-neuron-driver-2.19.64.0.deb
"