use std::collections::BTreeMap;
use crate::admin::store::{Circuit, CircuitNode, CircuitStatus, Service};
use crate::hex::to_hex;
#[derive(Debug, Serialize, Clone, PartialEq)]
pub(crate) struct CircuitResponse<'a> {
pub id: &'a str,
pub members: Vec<CircuitNodeResponse<'a>>,
pub roster: Vec<ServiceResponse<'a>>,
pub management_type: &'a str,
pub display_name: &'a Option<String>,
pub circuit_version: i32,
pub circuit_status: &'a CircuitStatus,
}
impl<'a> From<&'a Circuit> for CircuitResponse<'a> {
fn from(circuit: &'a Circuit) -> Self {
Self {
id: circuit.circuit_id(),
members: circuit
.members()
.iter()
.map(CircuitNodeResponse::from)
.collect(),
roster: circuit.roster().iter().map(ServiceResponse::from).collect(),
management_type: circuit.circuit_management_type(),
display_name: circuit.display_name(),
circuit_version: circuit.circuit_version(),
circuit_status: circuit.circuit_status(),
}
}
}
#[derive(Debug, Serialize, Clone, PartialEq)]
pub(crate) struct ServiceResponse<'a> {
pub service_id: &'a str,
pub service_type: &'a str,
pub node_id: &'a str,
pub arguments: BTreeMap<String, String>,
}
impl<'a> From<&'a Service> for ServiceResponse<'a> {
fn from(service_def: &'a Service) -> Self {
Self {
service_id: service_def.service_id(),
service_type: service_def.service_type(),
node_id: service_def.node_id(),
arguments: service_def
.arguments()
.iter()
.map(|(key, value)| (key.to_string(), value.to_string()))
.collect::<BTreeMap<String, String>>(),
}
}
}
#[derive(Debug, Serialize, Clone, PartialEq)]
pub(crate) struct CircuitNodeResponse<'a> {
pub node_id: &'a str,
pub endpoints: &'a [String],
pub public_key: Option<String>,
}
impl<'a> From<&'a CircuitNode> for CircuitNodeResponse<'a> {
fn from(node_def: &'a CircuitNode) -> Self {
Self {
node_id: node_def.node_id(),
endpoints: node_def.endpoints(),
public_key: node_def
.public_key()
.as_ref()
.map(|public_key| to_hex(public_key.as_slice())),
}
}
}