use crate::{
packet_route::link::{
InfoData, InfoKind, InfoVlan, VlanFlags, VlanProtocol, VlanQosMapping,
},
LinkMessageBuilder,
};
#[derive(Clone, Copy, Debug, PartialEq, Eq)]
pub struct QosMapping {
pub from: u32,
pub to: u32,
}
impl From<QosMapping> for VlanQosMapping {
fn from(QosMapping { from, to }: QosMapping) -> Self {
Self::Mapping(from, to)
}
}
#[derive(Debug)]
pub struct LinkVlan;
impl LinkVlan {
pub fn new(
name: &str,
base_iface_index: u32,
vlan_id: u16,
) -> LinkMessageBuilder<Self> {
LinkMessageBuilder::<LinkVlan>::new(name)
.id(vlan_id)
.link(base_iface_index)
}
}
impl LinkMessageBuilder<LinkVlan> {
pub fn new(name: &str) -> Self {
LinkMessageBuilder::<LinkVlan>::new_with_info_kind(InfoKind::Vlan)
.name(name.to_string())
}
pub fn append_info_data(mut self, info: InfoVlan) -> Self {
if let InfoData::Vlan(infos) = self
.info_data
.get_or_insert_with(|| InfoData::Vlan(Vec::new()))
{
infos.push(info);
}
self
}
pub fn id(self, vlan_id: u16) -> Self {
self.append_info_data(InfoVlan::Id(vlan_id))
}
pub fn protocol(self, protocol: VlanProtocol) -> Self {
self.append_info_data(InfoVlan::Protocol(protocol))
}
pub fn flags(self, flags: VlanFlags, mask: VlanFlags) -> Self {
self.append_info_data(InfoVlan::Flags((flags, mask)))
}
pub fn qos<I, E>(self, ingress_qos: I, egress_qos: E) -> Self
where
I: IntoIterator<Item = QosMapping>,
E: IntoIterator<Item = QosMapping>,
{
let mut ret = self;
let ingress: Vec<_> =
ingress_qos.into_iter().map(VlanQosMapping::from).collect();
if !ingress.is_empty() {
ret = ret.append_info_data(InfoVlan::IngressQos(ingress));
}
let egress: Vec<_> =
egress_qos.into_iter().map(VlanQosMapping::from).collect();
if !egress.is_empty() {
ret = ret.append_info_data(InfoVlan::EgressQos(egress));
}
ret
}
}