Charts V1

schaffer.go

@@ -7,6 +7,7 @@ import (
 	"strings"
 	"time"
 
+	"github.com/gookit/color"
 	"github.com/xujiajun/nutsdb"
 )
 
@@ -270,3 +271,40 @@ func CheckAnimeExistInDbAndUserWatches(animeId, userId int64) (bool, error) {
 	}
 	return false, err
 }
+
+func BuildMovieCharts() ([]MovieChart, error) {
+	key := "charts"
+	var charts []MovieChart
+	data, err := mmCache.Get(key)
+	if err == nil {
+		err = json.Unmarshal(data, &charts)
+		if err == nil {
+			return charts, err
+		}
+	}
+
+	movieList, err := MmReadCharts()
+	if err != nil {
+		return nil, err
+	}
+	charts = make([]MovieChart, 0)
+	for _, m := range movieList {
+		c := MovieChart{
+			MmId: m.Id,
+		}
+		anime, err := SearchAnime(m.Anime)
+		if err != nil {
+			color.Errorln(err.Error())
+		} else {
+			c.Data = *anime
+			charts = append(charts, c)
+		}
+	}
+
+	bytes, err := json.Marshal(charts)
+	if err == nil {
+		mmCache.Set(key, bytes)
+	}
+
+	return charts, nil
+}