113 lines
2.9 KiB
Rust
113 lines
2.9 KiB
Rust
use crate::core::client::Client;
|
|
|
|
use super::{
|
|
items::{ItemAcceptor, TypeInfo},
|
|
spatial::find_spatial,
|
|
Node,
|
|
};
|
|
use color_eyre::eyre::Result;
|
|
use glam::Mat4;
|
|
use parking_lot::Mutex;
|
|
use rustc_hash::FxHashMap;
|
|
use stardust_xr::schemas::flex::{deserialize, serialize};
|
|
use std::{
|
|
fmt::Debug,
|
|
sync::{Arc, Weak},
|
|
};
|
|
|
|
lazy_static::lazy_static! {
|
|
pub static ref STARTUP_SETTINGS: Mutex<FxHashMap<String, StartupSettings>> = Default::default();
|
|
}
|
|
|
|
#[derive(Default, Clone)]
|
|
pub struct StartupSettings {
|
|
pub transform: Mat4,
|
|
pub acceptors: FxHashMap<&'static TypeInfo, Weak<ItemAcceptor>>,
|
|
}
|
|
impl StartupSettings {
|
|
pub fn add_to(node: &Arc<Node>) {
|
|
node.startup_settings
|
|
.set(Mutex::new(StartupSettings::default()))
|
|
.unwrap();
|
|
}
|
|
|
|
fn set_root_flex(node: &Node, calling_client: Arc<Client>, data: &[u8]) -> Result<()> {
|
|
let spatial = find_spatial(&calling_client, "Root spatial", deserialize(data)?)?;
|
|
node.startup_settings.get().unwrap().lock().transform = spatial.global_transform();
|
|
|
|
Ok(())
|
|
}
|
|
|
|
fn add_automatic_acceptor_flex(
|
|
node: &Node,
|
|
calling_client: Arc<Client>,
|
|
data: &[u8],
|
|
) -> Result<()> {
|
|
let acceptor_node = calling_client.get_node("Item acceptor", deserialize(data)?)?;
|
|
let acceptor =
|
|
acceptor_node.get_aspect("Item acceptor", "item acceptor", |n| &n.item_acceptor)?;
|
|
let mut startup_settings = node.startup_settings.get().unwrap().lock();
|
|
startup_settings
|
|
.acceptors
|
|
.insert(acceptor.type_info, Arc::downgrade(acceptor));
|
|
|
|
Ok(())
|
|
}
|
|
|
|
fn generate_desktop_startup_id_flex(
|
|
node: &Node,
|
|
_calling_client: Arc<Client>,
|
|
_data: &[u8],
|
|
) -> Result<Vec<u8>> {
|
|
let id = nanoid::nanoid!();
|
|
let data = serialize(&id)?;
|
|
STARTUP_SETTINGS
|
|
.lock()
|
|
.insert(id, node.startup_settings.get().unwrap().lock().clone());
|
|
Ok(data)
|
|
}
|
|
}
|
|
impl Debug for StartupSettings {
|
|
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
|
f.debug_struct("StartupSettings")
|
|
.field("transform", &self.transform)
|
|
.field(
|
|
"acceptors",
|
|
&self
|
|
.acceptors
|
|
.iter()
|
|
.map(|(k, _)| k.type_name)
|
|
.collect::<Vec<_>>(),
|
|
)
|
|
.finish()
|
|
}
|
|
}
|
|
|
|
pub fn create_interface(client: &Arc<Client>) {
|
|
let node = Node::create(client, "", "startup", false);
|
|
node.add_local_signal("create_startup_settings", create_startup_settings_flex);
|
|
node.add_to_scenegraph();
|
|
}
|
|
|
|
pub fn create_startup_settings_flex(
|
|
_node: &Node,
|
|
calling_client: Arc<Client>,
|
|
data: &[u8],
|
|
) -> Result<()> {
|
|
let name = flexbuffers::Reader::get_root(data)?.get_str()?;
|
|
let node = Node::create(&calling_client, "/startup/settings", name, true).add_to_scenegraph();
|
|
StartupSettings::add_to(&node);
|
|
|
|
node.add_local_signal("set_root", StartupSettings::set_root_flex);
|
|
node.add_local_signal(
|
|
"add_automatic_acceptor",
|
|
StartupSettings::add_automatic_acceptor_flex,
|
|
);
|
|
node.add_local_method(
|
|
"generate_desktop_startup_id",
|
|
StartupSettings::generate_desktop_startup_id_flex,
|
|
);
|
|
|
|
Ok(())
|
|
}
|