beet_dom 0.0.8

Utilities for dom rendering and interaction
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55

use crate::prelude::*;
use beet_core::prelude::*;





/// An element containing code, this element will be replaced by the `<pre>`
/// generated by syntect.
#[derive(Debug, Default, Clone, PartialEq, Hash, Component, Reflect)]
#[reflect(Component)]
#[component(immutable)]
#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
#[cfg_attr(feature = "tokens", derive(ToTokens))]
pub struct CodeNode {
	/// Override the [`SyntectConfig::theme`] lang for this code node.
	pub lang: Option<String>,
	/// Override the [`SyntectConfig::theme`] theme for this code node.
	pub theme: Option<String>,
}
impl CodeNode {
	/// Create a new [`CodeNode`] with the given language.
	pub fn new(lang: impl Into<String>) -> Self {
		Self {
			lang: Some(lang.into()),
			theme: None,
		}
	}
}



pub fn extract_code_nodes(
	mut commands: Commands,
	query: Populated<Entity, Added<ElementNode>>,
	attributes: FindAttribute,
) {
	for node_ent in query.iter() {
		if let Some((attr_entity, _)) = attributes.find(node_ent, "node:code") {
			commands.entity(attr_entity).despawn();

			let lang =
				attributes.find_value(node_ent, "lang").map(|(entity, v)| {
					commands.entity(entity).despawn();
					v.0.clone()
				});
			let theme =
				attributes.find_value(node_ent, "theme").map(|(entity, v)| {
					commands.entity(entity).despawn();
					v.0.clone()
				});
			commands.entity(node_ent).insert(CodeNode { lang, theme });
		}
	}
}