automerge/rust/automerge/src/query/keys.rs
Alex Good dd3c6d1303
Move rust workspace into ./rust
After some discussion with PVH I realise that the repo structure in the
last reorg was very rust-centric. In an attempt to put each language on
a level footing move the rust code and project files into ./rust
2022-10-16 19:55:51 +01:00

54 lines
1.4 KiB
Rust

use crate::op_tree::OpTreeNode;
use crate::types::Key;
use std::fmt::Debug;
#[derive(Debug)]
pub(crate) struct Keys<'a> {
index: usize,
last_key: Option<Key>,
index_back: usize,
last_key_back: Option<Key>,
root_child: &'a OpTreeNode,
}
impl<'a> Keys<'a> {
pub(crate) fn new(root_child: &'a OpTreeNode) -> Self {
Self {
index: 0,
last_key: None,
index_back: root_child.len(),
last_key_back: None,
root_child,
}
}
}
impl<'a> Iterator for Keys<'a> {
type Item = Key;
fn next(&mut self) -> Option<Self::Item> {
for i in self.index..self.index_back {
let op = self.root_child.get(i)?;
self.index += 1;
if Some(op.elemid_or_key()) != self.last_key && op.visible() {
self.last_key = Some(op.elemid_or_key());
return Some(op.elemid_or_key());
}
}
None
}
}
impl<'a> DoubleEndedIterator for Keys<'a> {
fn next_back(&mut self) -> Option<Self::Item> {
for i in (self.index..self.index_back).rev() {
let op = self.root_child.get(i)?;
self.index_back -= 1;
if Some(op.elemid_or_key()) != self.last_key_back && op.visible() {
self.last_key_back = Some(op.elemid_or_key());
return Some(op.elemid_or_key());
}
}
None
}
}