hlfw.ca

ytcli

Download patch

ref: 88aeb8b6ebf3934430be9ab8e629201499eb1aba
parent: 37568a19faabb44f930b1b27dc1faa8c415d27dd
author: Halfwit <michaelmisch1985@gmail.com>
date: Fri Feb 16 14:58:06 PST 2018 

Add some curl and some jshon, let's do this

Signed-off-by: Halfwit <michaelmisch1985@gmail.com>


--- a/README.md
+++ b/README.md
@@ -3,12 +3,11 @@
 ## Usage
 
 ```
-ytcli [action]
+ytcli [-t] [action]
 
 Actions:
 	search	<keyword>
 		will return a list of URLs and descritions
-		using -t will also return the path to a thumbnail (in /tmp)
 	user	<username>
 		will return a list of videos by user
 	channel	<channel name>
@@ -15,6 +14,8 @@
 		will return a list of videos by channel
 	channel-id <channel name>
 		will return channel id of given channel
+	-t 
+		return thumbnail as well
 ```
 
 ## Configuration
@@ -28,5 +29,4 @@
 # Lines starting with # will be ignored
 key=4TNSHS543tn5hNSTH54354
 max_results=50
-tmpdir=/tmp/yt
 ```
--- a/ytcli
+++ b/ytcli
@@ -6,16 +6,89 @@
 # TODO: Fetch channel ID
 # TODO: Fetch images
 
+CFG="$XDG_CONFIG_HOME/ytcli/config"
+API="https://www.googleapis.com/youtube/v3/search?part=snippet"
+THUMB=
+
 usage() {
 cat <<USAGE
-Usage: 
-	ytctl [search|search -t] KEYWORD
-	ytctl [user|channel|channel-id] KEYWORD
-
 Will return a list of [thumbs], URLs, and Descriptions for a given search.
+'chanid' will return the ID of a named channel
+Usage: 
+	ytctl [-t] [search|playlist|user|channel|chanid] KEYWORD
 USAGE
 }
 
+# Helper functions
+key() {
+	sed -n 's/key=//p' "$CFG"
+}
+
+max_results() {
+	sed -n 's/max_results=//p' "$CFG"
+}
+
+parse () {
+	jshon -CQ -e items -a -e snippet -e title -uppe id -e "$1"Id -u
+}
+
+parse_t () {
+	jshon -CQ -e items -a -e snippet -e title -uppe id -e "$1"Id -uppe snippet -e thumbnails -e high -e url -u
+}
+
+# We get back just a token from the search, so append it to create a full url to the video
+fixurl_t() {
+	while {
+		read -r description
+		read -r url
+		read -r image
+	} do
+		printf '%s\n%s\n%s\n' "$description" "https://www.youtube.com/watch?v=$url" "$image"
+	done
+}
+
+fixurl() {
+	while {
+		read -r description
+		read -r url
+	} do
+		printf '%s\n%s\n' "$description" "https://www.youtube.com/watch?v=$url"
+	done
+}
+
+get() {
+	if test 0 -lt ${#THUMB} ; then 
+		curl -s "$1" | parse_t "$2" | fixurl_t
+	else
+		curl -s "$1" | parse "$2"   | fixurl
+	fi
+}
+
+chanid() {
+	url="https://www.googleapis.com/youtube/v3/search?part=id%2Csnippet&q=$@&type=channel&key=$(key)" 
+	curl -s "$url" | jshon -CQ -e items -e id -e id -e channelId -u
+}
+
+search() {
+	get "$API&maxResults=$(max_results)&key=$(key)&type=video&q=$@" video
+}
+
+playlist() {
+	get "$API&maxResults=$(max_results)&key=$(key)&type=playlist&q=$@" playlist
+}
+
+user() {
+	url="https://www.googleapis.com/youtube/v3/channels?part=contentDetails&forUsername=$@&key=$(key)"
+	pid="$(curl -s "$url" | jshon -CQ -e items -e -id -e id -u)"
+	get "https://www.googleapis.com/youtube/v3/playlistItems?part=snippet&maxResults=$(max_results)&playlistId=$pid&key=$(key)" video
+}
+
+channel() {
+	get "$API&maxResults=$(max_results)&key=$(key)&type=video&channelId=$(chanid "$@")" channel
+}
+
+## Main ## 
+
 # Scrub out our flag
 if test $2 = "-t"; then
 	THUMB="$1"
@@ -24,9 +97,10 @@
 fi
 
 case $1 in
-	search)		shift && search 	"$@" ;;
+	search)		shift && search		"$@" ;;
+	playlist)   shift && playlist	"$@" ;;
 	user)		shift && user   	"$@" ;;
 	channel) 	shift && channel 	"$@" ;;
-	channel-id)	shift && channel-id "$@" ;;
+	channel-id)	shift && chanid	"$@" ;;
 	*) usage ;;
 esac