97 lines
2.1 KiB
Go
97 lines
2.1 KiB
Go
|
package spotify
|
||
|
|
|
||
|
|
import (
|
||
|
|
"git.nobrain.org/r4/dischord/extractor"
|
||
|
|
"git.nobrain.org/r4/dischord/extractor/youtube"
|
||
|
|
|
||
|
|
"errors"
|
||
|
|
"net/url"
|
||
|
|
"strings"
|
||
|
|
)
|
||
|
|
|
||
|
|
func init() {
|
||
|
|
extractor.AddExtractor("spotify", NewExtractor())
|
||
|
|
}
|
||
|
|
|
||
|
|
type matchType int
|
||
|
|
|
||
|
|
const (
|
||
|
|
matchTypeNone matchType = iota
|
||
|
|
matchTypeTrack
|
||
|
|
matchTypeAlbum
|
||
|
|
matchTypePlaylist
|
||
|
|
)
|
||
|
|
|
||
|
|
var (
|
||
|
|
ErrInvalidInput = errors.New("invalid input")
|
||
|
|
)
|
||
|
|
|
||
|
|
func matches(input string) (string, matchType) {
|
||
|
|
u, err := url.Parse(input)
|
||
|
|
if err != nil {
|
||
|
|
return "", matchTypeNone
|
||
|
|
}
|
||
|
|
if u.Scheme != "http" && u.Scheme != "https" {
|
||
|
|
return "", matchTypeNone
|
||
|
|
}
|
||
|
|
if u.Host != "open.spotify.com" {
|
||
|
|
return "", matchTypeNone
|
||
|
|
}
|
||
|
|
sp := strings.Split(u.Path, "/")
|
||
|
|
if len(sp) != 3 || sp[0] != "" {
|
||
|
|
return "", matchTypeNone
|
||
|
|
}
|
||
|
|
switch sp[1] {
|
||
|
|
case "track":
|
||
|
|
return sp[2], matchTypeTrack
|
||
|
|
case "album":
|
||
|
|
return sp[2], matchTypeAlbum
|
||
|
|
case "playlist":
|
||
|
|
return sp[2], matchTypePlaylist
|
||
|
|
}
|
||
|
|
return "", matchTypeNone
|
||
|
|
}
|
||
|
|
|
||
|
|
type Extractor struct {
|
||
|
|
ytSearcher *youtube.Searcher
|
||
|
|
ytSearcherConfig extractor.ProviderConfig
|
||
|
|
ytExtractor *youtube.Extractor
|
||
|
|
ytExtractorConfig extractor.ProviderConfig
|
||
|
|
token apiToken
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewExtractor() *Extractor {
|
||
|
|
extractor := &Extractor{}
|
||
|
|
extractor.ytSearcher = &youtube.Searcher{}
|
||
|
|
extractor.ytSearcherConfig = extractor.ytSearcher.DefaultConfig()
|
||
|
|
extractor.ytExtractor = &youtube.Extractor{}
|
||
|
|
extractor.ytExtractorConfig = extractor.ytExtractor.DefaultConfig()
|
||
|
|
return extractor
|
||
|
|
}
|
||
|
|
|
||
|
|
func (e *Extractor) DefaultConfig() extractor.ProviderConfig {
|
||
|
|
return extractor.ProviderConfig{}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (e *Extractor) Matches(cfg extractor.ProviderConfig, input string) bool {
|
||
|
|
_, m := matches(input)
|
||
|
|
return m != matchTypeNone
|
||
|
|
}
|
||
|
|
|
||
|
|
func (e *Extractor) Extract(cfg extractor.ProviderConfig, input string) ([]extractor.Data, error) {
|
||
|
|
id, m := matches(input)
|
||
|
|
switch m {
|
||
|
|
case matchTypeTrack:
|
||
|
|
d, err := getTrack(e, id)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
return []extractor.Data{d}, nil
|
||
|
|
case matchTypeAlbum:
|
||
|
|
return getAlbum(e, id)
|
||
|
|
case matchTypePlaylist:
|
||
|
|
return getPlaylist(e, id)
|
||
|
|
}
|
||
|
|
return nil, ErrInvalidInput
|
||
|
|
}
|