Add YouTube search (#241)

youtube/handler/youtube.go

@@ -0,0 +1,66 @@
+package handler
+
+import (
+	"context"
+	"strings"
+
+	"github.com/micro/micro/v3/service/errors"
+	"github.com/micro/micro/v3/service/logger"
+	pb "github.com/micro/services/youtube/proto"
+	"google.golang.org/api/option"
+	"google.golang.org/api/youtube/v3"
+)
+
+type Youtube struct {
+	Client *youtube.Service
+}
+
+func New(apiKey string) *Youtube {
+	ctx := context.TODO()
+	yt, _ := youtube.NewService(ctx, option.WithAPIKey(apiKey))
+
+	return &Youtube{
+		Client: yt,
+	}
+}
+
+func (y *Youtube) Search(ctx context.Context, req *pb.SearchRequest, rsp *pb.SearchResponse) error {
+	if len(req.Query) == 0 {
+		return errors.BadRequest("youtube.search", "missing query")
+	}
+
+	resp, err := y.Client.Search.List([]string{"id", "snippet"}).Q(req.Query).MaxResults(25).Do()
+	if err != nil {
+		logger.Errorf("failed to search youtube for %v: %v", req.Query, err)
+		return errors.InternalServerError("youtube.search", "Failed to search for "+req.Query)
+	}
+
+	for _, item := range resp.Items {
+		var id, url string
+		kind := strings.Split(item.Id.Kind, "#")[1]
+		switch kind {
+		case "video":
+			id = item.Id.VideoId
+			url = "https://www.youtube.com/watch?v=" + id
+		case "playlist":
+			id = item.Id.PlaylistId
+			url = "https://www.youtube.com/playlist?list=" + id
+		case "channel":
+			id = item.Id.ChannelId
+			url = "https://www.youtube.com/channel/" + id
+		}
+		rsp.Results = append(rsp.Results, &pb.SearchResult{
+			Id:           id,
+			Kind:         kind,
+			Title:        item.Snippet.Title,
+			ChannelId:    item.Snippet.ChannelId,
+			ChannelTitle: item.Snippet.ChannelTitle,
+			Description:  item.Snippet.Description,
+			PublishedAt:  item.Snippet.PublishedAt,
+			Broadcasting: item.Snippet.LiveBroadcastContent,
+			Url:          url,
+		})
+	}
+
+	return nil
+}