1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73

//! Data retrieval related routes.
//!
//! To be mounted on "/api/v1/check".


use self::super::super::super::setup::{SpecificLeaderboardConfig, LeaderboardGroupSettings, LeaderboardSettings, DatabaseConnection, LeaderboardConfig,
                                       ActivityCache};
use self::super::super::super::{SanitisedUserData, SudokuSolution, LeaderboardOf, User};
use self::super::super::super::errors::{GenericErrorSeverity, GenericError};
use rocket::response::status::Custom;
use rocket::request::State;
use rocket_contrib::Json;
use rocket::http::Status;


/// Get default scores
#[get("/leaderboard")]
pub fn leaderboard_specless(db: DatabaseConnection, settings: State<LeaderboardSettings>)
                            -> Result<Result<Json<Vec<SudokuSolution>>, Json<Vec<SanitisedUserData>>>, Custom<Json<GenericError>>> {
    leaderboard(db, settings, None)
}

/// Get scores
#[get("/leaderboard?<spec>")]
pub fn leaderboard(db: DatabaseConnection, settings: State<LeaderboardSettings>, spec: Option<SpecificLeaderboardConfig>)
                   -> Result<Result<Json<Vec<SudokuSolution>>, Json<Vec<SanitisedUserData>>>, Custom<Json<GenericError>>> {
    match spec.as_ref().map(|s| s.for_whom).unwrap_or_else(|| LeaderboardOf::Default) {
        LeaderboardOf::Solutions => leaderboard_solutions(db, &settings.board, spec.map(|s| s.cfg)),
        LeaderboardOf::Users => leaderboard_users(db, &settings.person, spec.map(|s| s.cfg)),
    }
}

/// Get active user count
#[get("/active_users")]
pub fn active_users(ac: State<ActivityCache>) -> Json<usize> {
    Json(ac.active_users())
}

fn leaderboard_solutions(db: DatabaseConnection, settings: &LeaderboardGroupSettings, spec: Option<LeaderboardConfig>)
                         -> Result<Result<Json<Vec<SudokuSolution>>, Json<Vec<SanitisedUserData>>>, Custom<Json<GenericError>>> {
    SudokuSolution::leaders(&if let Some(spec) = spec {
                                if spec > settings.max {
                                    LeaderboardConfig { count: settings.max.count, ..spec }
                                } else {
                                    spec
                                }
                            } else {
                                settings.default
                            },
                            &db)
        .map(Json)
        .map(Ok)
        .map_err(|e| Custom(Status::InternalServerError, Json((e, GenericErrorSeverity::Danger).into())))
}


fn leaderboard_users(db: DatabaseConnection, settings: &LeaderboardGroupSettings, spec: Option<LeaderboardConfig>)
                     -> Result<Result<Json<Vec<SudokuSolution>>, Json<Vec<SanitisedUserData>>>, Custom<Json<GenericError>>> {
    User::leaders(&if let Some(spec) = spec {
                      if spec > settings.max {
                          LeaderboardConfig { count: settings.max.count, ..spec }
                      } else {
                          spec
                      }
                  } else {
                      settings.default
                  },
                  &db)
        .map(|ls| ls.into_iter().map(Into::into).collect())
        .map(Json)
        .map(Err)
        .map_err(|e| Custom(Status::InternalServerError, Json((e, GenericErrorSeverity::Danger).into())))
}