send notifications

This commit is contained in:
realaravinth 2021-05-10 00:55:47 +05:30
parent bd20b4238b
commit aa0c30f1bd
No known key found for this signature in database
GPG key ID: AD9F0F08E855ED88
6 changed files with 168 additions and 2 deletions

View file

@ -0,0 +1,10 @@
-- Add migration script here
CREATE TABLE IF NOT EXISTS mcaptcha_notifications (
id SERIAL PRIMARY KEY NOT NULL,
tx INTEGER NOT NULL references mcaptcha_users(ID) ON DELETE CASCADE,
rx INTEGER NOT NULL references mcaptcha_users(ID) ON DELETE CASCADE,
heading varchar(30) NOT NULL,
message varchar(250) NOT NULL,
read BOOLEAN DEFAULT NULL,
received timestamptz NOT NULL DEFAULT now()
);

View file

@ -21,6 +21,7 @@ pub mod account;
pub mod auth;
pub mod mcaptcha;
pub mod meta;
mod notifications;
pub mod pow;
mod routes;
@ -31,6 +32,7 @@ pub fn services(cfg: &mut ServiceConfig) {
auth::services(cfg);
account::services(cfg);
mcaptcha::services(cfg);
notifications::services(cfg);
pow::services(cfg);
}

View file

@ -0,0 +1,102 @@
/*
* Copyright (C) 2021 Aravinth Manivannan <realaravinth@batsense.net>
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU Affero General Public License as
* published by the Free Software Foundation, either version 3 of the
* License, or (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Affero General Public License for more details.
*
* You should have received a copy of the GNU Affero General Public License
* along with this program. If not, see <https://www.gnu.org/licenses/>.
*/
use actix_identity::Identity;
use actix_web::{web, HttpResponse, Responder};
use serde::{Deserialize, Serialize};
use crate::errors::*;
use crate::Data;
#[derive(Serialize, Deserialize)]
pub struct AddNotification {
pub to: String,
pub heading: String,
pub message: String,
}
/// route handler that adds a notification message
pub async fn add_notification(
payload: web::Json<AddNotification>,
data: web::Data<Data>,
id: Identity,
) -> ServiceResult<impl Responder> {
let sender = id.identity().unwrap();
// TODO handle error where payload.to doesnt exist
sqlx::query!(
"INSERT INTO mcaptcha_notifications (
heading, message, tx, rx)
VALUES (
$1, $2,
(SELECT ID FROM mcaptcha_users WHERE name = $3),
(SELECT ID FROM mcaptcha_users WHERE name = $4)
);",
&payload.heading,
&payload.message,
&sender,
&payload.to,
)
.execute(&data.db)
.await?;
Ok(HttpResponse::Ok())
}
#[cfg(test)]
mod tests {
use actix_web::http::{header, StatusCode};
use actix_web::test;
use super::*;
use crate::tests::*;
use crate::*;
#[actix_rt::test]
async fn notification_works() {
const NAME1: &str = "notifuser1";
const NAME2: &str = "notiuser2";
const PASSWORD: &str = "longpassworddomain";
const EMAIL1: &str = "testnotification1@a.com";
const EMAIL2: &str = "testnotification2@a.com";
{
let data = Data::new().await;
delete_user(NAME1, &data).await;
delete_user(NAME2, &data).await;
}
register_and_signin(NAME1, EMAIL1, PASSWORD).await;
register_and_signin(NAME2, EMAIL2, PASSWORD).await;
let (data, _creds, signin_resp) = signin(NAME1, PASSWORD).await;
let cookies = get_cookie!(signin_resp);
let mut app = get_app!(data).await;
let msg = AddNotification {
to: NAME2.into(),
heading: "Test notification".into(),
message: "Testeing notifications with a dummy message".into(),
};
let send_notification_resp = test::call_service(
&mut app,
post_request!(&msg, V1_API_ROUTES.notifications.add)
.cookie(cookies.clone())
.to_request(),
)
.await;
assert_eq!(send_notification_resp.status(), StatusCode::OK);
}
}

View file

@ -0,0 +1,49 @@
/*
* Copyright (C) 2021 Aravinth Manivannan <realaravinth@batsense.net>
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU Affero General Public License as
* published by the Free Software Foundation, either version 3 of the
* License, or (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Affero General Public License for more details.
*
* You should have received a copy of the GNU Affero General Public License
* along with this program. If not, see <https://www.gnu.org/licenses/>.
*/
mod add;
pub mod routes {
pub struct Notifications {
pub add: &'static str,
pub mark_read: &'static str,
pub get: &'static str,
}
impl Notifications {
pub const fn new() -> Notifications {
Notifications {
add: "/api/v1/notifications/add",
mark_read: "/api/v1/notifications/read/",
get: "/api/v1/notifications/get/",
}
}
}
}
pub fn services(cfg: &mut actix_web::web::ServiceConfig) {
use crate::define_resource;
use crate::V1_API_ROUTES;
define_resource!(
cfg,
V1_API_ROUTES.notifications.add,
Methods::ProtectPost,
add::add_notification
);
}

View file

@ -21,6 +21,7 @@ use super::mcaptcha::duration::routes::Duration;
use super::mcaptcha::levels::routes::Levels;
use super::mcaptcha::mcaptcha::routes::MCaptcha;
use super::meta::routes::Meta;
use super::notifications::routes::Notifications;
use super::pow::routes::PoW;
pub const ROUTES: Routes = Routes::new();
@ -33,6 +34,7 @@ pub struct Routes {
pub duration: Duration,
pub meta: Meta,
pub pow: PoW,
pub notifications: Notifications,
}
impl Routes {
@ -45,6 +47,7 @@ impl Routes {
duration: Duration::new(),
meta: Meta::new(),
pow: PoW::new(),
notifications: Notifications::new(),
}
}
}

View file

@ -39,9 +39,9 @@ impl Routes {
panel,
home,
errors: Errors::new(),
about: "/aboubt",
about: "/about",
thanks: "/thanks",
donate: "/donat",
donate: "/donate",
}
}
}