trakt: add person search to movies

cmd/movies.go

@@ -65,6 +65,8 @@ var moviesCmd = &cobra.Command{
 			"genre":    flagGenre,
 			"year":     flagYear,
 			"rating":   flagRating,
+
+			"query": flagQueryStr,
 		}
 
 		// retrieve media
@@ -104,6 +106,7 @@ func init() {
 
 	moviesCmd.Flags().IntVar(&flagLimit, "limit", 0, "Max accepted items to add.")
 
+	moviesCmd.Flags().StringVar(&flagQueryStr, "query", "", "Query for search.")
 	moviesCmd.Flags().StringVar(&flagCountry, "country", "", "Countries to filter results.")
 	moviesCmd.Flags().StringVar(&flagLanguage, "language", "", "Languages to filter results.")
 	moviesCmd.Flags().StringVar(&flagGenre, "genre", "", "Genres to filter results.")

cmd/root.go

@@ -29,6 +29,7 @@ var (
 	flagSearchType string
 	flagLimit      int
 
+	flagQueryStr string
 	flagCountry  string
 	flagLanguage string
 	flagGenre    string

provider/trakt.go

@@ -12,6 +12,7 @@ import (
 	"github.com/sirupsen/logrus"
 	"go.uber.org/ratelimit"
 	"strconv"
+	"strings"
 	"time"
 )
 
@@ -73,6 +74,10 @@ type TraktMoviesResponse struct {
 	Movie *TraktMovie `json:"movie"`
 }
 
+type TraktPersonCastResponse struct {
+	Cast []TraktMoviesResponse
+}
+
 type TraktShowIds struct {
 	Trakt int    `json:"trakt"`
 	Slug  string `json:"slug"`
@@ -133,6 +138,7 @@ func NewTrakt() *Trakt {
 			SearchTypePopular,
 			SearchTypeNow,
 			SearchTypeWatched,
+			SearchTypePerson,
 		},
 	}
 }
@@ -224,6 +230,13 @@ func (p *Trakt) GetMovies(searchType string, logic map[string]interface{}, param
 		return p.getMovies("/movies/boxoffice", logic, params)
 	case SearchTypeWatched:
 		return p.getMovies("/movies/watched", logic, params)
+	case SearchTypePerson:
+		queryStr, ok := params["query"]
+		if !ok || queryStr == "" {
+			return nil, errors.New("person search must have a --query string provided, e.g. bryan-cranston")
+		}
+
+		return p.getMovies(fmt.Sprintf("/people/%s/movies", queryStr), logic, params)
 	default:
 		break
 	}
@@ -375,9 +388,21 @@ func (p *Trakt) getMovies(endpoint string, logic map[string]interface{}, params
 
 		// decode response
 		var s []TraktMoviesResponse
-		if err := resp.ToJSON(&s); err != nil {
-			_ = resp.Response().Body.Close()
-			return nil, errors.WithMessage(err, "failed decoding movies api response")
+
+		if !strings.Contains(endpoint, "/people/") {
+			// non person search
+			if err := resp.ToJSON(&s); err != nil {
+				_ = resp.Response().Body.Close()
+				return nil, errors.WithMessage(err, "failed decoding movies api response")
+			}
+		} else {
+			// person search
+			var tmp TraktPersonCastResponse
+			if err := resp.ToJSON(&tmp); err != nil {
+				_ = resp.Response().Body.Close()
+				return nil, errors.WithMessage(err, "failed decoding movies api response")
+			}
+			s = tmp.Cast
 		}
 
 		_ = resp.Response().Body.Close()

provider/types.go

@@ -13,5 +13,6 @@ const (
 	SearchTypeUpcoming string = "upcoming"
 	SearchTypePopular  string = "popular"
 	SearchTypeTrending string = "trending"
-	SearchTypeWatched string = "watched"
+	SearchTypeWatched  string = "watched"
+	SearchTypePerson   string = "person"
 )