ReSet/src/components/output/inputStreamEntry.rs

use std::sync::Arc;
use std::time::{Duration, SystemTime};

use adw::glib;
use adw::glib::Object;
use adw::prelude::{ButtonExt, RangeExt};
use dbus::blocking::Connection;
use dbus::Error;
use glib::subclass::types::ObjectSubclassIsExt;
use glib::{clone, Cast, Propagation};
use gtk::{gio, StringObject};
use ReSet_Lib::audio::audio::InputStream;

use super::inputStreamEntryImpl;
use super::sinkBox::SinkBox;

glib::wrapper! {
    pub struct InputStreamEntry(ObjectSubclass<inputStreamEntryImpl::InputStreamEntry>)
    @extends gtk::Box, gtk::Widget,
    @implements gtk::Accessible, gtk::Buildable, gtk::ConstraintTarget, gtk::Orientable;
}

impl InputStreamEntry {
    pub fn new(sink_box: Arc<SinkBox>, stream: InputStream) -> Self {
        let obj: Self = Object::builder().build();
        // TODO use event callback for progress bar -> this is the "im speaking" indicator
        {
            let index = stream.sink_index;
            let box_imp = sink_box.imp();
            let imp = obj.imp();
            if stream.muted {
                imp.resetSinkMute
                    .set_icon_name("audio-volume-muted-symbolic");
            } else {
                imp.resetSinkMute
                    .set_icon_name("audio-volume-high-symbolic");
            }
            let name = stream.application_name.clone() + ": " + stream.name.as_str();
            imp.resetSinkName.set_text(name.as_str());
            let volume = stream.volume.first().unwrap_or(&0_u32);
            let fraction = (*volume as f64 / 655.36).round();
            let percentage = (fraction).to_string() + "%";
            imp.resetVolumePercentage.set_text(&percentage);
            imp.resetVolumeSlider.set_value(*volume as f64);
            imp.stream.replace(stream);
            {
                let sink = box_imp.resetDefaultSink.borrow();
                imp.associatedSink.replace((sink.index, sink.name.clone()));
            }
            imp.resetVolumeSlider.connect_change_value(
                clone!(@weak imp => @default-return Propagation::Stop, move |_, _, value| {
                    let fraction = (value / 655.36).round();
                    let percentage = (fraction).to_string() + "%";
                    imp.resetVolumePercentage.set_text(&percentage);
                    let mut stream = imp.stream.try_borrow();
                    while stream.is_err() {
                        stream = imp.stream.try_borrow();
                    }
                    let stream = stream.unwrap();
                    let index = stream.index;
                    let channels = stream.channels;
                    {
                        let mut time = imp.volumeTimeStamp.borrow_mut();
                        if time.is_some() && time.unwrap().elapsed().unwrap() < Duration::from_millis(50) {
                            return Propagation::Proceed;
                        }
                        *time = Some(SystemTime::now());
                    }
                    set_inputstream_volume(value, index, channels);
                    Propagation::Proceed
                }),
            );
            {
                let list = box_imp.resetModelList.read().unwrap();
                // while list.is_err() {
                //     list = box_imp.resetModelList.try_borrow();
                // }
                // let list = list.unwrap();
                imp.resetSelectedSink.set_model(Some(&*list));
                let map = box_imp.resetSinkMap.read().unwrap();
                let sink_list = box_imp.resetSinkList.read().unwrap();
                let name = sink_list.get(&index);
                if name.is_some() {
                    let name = &name.unwrap().2;
                    let index = map.get(name);
                    if index.is_some() {
                        imp.resetSelectedSink.set_selected(index.unwrap().1);
                    }
                } else {
                    let mut name = box_imp.resetDefaultSink.try_borrow();
                    while name.is_err() {
                        name = box_imp.resetDefaultSink.try_borrow();
                    }
                    let name = &name.unwrap().alias;
                    let index = map.get(name);
                    if index.is_some() {
                        imp.resetSelectedSink.set_selected(index.unwrap().1);
                    }
                }
            }
            imp.resetSelectedSink.connect_selected_notify(
                clone!(@weak imp, @weak box_imp => move |dropdown| {
                    let selected = dropdown.selected_item();
                    if selected.is_none() {
                        return;
                    }
                    let selected = selected.unwrap();
                    let selected = selected.downcast_ref::<StringObject>().unwrap();
                    let selected = selected.string().to_string();
                    let sink = box_imp.resetSinkMap.read().unwrap();
                    // if sink.is_err() {
                    //     return;
                    // }
                    // let sink = sink.unwrap();
                    let sink = sink.get(&selected);
                    if sink.is_none() {
                        return;
                    }
                    let mut stream = imp.stream.try_borrow();
                    while stream.is_err() {
                        stream = imp.stream.try_borrow();
                    }
                    let stream = stream.unwrap();
                let sink = sink.unwrap().0;
                    set_sink_of_input_stream(stream.index, sink);
                }),
            );
            imp.resetSinkMute
                .connect_clicked(clone!(@weak imp => move |_| {
                    let stream = imp.stream.clone();
                    let mut stream = stream.try_borrow_mut();
                    while stream.is_err() {
                        stream = imp.stream.try_borrow_mut();
                    }
                    let mut stream = stream.unwrap();
                    stream.muted = !stream.muted;
                    let muted = stream.muted;
                    let index = stream.index;
                    if muted {
                        imp.resetSinkMute
                           .set_icon_name("audio-volume-muted-symbolic");
                    } else {
                        imp.resetSinkMute
                           .set_icon_name("audio-volume-high-symbolic");
                    }
                    toggle_input_stream_mute(index, muted);
                }));
        }
        obj
    }
}

fn set_inputstream_volume(value: f64, index: u32, channels: u16) -> bool {
    gio::spawn_blocking(move || {
        let conn = Connection::new_session().unwrap();
        let proxy = conn.with_proxy(
            "org.xetibo.ReSet",
            "/org/xetibo/ReSet",
            Duration::from_millis(1000),
        );
        let _: Result<(), Error> = proxy.method_call(
            "org.xetibo.ReSet",
            "SetInputStreamVolume",
            (index, channels, value as u32),
        );
        // if res.is_err() {
        //     return false;
        // }
        // res.unwrap().0
    });
    true
}

fn toggle_input_stream_mute(index: u32, muted: bool) -> bool {
    gio::spawn_blocking(move || {
        let conn = Connection::new_session().unwrap();
        let proxy = conn.with_proxy(
            "org.xetibo.ReSet",
            "/org/xetibo/ReSet",
            Duration::from_millis(1000),
        );
        let _: Result<(), Error> =
            proxy.method_call("org.xetibo.ReSet", "SetInputStreamMute", (index, muted));
        // if res.is_err() {
        //     return false;
        // }
        // res.unwrap().0
    });
    true
}

fn set_sink_of_input_stream(stream: u32, sink: u32) -> bool {
    gio::spawn_blocking(move || {
        let conn = Connection::new_session().unwrap();
        let proxy = conn.with_proxy(
            "org.xetibo.ReSet",
            "/org/xetibo/ReSet",
            Duration::from_millis(1000),
        );
        let _: Result<(), Error> =
            proxy.method_call("org.xetibo.ReSet", "SetSinkOfInputStream", (stream, sink));
        // if res.is_err() {
        //     return false;
        // }
        // res.unwrap().0
    });
    true
}

// TODO propagate error from dbus