Fix crash on invalid playlist URL

pull/42/head
Matthieu Grieger 2015-02-02 19:45:17 -08:00
parent 00fb7ff05a
commit 7c57cbc3d9
2 changed files with 6 additions and 2 deletions

View File

@ -37,13 +37,15 @@ func NewPlaylist(user, id string) (*Playlist, error) {
if response, err := http.Get(jsonUrl); err == nil {
defer response.Body.Close()
if response.StatusCode != 400 {
if response.StatusCode != 400 && response.StatusCode != 404 {
if body, err := ioutil.ReadAll(response.Body); err == nil {
jsonString = string(body)
}
} else {
return nil, errors.New("Invalid YouTube ID supplied.")
}
} else {
return nil, errors.New("An error occurred while receiving HTTP GET request.")
}
jsonData := map[string]interface{}{}

View File

@ -39,13 +39,15 @@ func NewSong(user, id string) (*Song, error) {
if response, err := http.Get(jsonUrl); err == nil {
defer response.Body.Close()
if response.StatusCode != 400 {
if response.StatusCode != 400 && response.StatusCode != 404 {
if body, err := ioutil.ReadAll(response.Body); err == nil {
jsonString = string(body)
}
} else {
return nil, errors.New("Invalid YouTube ID supplied.")
}
} else {
return nil, errors.New("An error occurred while receiving HTTP GET request.")
}
jsonData := map[string]interface{}{}