PKGBUILD.template

# Maintainer: Hauke Rehfeld <aur@haukerehfeld.de>
pkgname="${_pkgbase}-${_model}"

# our own artificial version number, but per model
pkgver=4
pkgrel=4
pkgdesc="This is an autogenerated file, please see https://github.com/hrehfeld/archlinux-whisper.cpp-model"
arch=("i686" "x86_64")
url="${_baseurl}/tree/master/models"
license=("MIT")

makedepends=()
depends=()
conflicts=()
provides=()

_model_file="ggml-${_model}.bin"
_model_path="/usr/share/$pkgname/${_model_file}"

__model_url() {
    local model=$1
    # copied from https://github.com/ggerganov/whisper.cpp/raw/master/models/download-ggml-model.sh
    local src="https://huggingface.co/ggerganov/whisper.cpp"
    local pfx="resolve/main/ggml"
    if echo "$model" | grep -q "tdrz"; then
        local src="https://huggingface.co/akashmjn/tinydiarize-whisper.cpp"
        local pfx="resolve/main/ggml"
    fi

    echo $src/$pfx-"$model".bin

}

source=("$_model_file::$(__model_url $_model)")
sha256sums=("$_model_sha256sum")


package() {
  install -Dm644 "${srcdir}/${_model_file}" "$pkgdir$_model_path"

  wrapper="whisper.cpp-${_model}"
  echo "#!/bin/sh
/usr/bin/whisper.cpp --model ${_model_path} "\"'$@'\" > "$srcdir/$wrapper"
  install -Dm755 "${srcdir}/$wrapper" "$pkgdir/usr/bin/$wrapper"


}