Run 'cargo fmt'

pull/5/head
Charlotte Som 2021-07-18 01:33:22 +01:00
parent d7f0747090
commit 117ee9935e
3 changed files with 95 additions and 45 deletions

View File

@ -1,9 +1,8 @@
extern crate log; extern crate log;
use crate::file_io::{append_json, read_json, write_json};
use crate::user::User; use crate::user::User;
use crate::file_io::{read_json, append_json, write_json};
extern crate sha1; extern crate sha1;
#[get("/")] #[get("/")]
pub fn index() -> &'static str { pub fn index() -> &'static str {
"API Info: "API Info:
@ -25,20 +24,21 @@ pub fn index() -> &'static str {
#[post("/api/register/<name>/<pin>/<pronouns>")] #[post("/api/register/<name>/<pin>/<pronouns>")]
pub fn register_user(name: String, pin: i32, pronouns: String) -> String { pub fn register_user(name: String, pin: i32, pronouns: String) -> String {
let mut users: Vec<User> = read_json(); // Create an array of users out of parsed json let mut users: Vec<User> = read_json(); // Create an array of users out of parsed json
for i in &users { // loop through elements of the vector for i in &users {
// loop through elements of the vector
if i.name == name.to_lowercase() { if i.name == name.to_lowercase() {
warn!("Cannot create user {}! User is already in system.", i.name); warn!("Cannot create user {}! User is already in system.", i.name);
return "User already exists!".to_string(); return "User already exists!".to_string();
}; };
}; }
let pin_hashed = sha1::Sha1::from(&pin.to_string()).digest().to_string(); // hash the pin let pin_hashed = sha1::Sha1::from(&pin.to_string()).digest().to_string(); // hash the pin
users.push( User { users.push(User {
name: name.to_string().to_lowercase(), name: name.to_string().to_lowercase(),
pin_hashed: pin_hashed, pin_hashed: pin_hashed,
pronouns: pronouns.to_string().to_lowercase(), pronouns: pronouns.to_string().to_lowercase(),
session_token: "NULL".to_string() session_token: "NULL".to_string(),
}); // append the user to the vec }); // append the user to the vec
// append to the json file // append to the json file
@ -47,8 +47,16 @@ pub fn register_user(name: String, pin: i32, pronouns: String) -> String {
Ok(()) => info!("Succesfully appended to json"), Ok(()) => info!("Succesfully appended to json"),
}; };
info!("succesfully created user {} with pin hash {}", users[users.len()-1].name.to_string(), users[users.len()-1].pin_hashed); info!(
return format!("User {} registered with pin hash: {}", users[users.len()-1].name.to_string().to_lowercase(), users[users.len()-1].pin_hashed); "succesfully created user {} with pin hash {}",
users[users.len() - 1].name.to_string(),
users[users.len() - 1].pin_hashed
);
return format!(
"User {} registered with pin hash: {}",
users[users.len() - 1].name.to_string().to_lowercase(),
users[users.len() - 1].pin_hashed
);
} }
// Check if pin matches user // Check if pin matches user
@ -56,7 +64,8 @@ pub fn register_user(name: String, pin: i32, pronouns: String) -> String {
pub fn check_pin(name: String, pin: i32) -> String { pub fn check_pin(name: String, pin: i32) -> String {
let users: Vec<User> = read_json(); let users: Vec<User> = read_json();
let hashed_pin_input = sha1::Sha1::from(&pin.to_string()).digest().to_string(); let hashed_pin_input = sha1::Sha1::from(&pin.to_string()).digest().to_string();
for i in &users { // loop through the vector for i in &users {
// loop through the vector
if i.name == name.to_lowercase() { if i.name == name.to_lowercase() {
if i.pin_hashed == hashed_pin_input { if i.pin_hashed == hashed_pin_input {
info!("pin correct for user {}", i.name); info!("pin correct for user {}", i.name);
@ -66,8 +75,11 @@ pub fn check_pin(name: String, pin: i32) -> String {
return "Incorrect pin".to_string(); return "Incorrect pin".to_string();
}; };
}; };
}; }
warn!("cannot check pin for user {} as they do not exist", name.to_string().to_lowercase()); warn!(
"cannot check pin for user {} as they do not exist",
name.to_string().to_lowercase()
);
return format!("User {} does not exist.", name.to_string().to_lowercase()); return format!("User {} does not exist.", name.to_string().to_lowercase());
} }
@ -80,35 +92,60 @@ pub fn change(name: String, pin: i32, new_name: String, new_pin: i32) -> String
// Loop over elements in vector // Loop over elements in vector
for i in 0..users.len() { for i in 0..users.len() {
if users[i].name == name.to_lowercase() { // make sure name exists if users[i].name == name.to_lowercase() {
if users[i].pin_hashed == hashed_pin_input { // check if pin is correct // make sure name exists
if users[i].pin_hashed == hashed_pin_input {
// check if pin is correct
// Check wether to change name or name+pin // Check wether to change name or name+pin
if users[i].name == new_name.to_lowercase() { if users[i].name == new_name.to_lowercase() {
// check if new name already exists // check if new name already exists
users[i].pin_hashed = sha1::Sha1::from(&new_pin.to_string()).digest().to_string(); users[i].pin_hashed =
match write_json(&users) { sha1::Sha1::from(&new_pin.to_string()).digest().to_string();
Err(why) => panic!("Cannot write to json! {}", why), match write_json(&users) {
Ok(()) => info!("succesfully wrote to json file"), Err(why) => panic!("Cannot write to json! {}", why),
} Ok(()) => info!("succesfully wrote to json file"),
info!("Changed pin of {}", name.to_string().to_lowercase()); }
return format!("User {}'s new pin hash is {}.", name.to_string().to_lowercase(), users[i].pin_hashed); info!("Changed pin of {}", name.to_string().to_lowercase());
return format!(
"User {}'s new pin hash is {}.",
name.to_string().to_lowercase(),
users[i].pin_hashed
);
} else { } else {
// check if new name already exists // check if new name already exists
for n in &users { for n in &users {
if n.name == new_name.to_lowercase() { if n.name == new_name.to_lowercase() {
warn!("Could not change name of {} to {}, as new name is already taken.", name.to_lowercase(), new_name.to_lowercase()); warn!(
return format!("New name {} is already taken!", new_name.to_lowercase()); "Could not change name of {} to {}, as new name is already taken.",
name.to_lowercase(),
new_name.to_lowercase()
);
return format!(
"New name {} is already taken!",
new_name.to_lowercase()
);
} }
} }
users[i].name = new_name.to_string().to_lowercase(); users[i].name = new_name.to_string().to_lowercase();
users[i].pin_hashed = sha1::Sha1::from(&new_pin.to_string()).digest().to_string(); users[i].pin_hashed =
sha1::Sha1::from(&new_pin.to_string()).digest().to_string();
match write_json(&users) { match write_json(&users) {
Err(why) => panic!("couldn't write to json file! {}", why), Err(why) => panic!("couldn't write to json file! {}", why),
Ok(()) => info!("succesfully wrote to json file"), Ok(()) => info!("succesfully wrote to json file"),
} }
info!("Changed name of {} to {}. New pin hash is {}", name.to_string(), users[i].name.to_string(), users[i].pin_hashed.to_string()); info!(
return format!("User previously known as {} is now {}. Pin hash, if different, is {}", name.to_string(), users[i].name.to_string(), users[i].pin_hashed.to_string()); "Changed name of {} to {}. New pin hash is {}",
name.to_string(),
users[i].name.to_string(),
users[i].pin_hashed.to_string()
);
return format!(
"User previously known as {} is now {}. Pin hash, if different, is {}",
name.to_string(),
users[i].name.to_string(),
users[i].pin_hashed.to_string()
);
} }
} else { } else {
warn!("Incorrect pin given for user {}!", name.to_string()); warn!("Incorrect pin given for user {}!", name.to_string());
@ -116,14 +153,20 @@ pub fn change(name: String, pin: i32, new_name: String, new_pin: i32) -> String
} }
} }
} }
warn!("User {} not found, could not change pin and/or name.", name.to_string()); warn!(
"User {} not found, could not change pin and/or name.",
name.to_string()
);
return format!("User {} not found.", name.to_string()); return format!("User {} not found.", name.to_string());
} }
#[get("/api/users/<name>")] #[get("/api/users/<name>")]
pub fn get_user(name: String) -> String { pub fn get_user(name: String) -> String {
let users: Vec<User> = read_json(); let users: Vec<User> = read_json();
let found_user = users.iter().filter(|u| u.name == name.to_lowercase()).next(); let found_user = users
.iter()
.filter(|u| u.name == name.to_lowercase())
.next();
match found_user { match found_user {
Some(user) => format!("User {}", &user.name), Some(user) => format!("User {}", &user.name),
@ -135,7 +178,10 @@ pub fn get_user(name: String) -> String {
#[get("/api/about/name/<name>")] #[get("/api/about/name/<name>")]
pub fn get_user_name(name: String) -> String { pub fn get_user_name(name: String) -> String {
let users: Vec<User> = read_json(); let users: Vec<User> = read_json();
let found_user = users.iter().filter(|u| u.name == name.to_lowercase()).next(); let found_user = users
.iter()
.filter(|u| u.name == name.to_lowercase())
.next();
match found_user { match found_user {
Some(user) => user.name.to_string(), Some(user) => user.name.to_string(),
@ -146,7 +192,10 @@ pub fn get_user_name(name: String) -> String {
#[get("/api/about/pronouns/<name>")] #[get("/api/about/pronouns/<name>")]
pub fn get_user_pronouns(name: String) -> String { pub fn get_user_pronouns(name: String) -> String {
let users: Vec<User> = read_json(); let users: Vec<User> = read_json();
let found_user = users.iter().filter(|u| u.name == name.to_lowercase()).next(); let found_user = users
.iter()
.filter(|u| u.name == name.to_lowercase())
.next();
match found_user { match found_user {
Some(user) => user.pronouns.to_string(), Some(user) => user.pronouns.to_string(),

View File

@ -7,7 +7,9 @@ use crate::user::User;
use serde_json::Result; use serde_json::Result;
fn read_lines<P>(filename: P) -> io::Result<io::Lines<io::BufReader<File>>> fn read_lines<P>(filename: P) -> io::Result<io::Lines<io::BufReader<File>>>
where P: AsRef<Path>, { where
P: AsRef<Path>,
{
let file = File::open(filename)?; let file = File::open(filename)?;
Ok(io::BufReader::new(file).lines()) Ok(io::BufReader::new(file).lines())
} }
@ -35,22 +37,23 @@ pub fn read_json() -> Vec<User> {
} }
// Function to append the last value of the users vector to the file // Function to append the last value of the users vector to the file
pub fn append_json(users_list: &Vec<User>) -> Result<()> { pub fn append_json(users_list: &Vec<User>) -> Result<()> {
// Create a file to write to // Create a file to write to
let path = Path::new("users.json"); let path = Path::new("users.json");
let display = path.display(); let display = path.display();
let mut file = match OpenOptions::new() let mut file = match OpenOptions::new()
.write(true) .write(true)
.create(true) .create(true)
.append(true) .append(true)
.open(&path){ .open(&path)
{
Err(why) => panic!("couldn't create {}: {}", display, why), Err(why) => panic!("couldn't create {}: {}", display, why),
Ok(file) => file, Ok(file) => file,
}; };
// Serialize the last user value // Serialize the last user value
let users_json = serde_json::to_string(&users_list[users_list.len()-1])?; let users_json = serde_json::to_string(&users_list[users_list.len() - 1])?;
// Write to the file // Write to the file
match file.write_all(users_json.as_bytes()) { match file.write_all(users_json.as_bytes()) {
@ -71,19 +74,17 @@ pub fn write_json(users_list: &Vec<User>) -> Result<()> {
let path = Path::new("users.json"); let path = Path::new("users.json");
let display = path.display(); let display = path.display();
let mut file = match OpenOptions::new() let mut file = match OpenOptions::new().write(true).create(true).open(&path) {
.write(true)
.create(true)
.open(&path){
Err(why) => panic!("couldn't create {}: {}", display, why), Err(why) => panic!("couldn't create {}: {}", display, why),
Ok(file) => file, Ok(file) => file,
}; };
let mut users_json = String::new(); let mut users_json = String::new();
for i in 0..users_list.len() { for i in 0..users_list.len() {
// Serialize the users // Serialize the users
users_json += &serde_json::to_string(&users_list[i])?; users_json += &serde_json::to_string(&users_list[i])?;
if i != users_list.len()-1 { // don't append newline if it's the last element if i != users_list.len() - 1 {
// don't append newline if it's the last element
users_json += "\n"; users_json += "\n";
} }
} }

View File

@ -1,7 +1,7 @@
use uuid::Uuid;
use chrono::prelude::*;
use crate::user::User; use crate::user::User;
use chrono::prelude::*;
use serde::{Deserialize, Serialize}; use serde::{Deserialize, Serialize};
use uuid::Uuid;
#[derive(Deserialize, Serialize)] #[derive(Deserialize, Serialize)]
pub struct Message<'r> { pub struct Message<'r> {