140 lines
2.9 KiB
Go
140 lines
2.9 KiB
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 string
|
|
Album AlbumPreview
|
|
ArtistPageURL 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"`
|
|
PrimaryArtist struct {
|
|
URL string
|
|
} `json:"primary_artist"`
|
|
}
|
|
}
|
|
}
|
|
|
|
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 = songData.Description.Plain
|
|
s.Credits = make(map[string]string)
|
|
s.Album.Name = songData.Album.Name
|
|
s.ArtistPageURL = utils.TrimURL(songData.PrimaryArtist.URL)
|
|
s.Album.URL = utils.TrimURL(songData.Album.URL)
|
|
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 (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
|
|
}
|