serv/tube
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Improve video id handling in import to be more user friendly

Browse source
This commit is contained in:
James Mills 2020-04-01 22:13:02 +10:00
parent 9bbfb420f9
commit 4e203eb773
No known key found for this signature in database
GPG key ID: AC4C014F1440EBD6
4 changed files with 10 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
app/rice-box.go
View file

File diff suppressed because one or more lines are too long

4
importers/importer.go
View file

@ -23,9 +23,9 @@ type Importer interface {
}
func NewImporter(url string) (Importer, error) {
if strings.Contains(url, "youtube.com") || strings.HasPrefix(url, "youtube:") {
if strings.Contains(url, "youtube.com") || strings.HasPrefix(strings.ToLower(url), "youtube:") {
return &YoutubeImporter{}, nil
} else if strings.Contains(url, "vimeo.com") || strings.HasPrefix(url, "vimeo:") {
} else if strings.Contains(url, "vimeo.com") || strings.HasPrefix(strings.ToLower(url), "vimeo:") {
return &VimeoImporter{}, nil
} else {
return nil, ErrUnsupportedVideoURL

4
importers/vimeo_importer.go
View file

@ -10,8 +10,8 @@ import (
type VimeoImporter struct{}
func (i *VimeoImporter) GetVideoInfo(url string) (videoInfo VideoInfo, err error) {
if strings.HasPrefix(url, "vimeo:") {
url = strings.TrimPrefix(url, "vimeo:")
if strings.HasPrefix(strings.ToLower(url), "vimeo:") {
url = strings.TrimSpace(strings.SplitN(url, ":", 2)[1])
}
if !strings.HasPrefix(url, "http") {

4
importers/youtube_importer.go
View file

@ -10,8 +10,8 @@ import (
type YoutubeImporter struct{}
func (i *YoutubeImporter) GetVideoInfo(url string) (videoInfo VideoInfo, err error) {
if strings.HasPrefix(url, "youtube:") {
url = strings.TrimPrefix(url, "youtube:")
if strings.HasPrefix(strings.ToLower(url), "youtube:") {
url = strings.TrimSpace(strings.SplitN(url, ":", 2)[1])
}
info, err := ytdl.GetVideoInfo(url)