From 15e2e6b54695b133b7d47672ac792c01212f38ce Mon Sep 17 00:00:00 2001
From: Lucas BEE <pouulet@gmail.com>
Date: Tue, 3 Jan 2017 12:09:43 +0000
Subject: [PATCH] Add search to the party

---
 src/internal/config/canape.go            |  9 ++++++
 src/internal/external_medias/handlers.go |  1 -
 src/internal/movies/handlers.go          | 39 ++++++++++++++++++++++++
 src/main.go                              |  1 +
 4 files changed, 49 insertions(+), 1 deletion(-)

diff --git a/src/internal/config/canape.go b/src/internal/config/canape.go
index 3db79ae..505d3b9 100644
--- a/src/internal/config/canape.go
+++ b/src/internal/config/canape.go
@@ -24,6 +24,7 @@ type Config struct {
 	TmdbAPIKey      string `yaml:"tmdb_api_key"`
 	MovieDetailers  []polochon.Detailer
 	MovieTorrenters []polochon.Torrenter
+	MovieSearchers  []polochon.Searcher
 	ShowDetailers   []polochon.Detailer
 	ShowTorrenters  []polochon.Torrenter
 }
@@ -71,6 +72,14 @@ func Load(path string) (*Config, error) {
 	}
 	cf.MovieTorrenters = append(cf.MovieTorrenters, d)
 
+	// Default searchers
+	cf.MovieSearchers = []polochon.Searcher{}
+	s, err := yts.NewSearcher()
+	if err != nil {
+		return nil, err
+	}
+	cf.MovieSearchers = append(cf.MovieSearchers, s)
+
 	// Default detailers
 	cf.ShowDetailers = []polochon.Detailer{}
 	showDetailer, err := tvdb.NewDetailer()
diff --git a/src/internal/external_medias/handlers.go b/src/internal/external_medias/handlers.go
index 1ca4f6e..c60f603 100644
--- a/src/internal/external_medias/handlers.go
+++ b/src/internal/external_medias/handlers.go
@@ -78,7 +78,6 @@ func GetMediaIDs(env *web.Env, mediaType string, source string, category string,
 		IDs:      ids,
 	}
 
-	log.Debugf("Inserting int to DB %+v", media)
 	err = media.Upsert(env.Database)
 	if err != nil {
 		return nil, err
diff --git a/src/internal/movies/handlers.go b/src/internal/movies/handlers.go
index be12419..70e0e76 100644
--- a/src/internal/movies/handlers.go
+++ b/src/internal/movies/handlers.go
@@ -1,6 +1,7 @@
 package movies
 
 import (
+	"errors"
 	"fmt"
 	"net"
 	"net/http"
@@ -109,3 +110,41 @@ func GetDetailsHandler(env *web.Env, w http.ResponseWriter, r *http.Request) err
 
 	return env.RenderJSON(w, m)
 }
+
+// SearchMovie will search movie
+func SearchMovie(env *web.Env, w http.ResponseWriter, r *http.Request) error {
+	key := r.FormValue("key")
+	if key == "" {
+		return env.RenderError(w, errors.New("no given key"))
+	}
+
+	var movies []*polochon.Movie
+	searchers := env.Config.MovieSearchers
+	for _, searcher := range searchers {
+		result, err := searcher.SearchMovie(key, env.Log)
+		if err != nil {
+			env.Log.Errorf("error while searching movie : %s", err)
+			continue
+		}
+		movies = append(movies, result...)
+	}
+
+	env.Log.Debugf("got %d movies doing search %q", len(movies), key)
+	movieList := []*Movie{}
+	for _, m := range movies {
+		movie := New(m.ImdbID)
+		err := movie.GetDetails(env, false)
+		if err != nil {
+			env.Log.Errorf("error while getting movie details : %s", err)
+			continue
+		}
+		err = movie.GetTorrents(env, false)
+		if err != nil {
+			env.Log.Errorf("error while getting movie torrents : %s", err)
+			continue
+		}
+		movieList = append(movieList, movie)
+	}
+
+	return env.RenderJSON(w, movieList)
+}
diff --git a/src/main.go b/src/main.go
index e717a60..e55496c 100644
--- a/src/main.go
+++ b/src/main.go
@@ -80,6 +80,7 @@ func main() {
 	env.Handle("/movies/{id:tt[0-9]+}/get_details", movies.GetDetailsHandler).WithRole(users.UserRole)
 	env.Handle("/movies/explore", extmedias.Explore)
 	env.Handle("/movies/refresh", extmedias.Refresh)
+	env.Handle("/movies/search", movies.SearchMovie)
 
 	// env.Handle("/shows/polochon", shows.FromPolochon).WithRole(users.UserRole)
 	env.Handle("/shows/{id:tt[0-9]+}", shows.GetDetailsHandler)