package main import ( "fmt" "strings" "github.com/animenotifier/arn" "github.com/fatih/color" ) type stats map[string]float64 func main() { color.Yellow("Generating statistics") userStats := getUserStats() animeStats := getAnimeStats() arn.PanicOnError(arn.DB.Set("Cache", "user statistics", &arn.StatisticsCategory{ Name: "Users", PieCharts: userStats, })) arn.PanicOnError(arn.DB.Set("Cache", "anime statistics", &arn.StatisticsCategory{ Name: "Anime", PieCharts: animeStats, })) color.Green("Finished.") } func getUserStats() []*arn.PieChart { println("Generating user statistics") analytics, err := arn.AllAnalytics() arn.PanicOnError(err) screenSize := stats{} pixelRatio := stats{} browser := stats{} country := stats{} gender := stats{} os := stats{} notifications := stats{} activity := stats{} for _, info := range analytics { pixelRatio[fmt.Sprintf("%.1f", info.Screen.PixelRatio)]++ size := arn.ToString(info.Screen.Width) + " x " + arn.ToString(info.Screen.Height) screenSize[size]++ } for user := range arn.MustStreamUsers() { if user.Gender != "" { gender[user.Gender]++ } if user.Browser.Name != "" { browser[user.Browser.Name]++ } if user.Location.CountryName != "" { country[user.Location.CountryName]++ } if user.OS.Name != "" { if strings.HasPrefix(user.OS.Name, "CrOS") { user.OS.Name = "Chrome OS" } os[user.OS.Name]++ } if len(user.PushSubscriptions().Items) > 0 { notifications["Enabled"]++ } else { notifications["Disabled"]++ } if user.IsActive() { activity["Active last week"]++ } else { activity["Inactive"]++ } } println("Finished user statistics") return []*arn.PieChart{ arn.NewPieChart("OS", os), arn.NewPieChart("Screen size", screenSize), arn.NewPieChart("Browser", browser), arn.NewPieChart("Country", country), arn.NewPieChart("Activity", activity), arn.NewPieChart("Notifications", notifications), arn.NewPieChart("Gender", gender), arn.NewPieChart("Pixel ratio", pixelRatio), } } func getAnimeStats() []*arn.PieChart { println("Generating anime statistics") allAnime, err := arn.AllAnime() arn.PanicOnError(err) shoboi := stats{} anilist := stats{} mal := stats{} anidb := stats{} status := stats{} types := stats{} shoboiEdits := stats{} anilistEdits := stats{} malEdits := stats{} anidbEdits := stats{} rating := stats{} twist := stats{} for _, anime := range allAnime { for _, external := range anime.Mappings { if external.Service == "shoboi/anime" { if external.CreatedBy == "" { shoboiEdits["(auto-generated)"]++ } else { user, err := arn.GetUser(external.CreatedBy) arn.PanicOnError(err) shoboiEdits[user.Nick]++ } } if external.Service == "anilist/anime" { if external.CreatedBy == "" { anilistEdits["(auto-generated)"]++ } else { user, err := arn.GetUser(external.CreatedBy) arn.PanicOnError(err) anilistEdits[user.Nick]++ } } if external.Service == "myanimelist/anime" { if external.CreatedBy == "" { malEdits["(auto-generated)"]++ } else { user, err := arn.GetUser(external.CreatedBy) arn.PanicOnError(err) malEdits[user.Nick]++ } } if external.Service == "anidb/anime" { if external.CreatedBy == "" { anidbEdits["(auto-generated)"]++ } else { user, err := arn.GetUser(external.CreatedBy) arn.PanicOnError(err) anidbEdits[user.Nick]++ } } } if anime.GetMapping("shoboi/anime") != "" { shoboi["Connected with Shoboi"]++ } else { shoboi["Not connected with Shoboi"]++ } if anime.GetMapping("anilist/anime") != "" { anilist["Connected with AniList"]++ } else { anilist["Not connected with AniList"]++ } if anime.GetMapping("myanimelist/anime") != "" { mal["Connected with MyAnimeList"]++ } else { mal["Not connected with MyAnimeList"]++ } if anime.GetMapping("anidb/anime") != "" { anidb["Connected with AniDB"]++ } else { anidb["Not connected with AniDB"]++ } rating[arn.ToString(int(anime.Rating.Overall+0.5))]++ found := false for _, episode := range anime.Episodes().Items { if episode.Links != nil && episode.Links["twist.moe"] != "" { found = true break } } if found { twist["Connected with AnimeTwist"]++ } else { twist["Not connected with AnimeTwist"]++ } status[anime.Status]++ types[anime.Type]++ } println("Finished anime statistics") return []*arn.PieChart{ arn.NewPieChart("Type", types), arn.NewPieChart("Status", status), arn.NewPieChart("Rating", rating), arn.NewPieChart("MyAnimeList", mal), arn.NewPieChart("AniList", anilist), arn.NewPieChart("AniDB", anidb), arn.NewPieChart("Shoboi", shoboi), arn.NewPieChart("AnimeTwist", twist), // arn.NewPieChart("MyAnimeList Editors", malEdits), arn.NewPieChart("AniList Editors", anilistEdits), // arn.NewPieChart("AniDB Editors", anidbEdits), arn.NewPieChart("Shoboi Editors", shoboiEdits), } }