# Maintainer: Pieter Robyns <pieter.robyns@uhasselt.be>
# Upstream URL: https://github.com/tensorflow/tensorflow

pkgname=tensorflow-git
pkgver=0.5.0.r9.g41930f0
pkgrel=1
pkgdesc="Open source software library for numerical computation using data flow graphs."
arch=('i686' 'x86_64')
url="https://tensorflow.org/"
license=('Apache2')
provides=('tensorflow')
conflicts=('tensorflow' 'python2-tensorflow')
depends=('python2-numpy' 'swig' 'python2-wheel')
makedepends=('git' 'python2-pip')
source=("git+https://github.com/tensorflow/tensorflow"
        "git+https://github.com/bazelbuild/bazel")
md5sums=('SKIP' 'SKIP')

pkgver() {
  cd "$srcdir/tensorflow"
  git describe --long --tags | sed 's/\([^-]*-g\)/r\1/;s/-/./g;s/^v//'
}

prepare() {
  # Build bazel
  echo "Please note: currently, bazel version 0.1.0 is required to build this package. Building it temporarily..."
  cd "$srcdir/bazel"
  git checkout tags/0.1.0
  ./compile.sh
  export TMP_BAZEL_EXECUTABLE=`pwd`/output/bazel

  # Clean up tmp files and install tensorflow
  rm -rf /tmp/tensorflow_pkg/
  cd "$srcdir/tensorflow"
  git submodule init
  git config submodule.protobuf.url $srcdir/google/protobuf
  git submodule update
  ./configure
}

build() {
  # Build tensorflow
  cd "$srcdir/tensorflow"

  export PYTHON=python2
  ${TMP_BAZEL_EXECUTABLE} build -c opt //tensorflow/tools/pip_package:build_pip_package

  sed -i 's/python/python2/g' bazel-bin/tensorflow/tools/pip_package/build_pip_package
  bazel-bin/tensorflow/tools/pip_package/build_pip_package /tmp/tensorflow_pkg
}

package() {
  cd "$srcdir/tensorflow"

  TMP_PKG=`find /tmp/tensorflow_pkg -name "tensor*.whl"`
  pip2 install --root $pkgdir/ $TMP_PKG --no-dependencies
  install -Dm644 LICENSE "$pkgdir/usr/share/licenses/$pkgname/LICENSE"
}