wirectl 0.1.0

use crate::WireCtlError;
use crate::{ipc, types::*};
use futures::TryStreamExt;
use rtnetlink::new_connection;

cfg_if! {
    if #[cfg(target_os = "linux")] {
        pub const AVAILABLE_WG_APIS: [WgApi; 1] = [WgApi::IPC];
    } else if #[cfg(any(target_os = "openbsd", target_os = "freebsd"))] {
        pub const AVAILABLE_WG_APIS: [WgApi; 1] = [WgApi::IPC];
    } else {
        pub const AVAILABLE_WG_APIS: [WgApi; 1] = [WgApi::IPC];
    }
}

/// Used by [`crate::interface::WgInterface`] to determine which underlaying API to use
#[allow(clippy::upper_case_acronyms)]
#[derive(Clone, Copy, Debug, PartialEq, Eq)]
pub enum WgApi {
    IPC,
    #[cfg(target_os = "linux")]
    Linux,
    #[cfg(any(target_os = "openbsd", target_os = "freebsd"))]
    BSD,
}

impl WgApi {
    pub(crate) async fn list_interfaces(self) -> Result<Vec<String>, WireCtlError> {
        match self {
            WgApi::IPC => ipc::list_interfaces().await,
            #[cfg(target_os = "linux")]
            WgApi::Linux => todo!(),
            #[cfg(any(target_os = "openbsd", target_os = "freebsd"))]
            WgApi::BSD => todo!(),
        }
    }

    pub(crate) async fn get_config(self, ifname: &str) -> Result<WgDevice, WireCtlError> {
        match self {
            WgApi::IPC => ipc::get_config(ifname).await,
            #[cfg(target_os = "linux")]
            WgApi::Linux => todo!(),
            #[cfg(any(target_os = "openbsd", target_os = "freebsd"))]
            WgApi::BSD => todo!(),
        }
    }

    pub(crate) async fn set_config(
        self,
        ifname: &str,
        conf: WgDeviceSettings,
    ) -> Result<(), WireCtlError> {
        match self {
            WgApi::IPC => ipc::set_config(ifname, conf).await,
            #[cfg(target_os = "linux")]
            WgApi::Linux => todo!(),
            #[cfg(any(target_os = "openbsd", target_os = "freebsd"))]
            WgApi::BSD => todo!(),
        }
    }

    pub(crate) async fn add_interface(self, ifname: &str) -> Result<(), WireCtlError> {
        match self {
            WgApi::IPC => ipc::create_interface(ifname).await,
            #[cfg(target_os = "linux")]
            WgApi::Linux => todo!(),
            #[cfg(any(target_os = "openbsd", target_os = "freebsd"))]
            WgApi::BSD => todo!(),
        }
    }

    pub(crate) async fn del_interface(self, ifname: &str) -> Result<(), WireCtlError> {
        let is_wg_if = match self {
            WgApi::IPC => ipc::check_device(ifname).await,
            #[cfg(target_os = "linux")]
            WgApi::Linux => todo!(),
            #[cfg(any(target_os = "openbsd", target_os = "freebsd"))]
            WgApi::BSD => todo!(),
        };
        if !is_wg_if {
            return Err(WireCtlError::NotFound);
        }

        let (connection, handle, _) = new_connection()?;
        smol::spawn(connection).detach();

        let mut links = handle
            .link()
            .get()
            .set_name_filter(ifname.to_owned())
            .execute();

        if let Some(msg) = links.try_next().await? {
            handle.link().del(msg.header.index).execute().await?;
        } else {
            return Err(WireCtlError::NotFound);
        }
        Ok(())
    }
}