ref: b6c4030d1499955944c9977f4c2accd4dcabbace
dir: /call.go/
package main
import (
"flag"
"log"
"regexp"
"strings"
"google.golang.org/api/youtube/v3"
)
var yt = "https://www.youtube.com/watch?v="
func playlistSearch(query string, service *youtube.Service) map[string]string {
results := make(map[string]string)
call := service.Search.List([]string{"id,snippet"}).Q(query).MaxResults(*nresults).Fields("items(id(kind, playlistId), snippet(thumbnails/default/url, title))")
response, err := call.Do()
if err != nil {
log.Fatal(err)
}
for _, item := range response.Items {
if item.Id.Kind == "youtube#playlist" {
if *thumbs {
key := yt + item.Id.PlaylistId + " " + item.Snippet.Thumbnails.Default.Url
results[key] = item.Snippet.Title
} else {
results[yt + item.Id.PlaylistId] = item.Snippet.Title
}
}
}
return results
}
func channelSearch(query string, service *youtube.Service) map[string]string {
// Fetch channel ID
results := make(map[string]string)
call := service.Search.List([]string{"id,snippet"}).Q(query).MaxResults(1).Fields("items/id/channelId")
id, err := call.Do()
if err != nil {
log.Fatal(err)
}
call2 := service.Search.List([]string{"id,snippet"}).MaxResults(*nresults).ChannelId(id.Items[0].Id.ChannelId).Fields("items(id/kind, id/videoId, snippet/thumbnails/default/url, snippet/title)")
response, err := call2.Do()
if err != nil {
log.Fatal(err)
}
for _, item := range response.Items {
if item.Id.Kind == "youtube#video" {
if *thumbs {
key := yt + item.Id.VideoId + " " + item.Snippet.Thumbnails.Default.Url
results[key] = item.Snippet.Title
} else {
results[yt + item.Id.VideoId] = item.Snippet.Title
}
}
}
return results
}
func normalSearch(query string, service *youtube.Service) map[string]string {
results := make(map[string]string)
call := service.Search.List([]string{"id,snippet"}).Q(query).MaxResults(*nresults).Fields("items(id/kind, id/videoId, snippet(title, thumbnails/default/url))")
response, err := call.Do()
if err != nil {
log.Fatal(err)
}
for _, item := range response.Items {
if item.Id.Kind == "youtube#video" {
if *thumbs {
key := yt + item.Id.VideoId + " " + item.Snippet.Thumbnails.Default.Url
results[key] = item.Snippet.Title
} else {
results[yt + item.Id.VideoId] = item.Snippet.Title
}
}
}
return results
}
func feedSearch(query string, service *youtube.Service) map[string]string {
results := make(map[string]string)
feed := "https://youtube.com/feeds/video.xml?channel_id="
if *channel {
call := service.Search.List([]string{"id,snippet"}).Q(query).MaxResults(*nresults).Fields("items(id/kind, id/channelId, snippet/title)")
response, err := call.Do()
if err != nil {
log.Fatal(err)
}
for _, item := range response.Items {
if item.Id.Kind == "youtube#channel" {
results[feed + item.Id.ChannelId] = item.Snippet.Title
}
}
} else {
// TODO: Remove unwanted group matches from this regex
// This was slapped out pretty quickly
r := regexp.MustCompile(`^((?:https?:)?\/\/)?((?:www|m)\.)?((?:youtube\.com|youtu.be))(\/(?:[\w\-]+\?v=|embed\/|v\/)?)([\w\-]+)(\S+)?$`)
vid := r.FindStringSubmatch(query)
call := service.Videos.List([]string{"id,snippet"}).Id(vid[5]).Fields("item/snippet(channelId, channelTitle)")
response, err := call.Do()
if err != nil {
log.Fatal(err)
}
for _, item := range response.Items {
results[feed + item.Snippet.ChannelId] = item.Snippet.ChannelTitle
}
}
return results
}
func relatedSearch(query string, service *youtube.Service) map[string]string{
results := make(map[string]string)
r := regexp.MustCompile(`^((?:https?:)?\/\/)?((?:www|m)\.)?((?:youtube\.com|youtu.be))(\/(?:[\w\-]+\?v=|embed\/|v\/)?)([\w\-]+)(\S+)?$`)
vid := r.FindStringSubmatch(query)
call := service.Search.List([]string{"id,snippet"}).RelatedToVideoId(vid[5]).MaxResults(*nresults).Type("video").Fields("items(id/videoId, snippet(thumbnails/default/url, title))")
response, err := call.Do()
if err != nil {
log.Fatal(err)
}
for _, item := range response.Items {
if *thumbs {
key := yt + item.Id.VideoId + " " + item.Snippet.Thumbnails.Default.Url
results[key] = item.Snippet.Title
} else {
results[yt + item.Id.VideoId] = item.Snippet.Title
}
}
return results
}
func runCommand(service *youtube.Service) map[string]string {
query := strings.Join(flag.Args(), " ")
switch {
case *feed:
return feedSearch(query, service)
case *playlist:
return playlistSearch(query, service)
case *channel:
return channelSearch(query, service)
case *related:
return relatedSearch(query, service)
case *username:
}
return normalSearch(query, service)
}