125 lines
3.2 KiB
Rust
125 lines
3.2 KiB
Rust
|
/*
|
||
|
* Copyright (C) 2021 Aravinth Manivannan <realaravinth@batsense.net>
|
||
|
*
|
||
|
* This program is free software: you can redistribute it and/or modify
|
||
|
* it under the terms of the GNU Affero General Public License as
|
||
|
* published by the Free Software Foundation, either version 3 of the
|
||
|
* License, or (at your option) any later version.
|
||
|
*
|
||
|
* This program is distributed in the hope that it will be useful,
|
||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||
|
* GNU Affero General Public License for more details.
|
||
|
*
|
||
|
* You should have received a copy of the GNU Affero General Public License
|
||
|
* along with this program. If not, see <https://www.gnu.org/licenses/>.
|
||
|
*/
|
||
|
use std::str::FromStr;
|
||
|
|
||
|
use actix_identity::Identity;
|
||
|
use actix_web::{web, HttpResponse, Responder};
|
||
|
use futures::future::try_join_all;
|
||
|
use serde::{Deserialize, Serialize};
|
||
|
use uuid::Uuid;
|
||
|
|
||
|
use super::get_random;
|
||
|
use super::get_uuid;
|
||
|
use crate::errors::*;
|
||
|
use crate::AppData;
|
||
|
|
||
|
pub mod routes {
|
||
|
pub struct Benches {
|
||
|
pub submit: &'static str,
|
||
|
}
|
||
|
|
||
|
impl Benches {
|
||
|
pub const fn new() -> Benches {
|
||
|
let submit = "/api/v1/benches";
|
||
|
Benches { submit }
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
pub fn services(cfg: &mut web::ServiceConfig) {
|
||
|
cfg.service(submit);
|
||
|
}
|
||
|
|
||
|
#[derive(Serialize, Deserialize)]
|
||
|
struct Bench {
|
||
|
duration: f32,
|
||
|
difficulty: i32,
|
||
|
}
|
||
|
|
||
|
#[derive(Serialize, Deserialize)]
|
||
|
struct Submission {
|
||
|
device_user_provided: String,
|
||
|
device_software_recognised: String,
|
||
|
threads: i32,
|
||
|
benches: Vec<Bench>,
|
||
|
}
|
||
|
|
||
|
#[my_codegen::post(
|
||
|
path = "crate::V1_API_ROUTES.benches.submit",
|
||
|
wrap = "crate::CheckLogin"
|
||
|
)]
|
||
|
async fn submit(
|
||
|
data: AppData,
|
||
|
id: Identity,
|
||
|
payload: web::Json<Submission>,
|
||
|
) -> ServiceResult<impl Responder> {
|
||
|
let username = id.identity().unwrap();
|
||
|
|
||
|
let user_id = Uuid::from_str(&username).unwrap();
|
||
|
let payload = payload.into_inner();
|
||
|
|
||
|
sqlx::query!("INSERT INTO survey_responses (user_id, device_user_provided, device_software_recognised,
|
||
|
threads) VALUES ($1, $2, $3, $4)", &user_id, &payload.device_user_provided,
|
||
|
&payload.device_software_recognised, &payload.threads).execute(&data.db).await?;
|
||
|
|
||
|
struct ID {
|
||
|
id: i32,
|
||
|
}
|
||
|
|
||
|
let resp_id = sqlx::query_as!(
|
||
|
ID,
|
||
|
"SELECT ID FROM survey_responses
|
||
|
WHERE user_id = $1 AND device_software_recognised = $2",
|
||
|
&user_id,
|
||
|
&payload.device_software_recognised
|
||
|
)
|
||
|
.fetch_one(&data.db)
|
||
|
.await?;
|
||
|
|
||
|
let mut futs = Vec::with_capacity(payload.benches.len());
|
||
|
|
||
|
for bench in payload.benches.iter() {
|
||
|
let fut = sqlx::query!(
|
||
|
"INSERT INTO survey_benches
|
||
|
(resp_id, difficulty, duration) VALUES ($1, $2, $3)",
|
||
|
&resp_id.id,
|
||
|
&bench.difficulty,
|
||
|
bench.duration
|
||
|
)
|
||
|
.execute(&data.db);
|
||
|
|
||
|
futs.push(fut);
|
||
|
}
|
||
|
|
||
|
let submitions_id = get_uuid();
|
||
|
|
||
|
let fut = sqlx::query!(
|
||
|
"INSERT INTO survey_response_tokens (resp_id, user_id, id)
|
||
|
VALUES ($1, $2, $3)",
|
||
|
&resp_id.id,
|
||
|
&user_id,
|
||
|
&submitions_id
|
||
|
)
|
||
|
.execute(&data.db);
|
||
|
|
||
|
futs.push(fut);
|
||
|
|
||
|
try_join_all(futs).await?;
|
||
|
|
||
|
Ok(HttpResponse::Ok())
|
||
|
}
|