From 651b9487c1467a5be5aa44726a6b4d1f606dcc70 Mon Sep 17 00:00:00 2001 From: Santo Cariotti Date: Tue, 16 Mar 2021 14:17:45 +0100 Subject: feat: create and search email --- Cargo.lock | 19 ++++++++++++++ Cargo.toml | 2 ++ src/email/models.rs | 75 ++++++++++++++++++++++++++++++++++++++++++++++++++++- src/email/routes.rs | 55 ++++++++++++++++++++++++++++++++++++--- 4 files changed, 147 insertions(+), 4 deletions(-) diff --git a/Cargo.lock b/Cargo.lock index ad4fcfa..bcdd80e 100644 --- a/Cargo.lock +++ b/Cargo.lock @@ -913,6 +913,8 @@ dependencies = [ "config", "deadpool-postgres", "dotenv", + "hex", + "md-5", "regex", "serde 1.0.124", "slog", @@ -974,6 +976,12 @@ dependencies = [ "libc", ] +[[package]] +name = "hex" +version = "0.4.3" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "7f24254aa9a54b5c858eaee2f5bccdb46aaf0e486a595ed5fd8f86ba55232a70" + [[package]] name = "hmac" version = "0.9.0" @@ -1171,6 +1179,17 @@ version = "2.0.0" source = "registry+https://github.com/rust-lang/crates.io-index" checksum = "60302e4db3a61da70c0cb7991976248362f30319e88850c487b9b95bbf059e00" +[[package]] +name = "md-5" +version = "0.9.1" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "7b5a279bb9607f9f53c22d496eade00d138d1bdcccd07d74650387cf94942a15" +dependencies = [ + "block-buffer", + "digest", + "opaque-debug", +] + [[package]] name = "md5" version = "0.7.0" diff --git a/Cargo.toml b/Cargo.toml index ae7576f..8502fcb 100644 --- a/Cargo.toml +++ b/Cargo.toml @@ -25,3 +25,5 @@ serde = { version = "1.0.104", features = ["derive"] } chrono = { version = "0.4.19", features = ["serde"] } uuid = { version = "0.8.2", features = ["serde", "v4"] } regex = "1" +md-5 = "0.9.1" +hex = "0.4.3" diff --git a/src/email/models.rs b/src/email/models.rs index 477092d..ed46026 100644 --- a/src/email/models.rs +++ b/src/email/models.rs @@ -1,11 +1,14 @@ use crate::db::get_client; -use crate::errors::AppError; +use crate::errors::{AppError, AppErrorType}; use deadpool_postgres::Pool; use serde::{Deserialize, Serialize}; use tokio_pg_mapper::FromTokioPostgresRow; use tokio_pg_mapper_derive::PostgresMapper; +use hex; +use md5::{Digest, Md5}; + #[derive(Serialize, Deserialize, PostgresMapper)] #[pg_mapper(table = "email")] /// Emails model @@ -35,4 +38,74 @@ impl Email { Ok(emails) } + + /// Search an email + pub async fn search( + pool: Pool, + email: &String, + ) -> Result { + let client = get_client(pool.clone()).await.unwrap(); + + let statement = + client.prepare("SELECT * FROM email WHERE email=$1").await?; + + let email = client + .query_opt(&statement, &[&email]) + .await? + .map(|row| Email::from_row_ref(&row).unwrap()); + + match email { + Some(email) => Ok(email), + None => Err(AppError { + error_type: AppErrorType::NotFoundError, + cause: None, + message: Some("Email not found".to_string()), + }), + } + } + + /// Create new email + pub async fn create( + pool: Pool, + data: &EmailData, + ) -> Result { + // Search an email that matches with that string, because if it's + // exists, the server cannot create a clone + match Email::search(pool.clone(), &data.email).await { + Ok(_) => { + return Err(AppError { + message: Some("Email already exists".to_string()), + cause: Some("".to_string()), + error_type: AppErrorType::AuthorizationError, + }); + } + Err(_) => {} + }; + + let client = get_client(pool.clone()).await.unwrap(); + + let mut hasher = Md5::new(); + hasher.update(&data.email.as_bytes()); + let hash_final = hasher.finalize(); + + let digest = hex::encode(&hash_final.as_slice()); + + let statement = client + .prepare("INSERT INTO email VALUES ($1, $2) RETURNING *") + .await?; + + let email = client + .query_opt(&statement, &[&data.email, &digest]) + .await? + .map(|row| Email::from_row_ref(&row).unwrap()); + + match email { + Some(email) => Ok(email), + None => Err(AppError { + message: Some("Error creating a new email".to_string()), + cause: Some("Unknown error".to_string()), + error_type: AppErrorType::DbError, + }), + } + } } diff --git a/src/email/routes.rs b/src/email/routes.rs index 06b74fb..14299eb 100644 --- a/src/email/routes.rs +++ b/src/email/routes.rs @@ -1,7 +1,9 @@ +use std::collections::HashMap; + use crate::config::AppState; -use crate::email::models::Email; +use crate::email::models::{Email, EmailData}; use crate::errors::AppErrorResponse; -use actix_web::{web, HttpResponse, Responder}; +use actix_web::{web, HttpRequest, HttpResponse, Responder}; use slog::info; /// Endpoint used for retrieve all emails @@ -18,9 +20,56 @@ async fn index(state: web::Data) -> impl Responder { } } +// Endpoint used for create new email +async fn create_email( + payload: web::Json, + state: web::Data, +) -> impl Responder { + info!(state.log, "POST /email/"); + let result = Email::create(state.pool.clone(), &payload).await; + + result + .map(|email| HttpResponse::Created().json(email)) + .map_err(|e| e) +} + +// Endpoint used for email search +async fn search_email( + req: HttpRequest, + state: web::Data, +) -> impl Responder { + let query = + web::Query::>::from_query(req.query_string()) + .unwrap(); + let email = match query.get("q") { + Some(x) => x, + None => { + return HttpResponse::NotFound().json(AppErrorResponse { + detail: "No email found".to_string(), + }); + } + }; + let result = Email::search(state.pool.clone(), email).await; + info!(state.log, "GET /email/search?q={}", email); + + match result { + Ok(email) => HttpResponse::Ok().json(email), + _ => HttpResponse::NotFound().json(AppErrorResponse { + detail: "No email found".to_string(), + }), + } +} pub fn config(cfg: &mut web::ServiceConfig) { cfg.service( web::scope("/email") - .service(web::resource("{_:/?}").route(web::get().to(index))), + .service( + web::resource("{_:/?}") + .route(web::get().to(index)) + .route(web::post().to(create_email)), + ) + .service( + web::resource("/search{_:/?}") + .route(web::get().to(search_email)), + ), ); } -- cgit v1.2.3-18-g5258