Add database messages

jirs-server/src/db/invitations.rs

@@ -0,0 +1,131 @@
+use actix::{Handler, Message};
+use diesel::pg::Pg;
+use diesel::prelude::*;
+
+use jirs_data::{
+    EmailString, Invitation, InvitationId, InvitationState, ProjectId, UserId, UsernameString,
+};
+
+use crate::db::DbExecutor;
+use crate::errors::ServiceErrors;
+use crate::models::InvitationForm;
+
+pub struct ListInvitation {
+    pub user_id: UserId,
+}
+
+impl Message for ListInvitation {
+    type Result = Result<Vec<Invitation>, ServiceErrors>;
+}
+
+impl Handler<ListInvitation> for DbExecutor {
+    type Result = Result<Vec<Invitation>, ServiceErrors>;
+
+    fn handle(&mut self, msg: ListInvitation, _ctx: &mut Self::Context) -> Self::Result {
+        use crate::schema::invitations::dsl::*;
+
+        let conn = &self
+            .pool
+            .get()
+            .map_err(|_| ServiceErrors::DatabaseConnectionLost)?;
+
+        let query = invitations.filter(invited_by_id.eq(msg.user_id));
+        debug!("{}", diesel::debug_query::<Pg, _>(&query).to_string());
+        query
+            .load(conn)
+            .map_err(|_| ServiceErrors::DatabaseConnectionLost)
+    }
+}
+
+pub struct CreateInvitation {
+    pub user_id: UserId,
+    pub project_id: ProjectId,
+    pub email: EmailString,
+    pub name: UsernameString,
+}
+
+impl Message for CreateInvitation {
+    type Result = Result<Invitation, ServiceErrors>;
+}
+
+impl Handler<CreateInvitation> for DbExecutor {
+    type Result = Result<Invitation, ServiceErrors>;
+
+    fn handle(&mut self, msg: CreateInvitation, _ctx: &mut Self::Context) -> Self::Result {
+        use crate::schema::invitations::dsl::*;
+
+        let conn = &self
+            .pool
+            .get()
+            .map_err(|_| ServiceErrors::DatabaseConnectionLost)?;
+
+        let form = InvitationForm {
+            name: msg.name,
+            email: msg.email,
+            state: InvitationState::Sent,
+            project_id: msg.project_id,
+            invited_by_id: msg.user_id,
+        };
+        let query = diesel::insert_into(invitations).values(form);
+        debug!("{}", diesel::debug_query::<Pg, _>(&query).to_string());
+        query
+            .get_result(conn)
+            .map_err(|_| ServiceErrors::DatabaseConnectionLost)
+    }
+}
+
+pub struct DeleteInvitation {
+    pub id: InvitationId,
+}
+
+impl Message for DeleteInvitation {
+    type Result = Result<(), ServiceErrors>;
+}
+
+impl Handler<DeleteInvitation> for DbExecutor {
+    type Result = Result<(), ServiceErrors>;
+
+    fn handle(&mut self, msg: DeleteInvitation, _ctx: &mut Self::Context) -> Self::Result {
+        use crate::schema::invitations::dsl::*;
+
+        let conn = &self
+            .pool
+            .get()
+            .map_err(|_| ServiceErrors::DatabaseConnectionLost)?;
+        let query = diesel::delete(invitations).filter(id.eq(msg.id));
+        debug!("{}", diesel::debug_query::<Pg, _>(&query).to_string());
+        query
+            .execute(conn)
+            .map_err(|_| ServiceErrors::DatabaseConnectionLost)?;
+        Ok(())
+    }
+}
+
+pub struct RevokeInvitation {
+    pub id: InvitationId,
+}
+
+impl Message for RevokeInvitation {
+    type Result = Result<(), ServiceErrors>;
+}
+
+impl Handler<RevokeInvitation> for DbExecutor {
+    type Result = Result<(), ServiceErrors>;
+
+    fn handle(&mut self, msg: RevokeInvitation, _ctx: &mut Self::Context) -> Self::Result {
+        use crate::schema::invitations::dsl::*;
+
+        let conn = &self
+            .pool
+            .get()
+            .map_err(|_| ServiceErrors::DatabaseConnectionLost)?;
+        let query = diesel::update(invitations)
+            .set(state.eq(InvitationState::Revoked))
+            .filter(id.eq(msg.id));
+        debug!("{}", diesel::debug_query::<Pg, _>(&query).to_string());
+        query
+            .execute(conn)
+            .map_err(|_| ServiceErrors::DatabaseConnectionLost)?;
+        Ok(())
+    }
+}

jirs-server/src/db/mod.rs

@@ -11,6 +11,7 @@ use crate::db::dev::VerboseConnection;
 
 pub mod authorize_user;
 pub mod comments;
+pub mod invitations;
 pub mod issue_assignees;
 pub mod issues;
 pub mod projects;