# This file is part of BlackArch Linux ( https://www.blackarch.org/ ).
# See COPYING for license details.

pkgname=artillery
pkgver=1.5.1.r218.g805a5d8
pkgrel=1
epoch=2
groups=('blackarch' 'blackarch-defensive' 'blackarch-honeypot')
pkgdesc='Blue team tool designed to protect Linux and Windows operating systems through multiple methods.'
arch=('any')
url='https://www.trustedsec.com/downloads/artillery/'
license=('BSD-3-Clause')
depends=('python' 'python-yaml')
makedepends=('git')
options=('!emptydirs')
source=("git+https://github.com/BinaryDefense/$pkgname.git"
        'artillery.service')
sha512sums=('SKIP'
            '13cac51ee7660f4605c1e4d1c4ae1988b221512fd68218bcde39539d584b986578a81180a8221b32438945e3136cf96d90dee2667b21c7174c3cc5b60559011c')

pkgver() {
  cd $pkgname

  ( set -o pipefail
    git describe --long --tags --abbrev=7 2>/dev/null |
      sed 's/\([^-]*-g\)/r\1/;s/-/./g' ||
    printf "%s.%s" "$(git rev-list --count HEAD)" \
      "$(git rev-parse --short=7 HEAD)"
  )
}

prepare() {
  cd $pkgname

  sed -i "s|/var|$pkgdir/var|g" setup.py
  sed -i "s|/etc|$pkgdir/etc|g" setup.py
  sed -i "s|/var/artillery|$pkgdir/usr/share/artillery|g" artillery.py \
    src/ssh_monitor.py src/monitor.py restart_server.py src/core.py \
    src/email_handler.py src/ftp_monitor.py src/monitor.py src/startup_artillery
  sed -i "s|/var/log/apache|/var/log/httpd|g" src/apache_monitor.py
}

package() {
  cd $pkgname

  install -dm 755 "$pkgdir/usr/bin"
  install -dm 755 "$pkgdir/usr/share/$pkgname"

  python setup.py <<< $'y\nn\nn'

  install -Dm 644 -t "$pkgdir/usr/share/doc/$pkgname" readme/*
  install -Dm 644 readme/LICENSE "$pkgdir/usr/share/licenses/$pkgname/LICENSE"

  rm readme/LICENSE

  mv "$pkgdir/var/artillery/"* "$pkgdir/usr/share/$pkgname"

  cat > "$pkgdir/usr/bin/$pkgname" << EOF
#!/bin/sh
if test \$EUID != 0 ; then
  echo 'Must be run as root.'
  exit 1
fi
exec python /usr/share/$pkgname/$pkgname.py "\$@"
EOF

  chmod +x "$pkgdir/usr/bin/$pkgname"

  # clean-up
  sed -i "s|$pkgdir/var|/var|g" setup.py
  sed -i "s|$pkgdir/etc|/etc|g" setup.py
  sed -i "s|$pkgdir/usr/share/artillery|/usr/share/artillery|g" \
    "$pkgdir/usr/share/artillery/artillery.py" \
    "$pkgdir/usr/share/artillery/src/ssh_monitor.py" \
    "$pkgdir/usr/share/artillery/src/monitor.py" \
    "$pkgdir/usr/share/artillery/restart_server.py" \
    "$pkgdir/usr/share/artillery/src/core.py" \
    "$pkgdir/usr/share/artillery/src/email_handler.py" \
    "$pkgdir/usr/share/artillery/src/ftp_monitor.py" \
    "$pkgdir/usr/share/artillery/src/monitor.py" \
    "$pkgdir/usr/share/artillery/src/startup_artillery"
}