Compare commits

...

2 Commits

Author SHA1 Message Date
Sebastian H. Gabrielli
cde27b56d9 Add serde renames and make structs use proper rust variable names 2023-12-25 15:58:18 +01:00
Sebastian H. Gabrielli
56f1a1375e Move index to main & make ErrorResponder public 2023-12-25 15:51:31 +01:00
3 changed files with 54 additions and 41 deletions

View File

@ -7,6 +7,11 @@ use database::set_up_db;
mod webserver_member;
use webserver_member::*;
#[get("/")]
fn index() -> &'static str {
"Hello, world!\nNothing useful is served here."
}
#[launch]
async fn rocket() -> _ {
let db = match set_up_db().await {

View File

@ -1,20 +1,22 @@
use rocket::serde::{Serialize, Deserialize};
use sea_orm::*;
use super::entities::{prelude::*, *};
use super::entities::{*};
#[derive(Serialize, Deserialize)]
pub struct RfidCard {
pub cardId: String,
pub cardComment: String
#[serde(rename = "cardId")]
pub card_id: String,
#[serde(rename = "cardComment")]
pub card_comment: String
}
// Implement `.into()` to auto map values from the database model to the local struct
impl From<rfid_cards::Model> for RfidCard {
fn from(rfid_card: rfid_cards::Model) -> RfidCard {
RfidCard {
cardId: rfid_card.card_id,
cardComment: rfid_card.card_comment,
card_id: rfid_card.card_id,
card_comment: rfid_card.card_comment,
}
}
}
@ -22,13 +24,18 @@ impl From<rfid_cards::Model> for RfidCard {
#[derive(Serialize, Deserialize)]
pub struct Member {
pub id: i32,
pub ntnuUsername: String,
pub firstName: String,
pub lastName: String,
#[serde(rename = "ntnuUsername")]
pub ntnu_username: String,
#[serde(rename = "firstName")]
pub first_name: String,
#[serde(rename = "lastName")]
pub last_name: String,
pub email: String,
pub balance: i32,
pub imagePreference: String,
pub rfidCards: Vec<RfidCard>
#[serde(rename = "imagePreference")]
pub image_preference: String,
#[serde(rename = "lastName")]
pub rfid_cards: Vec<RfidCard>
}
// Create the `.into()` functionality to auto map values from the database model to the local struct
@ -37,13 +44,13 @@ impl From<members::Model> for Member {
fn from(member: members::Model) -> Member {
Member {
id: member.id,
ntnuUsername: member.ntnu_username,
firstName: member.first_name,
lastName: member.last_name,
ntnu_username: member.ntnu_username,
first_name: member.first_name,
last_name: member.last_name,
email: member.email,
balance: member.balance,
imagePreference: member.image_preference,
rfidCards: Vec::new(),
image_preference: member.image_preference,
rfid_cards: Vec::new(),
}
}
}
@ -53,9 +60,9 @@ impl From<members::Model> for MinimalMember {
fn from(member: members::Model) -> MinimalMember {
MinimalMember {
id: member.id,
ntnuUsername: member.ntnu_username,
firstName: member.first_name,
lastName: member.last_name,
ntnu_username: member.ntnu_username,
first_name: member.first_name,
last_name: member.last_name,
email: member.email,
}
}
@ -65,9 +72,9 @@ impl From<members::Model> for MinimalMember {
impl Into<members::ActiveModel> for rocket::serde::json::Json<MinimalMemberWithoutId> {
fn into(self) -> members::ActiveModel {
members::ActiveModel {
ntnu_username: ActiveValue::Set(self.ntnuUsername.to_owned()),
first_name: ActiveValue::Set(self.firstName.to_owned()),
last_name: ActiveValue::Set(self.lastName.to_owned()),
ntnu_username: ActiveValue::Set(self.ntnu_username.to_owned()),
first_name: ActiveValue::Set(self.first_name.to_owned()),
last_name: ActiveValue::Set(self.last_name.to_owned()),
email: ActiveValue::Set(self.email.to_owned()),
balance: ActiveValue::Set(0),
image_preference: ActiveValue::Set("".to_string()),
@ -79,17 +86,23 @@ impl Into<members::ActiveModel> for rocket::serde::json::Json<MinimalMemberWitho
#[derive(Serialize, Deserialize)]
pub struct MinimalMember {
pub id: i32,
pub ntnuUsername: String,
pub firstName: String,
pub lastName: String,
#[serde(rename = "ntnuUsername")]
pub ntnu_username: String,
#[serde(rename = "firstName")]
pub first_name: String,
#[serde(rename = "lastName")]
pub last_name: String,
pub email: String,
}
#[derive(Serialize, Deserialize)]
pub struct MinimalMemberWithoutId {
pub ntnuUsername: String,
pub firstName: String,
pub lastName: String,
#[serde(rename = "ntnuUsername")]
pub ntnu_username: String,
#[serde(rename = "firstName")]
pub first_name: String,
#[serde(rename = "lastName")]
pub last_name: String,
pub email: String,
}

View File

@ -15,8 +15,8 @@ use sea_orm::*;
#[derive(Responder)]
#[response(status = 500, content_type = "text/plain")]
struct ErrorResponder {
message: String
pub struct ErrorResponder {
pub message: String
}
impl From<DbErr> for ErrorResponder {
@ -35,11 +35,6 @@ impl From<&str> for ErrorResponder {
}
}
#[get("/")]
pub fn index() -> &'static str {
"Hello, world!\nNothing useful is served here."
}
#[get("/members")]
pub async fn get_members(db: &State<DatabaseConnection>) -> Result<Json<MultipleMembersStruct>, ErrorResponder> {
let db = db as &DatabaseConnection;
@ -62,7 +57,7 @@ pub async fn get_members(db: &State<DatabaseConnection>) -> Result<Json<Multiple
let rfid_cards: Vec<RfidCard> = rfid_cards.into_iter().map(RfidCard::from).collect();
// Add these cards to the member's rfidCards field
member.rfidCards = rfid_cards;
member.rfid_cards = rfid_cards;
}
@ -80,7 +75,7 @@ pub async fn add_member(db: &State<DatabaseConnection>, minimal_member_without_i
// Check if a member with the same NTNU username already exists
let matching_member: Option<members::Model> = Members::find() // Find a member in the "Members" table
.filter(members::Column::NtnuUsername.eq(minimal_member_without_id.ntnuUsername.to_owned())) // Filter by the provided username in the NtnuUsername column
.filter(members::Column::NtnuUsername.eq(minimal_member_without_id.ntnu_username.to_owned())) // Filter by the provided username in the NtnuUsername column
.one(db) // We only care about one result
.await?; // Wait for the result
@ -118,19 +113,19 @@ pub async fn add_member(db: &State<DatabaseConnection>, minimal_member_without_i
Ok(Json(member))
}
#[get("/member/<memberId>")]
pub async fn get_member_by_id(db: &State<DatabaseConnection>, memberId: i32) -> Result<Json<Member>, ErrorResponder> {
#[get("/member/<member_id>")]
pub async fn get_member_by_id(db: &State<DatabaseConnection>, member_id: i32) -> Result<Json<Member>, ErrorResponder> {
let db = db as &DatabaseConnection;
// Create an empty variable to store the resulting member info
let database_member;
// Search the database for the member based on the ID
match Members::find_by_id(memberId).one(db).await? {
match Members::find_by_id(member_id).one(db).await? {
Some(model) => database_member = model, // If the member is found, add the info to the previously created variable
None => {
return Err(
ErrorResponder {
message: format!("Failed to fetch member by ID. memberId: {}", memberId) // If it is not found return with this error
message: format!("Failed to fetch member by ID. memberId: {}", member_id) // If it is not found return with this error
}
);
},
@ -142,7 +137,7 @@ pub async fn get_member_by_id(db: &State<DatabaseConnection>, memberId: i32) ->
// Create a member from the database model and append the RFID cards
let mut member: Member = database_member.into();
member.rfidCards = rfids;
member.rfid_cards = rfids;
Ok(Json(member))
}