last_order 0.1.6

use std::net::ToSocketAddrs;

use strum::EnumString;

pub mod generic;
use generic::device::NetworkDevice;

pub mod vendor;

pub mod error;
use error::Error;

#[derive(Debug, Clone, Copy, PartialEq, EnumString)]
#[strum(serialize_all = "snake_case")]
pub enum Vendor {
    Aruba,
    Cisco,
    H3C,
    Huawei,
    Ruijie,
}

/// Creates a new network device based on the specified vendor.
///
/// # Arguments
/// * `vendor` - The vendor of the device (e.g., Huawei, H3C).
/// * `addr` - The address of the device (e.g., IP and port).
/// * `username` - Optional username for authentication.
/// * `password` - Optional password for authentication.
///
/// # Returns
/// A `Result` containing a `Box<dyn NetworkDevice>` on success, or an error if the connection fails.
pub fn create_network_device<A: ToSocketAddrs>(
    vendor: Vendor,
    addr: A,
    username: Option<&str>,
    password: Option<&str>,
) -> Result<Box<dyn NetworkDevice>, Error> {
    Ok(match vendor {
        Vendor::Aruba => vendor::aruba::ArubaSSH::connect(addr, username, password)?.into_dyn(),
        Vendor::Cisco => vendor::cisco::CiscoSSH::connect(addr, username, password)?.into_dyn(),
        Vendor::H3C => vendor::h3c::H3cSSH::connect(addr, username, password)?.into_dyn(),
        Vendor::Huawei => vendor::huawei::HuaweiSSH::connect(addr, username, password)?.into_dyn(),
        Vendor::Ruijie => vendor::ruijie::RuijieSSH::connect(addr, username, password)?.into_dyn(),
    })
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn dev() -> anyhow::Result<()> {
        env_logger::try_init().ok();

        let ssh = (Vendor::H3C, format!("{}:22", "10.123.0.24"));
        // let ssh = (Vendor::Huawei, format!("{}:22", "10.123.255.11"));
        // let ssh = (Vendor::Aruba, format!("{}:22", "10.123.0.15"));

        let user = Some("HBSpy");
        let pass = Some(std::env::var("LO_TESTPASS").expect("LO_TESTPASS not set"));

        let mut ssh = create_network_device(ssh.0, ssh.1, user, pass.as_deref())?;

        let result = ssh.version()?;

        dbg!(&result);

        Ok(())
    }
}