1 files changed, 14 insertions, 41 deletions
diff --git a/rmt-search b/rmt-search
index ecf3594..b5db078 100755
--- a/rmt-search
+++ b/rmt-search
@@ -1,46 +1,19 @@
-#! /bin/bash
+#! /bin/zsh
 
-CHOICES=('YouTube' \
-         'Odysee' \
-         'PeerTube' \
-         '1337x' \
-         'Wallhaven')
+query_type='video'
+CH_SUFFIX='\C'
 
-[ ! -z "$1" ] && selected="$1" \
-              || selected="$(echo "${CHOICES[@]}" | tr ' ' '\n' \
-                                                  | dmenu -i \
-                                                          -nb '#ffffff' \
-                                                          -nf '#000080' \
-                                                          -sb '#000080' \
-                                                          -sf '#ffffff')"
+[ ! -z "$1" ] && query="$1" \
+              || query="$(print | dmenu -p 'Search Query')"
 
-case "$selected" in
-    "${CHOICES[0]}")
-        ./lib/search/"${CHOICES[0]}" "$2" "$3"
-    ;;
+if [[ "$query" == *"$CH_SUFFIX" ]]
+    then
+        query=${query%"$CH_SUFFIX"}
+        query_type='channel'
+fi
 
-    "${CHOICES[1]}")
-        ./lib/search/"${CHOICES[1]}" "$2" "$3"
-    ;;
-
-    "${CHOICES[2]}")
-        ./lib/search/"${CHOICES[2]}" "$2" "$3"
-    ;;
-
-    "${CHOICES[3]}")
-        ./lib/search/"${CHOICES[3]}" "$2" "$3"
-    ;;
-
-    "${CHOICES[4]}")
-        ./lib/search/"${CHOICES[4]}" "$2" "$3"
-    ;;
-
-    '')
-        exit 0
-    ;;
-
-    *)
-        ./lib/search/"${CHOICES[0]}" "$selected" "$2"
-    ;;
-esac
+[ -z "$query" ] && exit 0
 
+selected="$(ytfzf -D -L --type="$query_type" \
+                        --ii='https://invidious.flokinet.to' "$query")"
+[ ! -z "$selected" ] && rmt-action "$2" "$selected"