Add playlist search support

lib/yt-search

@@ -61,7 +61,12 @@ function get_input() {
         get_input
     elif [ $n -gt 0 ] && [ $n -lt 20 ]
     then
+        if sed -n ${n}p $temp_file | grep "View full playlist" > /dev/null
+        then
+           sed -n ${n}p $temp_file | sed 's/.*&list=//' | sed 's/".*//' > $output
+        else
             sed -n ${n}p $temp_file | sed 's/.*<a href="\/watch?v=//' | sed 's/".*//' > $output
+        fi
     else
         echo "Bad input, mate. Type in a valid number or 'exit'."
         get_input
@@ -107,6 +112,7 @@ do
         type="Playlist"
         title=$(echo "$LINE" | sed 's/.*title="//' | sed 's/".*//')
         items=$(echo "$LINE" | sed 's/.*View full playlist (//' | sed 's/).*//')
+        itemid=$(echo "$LINE" | sed 's/.*&amp;list=//' | sed 's/".*//')
         if [ $item_num -ge $((results+1)) ]
         then
             break
@@ -118,8 +124,7 @@ do
         fi
         printf '%s\n' "$title"
         printf '    '
-        printf '%s | %s\n' "$type" "$items"
+        printf '%s | %s | %s\n' "$type" "$items" "$itemid"
-        printf '    '
     else
         type="Video"
         duration=$(echo "$LINE" | sed 's/.*Duration: //' | sed 's/\..*//')
@@ -145,7 +150,6 @@ do
         fi
         printf '%s\n' "$title"
         printf '    '
-        fi
     i=0
     while [ $i -lt 16 ]
     do
@@ -172,6 +176,7 @@ do
         fi
     done
     printf ' | %s\n' "$itemid"
+    fi
 done
 
 printf '\033[0m'

shelltube

@@ -39,8 +39,13 @@ fi
 search() {
     output="$(mktemp -u /tmp/yt_XXXXXX)"
     yt-search -i "$1" "$output"
+    if grep "^PL" "$output" > /dev/null
+    then
+        playlist "$(cat "$output")"
+    else
         selected_video="$(cat "$output")"
         yt-meta "$selected_video"
+    fi
 }
 
 playlist() {
@@ -151,7 +156,8 @@ parse() {
         ;;
     '' ) ;;
     * )
-        printf 'unknown command: "%s"\ntry "help"\n' "$command"
+        echo "Unknown command: \"$command\""
+        echo "Try \"help\" or \"?\""
         ;;
     esac
 }