1 files changed, 20 insertions, 9 deletions
diff --git a/PKGBUILD b/PKGBUILD
index baa64f12938e..cc63bfaa968f 100644
--- a/PKGBUILD
+++ b/PKGBUILD
@@ -9,15 +9,26 @@ license=('LicenseRef-WDDisclaimerofLiability')
 source=("$pkgname-$pkgver.zip::https://downloads.wdc.com/sata/${pkgname^^}v4.zip")
 sha512sums=('cc81b0f1ea0e92042b870d8484449d30a89b4d55230d24944f560a0cfb8fbd34cb754bd719cf5593ba79d022ef50718b8c4d9c37381d752e141d6e3c90b1f3f9')
 
-package() {
-	case "$CARCH" in
-		i686|pentium4)
-			install -Dm755 "$srcdir/${pkgname}x32" "$pkgdir/usr/bin/$pkgname"
-		;;
-		*)
-			install -Dm755 "$srcdir/${pkgname}x64" "$pkgdir/usr/bin/$pkgname"
-		;;
-	esac
+_sourcedirectory="$_projectname-$pkgver"
+
+case "$CARCH" in
+	i686|pentium4)
+		_binaryname="${pkgname}x32"
+	;;
+	*)
+		_binaryname="${pkgname}x64"
+	;;
+esac
+
+prepare() {
+	chmod +x "$srcdir/$_binaryname"
+}
 
+check() {
+	"$srcdir/$_binaryname" -? | tee '/dev/stderr' | grep -q "^${pkgname^^} Version"
+}
+
+package() {
+	install -Dm755 "$srcdir/$_binaryname" "$pkgdir/usr/bin/$pkgname"
 	install -Dm644 'Disclaimer of Liability - As-Is.txt' "$pkgdir/usr/share/licenses/$pkgname/WDDisclaimerofLiability"
 }