use std::sync::mpsc::{channel, Receiver}; use gdnative::prelude::*; use tts::{Features, UtteranceId, TTS as Tts}; #[derive(NativeClass)] #[inherit(Reference)] struct Utterance(pub(crate) Option); #[methods] impl Utterance { fn new(_owner: &Reference) -> Self { Self(None) } } #[allow(clippy::enum_variant_names)] enum Msg { UtteranceBegin(UtteranceId), UtteranceEnd(UtteranceId), UtteranceStop(UtteranceId), } #[derive(NativeClass)] #[inherit(Node)] #[register_with(Self::register)] struct TTS(Tts, Receiver); #[methods] impl TTS { fn new(owner: &Node) -> Self { owner.set_pause_mode(2); let tts = Tts::default().expect("Failed to initialize TTS"); let (tx, rx) = channel(); let Features { utterance_callbacks, .. } = tts.supported_features(); if utterance_callbacks { let tx_end = tx.clone(); let tx_stop = tx.clone(); tts.on_utterance_begin(Some(Box::new(move |utterance| { tx.send(Msg::UtteranceBegin(utterance)) .expect("Failed to send UtteranceBegin"); }))) .expect("Failed to set utterance_begin callback"); tts.on_utterance_end(Some(Box::new(move |utterance| { tx_end .send(Msg::UtteranceEnd(utterance)) .expect("Failed to send UtteranceEnd"); }))) .expect("Failed to set utterance_end callback"); tts.on_utterance_stop(Some(Box::new(move |utterance| { tx_stop .send(Msg::UtteranceStop(utterance)) .expect("Failed to send UtteranceStop"); }))) .expect("Failed to set utterance_stop callback"); } Self(tts, rx) } fn register(builder: &ClassBuilder) { builder .add_property("rate") .with_getter(|this: &TTS, _| match this.0.get_rate() { Ok(rate) => rate, _ => 0., }) .with_setter(|this: &mut TTS, _, v: f32| { let Features { rate: rate_supported, .. } = this.0.supported_features(); if rate_supported { let mut v = v; if v < this.0.min_rate() { v = this.0.min_rate(); } else if v > this.0.max_rate() { v = this.0.max_rate(); } this.0.set_rate(v).expect("Failed to set rate"); } }) .done(); builder .add_property("min_rate") .with_getter(|this: &TTS, _| { let Features { rate: rate_supported, .. } = this.0.supported_features(); if rate_supported { this.0.min_rate() } else { 0. } }) .done(); builder .add_property("max_rate") .with_getter(|this: &TTS, _| { let Features { rate: rate_supported, .. } = this.0.supported_features(); if rate_supported { this.0.max_rate() } else { 0. } }) .done(); builder .add_property("normal_rate") .with_getter(|this: &TTS, _| { let Features { rate: rate_supported, .. } = this.0.supported_features(); if rate_supported { this.0.normal_rate() } else { 0. } }) .done(); builder .add_property("can_detect_screen_reader") .with_getter(|_: &TTS, _| cfg!(all(windows, features = "use_tolk"))) .done(); #[allow(unreachable_code)] builder .add_property("has_screen_reader") .with_getter(|_: &TTS, _| { #[cfg(all(windows, features = "use_tolk"))] { let tolk = tolk::Tolk::new(); return tolk.detect_screen_reader().is_some(); } false }) .done(); builder .add_property("can_detect_is_speaking") .with_getter(|this: &TTS, _| { let Features { is_speaking: is_speaking_supported, .. } = this.0.supported_features(); is_speaking_supported }) .done(); builder .add_property("is_speaking") .with_getter(|this: &TTS, _| { let Features { is_speaking: is_speaking_supported, .. } = this.0.supported_features(); if is_speaking_supported { this.0 .is_speaking() .expect("Failed to determine if speaking") } else { false } }) .done(); builder.add_signal(Signal { name: "utterance_begin", args: &[SignalArgument { name: "utterance", default: Variant::default(), export_info: ExportInfo::new(VariantType::Object), usage: PropertyUsage::DEFAULT, }], }); builder.add_signal(Signal { name: "utterance_end", args: &[SignalArgument { name: "utterance", default: Variant::default(), export_info: ExportInfo::new(VariantType::Object), usage: PropertyUsage::DEFAULT, }], }); builder.add_signal(Signal { name: "utterance_stop", args: &[SignalArgument { name: "utterance", default: Variant::default(), export_info: ExportInfo::new(VariantType::Object), usage: PropertyUsage::DEFAULT, }], }); } #[export] fn speak(&mut self, _owner: &Node, message: GodotString, interrupt: bool) -> Variant { let message = message.to_string(); if let Ok(id) = self.0.speak(message, interrupt) { let utterance: Instance = Instance::new(); if id.is_some() { utterance .map_mut(|u, _| u.0 = id) .expect("Failed to set utterance ID"); } utterance.owned_to_variant() } else { Variant::default() } } #[export] fn stop(&mut self, _owner: &Node) { self.0.stop().expect("Failed to stop"); } #[export] fn is_rate_supported(&mut self, _owner: &Node) -> bool { let Features { rate: rate_supported, .. } = self.0.supported_features(); rate_supported } #[export] fn are_utterance_callbacks_supported(&mut self, _owner: &Node) -> bool { let Features { utterance_callbacks: supported, .. } = self.0.supported_features(); supported } #[export] fn _process(&mut self, owner: &Node, _delta: f32) { if let Ok(msg) = self.1.try_recv() { match msg { Msg::UtteranceBegin(utterance_id) => { let utterance: Instance = Instance::new(); utterance .map_mut(|u, _| u.0 = Some(utterance_id)) .expect("Failed to set utterance ID"); owner.emit_signal("utterance_begin", &[utterance.owned_to_variant()]); } Msg::UtteranceEnd(utterance_id) => { let utterance: Instance = Instance::new(); utterance .map_mut(|u, _| u.0 = Some(utterance_id)) .expect("Failed to set utterance ID"); owner.emit_signal("utterance_end", &[utterance.owned_to_variant()]); } Msg::UtteranceStop(utterance_id) => { let utterance: Instance = Instance::new(); utterance .map_mut(|u, _| u.0 = Some(utterance_id)) .expect("Failed to set utterance ID"); owner.emit_signal("utterance_stop", &[utterance.owned_to_variant()]); } } } } } fn init(handle: InitHandle) { env_logger::init(); handle.add_tool_class::(); handle.add_class::(); } godot_gdnative_init!(); godot_nativescript_init!(init); godot_gdnative_terminate!();