pkgs/aumo: Correctly deal with an unset argument

1 files changed, 11 insertions, 2 deletions

diff --git a/pkgs/by-name/au/aumo/aumo.sh b/pkgs/by-name/au/aumo/aumo.sh
index d86165b2..dd930178 100755
--- a/pkgs/by-name/au/aumo/aumo.sh
+++ b/pkgs/by-name/au/aumo/aumo.sh
@@ -15,6 +15,10 @@ die() {
     fi
 }
 
+usage() {
+    echo "Usage: $NAME mount|unmount"
+}
+
 get_mounted_labels() {
     findmnt --output label --json | jq '.filesystems | map(.label) | sort | unique | map(select(. != null))'
 }
@@ -45,14 +49,19 @@ mounting() {
     udisksctl mount --block-device "/dev/disk/by-label/$disk_name"
 }
 
-case "$1" in
+case "${1-unset}" in
 "mount")
     mounting
     ;;
 "unmount" | "umount")
     unmounting
     ;;
+"unset")
+    usage
+    die "You need to provide one argument."
+    ;;
 *)
-    die "Usage: $NAME mount|unmount"
+    usage
+    die "Unknown command: '$1'"
     ;;
 esac