summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorSanto Cariotti <santo@dcariotti.me>2021-03-16 14:17:45 +0100
committerSanto Cariotti <santo@dcariotti.me>2021-03-16 14:17:45 +0100
commit651b9487c1467a5be5aa44726a6b4d1f606dcc70 (patch)
tree8ca64dd0c2fa37f3bccf9dc8bc15527d522c3964
parent2689fe31f9f86ba190faea99635e62f624e3e02d (diff)
feat: create and search email
-rw-r--r--Cargo.lock19
-rw-r--r--Cargo.toml2
-rw-r--r--src/email/models.rs75
-rw-r--r--src/email/routes.rs55
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",
@@ -975,6 +977,12 @@ dependencies = [
]
[[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"
source = "registry+https://github.com/rust-lang/crates.io-index"
@@ -1172,6 +1180,17 @@ 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"
source = "registry+https://github.com/rust-lang/crates.io-index"
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<Email, AppError> {
+ 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<Email, AppError> {
+ // 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<AppState>) -> impl Responder {
}
}
+// Endpoint used for create new email
+async fn create_email(
+ payload: web::Json<EmailData>,
+ state: web::Data<AppState>,
+) -> 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<AppState>,
+) -> impl Responder {
+ let query =
+ web::Query::<HashMap<String, String>>::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)),
+ ),
);
}