cosmic-applets/applets/cosmic-applet-network/src/ui/available_wifi.rs

// SPDX-License-Identifier: LGPL-3.0-or-later

use crate::{task, widgets::SettingsEntry};
use cosmic_dbus_networkmanager::{
    device::{wireless::WirelessDevice, SpecificDevice},
    nm::NetworkManager,
};
use futures_util::StreamExt;
use gtk4::{
    glib::{self, clone, source::PRIORITY_DEFAULT, MainContext, Sender},
    prelude::*,
    Align, Image,
};
use std::{cell::RefCell, rc::Rc};
use zbus::Connection;

pub fn add_available_wifi(target: &gtk4::Box) {
    let ap_entries = Rc::<RefCell<Vec<SettingsEntry>>>::default();
    let (tx, rx) = MainContext::channel::<Vec<AccessPoint>>(PRIORITY_DEFAULT);
    task::spawn(async move {
        if let Err(err) = scan_for_wifi(tx).await {
            eprintln!("scan_for_wifi failed: {}", err);
        }
    });
    rx.attach(
        None,
        clone!(@strong ap_entries, @weak target => @default-return Continue(true), move |aps| {
            build_aps_list(ap_entries.clone(), target, dbg!(aps));
            Continue(true)
        }),
    );
}

fn build_aps_list(
    ap_entries: Rc<RefCell<Vec<SettingsEntry>>>,
    target: gtk4::Box,
    aps: Vec<AccessPoint>,
) {
    let mut ap_entries = ap_entries.borrow_mut();
    for old_ap_box in ap_entries.drain(..) {
        target.remove(&old_ap_box);
    }
    for ap in aps {
        view! {
            ap_entry = SettingsEntry {
                align_child: Align::Start,
                set_title: &ap.ssid,
                set_child: ap_icon = &Image {
                    set_icon_name: Some("network-wireless-symbolic"),
                }
            }
        }
        target.append(&ap_entry);
        ap_entries.push(ap_entry);
    }
}

async fn scan_for_wifi(tx: Sender<Vec<AccessPoint>>) -> zbus::Result<()> {
    let conn = Connection::system().await?;
    let network_manager = NetworkManager::new(&conn).await?;
    loop {
        let devices = network_manager.devices().await?;
        for device in devices {
            if let Ok(Some(SpecificDevice::Wireless(wireless_device))) =
                device.downcast_to_device().await
            {
                handle_wireless_device(wireless_device, tx.clone()).await?;
            }
        }
        tokio::time::sleep(std::time::Duration::from_secs(5)).await;
    }
}

async fn handle_wireless_device(
    device: WirelessDevice<'_>,
    tx: Sender<Vec<AccessPoint>>,
) -> zbus::Result<()> {
    device
        .request_scan(std::collections::HashMap::new())
        .await?;
    let mut scan_changed = device.receive_last_scan_changed().await;
    if let Some(t) = scan_changed.next().await {
        if let Ok(-1) = t.get().await {
            eprintln!("scan errored");
            return Ok(());
        }
    }
    let access_points = device.get_access_points().await?;
    let mut aps = Vec::with_capacity(access_points.len());
    for ap in access_points {
        aps.push(AccessPoint {
            ssid: String::from_utf8_lossy(&ap.ssid().await?.clone()).into_owned(),
            strength: ap.strength().await?,
        });
    }
    tx.send(aps).expect("failed to send back to main thread");
    Ok(())
}

#[derive(Debug)]
struct AccessPoint {
    ssid: String,
    strength: u8,
}