fix: use getter for all_opened

dont allow access to struct field
This commit is contained in:
EdJoPaTo
2020-10-31 21:31:54 +01:00
parent b585b9dde9
commit 5265bf1d1e
3 changed files with 13 additions and 14 deletions
+6 -11
View File
@@ -2,7 +2,6 @@
use crate::identifier::{TreeIdentifier, TreeIdentifierVec};
use crate::TreeItem;
use std::collections::HashSet;
pub struct Flattened<'a> {
pub identifier: Vec<usize>,
@@ -15,15 +14,12 @@ impl<'a> Flattened<'a> {
}
}
pub fn flatten<'a>(
opened: &HashSet<TreeIdentifierVec>,
items: &'a [TreeItem<'a>],
) -> Vec<Flattened<'a>> {
pub fn flatten<'a>(opened: &[TreeIdentifierVec], items: &'a [TreeItem<'a>]) -> Vec<Flattened<'a>> {
internal(opened, items, &[])
}
fn internal<'a>(
opened: &HashSet<TreeIdentifierVec>,
opened: &[TreeIdentifierVec],
items: &'a [TreeItem<'a>],
current: TreeIdentifier,
) -> Vec<Flattened<'a>> {
@@ -78,8 +74,7 @@ fn get_example_tree_items<'a>() -> Vec<TreeItem<'a>> {
#[test]
fn get_opened_nothing_opened_is_top_level() {
let items = get_example_tree_items();
let opened = HashSet::new();
let result = flatten(&opened, &items);
let result = flatten(&[], &items);
let result_text: Vec<_> = result
.iter()
.map(|o| get_naive_string_from_text(&o.item.text))
@@ -90,7 +85,7 @@ fn get_opened_nothing_opened_is_top_level() {
#[test]
fn get_opened_wrong_opened_is_only_top_level() {
let items = get_example_tree_items();
let opened = [vec![0], vec![1, 1]].iter().cloned().collect();
let opened = [vec![0], vec![1, 1]];
let result = flatten(&opened, &items);
let result_text: Vec<_> = result
.iter()
@@ -102,7 +97,7 @@ fn get_opened_wrong_opened_is_only_top_level() {
#[test]
fn get_opened_one_is_opened() {
let items = get_example_tree_items();
let opened = [vec![1]].iter().cloned().collect();
let opened = [vec![1]];
let result = flatten(&opened, &items);
let result_text: Vec<_> = result
.iter()
@@ -114,7 +109,7 @@ fn get_opened_one_is_opened() {
#[test]
fn get_opened_all_opened() {
let items = get_example_tree_items();
let opened = [vec![1], vec![1, 1]].iter().cloned().collect();
let opened = [vec![1], vec![1, 1]];
let result = flatten(&opened, &items);
let result_text: Vec<_> = result
.iter()
+6 -2
View File
@@ -19,7 +19,7 @@ pub use self::flatten::flatten;
pub struct TreeState {
offset: usize,
selected: TreeIdentifierVec,
pub opened: HashSet<TreeIdentifierVec>,
opened: HashSet<TreeIdentifierVec>,
}
impl Default for TreeState {
@@ -64,6 +64,10 @@ impl TreeState {
pub fn close_all(&mut self) {
self.opened.clear();
}
pub fn get_all_opened(&self) -> Vec<TreeIdentifierVec> {
self.opened.iter().cloned().collect()
}
}
#[derive(Debug, Clone)]
@@ -184,7 +188,7 @@ impl<'a> StatefulWidget for Tree<'a> {
return;
}
let visible = flatten(&state.opened, &self.items);
let visible = flatten(&state.get_all_opened(), &self.items);
if visible.is_empty() {
return;
}