use std::sync::Arc;
use async_std::sync::Mutex;
use serde::{Deserialize, Serialize};
use ng_repo::errors::*;
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 ListUsersV0 {
pub admins: bool,
}
#[derive(Clone, Copy, Debug, Serialize, Deserialize)]
pub enum ListUsers {
V0(ListUsersV0),
}
impl ListUsers {
pub fn admins(&self) -> bool {
match self {
Self::V0(o) => o.admins,
}
}
pub fn get_actor(&self) -> Box<dyn EActor> {
Actor::<ListUsers, AdminResponse>::new_responder(0)
}
}
impl TryFrom<ProtocolMessage> for ListUsers {
type Error = ProtocolError;
fn try_from(msg: ProtocolMessage) -> Result<Self, Self::Error> {
if let ProtocolMessage::Start(StartProtocol::Admin(AdminRequest::V0(AdminRequestV0 {
content: AdminRequestContentV0::ListUsers(a),
..
}))) = msg
{
Ok(a)
} else {
Err(ProtocolError::InvalidValue)
}
}
}
impl From<ListUsers> for ProtocolMessage {
fn from(_msg: ListUsers) -> ProtocolMessage {
unimplemented!();
}
}
impl From<ListUsers> for AdminRequestContentV0 {
fn from(msg: ListUsers) -> AdminRequestContentV0 {
AdminRequestContentV0::ListUsers(msg)
}
}
impl Actor<'_, ListUsers, AdminResponse> {}
#[async_trait::async_trait]
impl EActor for Actor<'_, ListUsers, AdminResponse> {
async fn respond(
&mut self,
msg: ProtocolMessage,
fsm: Arc<Mutex<NoiseFSM>>,
) -> Result<(), ProtocolError> {
let req = ListUsers::try_from(msg)?;
let sb = { BROKER.read().await.get_server_broker()? };
let res = { sb.read().await.list_users(req.admins()) };
let response: AdminResponseV0 = res.into();
fsm.lock().await.send(response.into()).await?;
Ok(())
}
}