174 lines
4.3 KiB
Go
174 lines
4.3 KiB
Go
|
package imdb
|
||
|
|
||
|
import (
|
||
|
"encoding/json"
|
||
|
"errors"
|
||
|
"fmt"
|
||
|
//"log"
|
||
|
"net/http"
|
||
|
"strings"
|
||
|
)
|
||
|
|
||
|
//=======================================================================
|
||
|
// Const
|
||
|
//=======================================================================
|
||
|
const baseUrl string = "http://www.omdbapi.com/?"
|
||
|
const plot string = "full"
|
||
|
const tomatoes string = "true"
|
||
|
|
||
|
//=======================================================================
|
||
|
// Global vars
|
||
|
//=======================================================================
|
||
|
|
||
|
//=======================================================================
|
||
|
// Structs
|
||
|
//=======================================================================
|
||
|
|
||
|
//Type for the Search Response
|
||
|
type SearchResult struct {
|
||
|
Title string
|
||
|
Year string
|
||
|
ImdbId string
|
||
|
Type string
|
||
|
}
|
||
|
|
||
|
//Type that respond Search
|
||
|
type SearchResponse struct {
|
||
|
Search []SearchResult
|
||
|
Response string
|
||
|
Error string
|
||
|
}
|
||
|
|
||
|
//Type for searching a specific movie
|
||
|
type MovieResult struct {
|
||
|
Title string
|
||
|
Year string
|
||
|
Rated string
|
||
|
Released string
|
||
|
Runtime string
|
||
|
Genre string
|
||
|
Director string
|
||
|
Writer string
|
||
|
Actors string
|
||
|
Plot string
|
||
|
Language string
|
||
|
Country string
|
||
|
Awards string
|
||
|
Poster string
|
||
|
Metascore string
|
||
|
imdbRating string
|
||
|
imdbVotes string
|
||
|
imdbID string
|
||
|
Type string
|
||
|
tomatoMeter string
|
||
|
tomatoImage string
|
||
|
tomatoRating string
|
||
|
tomatoReviews string
|
||
|
tomatoFresh string
|
||
|
tomatoRotten string
|
||
|
tomatoConsensus string
|
||
|
tomatoUserMeter string
|
||
|
tomatoUserRating string
|
||
|
tomatoUserReviews string
|
||
|
DVD string
|
||
|
BoxOffice string
|
||
|
Production string
|
||
|
Website string
|
||
|
Response string
|
||
|
Error string
|
||
|
}
|
||
|
|
||
|
//=======================================================================
|
||
|
// Funcs
|
||
|
//=======================================================================
|
||
|
|
||
|
//Search for movies given a Title and year, Year is optional you can pass nil
|
||
|
func SearchMovies(title string, year string) (*SearchResponse, error) {
|
||
|
resp, err := omdbApiRequest(title, "", "", year)
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
|
||
|
r := new(SearchResponse)
|
||
|
err = json.NewDecoder(resp.Body).Decode(r)
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
if r.Response == "False" {
|
||
|
return r, errors.New(r.Error)
|
||
|
}
|
||
|
|
||
|
return r, nil
|
||
|
}
|
||
|
|
||
|
//returns a MovieResult given Title
|
||
|
func GetMovieByTitle(title string, year string) (*MovieResult, error) {
|
||
|
resp, err := omdbApiRequest("", "", title, year)
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
|
||
|
r := new(MovieResult)
|
||
|
err = json.NewDecoder(resp.Body).Decode(r)
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
if r.Response == "False" {
|
||
|
return r, errors.New(r.Error)
|
||
|
}
|
||
|
return r, nil
|
||
|
}
|
||
|
|
||
|
// returns a MovieResult given a ImdbId ex:"tt2015381"
|
||
|
func GetMovieByImdbId(id string) (*MovieResult, error) {
|
||
|
resp, err := omdbApiRequest("", id, "", "")
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
|
||
|
r := new(MovieResult)
|
||
|
err = json.NewDecoder(resp.Body).Decode(r)
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
if r.Response == "False" {
|
||
|
return r, errors.New(r.Error)
|
||
|
}
|
||
|
return r, nil
|
||
|
}
|
||
|
|
||
|
func omdbApiRequest(s string, i string, t string, y string) (resp *http.Response, err error) {
|
||
|
//s = Search Parameter, if this is != nil then its a searchMovies
|
||
|
//i = Id Parameter, if this is != nil then its a getMovieByImdbId
|
||
|
//t = Title Parameter, if this is != nil then its a getMovieByTitle
|
||
|
//y = Year Parameter, Optional data for s and t search
|
||
|
//var res http.Response
|
||
|
var url string
|
||
|
if s != "" {
|
||
|
s = strings.Replace(s, " ", "%20", -1)
|
||
|
url = fmt.Sprintf(baseUrl+"s=%s&y=%s", s, y)
|
||
|
} else if i != "" {
|
||
|
url = fmt.Sprintf(baseUrl+"i=%s&plot=%s&tomatoes=%s", i, plot, tomatoes)
|
||
|
} else if t != "" {
|
||
|
t = strings.Replace(t, " ", "%20", -1)
|
||
|
url = fmt.Sprintf(baseUrl+"t=%s&plot=%s&tomatoes=%s&y=%s", t, plot, tomatoes, y)
|
||
|
} else {
|
||
|
return nil, errors.New("Invalid Request")
|
||
|
}
|
||
|
//log.Print(url) //DEBUG
|
||
|
res, err := http.Get(url)
|
||
|
err = checkErrorStatus(res.StatusCode)
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
return res, nil
|
||
|
}
|
||
|
|
||
|
func checkErrorStatus(status int) error {
|
||
|
if status != 200 {
|
||
|
return errors.New(fmt.Sprintf("Status Code %d received from IMDB", status))
|
||
|
} else {
|
||
|
return nil
|
||
|
}
|
||
|
}
|