81 lines
2.4 KiB
Rust
81 lines
2.4 KiB
Rust
use crate::{
|
|
any_props::AnyProps,
|
|
innerlude::DirtyScope,
|
|
nodes::RenderReturn,
|
|
scope_context::ScopeContext,
|
|
scopes::{ScopeId, ScopeState},
|
|
virtual_dom::VirtualDom,
|
|
};
|
|
|
|
impl VirtualDom {
|
|
pub(super) fn new_scope(
|
|
&mut self,
|
|
props: Box<dyn AnyProps>,
|
|
name: &'static str,
|
|
) -> &ScopeState {
|
|
let parent_id = self.runtime.current_scope_id();
|
|
let height = parent_id
|
|
.and_then(|parent_id| self.get_scope(parent_id).map(|f| f.context().height + 1))
|
|
.unwrap_or(0);
|
|
let entry = self.scopes.vacant_entry();
|
|
let id = ScopeId(entry.key());
|
|
|
|
let scope = entry.insert(Box::new(ScopeState {
|
|
runtime: self.runtime.clone(),
|
|
context_id: id,
|
|
|
|
props,
|
|
last_rendered_node: Default::default(),
|
|
}));
|
|
|
|
let context =
|
|
ScopeContext::new(name, id, parent_id, height, self.runtime.scheduler.clone());
|
|
self.runtime.create_context_at(id, context);
|
|
|
|
scope
|
|
}
|
|
|
|
pub(crate) fn run_scope(&mut self, scope_id: ScopeId) -> RenderReturn {
|
|
self.runtime.scope_stack.borrow_mut().push(scope_id);
|
|
|
|
let new_nodes = unsafe {
|
|
let scope = &self.scopes[scope_id.0];
|
|
|
|
let context = scope.context();
|
|
context.suspended.set(false);
|
|
context.hook_index.set(0);
|
|
|
|
// safety: due to how we traverse the tree, we know that the scope is not currently aliased
|
|
let props: &dyn AnyProps = &*scope.props;
|
|
|
|
let _span = tracing::trace_span!("render", scope = %scope.context().name);
|
|
props.render()
|
|
};
|
|
|
|
let scope = &mut self.scopes[scope_id.0];
|
|
|
|
let context = scope.context();
|
|
|
|
// And move the render generation forward by one
|
|
context.render_count.set(context.render_count.get() + 1);
|
|
|
|
// remove this scope from dirty scopes
|
|
self.dirty_scopes.remove(&DirtyScope {
|
|
height: context.height,
|
|
id: context.id,
|
|
});
|
|
|
|
if context.suspended.get() {
|
|
if matches!(new_nodes, RenderReturn::Aborted(_)) {
|
|
self.suspended_scopes.insert(context.id);
|
|
}
|
|
} else if !self.suspended_scopes.is_empty() {
|
|
_ = self.suspended_scopes.remove(&context.id);
|
|
}
|
|
|
|
self.runtime.scope_stack.borrow_mut().pop();
|
|
|
|
new_nodes
|
|
}
|
|
}
|