Add search order and video length

1 files changed, 11 insertions, 2 deletions

diff --git a/grimtube.go b/grimtube.go
index 17de1b6..8c27490 100644
--- a/grimtube.go
+++ b/grimtube.go
@@ -17,6 +17,10 @@ func getLangs() []string {
 	return []string{"en", "fr", "de", "ja", "ru"}
 }
 
+func getOrders() []string {
+	return []string{"relevance", "date", "views", "rating"}
+}
+
 func serve(w http.ResponseWriter, templatePath string, data interface{}) {
 	funcMap := template.FuncMap {
 		"inc": func(i int) int {
@@ -52,6 +56,7 @@ func search(w http.ResponseWriter, r *http.Request) {
 		term := query.Get("term")
 		sPage := query.Get("page")
 		lang := query.Get("lang")
+		order := query.Get("order")
 		var page int
 		if sPage == "" {
 			page = 0
@@ -63,8 +68,8 @@ func search(w http.ResponseWriter, r *http.Request) {
 				page = p
 			}
 		}
-		log.Printf("searching: %s, page: %d, lang: %s\n", term, page, lang)
-		items, err := ytparser.Search(term, page, lang)
+		log.Printf("searching: %s, page: %d, lang: %s, order: %s\n", term, page, lang, order)
+		items, err := ytparser.Search(term, page, lang, order)
 		if err != nil {
 			log.Println(err)
 			data := struct {
@@ -80,12 +85,16 @@ func search(w http.ResponseWriter, r *http.Request) {
 				Page int
 				Lang string
 				Langs []string
+				Order string
+				Orders []string
 			}{
 				items,
 				term,
 				page,
 				lang,
 				getLangs(),
+				order,
+				getOrders(),
 			}
 			serve(w, "templates/search.html", data)
 		}