mirror of
https://github.com/ratatui-org/ratatui
synced 2024-11-10 07:04:17 +00:00
feat(serde): support TableState, ListState, and ScrollbarState (#723)
TableState, ListState, and ScrollbarState can now be serialized and deserialized using serde. ```rust #[derive(Debug, Clone, serde::Serialize, serde::Deserialize)] struct AppState { list_state: ListState, table_state: TableState, scrollbar_state: ScrollbarState, } let app_state = AppState::default(); let serialized = serde_json::to_string(app_state); let app_state = serde_json::from_str(serialized); ```
This commit is contained in:
parent
43b2b57191
commit
fe06f0c7b0
5 changed files with 212 additions and 0 deletions
|
@ -55,6 +55,7 @@ palette = "0.7.3"
|
|||
pretty_assertions = "1.4.0"
|
||||
rand = "0.8.5"
|
||||
rstest = "0.18.2"
|
||||
serde_json = "1.0.109"
|
||||
|
||||
[features]
|
||||
#! The crate provides a set of optional features that can be enabled in your `cargo.toml` file.
|
||||
|
@ -268,3 +269,7 @@ doc-scrape-examples = true
|
|||
name = "inline"
|
||||
required-features = ["crossterm"]
|
||||
doc-scrape-examples = true
|
||||
|
||||
[[test]]
|
||||
name = "state_serde"
|
||||
required-features = ["serde"]
|
||||
|
|
|
@ -45,6 +45,7 @@ use crate::{
|
|||
/// # }
|
||||
/// ```
|
||||
#[derive(Debug, Default, Clone, Eq, PartialEq, Hash)]
|
||||
#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
|
||||
pub struct ListState {
|
||||
offset: usize,
|
||||
selected: Option<usize>,
|
||||
|
|
|
@ -42,6 +42,7 @@ pub enum ScrollDirection {
|
|||
/// If you don't have multi-line content, you can leave the `viewport_content_length` set to the
|
||||
/// default of 0 and it'll use the track size as a `viewport_content_length`.
|
||||
#[derive(Debug, Default, Clone, Copy, Eq, PartialEq, Hash)]
|
||||
#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
|
||||
pub struct ScrollbarState {
|
||||
// The total length of the scrollable content.
|
||||
content_length: usize,
|
||||
|
|
|
@ -45,6 +45,7 @@
|
|||
/// [`Table::widths`]: crate::widgets::Table::widths
|
||||
/// [`Frame::render_stateful_widget`]: crate::Frame::render_stateful_widget
|
||||
#[derive(Debug, Default, Clone, Eq, PartialEq, Hash)]
|
||||
#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
|
||||
pub struct TableState {
|
||||
pub(crate) offset: usize,
|
||||
pub(crate) selected: Option<usize>,
|
||||
|
|
204
tests/state_serde.rs
Normal file
204
tests/state_serde.rs
Normal file
|
@ -0,0 +1,204 @@
|
|||
//! State like [`ListState`], [`TableState`] and [`ScrollbarState`] can be serialized and
|
||||
//! deserialized through serde. This allows saving your entire state to disk when the user exits the
|
||||
//! the app, and restore it again upon re-opening the app.
|
||||
//! This way, they get right back to where they were, without having to re-seek to their previous
|
||||
//! position, if that's applicable for the app at hand.
|
||||
//!
|
||||
//! **Note**: For this pattern to work easily, you need to have some toplevel struct which stores
|
||||
//! _only_ state and not any draw commands.
|
||||
//!
|
||||
//! **Note**: For many applications, it might be beneficial to instead keep your own state and
|
||||
//! instead construct the state for widgets on the fly instead, if that allows you to express you
|
||||
//! the semantic meaning of your state better or only fetch part of a dataset.
|
||||
|
||||
// not too happy about the redundancy in these tests,
|
||||
// but if that helps readability then it's ok i guess /shrug
|
||||
|
||||
use ratatui::{backend::TestBackend, prelude::*, widgets::*};
|
||||
|
||||
#[derive(Debug, Clone, PartialEq, Eq, serde::Serialize, serde::Deserialize)]
|
||||
struct AppState {
|
||||
list_state: ListState,
|
||||
table_state: TableState,
|
||||
scrollbar_state: ScrollbarState,
|
||||
}
|
||||
|
||||
impl Default for AppState {
|
||||
fn default() -> Self {
|
||||
Self {
|
||||
list_state: ListState::default(),
|
||||
table_state: TableState::default(),
|
||||
scrollbar_state: ScrollbarState::new(10),
|
||||
}
|
||||
}
|
||||
}
|
||||
impl AppState {
|
||||
fn select(&mut self, index: usize) {
|
||||
self.list_state.select(Some(index));
|
||||
self.table_state.select(Some(index));
|
||||
self.scrollbar_state = self.scrollbar_state.position(index);
|
||||
}
|
||||
}
|
||||
|
||||
/// Renders the list to a TestBackend and asserts that the result matches the expected buffer.
|
||||
#[track_caller]
|
||||
fn assert_buffer(state: &mut AppState, expected: &Buffer) {
|
||||
let backend = TestBackend::new(21, 5);
|
||||
let mut terminal = Terminal::new(backend).unwrap();
|
||||
terminal
|
||||
.draw(|f| {
|
||||
let items = vec![
|
||||
"awa", "banana", "Cats!!", "d20", "Echo", "Foxtrot", "Golf", "Hotel", "IwI",
|
||||
"Juliett",
|
||||
];
|
||||
|
||||
use Constraint::*;
|
||||
let layout = Layout::default()
|
||||
.direction(Direction::Horizontal)
|
||||
.constraints([Length(10), Length(10), Length(1)])
|
||||
.split(f.size());
|
||||
let list = List::new(items.clone())
|
||||
.highlight_symbol(">>")
|
||||
.block(Block::default().borders(Borders::RIGHT));
|
||||
f.render_stateful_widget(list, layout[0], &mut state.list_state);
|
||||
|
||||
let table = Table::new(
|
||||
items.iter().map(|i| Row::new(vec![*i])),
|
||||
[Constraint::Length(10); 5],
|
||||
)
|
||||
.highlight_symbol(">>");
|
||||
f.render_stateful_widget(table, layout[1], &mut state.table_state);
|
||||
|
||||
let scrollbar = Scrollbar::new(ScrollbarOrientation::VerticalRight);
|
||||
f.render_stateful_widget(scrollbar, layout[2], &mut state.scrollbar_state);
|
||||
})
|
||||
.unwrap();
|
||||
terminal.backend().assert_buffer(expected);
|
||||
}
|
||||
|
||||
const DEFAULT_STATE_BUFFER: [&str; 5] = [
|
||||
"awa │awa ▲",
|
||||
"banana │banana █",
|
||||
"Cats!! │Cats!! ║",
|
||||
"d20 │d20 ║",
|
||||
"Echo │Echo ▼",
|
||||
];
|
||||
|
||||
const DEFAULT_STATE_REPR: &str = r#"{
|
||||
"list_state": {
|
||||
"offset": 0,
|
||||
"selected": null
|
||||
},
|
||||
"table_state": {
|
||||
"offset": 0,
|
||||
"selected": null
|
||||
},
|
||||
"scrollbar_state": {
|
||||
"content_length": 10,
|
||||
"position": 0,
|
||||
"viewport_content_length": 0
|
||||
}
|
||||
}"#;
|
||||
|
||||
#[test]
|
||||
fn default_state_serialize() {
|
||||
let mut state = AppState::default();
|
||||
|
||||
let expected = Buffer::with_lines(DEFAULT_STATE_BUFFER.to_vec());
|
||||
assert_buffer(&mut state, &expected);
|
||||
|
||||
let state = serde_json::to_string_pretty(&state).unwrap();
|
||||
assert_eq!(state, DEFAULT_STATE_REPR);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn default_state_deserialize() {
|
||||
let expected = Buffer::with_lines(DEFAULT_STATE_BUFFER.to_vec());
|
||||
let mut state: AppState = serde_json::from_str(DEFAULT_STATE_REPR).unwrap();
|
||||
assert_buffer(&mut state, &expected);
|
||||
}
|
||||
|
||||
const SELECTED_STATE_BUFFER: [&str; 5] = [
|
||||
" awa │ awa ▲",
|
||||
">>banana │>>banana █",
|
||||
" Cats!! │ Cats!! ║",
|
||||
" d20 │ d20 ║",
|
||||
" Echo │ Echo ▼",
|
||||
];
|
||||
const SELECTED_STATE_REPR: &str = r#"{
|
||||
"list_state": {
|
||||
"offset": 0,
|
||||
"selected": 1
|
||||
},
|
||||
"table_state": {
|
||||
"offset": 0,
|
||||
"selected": 1
|
||||
},
|
||||
"scrollbar_state": {
|
||||
"content_length": 10,
|
||||
"position": 1,
|
||||
"viewport_content_length": 0
|
||||
}
|
||||
}"#;
|
||||
|
||||
#[test]
|
||||
fn selected_state_serialize() {
|
||||
let mut state = AppState::default();
|
||||
state.select(1);
|
||||
|
||||
let expected = Buffer::with_lines(SELECTED_STATE_BUFFER.to_vec());
|
||||
assert_buffer(&mut state, &expected);
|
||||
|
||||
let state = serde_json::to_string_pretty(&state).unwrap();
|
||||
assert_eq!(state, SELECTED_STATE_REPR);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn selected_state_deserialize() {
|
||||
let expected = Buffer::with_lines(SELECTED_STATE_BUFFER.to_vec());
|
||||
let mut state: AppState = serde_json::from_str(SELECTED_STATE_REPR).unwrap();
|
||||
assert_buffer(&mut state, &expected);
|
||||
}
|
||||
|
||||
const SCROLLED_STATE_BUFFER: [&str; 5] = [
|
||||
" Echo │ Echo ▲",
|
||||
" Foxtrot│ Foxtrot ║",
|
||||
" Golf │ Golf ║",
|
||||
" Hotel │ Hotel █",
|
||||
">>IwI │>>IwI ▼",
|
||||
];
|
||||
|
||||
const SCROLLED_STATE_REPR: &str = r#"{
|
||||
"list_state": {
|
||||
"offset": 4,
|
||||
"selected": 8
|
||||
},
|
||||
"table_state": {
|
||||
"offset": 4,
|
||||
"selected": 8
|
||||
},
|
||||
"scrollbar_state": {
|
||||
"content_length": 10,
|
||||
"position": 8,
|
||||
"viewport_content_length": 0
|
||||
}
|
||||
}"#;
|
||||
|
||||
#[test]
|
||||
fn scrolled_state_serialize() {
|
||||
let mut state = AppState::default();
|
||||
state.select(8);
|
||||
|
||||
let expected = Buffer::with_lines(SCROLLED_STATE_BUFFER.to_vec());
|
||||
assert_buffer(&mut state, &expected);
|
||||
|
||||
let state = serde_json::to_string_pretty(&state).unwrap();
|
||||
assert_eq!(state, SCROLLED_STATE_REPR);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn scrolled_state_deserialize() {
|
||||
let expected = Buffer::with_lines(SCROLLED_STATE_BUFFER.to_vec());
|
||||
let mut state: AppState = serde_json::from_str(SCROLLED_STATE_REPR).unwrap();
|
||||
assert_buffer(&mut state, &expected);
|
||||
}
|
Loading…
Reference in a new issue