dumb/data/lyrics.go

package data

import (
	"encoding/json"
	"fmt"
	"strings"

	"github.com/PuerkitoBio/goquery"
	"github.com/rramiachraf/dumb/utils"
)

type Song struct {
	Artist  string
	Title   string
	Image   string
	Lyrics  string
	Credits map[string]string
	About   [2]string
	Album   struct {
		URL   string
		Name  string
		Image string
	}
}

type songResponse struct {
	Response struct {
		Song struct {
			ArtistNames string `json:"artist_names"`
			Image       string `json:"song_art_image_thumbnail_url"`
			Title       string
			Description struct {
				Plain string
			}
			Album struct {
				URL   string `json:"url"`
				Name  string `json:"name"`
				Image string `json:"cover_art_url"`
			}
			CustomPerformances []customPerformance `json:"custom_performances"`
			WriterArtists []struct {
				Name string
			} `json:"writer_artists"`
			ProducerArtists    []struct {
				Name string
			} `json:"producer_artists"`
		}
	}
}

type customPerformance struct {
	Label   string
	Artists []struct {
		Name string
	}
}

func (s *Song) parseLyrics(doc *goquery.Document) error {
	var htmlError error

	doc.Find("[data-lyrics-container='true']").Each(func(i int, ss *goquery.Selection) {
		h, err := ss.Html()
		if err != nil {
			htmlError = err
		}
		s.Lyrics += h
	})

	if htmlError != nil {
		return htmlError
	}

	return nil
}

func (s *Song) parseSongData(doc *goquery.Document) error {
	attr, exists := doc.Find("meta[property='twitter:app:url:iphone']").Attr("content")
	if exists {
		songID := strings.Replace(attr, "genius://songs/", "", 1)

		u := fmt.Sprintf("https://genius.com/api/songs/%s?text_format=plain", songID)

		res, err := utils.SendRequest(u)
		if err != nil {
			return err
		}

		defer res.Body.Close()

		var data songResponse
		decoder := json.NewDecoder(res.Body)
		err = decoder.Decode(&data)
		if err != nil {
			return err
		}

		songData := data.Response.Song
		s.Artist = songData.ArtistNames
		s.Image = songData.Image
		s.Title = songData.Title
		s.About[0] = songData.Description.Plain
		s.About[1] = truncateText(songData.Description.Plain)
		s.Credits = make(map[string]string)
		s.Album.Name = songData.Album.Name
		s.Album.URL = strings.Replace(songData.Album.URL, "https://genius.com", "", -1)
		s.Album.Image = ExtractImageURL(songData.Album.Image)

		s.Credits["Writers"] = joinNames(songData.WriterArtists)
		s.Credits["Producers"] = joinNames(songData.ProducerArtists)
		for _, perf := range songData.CustomPerformances {
			s.Credits[perf.Label] = joinNames(perf.Artists)
		}
	}

	return nil
}

func joinNames(data []struct {
	Name string
}) string {
	var names []string
	for _, hasName := range data {
		names = append(names, hasName.Name)
	}
	return strings.Join(names, ", ")
}

func truncateText(text string) string {
	textArr := strings.Split(text, "")

	if len(textArr) > 250 {
		return strings.Join(textArr[0:250], "") + "..."
	}

	return text
}

func (s *Song) Parse(doc *goquery.Document) error {
	if err := s.parseLyrics(doc); err != nil {
		return err
	}

	if err := s.parseSongData(doc); err != nil {
		return err
	}

	return nil
}