use tuirealm::command::{Cmd, CmdResult};
use tuirealm::component::Component;
use tuirealm::props::{
AttrValue, Attribute, Color, HorizontalAlignment, PropPayload, PropValue, Props, QueryResult,
SpanStatic, Style, TextModifiers,
};
use tuirealm::ratatui::Frame;
use tuirealm::ratatui::layout::Rect;
use tuirealm::ratatui::text::{Line, Span as RSpan, Text};
use tuirealm::ratatui::widgets::Paragraph;
use tuirealm::state::State;
use crate::prop_ext::CommonProps;
use crate::utils;
#[derive(Default)]
#[must_use]
pub struct Span {
common: CommonProps,
props: Props,
}
impl Span {
pub fn foreground(mut self, fg: Color) -> Self {
self.attr(Attribute::Foreground, AttrValue::Color(fg));
self
}
pub fn background(mut self, bg: Color) -> Self {
self.attr(Attribute::Background, AttrValue::Color(bg));
self
}
pub fn modifiers(mut self, m: TextModifiers) -> Self {
self.attr(Attribute::TextProps, AttrValue::TextModifiers(m));
self
}
pub fn style(mut self, style: Style) -> Self {
self.attr(Attribute::Style, AttrValue::Style(style));
self
}
pub fn alignment_horizontal(mut self, a: HorizontalAlignment) -> Self {
self.attr(
Attribute::AlignmentHorizontal,
AttrValue::AlignmentHorizontal(a),
);
self
}
pub fn spans<T>(mut self, s: impl IntoIterator<Item = T>) -> Self
where
T: Into<SpanStatic>,
{
self.attr(
Attribute::Text,
AttrValue::Payload(PropPayload::Vec(
s.into_iter()
.map(Into::into)
.map(PropValue::TextSpan)
.collect(),
)),
);
self
}
}
impl Component for Span {
fn view(&mut self, render: &mut Frame, area: Rect) {
if !self.common.display {
return;
}
let payload = self.props.get(Attribute::Text).and_then(|x| x.as_payload());
let text = match payload {
Some(PropPayload::Vec(lines)) => {
let lines: Vec<RSpan> = lines
.iter()
.filter_map(|x| x.as_textspan())
.map(utils::borrow_clone_span)
.collect();
Text::from(Line::from(lines))
}
_ => Text::default(),
};
let alignment: HorizontalAlignment = self
.props
.get(Attribute::AlignmentHorizontal)
.and_then(AttrValue::as_alignment_horizontal)
.unwrap_or(HorizontalAlignment::Left);
render.render_widget(
Paragraph::new(text)
.alignment(alignment)
.style(self.common.style),
area,
);
}
fn query<'a>(&'a self, attr: Attribute) -> Option<QueryResult<'a>> {
if let Some(value) = self.common.get_for_query(attr) {
return Some(value);
}
self.props.get_for_query(attr)
}
fn attr(&mut self, attr: Attribute, value: AttrValue) {
if let Some(value) = self.common.set(attr, value) {
self.props.set(attr, value);
}
}
fn state(&self) -> State {
State::None
}
fn perform(&mut self, cmd: Cmd) -> CmdResult {
CmdResult::Invalid(cmd)
}
}
#[cfg(test)]
mod tests {
use pretty_assertions::assert_eq;
use tuirealm::props::SpanStatic;
use tuirealm::ratatui::style::Stylize;
use super::*;
#[test]
fn test_components_span() {
let component = Span::default()
.background(Color::Blue)
.foreground(Color::Red)
.modifiers(TextModifiers::BOLD)
.alignment_horizontal(HorizontalAlignment::Center)
.spans([
SpanStatic::from("Press "),
SpanStatic::from("<ESC>").fg(Color::Cyan).bold(),
SpanStatic::from(" to quit"),
]);
assert_eq!(component.state(), State::None);
}
#[test]
fn various_spans_types() {
let _ = Span::default().spans(vec![SpanStatic::raw("hello")]);
let _ = Span::default().spans([SpanStatic::raw("hello")]);
let _ = Span::default().spans(vec![SpanStatic::raw("hello")].into_boxed_slice());
let _ = Span::default().spans(["Hello"].map(SpanStatic::raw));
}
}