Add weekly rankings and ranking pagination support

api/daily.go

@@ -81,8 +81,19 @@ func (s *Server) HandleSeed(w http.ResponseWriter, r *http.Request) {
 
 func (s *Server) HandleRankings(w http.ResponseWriter, r *http.Request) {
 	var err error
+	var category int
 	var page int
 
+	if r.URL.Query().Has("category") {
+		category, err = strconv.Atoi(r.URL.Query().Get("category"))
+		if err != nil {
+			http.Error(w, fmt.Sprintf("failed to convert category: %s", err), http.StatusBadRequest)
+			return
+		}
+	} else {
+		category = 0
+	}
+
 	if r.URL.Query().Has("page") {
 		page, err = strconv.Atoi(r.URL.Query().Get("page"))
 		if err != nil {
@@ -93,7 +104,7 @@ func (s *Server) HandleRankings(w http.ResponseWriter, r *http.Request) {
 		page = 1
 	}
 
-	rankings, err := db.FetchRankings(page)
+	rankings, err := db.FetchRankings(category, page)
 	if err != nil {
 		log.Print("failed to retrieve rankings")
 	}
@@ -106,3 +117,33 @@ func (s *Server) HandleRankings(w http.ResponseWriter, r *http.Request) {
 
 	w.Write(response)
 }
+
+// /daily/rankingpagecount - fetch daily ranking page count
+
+func (s *Server) HandleRankingPageCount(w http.ResponseWriter, r *http.Request) {
+	var err error
+	var category int
+
+	if r.URL.Query().Has("category") {
+		category, err = strconv.Atoi(r.URL.Query().Get("category"))
+		if err != nil {
+			http.Error(w, fmt.Sprintf("failed to convert category: %s", err), http.StatusBadRequest)
+			return
+		}
+	} else {
+		category = 0
+	}
+
+	pageCount, err := db.FetchRankingPageCount(category)
+	if err != nil {
+		log.Print("failed to retrieve ranking page count")
+	}
+
+	response, err := json.Marshal(pageCount)
+	if err != nil {
+		http.Error(w, fmt.Sprintf("failed to marshal response json: %s", err), http.StatusInternalServerError)
+		return
+	}
+
+	w.Write(response)
+}

api/generic.go

@@ -47,6 +47,8 @@ func (s *Server) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 		s.HandleSeed(w, r)
 	case "/daily/rankings":
 		s.HandleRankings(w, r)
+	case "/daily/rankingpagecount":
+		s.HandleRankingPageCount(w, r)
 	}
 }