beet_build 0.0.8

Codegen and compilation tooling for beet
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
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93

use crate::prelude::*;
use beet_core::prelude::*;
use beet_dom::prelude::*;
use beet_parse::prelude::*;
use syn::visit::Visit;


/// For a given rust file, extract tokens from `rsx!` macros and insert them
/// as [`RstmlTokens`].
pub fn import_rsx_snippets_rs(
	// even though our tokens are Unspan, they will be parsed by ParseRsxTokens
	// which also handles !Send tokens, so we must ensure main thread.
	_: TempNonSendMarker,
	macros: Res<TemplateMacros>,
	mut commands: Commands,
	query: Populated<(Entity, &SourceFile), Added<SourceFile>>,
) -> Result {
	for (entity, path) in query.iter() {
		if let Some(ex) = path.extension()
			&& ex == "rs"
		{
			trace!("rust source file changed: {}", path.display());

			let file = fs_ext::read_to_string(path)?;
			let file = syn::parse_file(&file)?;
			RsxSynVisitor {
				source_file: entity,
				commands: &mut commands,
				file: &path.into_ws_path()?,
				macros: &*macros,
			}
			.visit_file(&file);
		}
	}
	Ok(())
}

/// Spawn an [`RstmlTokens`] for each `rsx!` macro in the file.
struct RsxSynVisitor<'a, 'w, 's> {
	source_file: Entity,
	commands: &'a mut Commands<'w, 's>,
	/// Used for creating [`FileSpan`] in several places.
	/// We must use workspace relative paths because locations are created
	/// via the `file!()` macro.
	file: &'a WsPathBuf,
	macros: &'a TemplateMacros,
}

impl<'a, 'w, 's> Visit<'a> for RsxSynVisitor<'a, 'w, 's> {
	fn visit_macro(&mut self, mac: &syn::Macro) {
		if mac
			.path
			.segments
			.last()
			.map_or(false, |seg| *&seg.ident == *self.macros.rstml)
		{
			// mac.tokens is the inner tokens of the macro, ie the foo in rsx!{foo}
			// important for tracking exact span of the macro
			let tokens = mac.tokens.clone();
			self.commands.spawn((
				SnippetRoot::new_from_tokens(self.file.clone(), &tokens),
				StaticRoot,
				ChildOf(self.source_file),
				RstmlTokens::new(tokens),
			));
		}
	}
}


#[cfg(test)]
mod test {
	use crate::prelude::*;
	use beet_core::prelude::*;
	use beet_rsx::prelude::*;

	#[test]
	fn works() {
		let mut world = BuildPlugin::world();
		let test_site_index = WsPathBuf::new("tests/test_site/pages/index.rs");
		let entity = world
			.spawn(SourceFile::new(test_site_index.into_abs()))
			.id();

		world.run_schedule(ParseSourceFiles);
		let child = world.entity(entity).get::<Children>().unwrap()[0];
		world
			.run_system_cached_with(render_fragment, child)
			.unwrap()
			// only the output of the snippet, not the instance
			.xpect_str("party time!");
	}
}