retiretui 0.1.0

Retirement planning application for the terminal.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67

//! The theme's ground under everything: a cell a widget left at the
//! terminal's default is drawn in the theme's own foreground and ground.
//!
//! A pass over the composed frame rather than a style at every widget, so
//! that the rule holds for a widget built with no theme in reach, a stock
//! widget, and one not yet written alike. Under a theme that leaves its
//! grounds to the terminal it does nothing.

use bevy_app::App;
use bevy_ecs::prelude::{Commands, IntoScheduleConfigs, Res, ResMut, Resource};
use bevy_ecs::schedule::SystemSet;
use plurimus::core::ratatui_core::style::Color;
use plurimus::core::{
    CompositeSystems, FrameBuffer, MainWorld, TerminalRenderAppExt, TerminalRenderSystems,
};

use super::Theme;

pub fn plugin(app: &mut App) {
    app.add_extract_systems(extract);
    app.add_terminal_systems(
        TerminalRenderSystems::Composite,
        paint_ground
            .in_set(CompositeSystems::PostProcess)
            .in_set(Grounded),
    );
}

/// The ground being painted. What eases between a cell's colours runs
/// after it, so it starts from the colours the cell is drawn in.
#[derive(SystemSet, Clone, Copy, PartialEq, Eq, Hash, Debug)]
pub struct Grounded;

/// What the theme draws an unstyled cell in, as the render world holds it.
#[derive(Resource, Clone, Copy, PartialEq, Eq, Debug)]
struct Ground {
    fg: Color,
    bg: Option<Color>,
}

fn extract(main_world: Res<MainWorld>, mut commands: Commands) {
    let theme = main_world.resource::<Theme>();
    commands.insert_resource(Ground {
        fg: theme.fg,
        bg: theme.bg,
    });
}

fn paint_ground(ground: Option<Res<Ground>>, mut frame: ResMut<FrameBuffer>) {
    let Some(ground) = ground else {
        return;
    };
    let paints_fg = ground.fg != Color::Reset;
    if !paints_fg && ground.bg.is_none() {
        return;
    }
    for cell in &mut frame.0.content {
        if paints_fg && cell.fg == Color::Reset {
            cell.fg = ground.fg;
        }
        if let Some(bg) = ground.bg
            && cell.bg == Color::Reset
        {
            cell.bg = bg;
        }
    }
}