use crate::{
cached_set::{CacheId, CachedSet},
change_list::ChangeListBuilder,
events::EventsRegistry,
node::{Attribute, ElementNode, Listener, Node, NodeKind, TextNode},
};
use fxhash::{FxHashMap, FxHashSet};
use std::cmp::Ordering;
use std::u32;
use wasm_bindgen::UnwrapThrowExt;
pub(crate) fn diff(
cached_set: &CachedSet,
change_list: &mut ChangeListBuilder,
registry: &mut EventsRegistry,
old: &Node,
new: &Node,
cached_roots: &mut FxHashSet<CacheId>,
) {
match (&new.kind, &old.kind) {
(
&NodeKind::Text(TextNode { text: new_text }),
&NodeKind::Text(TextNode { text: old_text }),
) => {
if new_text != old_text {
change_list.commit_traversal();
change_list.set_text(new_text);
}
}
(&NodeKind::Text(_), &NodeKind::Element(_)) => {
change_list.commit_traversal();
create(cached_set, change_list, registry, new, cached_roots);
registry.remove_subtree(&old);
change_list.replace_with();
}
(&NodeKind::Element(_), &NodeKind::Text(_)) => {
change_list.commit_traversal();
create(cached_set, change_list, registry, new, cached_roots);
change_list.replace_with();
}
(
&NodeKind::Element(ElementNode {
key: _,
tag_name: new_tag_name,
listeners: new_listeners,
attributes: new_attributes,
children: new_children,
namespace: new_namespace,
}),
&NodeKind::Element(ElementNode {
key: _,
tag_name: old_tag_name,
listeners: old_listeners,
attributes: old_attributes,
children: old_children,
namespace: old_namespace,
}),
) => {
if new_tag_name != old_tag_name || new_namespace != old_namespace {
change_list.commit_traversal();
create(cached_set, change_list, registry, new, cached_roots);
registry.remove_subtree(&old);
change_list.replace_with();
return;
}
diff_listeners(change_list, registry, old_listeners, new_listeners);
diff_attributes(change_list, old_attributes, new_attributes, new_namespace.is_some());
diff_children(
cached_set,
change_list,
registry,
old_children,
new_children,
cached_roots,
);
}
(&NodeKind::Cached(ref new), &NodeKind::Cached(ref old)) => {
cached_roots.insert(new.id);
if new.id == old.id {
return;
}
let (new, new_template) = cached_set.get(new.id);
let (old, old_template) = cached_set.get(old.id);
if new_template == old_template {
diff(cached_set, change_list, registry, old, new, cached_roots);
} else {
change_list.commit_traversal();
create_and_replace(
cached_set,
change_list,
registry,
new_template,
old,
new,
cached_roots,
);
}
}
(&NodeKind::Cached(ref c), _) => {
change_list.commit_traversal();
cached_roots.insert(c.id);
let (new, new_template) = cached_set.get(c.id);
create_and_replace(
cached_set,
change_list,
registry,
new_template,
old,
new,
cached_roots,
);
}
(_, &NodeKind::Cached(_)) => {
change_list.commit_traversal();
create(cached_set, change_list, registry, new, cached_roots);
registry.remove_subtree(&old);
change_list.replace_with();
}
}
}
fn diff_listeners(
change_list: &mut ChangeListBuilder,
registry: &mut EventsRegistry,
old: &[Listener],
new: &[Listener],
) {
if !old.is_empty() || !new.is_empty() {
change_list.commit_traversal();
}
'outer1: for new_l in new {
unsafe {
registry.add(new_l);
}
for old_l in old {
if new_l.event == old_l.event {
change_list.update_event_listener(new_l);
continue 'outer1;
}
}
change_list.new_event_listener(new_l);
}
'outer2: for old_l in old {
registry.remove(old_l);
for new_l in new {
if new_l.event == old_l.event {
continue 'outer2;
}
}
change_list.remove_event_listener(old_l.event);
}
}
fn diff_attributes(change_list: &mut ChangeListBuilder, old: &[Attribute], new: &[Attribute], is_namespaced: bool) {
'outer: for new_attr in new {
if new_attr.is_volatile() {
change_list.commit_traversal();
change_list.set_attribute(new_attr.name, new_attr.value, is_namespaced);
} else {
for old_attr in old {
if old_attr.name == new_attr.name {
if old_attr.value != new_attr.value {
change_list.commit_traversal();
change_list.set_attribute(new_attr.name, new_attr.value, is_namespaced);
}
continue 'outer;
}
}
change_list.commit_traversal();
change_list.set_attribute(new_attr.name, new_attr.value, is_namespaced);
}
}
'outer2: for old_attr in old {
for new_attr in new {
if old_attr.name == new_attr.name {
continue 'outer2;
}
}
change_list.commit_traversal();
change_list.remove_attribute(old_attr.name);
}
}
fn diff_children(
cached_set: &CachedSet,
change_list: &mut ChangeListBuilder,
registry: &mut EventsRegistry,
old: &[Node],
new: &[Node],
cached_roots: &mut FxHashSet<CacheId>,
) {
if new.is_empty() {
if !old.is_empty() {
change_list.commit_traversal();
remove_all_children(change_list, registry, old);
}
return;
}
if new.len() == 1 {
match (old.first(), &new[0]) {
(
Some(&Node {
kind: NodeKind::Text(TextNode { text: old_text }),
}),
&Node {
kind: NodeKind::Text(TextNode { text: new_text }),
},
) if old_text == new_text => {
}
(
_,
&Node {
kind: NodeKind::Text(TextNode { text }),
},
) => {
change_list.commit_traversal();
change_list.set_text(text);
for o in old {
registry.remove_subtree(o);
}
return;
}
(_, _) => {}
}
}
if old.is_empty() {
if !new.is_empty() {
change_list.commit_traversal();
create_and_append_children(cached_set, change_list, registry, new, cached_roots);
}
return;
}
let new_is_keyed = new[0].key().is_some();
let old_is_keyed = old[0].key().is_some();
debug_assert!(
new.iter().all(|n| n.key().is_some() == new_is_keyed),
"all siblings must be keyed or all siblings must be non-keyed"
);
debug_assert!(
old.iter().all(|o| o.key().is_some() == old_is_keyed),
"all siblings must be keyed or all siblings must be non-keyed"
);
if new_is_keyed && old_is_keyed {
let t = change_list.next_temporary();
diff_keyed_children(cached_set, change_list, registry, old, new, cached_roots);
change_list.set_next_temporary(t);
} else {
diff_non_keyed_children(cached_set, change_list, registry, old, new, cached_roots);
}
}
fn diff_keyed_children(
cached_set: &CachedSet,
change_list: &mut ChangeListBuilder,
registry: &mut EventsRegistry,
old: &[Node],
new: &[Node],
cached_roots: &mut FxHashSet<CacheId>,
) {
if cfg!(debug_assertions) {
let mut keys = FxHashSet::default();
let mut assert_unique_keys = |children: &[Node]| {
keys.clear();
for child in children {
let key = child.key();
debug_assert!(
key.is_some(),
"if any sibling is keyed, all siblings must be keyed"
);
keys.insert(key);
}
debug_assert_eq!(
children.len(),
keys.len(),
"keyed siblings must each have a unique key"
);
};
assert_unique_keys(old);
assert_unique_keys(new);
}
let shared_prefix_count =
match diff_keyed_prefix(cached_set, change_list, registry, old, new, cached_roots) {
KeyedPrefixResult::Finished => return,
KeyedPrefixResult::MoreWorkToDo(count) => count,
};
let shared_suffix_count = old[shared_prefix_count..]
.iter()
.rev()
.zip(new[shared_prefix_count..].iter().rev())
.take_while(|&(old, new)| old.key() == new.key())
.count();
let old_shared_suffix_start = old.len() - shared_suffix_count;
let new_shared_suffix_start = new.len() - shared_suffix_count;
diff_keyed_middle(
cached_set,
change_list,
registry,
&old[shared_prefix_count..old_shared_suffix_start],
&new[shared_prefix_count..new_shared_suffix_start],
cached_roots,
shared_prefix_count,
shared_suffix_count,
old_shared_suffix_start,
);
let old_suffix = &old[old_shared_suffix_start..];
let new_suffix = &new[new_shared_suffix_start..];
debug_assert_eq!(old_suffix.len(), new_suffix.len());
if !old_suffix.is_empty() {
diff_keyed_suffix(
cached_set,
change_list,
registry,
old_suffix,
new_suffix,
cached_roots,
new_shared_suffix_start,
);
}
}
enum KeyedPrefixResult {
Finished,
MoreWorkToDo(usize),
}
fn diff_keyed_prefix(
cached_set: &CachedSet,
change_list: &mut ChangeListBuilder,
registry: &mut EventsRegistry,
old: &[Node],
new: &[Node],
cached_roots: &mut FxHashSet<CacheId>,
) -> KeyedPrefixResult {
change_list.go_down();
let mut shared_prefix_count = 0;
for (i, (old, new)) in old.iter().zip(new.iter()).enumerate() {
if old.key() != new.key() {
break;
}
change_list.go_to_sibling(i);
diff(cached_set, change_list, registry, old, new, cached_roots);
shared_prefix_count += 1;
}
if shared_prefix_count == old.len() {
change_list.go_up();
change_list.commit_traversal();
create_and_append_children(
cached_set,
change_list,
registry,
&new[shared_prefix_count..],
cached_roots,
);
return KeyedPrefixResult::Finished;
}
if shared_prefix_count == new.len() {
change_list.go_to_sibling(shared_prefix_count);
change_list.commit_traversal();
remove_self_and_next_siblings(change_list, registry, &old[shared_prefix_count..]);
return KeyedPrefixResult::Finished;
}
change_list.go_up();
KeyedPrefixResult::MoreWorkToDo(shared_prefix_count)
}
fn diff_keyed_middle(
cached_set: &CachedSet,
change_list: &mut ChangeListBuilder,
registry: &mut EventsRegistry,
old: &[Node],
mut new: &[Node],
cached_roots: &mut FxHashSet<CacheId>,
shared_prefix_count: usize,
shared_suffix_count: usize,
old_shared_suffix_start: usize,
) {
debug_assert_ne!(new.first().map(|n| n.key()), old.first().map(|o| o.key()));
debug_assert_ne!(new.last().map(|n| n.key()), old.last().map(|o| o.key()));
debug_assert!(new.len() < u32::MAX as usize);
let mut old_key_to_old_index = FxHashMap::default();
old_key_to_old_index.reserve(old.len());
old_key_to_old_index.extend(old.iter().enumerate().map(|(i, o)| (o.key(), i)));
let mut shared_keys = FxHashSet::default();
let mut new_index_to_old_index = Vec::with_capacity(new.len());
new_index_to_old_index.extend(new.iter().map(|n| {
let key = n.key();
if let Some(&i) = old_key_to_old_index.get(&key) {
shared_keys.insert(key);
i
} else {
u32::MAX as usize
}
}));
if shared_suffix_count == 0 && shared_keys.is_empty() {
if shared_prefix_count == 0 {
change_list.commit_traversal();
remove_all_children(change_list, registry, old);
} else {
change_list.go_down_to_child(shared_prefix_count);
change_list.commit_traversal();
remove_self_and_next_siblings(change_list, registry, &old[shared_prefix_count..]);
}
create_and_append_children(cached_set, change_list, registry, new, cached_roots);
return;
}
let mut old_index_to_temp = vec![u32::MAX; old.len()];
let mut start = 0;
loop {
let end = (start..old.len())
.find(|&i| {
let key = old[i].key();
!shared_keys.contains(&key)
})
.unwrap_or(old.len());
if end - start > 0 {
change_list.commit_traversal();
let mut t = change_list.save_children_to_temporaries(
shared_prefix_count + start,
shared_prefix_count + end,
);
for i in start..end {
old_index_to_temp[i] = t;
t += 1;
}
}
debug_assert!(end <= old.len());
if end == old.len() {
break;
} else {
start = end + 1;
}
}
let mut removed_count = 0;
for (i, old_child) in old.iter().enumerate().rev() {
if !shared_keys.contains(&old_child.key()) {
registry.remove_subtree(old_child);
change_list.commit_traversal();
change_list.remove_child(i + shared_prefix_count);
removed_count += 1;
}
}
if new.is_empty() {
return;
}
let mut new_index_is_in_lis = FxHashSet::default();
new_index_is_in_lis.reserve(new_index_to_old_index.len());
let mut predecessors = vec![0; new_index_to_old_index.len()];
let mut starts = vec![0; new_index_to_old_index.len()];
longest_increasing_subsequence::lis_with(
&new_index_to_old_index,
&mut new_index_is_in_lis,
|a, b| a < b,
&mut predecessors,
&mut starts,
);
if shared_suffix_count > 0 {
change_list.go_down_to_child(old_shared_suffix_start - removed_count);
} else {
let last_index = new.len() - 1;
let last = new.last().unwrap_throw();
new = &new[..new.len() - 1];
if shared_keys.contains(&last.key()) {
let old_index = new_index_to_old_index[last_index];
let temp = old_index_to_temp[old_index];
change_list.go_down_to_temp_child(temp);
diff(
cached_set,
change_list,
registry,
&old[old_index],
last,
cached_roots,
);
if new_index_is_in_lis.contains(&last_index) {
} else {
change_list.commit_traversal();
change_list.append_child();
change_list.go_down_to_temp_child(temp);
}
} else {
change_list.commit_traversal();
create(cached_set, change_list, registry, last, cached_roots);
change_list.append_child();
change_list.go_down_to_reverse_child(0);
}
}
for (new_index, new_child) in new.iter().enumerate().rev() {
let old_index = new_index_to_old_index[new_index];
if old_index == u32::MAX as usize {
debug_assert!(!shared_keys.contains(&new_child.key()));
change_list.commit_traversal();
create(cached_set, change_list, registry, new_child, cached_roots);
change_list.insert_before();
} else {
debug_assert!(shared_keys.contains(&new_child.key()));
let temp = old_index_to_temp[old_index];
debug_assert_ne!(temp, u32::MAX);
if new_index_is_in_lis.contains(&new_index) {
change_list.go_to_temp_sibling(temp);
} else {
change_list.commit_traversal();
change_list.push_temporary(temp);
change_list.insert_before();
}
diff(
cached_set,
change_list,
registry,
&old[old_index],
new_child,
cached_roots,
);
}
}
change_list.go_up();
}
fn diff_keyed_suffix(
cached_set: &CachedSet,
change_list: &mut ChangeListBuilder,
registry: &mut EventsRegistry,
old: &[Node],
new: &[Node],
cached_roots: &mut FxHashSet<CacheId>,
new_shared_suffix_start: usize,
) {
debug_assert_eq!(old.len(), new.len());
debug_assert!(!old.is_empty());
change_list.go_down();
for (i, (old_child, new_child)) in old.iter().zip(new.iter()).enumerate() {
change_list.go_to_sibling(new_shared_suffix_start + i);
diff(
cached_set,
change_list,
registry,
old_child,
new_child,
cached_roots,
);
}
change_list.go_up();
}
fn diff_non_keyed_children(
cached_set: &CachedSet,
change_list: &mut ChangeListBuilder,
registry: &mut EventsRegistry,
old: &[Node],
new: &[Node],
cached_roots: &mut FxHashSet<CacheId>,
) {
debug_assert!(!new.is_empty());
debug_assert!(!old.is_empty());
change_list.go_down();
for (i, (new_child, old_child)) in new.iter().zip(old.iter()).enumerate() {
change_list.go_to_sibling(i);
diff(
cached_set,
change_list,
registry,
old_child,
new_child,
cached_roots,
);
}
match old.len().cmp(&new.len()) {
Ordering::Greater => {
change_list.go_to_sibling(new.len());
change_list.commit_traversal();
remove_self_and_next_siblings(change_list, registry, &old[new.len()..]);
}
Ordering::Less => {
change_list.go_up();
change_list.commit_traversal();
create_and_append_children(
cached_set,
change_list,
registry,
&new[old.len()..],
cached_roots,
);
}
Ordering::Equal => {
change_list.go_up();
}
}
}
fn create_and_append_children(
cached_set: &CachedSet,
change_list: &mut ChangeListBuilder,
registry: &mut EventsRegistry,
new: &[Node],
cached_roots: &mut FxHashSet<CacheId>,
) {
debug_assert!(change_list.traversal_is_committed());
for child in new {
create(cached_set, change_list, registry, child, cached_roots);
change_list.append_child();
}
}
fn remove_all_children(
change_list: &mut ChangeListBuilder,
registry: &mut EventsRegistry,
old: &[Node],
) {
debug_assert!(change_list.traversal_is_committed());
for child in old {
registry.remove_subtree(child);
}
change_list.set_text("");
}
fn remove_self_and_next_siblings(
change_list: &mut ChangeListBuilder,
registry: &mut EventsRegistry,
old: &[Node],
) {
debug_assert!(change_list.traversal_is_committed());
for child in old {
registry.remove_subtree(child);
}
change_list.remove_self_and_next_siblings();
}
fn create(
cached_set: &CachedSet,
change_list: &mut ChangeListBuilder,
registry: &mut EventsRegistry,
node: &Node,
cached_roots: &mut FxHashSet<CacheId>,
) {
debug_assert!(change_list.traversal_is_committed());
match node.kind {
NodeKind::Text(TextNode { text }) => {
change_list.create_text_node(text);
}
NodeKind::Element(&ElementNode {
key: _,
tag_name,
listeners,
attributes,
children,
namespace,
}) => {
if let Some(namespace) = namespace {
change_list.create_element_ns(tag_name, namespace);
} else {
change_list.create_element(tag_name);
}
for l in listeners {
unsafe {
registry.add(l);
}
change_list.new_event_listener(l);
}
for attr in attributes {
change_list.set_attribute(&attr.name, &attr.value, namespace.is_some());
}
if children.len() == 1 {
if let Node {
kind: NodeKind::Text(TextNode { text }),
} = children[0]
{
change_list.set_text(text);
return;
}
}
for child in children {
create(cached_set, change_list, registry, child, cached_roots);
change_list.append_child();
}
}
NodeKind::Cached(ref c) => {
cached_roots.insert(c.id);
let (node, template) = cached_set.get(c.id);
if let Some(template) = template {
create_with_template(
cached_set,
change_list,
registry,
template,
node,
cached_roots,
);
} else {
create(cached_set, change_list, registry, node, cached_roots);
}
}
}
}
#[inline]
fn get_or_create_template<'a>(
cached_set: &'a CachedSet,
change_list: &mut ChangeListBuilder,
registry: &mut EventsRegistry,
cached_roots: &mut FxHashSet<CacheId>,
template_id: CacheId,
) -> (&'a Node<'a>, bool) {
let (template, template_template) = cached_set.get(template_id);
debug_assert!(
template_template.is_none(),
"templates should not be templated themselves"
);
if change_list.has_template(template_id) {
change_list.push_template(template_id);
(template, true)
} else {
create(cached_set, change_list, registry, template, cached_roots);
change_list.save_template(template_id);
(template, false)
}
}
fn create_and_replace(
cached_set: &CachedSet,
change_list: &mut ChangeListBuilder,
registry: &mut EventsRegistry,
new_template: Option<CacheId>,
old: &Node,
new: &Node,
cached_roots: &mut FxHashSet<CacheId>,
) {
debug_assert!(change_list.traversal_is_committed());
if let Some(template_id) = new_template {
let (template, needs_listeners) =
get_or_create_template(cached_set, change_list, registry, cached_roots, template_id);
change_list.replace_with();
let mut old_forcing = None;
if needs_listeners {
old_forcing = Some(change_list.push_force_new_listeners());
}
diff(
cached_set,
change_list,
registry,
template,
new,
cached_roots,
);
if let Some(old) = old_forcing {
change_list.pop_force_new_listeners(old);
}
change_list.commit_traversal();
} else {
create(cached_set, change_list, registry, new, cached_roots);
change_list.replace_with();
}
registry.remove_subtree(old);
}
fn create_with_template(
cached_set: &CachedSet,
change_list: &mut ChangeListBuilder,
registry: &mut EventsRegistry,
template_id: CacheId,
node: &Node,
cached_roots: &mut FxHashSet<CacheId>,
) {
debug_assert!(change_list.traversal_is_committed());
let (template, needs_listeners) =
get_or_create_template(cached_set, change_list, registry, cached_roots, template_id);
let mut old_forcing = None;
if needs_listeners {
old_forcing = Some(change_list.push_force_new_listeners());
}
diff(
cached_set,
change_list,
registry,
template,
node,
cached_roots,
);
if let Some(old) = old_forcing {
change_list.pop_force_new_listeners(old);
}
change_list.commit_traversal();
}