use std::sync::Arc;
use async_std::sync::Mutex;
use serde::{Deserialize, Serialize};
use ng_repo::errors::*;
#[allow(unused_imports)]
use ng_repo::log::*;
use super::super::StartProtocol;
use crate::broker::BROKER;
use crate::connection::NoiseFSM;
use crate::types::*;
use crate::{actor::*, types::ProtocolMessage};
#[derive(Clone, Copy, Debug, Serialize, Deserialize)]
pub struct ListInvitationsV0 {
pub admin: bool,
pub unique: bool,
pub multi: bool,
}
#[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(0)
}
}
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 {
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 sb = { BROKER.read().await.get_server_broker()? };
let res = {
sb.read()
.await
.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![],
},
}
}
}