Skip to content

Add some missing fields in Playlist object #231

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
corny merged 3 commits into from
Feb 4, 2022
Merged

Add some missing fields in Playlist object #231

Changes from 2 commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
fd84986
Add thumbnails to PlaylistEntry
ItsClairton Jan 10, 2022
414f0a7
Add description field in Playlist object
ItsClairton Jan 11, 2022
d5b0d4d
Add description and thumbnail in playlist tests
ItsClairton Jan 23, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
40 changes: 23 additions & 17 deletions playlist.go
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -26,17 +26,19 @@ var (
)

type Playlist struct {
ID string
Title string
Author string
Videos []*PlaylistEntry
ID string
Title string
Description string
Author string
Videos []*PlaylistEntry
}

type PlaylistEntry struct {
ID string
Title string
Author string
Duration time.Duration
ID string
Title string
Author string
Duration time.Duration
Thumbnails Thumbnails
}

func extractPlaylistID(url string) (string, error) {
Expand Down Expand Up @@ -90,7 +92,6 @@ func extractPlaylistJSON(r io.Reader) ([]byte, error) {
// Duration: .lengthSeconds

// TODO?: Author thumbnails: sidebar.playlistSidebarRenderer.items[0].playlistSidebarPrimaryInfoRenderer.thumbnailRenderer.playlistVideoThumbnailRenderer.thumbnail.thumbnails
// TODO? Video thumbnails: .thumbnail.thumbnails

func (p *Playlist) UnmarshalJSON(b []byte) (err error) {
var j *sjson.Json
Expand All @@ -104,6 +105,7 @@ func (p *Playlist) UnmarshalJSON(b []byte) (err error) {
}
}()
p.Title = j.GetPath("metadata", "playlistMetadataRenderer", "title").MustString()
p.Description = j.GetPath("metadata", "playlistMetadataRenderer", "description").MustString()
p.Author = j.GetPath("sidebar", "playlistSidebarRenderer", "items").GetIndex(1).
GetPath("playlistSidebarSecondaryInfoRenderer", "videoOwner", "videoOwnerRenderer", "title", "runs").
GetIndex(0).Get("text").MustString()
Expand All @@ -128,10 +130,13 @@ func (p *Playlist) UnmarshalJSON(b []byte) (err error) {

type videosJSONExtractor struct {
Renderer *struct {
ID string `json:"videoId"`
Title withRuns `json:"title"`
Author withRuns `json:"shortBylineText"`
Duration string `json:"lengthSeconds"`
ID string `json:"videoId"`
Title withRuns `json:"title"`
Author withRuns `json:"shortBylineText"`
Duration string `json:"lengthSeconds"`
Thumbnail struct {
Thumbnails []Thumbnail `json:"thumbnails"`
} `json:"thumbnail"`
} `json:"playlistVideoRenderer"`
}

Expand All @@ -141,10 +146,11 @@ func (vje videosJSONExtractor) PlaylistEntry() *PlaylistEntry {
panic("invalid video duration: " + vje.Renderer.Duration)
}
return &PlaylistEntry{
ID: vje.Renderer.ID,
Title: vje.Renderer.Title.String(),
Author: vje.Renderer.Author.String(),
Duration: time.Second * time.Duration(ds),
ID: vje.Renderer.ID,
Title: vje.Renderer.Title.String(),
Author: vje.Renderer.Author.String(),
Duration: time.Second * time.Duration(ds),
Thumbnails: vje.Renderer.Thumbnail.Thumbnails,
}
}

Expand Down