parent
b9764e7983
commit
245db1807b
@ -0,0 +1,123 @@ |
||||
/* |
||||
* Copyright (c) 2022-2023 Niko Bonnieure, Par le Peuple, NextGraph.org developers |
||||
* All rights reserved. |
||||
* Licensed under the Apache License, Version 2.0 |
||||
* <LICENSE-APACHE2 or http://www.apache.org/licenses/LICENSE-2.0>
|
||||
* or the MIT license <LICENSE-MIT or http://opensource.org/licenses/MIT>,
|
||||
* at your option. All files in the project carrying such |
||||
* notice may not be copied, modified, or distributed except |
||||
* according to those terms. |
||||
*/ |
||||
use crate::broker::{ServerConfig, BROKER}; |
||||
use crate::connection::NoiseFSM; |
||||
use crate::types::*; |
||||
use crate::{actor::*, errors::ProtocolError, types::ProtocolMessage}; |
||||
|
||||
use async_std::sync::Mutex; |
||||
use p2p_repo::log::*; |
||||
use p2p_repo::types::PubKey; |
||||
use serde::{Deserialize, Serialize}; |
||||
use std::sync::Arc; |
||||
|
||||
use super::StartProtocol; |
||||
|
||||
/// Add invitation
|
||||
#[derive(Clone, Debug, Serialize, Deserialize)] |
||||
pub struct AddInvitationV0 { |
||||
pub invite_code: InvitationCode, |
||||
pub expiry: u32, |
||||
pub memo: Option<String>, |
||||
} |
||||
|
||||
/// Add invitation
|
||||
#[derive(Clone, Debug, Serialize, Deserialize)] |
||||
pub enum AddInvitation { |
||||
V0(AddInvitationV0), |
||||
} |
||||
|
||||
impl AddInvitation { |
||||
pub fn code(&self) -> &InvitationCode { |
||||
match self { |
||||
AddInvitation::V0(o) => &o.invite_code, |
||||
} |
||||
} |
||||
pub fn expiry(&self) -> u32 { |
||||
match self { |
||||
AddInvitation::V0(o) => o.expiry, |
||||
} |
||||
} |
||||
pub fn memo(&self) -> &Option<String> { |
||||
match self { |
||||
AddInvitation::V0(o) => &o.memo, |
||||
} |
||||
} |
||||
pub fn get_actor(&self) -> Box<dyn EActor> { |
||||
Actor::<AddInvitation, AdminResponse>::new_responder() |
||||
} |
||||
} |
||||
|
||||
impl TryFrom<ProtocolMessage> for AddInvitation { |
||||
type Error = ProtocolError; |
||||
fn try_from(msg: ProtocolMessage) -> Result<Self, Self::Error> { |
||||
if let ProtocolMessage::Start(StartProtocol::Admin(AdminRequest::V0(AdminRequestV0 { |
||||
content: AdminRequestContentV0::AddInvitation(a), |
||||
.. |
||||
}))) = msg |
||||
{ |
||||
Ok(a) |
||||
} else { |
||||
log_debug!("INVALID {:?}", msg); |
||||
Err(ProtocolError::InvalidValue) |
||||
} |
||||
} |
||||
} |
||||
|
||||
impl From<AddInvitation> for ProtocolMessage { |
||||
fn from(msg: AddInvitation) -> ProtocolMessage { |
||||
unimplemented!(); |
||||
} |
||||
} |
||||
|
||||
impl From<AddInvitation> for AdminRequestContentV0 { |
||||
fn from(msg: AddInvitation) -> AdminRequestContentV0 { |
||||
AdminRequestContentV0::AddInvitation(msg) |
||||
} |
||||
} |
||||
|
||||
impl Actor<'_, AddInvitation, AdminResponse> {} |
||||
|
||||
#[async_trait::async_trait] |
||||
impl EActor for Actor<'_, AddInvitation, AdminResponse> { |
||||
async fn respond( |
||||
&mut self, |
||||
msg: ProtocolMessage, |
||||
fsm: Arc<Mutex<NoiseFSM>>, |
||||
) -> Result<(), ProtocolError> { |
||||
let req = AddInvitation::try_from(msg)?; |
||||
let broker = BROKER.read().await; |
||||
broker |
||||
.get_storage()? |
||||
.add_invitation(req.code(), req.expiry(), req.memo())?; |
||||
|
||||
let invitation = crate::types::Invitation::V0(InvitationV0::new( |
||||
broker.get_bootstrap()?.clone(), |
||||
Some(req.code().get_symkey()), |
||||
None, |
||||
broker.get_registration_url().map(|s| s.clone()), |
||||
)); |
||||
let response: AdminResponseV0 = invitation.into(); |
||||
fsm.lock().await.send(response.into()).await?; |
||||
Ok(()) |
||||
} |
||||
} |
||||
|
||||
impl From<Invitation> for AdminResponseV0 { |
||||
fn from(res: Invitation) -> AdminResponseV0 { |
||||
AdminResponseV0 { |
||||
id: 0, |
||||
result: 0, |
||||
content: AdminResponseContentV0::Invitation(res), |
||||
padding: vec![], |
||||
} |
||||
} |
||||
} |
||||
@ -0,0 +1,130 @@ |
||||
/* |
||||
* Copyright (c) 2022-2023 Niko Bonnieure, Par le Peuple, NextGraph.org developers |
||||
* All rights reserved. |
||||
* Licensed under the Apache License, Version 2.0 |
||||
* <LICENSE-APACHE2 or http://www.apache.org/licenses/LICENSE-2.0>
|
||||
* or the MIT license <LICENSE-MIT or http://opensource.org/licenses/MIT>,
|
||||
* at your option. All files in the project carrying such |
||||
* notice may not be copied, modified, or distributed except |
||||
* according to those terms. |
||||
*/ |
||||
use crate::broker::BROKER; |
||||
use crate::connection::NoiseFSM; |
||||
use crate::types::*; |
||||
use crate::{actor::*, errors::ProtocolError, types::ProtocolMessage}; |
||||
|
||||
use async_std::sync::Mutex; |
||||
use p2p_repo::log::*; |
||||
use p2p_repo::types::PubKey; |
||||
use serde::{Deserialize, Serialize}; |
||||
use std::sync::Arc; |
||||
|
||||
use super::StartProtocol; |
||||
|
||||
/// List invitations registered on this broker
|
||||
#[derive(Clone, Copy, Debug, Serialize, Deserialize)] |
||||
pub struct ListInvitationsV0 { |
||||
/// should list only the admin invitations.
|
||||
pub admin: bool, |
||||
/// should list only the unique invitations.
|
||||
pub unique: bool, |
||||
/// should list only the multi invitations.
|
||||
pub multi: bool, |
||||
} |
||||
|
||||
/// List invitations registered on this broker
|
||||
#[derive(Clone, Copy, Debug, Serialize, Deserialize)] |
||||
pub enum ListInvitations { |
||||
V0(ListInvitationsV0), |
||||
} |
||||
|
||||
impl ListInvitations { |
||||
pub fn admin(&self) -> bool { |
||||
match self { |
||||
Self::V0(o) => o.admin, |
||||
} |
||||
} |
||||
pub fn unique(&self) -> bool { |
||||
match self { |
||||
Self::V0(o) => o.unique, |
||||
} |
||||
} |
||||
pub fn multi(&self) -> bool { |
||||
match self { |
||||
Self::V0(o) => o.multi, |
||||
} |
||||
} |
||||
pub fn get_actor(&self) -> Box<dyn EActor> { |
||||
Actor::<ListInvitations, AdminResponse>::new_responder() |
||||
} |
||||
} |
||||
|
||||
impl TryFrom<ProtocolMessage> for ListInvitations { |
||||
type Error = ProtocolError; |
||||
fn try_from(msg: ProtocolMessage) -> Result<Self, Self::Error> { |
||||
if let ProtocolMessage::Start(StartProtocol::Admin(AdminRequest::V0(AdminRequestV0 { |
||||
content: AdminRequestContentV0::ListInvitations(a), |
||||
.. |
||||
}))) = msg |
||||
{ |
||||
Ok(a) |
||||
} else { |
||||
//log_debug!("INVALID {:?}", msg);
|
||||
Err(ProtocolError::InvalidValue) |
||||
} |
||||
} |
||||
} |
||||
|
||||
impl From<ListInvitations> for ProtocolMessage { |
||||
fn from(msg: ListInvitations) -> ProtocolMessage { |
||||
unimplemented!(); |
||||
} |
||||
} |
||||
|
||||
impl From<ListInvitations> for AdminRequestContentV0 { |
||||
fn from(msg: ListInvitations) -> AdminRequestContentV0 { |
||||
AdminRequestContentV0::ListInvitations(msg) |
||||
} |
||||
} |
||||
|
||||
impl Actor<'_, ListInvitations, AdminResponse> {} |
||||
|
||||
#[async_trait::async_trait] |
||||
impl EActor for Actor<'_, ListInvitations, AdminResponse> { |
||||
async fn respond( |
||||
&mut self, |
||||
msg: ProtocolMessage, |
||||
fsm: Arc<Mutex<NoiseFSM>>, |
||||
) -> Result<(), ProtocolError> { |
||||
let req = ListInvitations::try_from(msg)?; |
||||
let res = BROKER.read().await.get_storage()?.list_invitations( |
||||
req.admin(), |
||||
req.unique(), |
||||
req.multi(), |
||||
); |
||||
let response: AdminResponseV0 = res.into(); |
||||
fsm.lock().await.send(response.into()).await?; |
||||
Ok(()) |
||||
} |
||||
} |
||||
|
||||
impl From<Result<Vec<(InvitationCode, u32, Option<String>)>, ProtocolError>> for AdminResponseV0 { |
||||
fn from( |
||||
res: Result<Vec<(InvitationCode, u32, Option<String>)>, ProtocolError>, |
||||
) -> AdminResponseV0 { |
||||
match res { |
||||
Err(e) => AdminResponseV0 { |
||||
id: 0, |
||||
result: e.into(), |
||||
content: AdminResponseContentV0::EmptyResponse, |
||||
padding: vec![], |
||||
}, |
||||
Ok(vec) => AdminResponseV0 { |
||||
id: 0, |
||||
result: 0, |
||||
content: AdminResponseContentV0::Invitations(vec), |
||||
padding: vec![], |
||||
}, |
||||
} |
||||
} |
||||
} |
||||
Loading…
Reference in new issue