Initial Commit

parser.go

@@ -0,0 +1,167 @@
+package main
+
+import (
+	"encoding/json"
+	"log"
+	"strings"
+	"sync"
+
+	"github.com/tidwall/gjson"
+)
+
+func RunsText(j gjson.Result) string {
+
+	var s []string
+
+	a := j.Get("runs.#.text").Array()
+
+	for i := 0; i < len(a); i++ {
+		s = append(s, a[i].String())
+	}
+
+	return strings.Join(s, "")
+}
+
+func GetThumbnails(j gjson.Result) []Thumbnail {
+
+	t := []Thumbnail{}
+
+	wg := sync.WaitGroup{}
+
+	j.ForEach(
+		func(_, v gjson.Result) bool {
+
+			wg.Add(1)
+
+			go func() {
+				defer wg.Done()
+
+				u, err := ParseUrl(v.Get("url").String())
+				if err != nil {
+					log.Fatal(err)
+				}
+
+				t = append(t, Thumbnail{
+					Url:    u,
+					Width:  v.Get("width").Int(),
+					Height: v.Get("height").Int(),
+				})
+			}()
+
+			wg.Wait()
+
+			return true
+		},
+	)
+
+	return t
+}
+
+func GetSongs(s gjson.Result) []Item {
+
+	r := []Item{}
+
+	wg := sync.WaitGroup{}
+
+	s.ForEach(
+		func(_, v gjson.Result) bool {
+
+			wg.Add(1)
+
+			go func() {
+				defer wg.Done()
+
+				j := v.Get("musicResponsiveListItemRenderer")
+				f := j.Get("flexColumns.#.musicResponsiveListItemFlexColumnRenderer.text.runs.0")
+
+				r = append(r, Item{
+					Id:         j.Get("playlistItemData.videoId").String(),
+					Title:      f.Get("#(navigationEndpoint.watchEndpoint.videoId).text").String(),
+					Thumbnails: GetThumbnails(j.Get("thumbnail.musicThumbnailRenderer.thumbnail.thumbnails")),
+				})
+
+			}()
+
+			wg.Wait()
+
+			return true
+		},
+	)
+
+	return r
+}
+
+func GetTwoRowItem(t string, a gjson.Result) []Item {
+
+	r := []Item{}
+
+	var id string
+
+	if t == "album" {
+		id = "menu.menuRenderer.items.#(menuNavigationItemRenderer.text.runs.0.text == Shuffle play).menuNavigationItemRenderer.navigationEndpoint.watchPlaylistEndpoint.playlistId"
+	} else if t == "artist" {
+		id = "navigationEndpoint.browseEndpoint.browseId"
+	}
+
+	wg := sync.WaitGroup{}
+
+	a.ForEach(
+		func(_, v gjson.Result) bool {
+
+			wg.Add(1)
+
+			go func() {
+				defer wg.Done()
+
+				j := v.Get("musicTwoRowItemRenderer")
+
+				r = append(r, Item{
+					Id:         j.Get(id).String(),
+					Title:      RunsText(j.Get("title")),
+					Sub:        RunsText(j.Get("subtitle")),
+					Thumbnails: GetThumbnails(j.Get("thumbnailRenderer.musicThumbnailRenderer.thumbnail.thumbnails")),
+				})
+
+			}()
+
+			wg.Wait()
+
+			return true
+		},
+	)
+
+	return r
+}
+
+func ParseArtist(raw string) (string, error) {
+
+	j := gjson.Parse(raw)
+
+	h := j.Get("header.musicImmersiveHeaderRenderer")
+	c := j.Get("contents.singleColumnBrowseResultsRenderer.tabs.0.tabRenderer.content.sectionListRenderer.contents")
+
+	s := c.Get("#(musicShelfRenderer.title.runs.0.text == Songs).musicShelfRenderer")
+	a := c.Get("#(musicCarouselShelfRenderer.header.musicCarouselShelfBasicHeaderRenderer.title.runs.0.text == Albums).musicCarouselShelfRenderer")
+	u := c.Get("#(musicCarouselShelfRenderer.header.musicCarouselShelfBasicHeaderRenderer.title.runs.0.text == Fans might also like).musicCarouselShelfRenderer")
+
+	val := Artist{
+		Title:            RunsText(h.Get("title")),
+		Description:      RunsText(h.Get("description")),
+		SubscriberCount:  RunsText(h.Get("subscriptionButton.subscribeButtonRenderer.subscriberCountText")),
+		Thumbnails:       GetThumbnails(h.Get("thumbnail.musicThumbnailRenderer.thumbnail.thumbnails")),
+		BrowsePlaylistId: h.Get("playButton.buttonRenderer.navigationEndpoint.watchEndpoint.playlistId").String(),
+		PlaylistId:       s.Get("contents.0.musicResponsiveListItemRenderer.flexColumns.0.musicResponsiveListItemFlexColumnRenderer.text.runs.0.navigationEndpoint.watchEndpoint.playlistId").String(),
+		Items: Items{
+			Songs:   GetSongs(s.Get("contents")),
+			Albums:  GetTwoRowItem("album", a.Get("contents")),
+			Artists: GetTwoRowItem("artist", u.Get("contents")),
+		},
+	}
+
+	res, err := json.Marshal(val)
+	if err != nil {
+		return "", err
+	}
+
+	return string(res), nil
+}