jjj 0.2.0

A modal interface for Jujutsu.
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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93

//! Rendering logic for the general interface.

use anyhow::Result;
use bevy::prelude::*;
use bevy_ratatui::terminal::RatatuiContext;
use ratatui::prelude::*;
use ratatui::widgets::Clear;

use crate::app::AppSet;
use crate::events::prelude::*;
use crate::frontend::prelude::*;

use super::Screen;

#[tracing::instrument(skip_all)]
pub fn plugin(app: &mut App) {
    trace!("Initializing plugin...");

    app.add_systems(OnEnter(Screen::Interface), init);

    app.add_systems(
        Update,
        draw.map(bevy::utils::error)
            .in_set(AppSet::Render)
            .run_if(in_state(Screen::Interface)),
    );

    trace!("Plugin initialized.");
}

fn init(
    mut ev_read_log: EventWriter<LogRevsetEvent>,
    mut navigation: Navigation,
    mut commands: Commands,
) {
    let root = commands.spawn(ChangeBuffer::default()).id();
    commands.spawn(CommandLine::default());
    commands.spawn(StatusLine::default());

    navigation.focus_as_root(root);

    ev_read_log.send(LogRevsetEvent(
        "present(@) | ancestors(immutable_heads().., 2) | present(trunk())".to_string(),
    ));
}

fn draw(
    mut change_buffer: Query<&mut ChangeBuffer>,
    mut context: ResMut<RatatuiContext>,
    revset_prompt: Query<&RevsetPrompt>,
    command_line: Query<&CommandLine>,
    error_popup: Query<&ErrorPopup>,
    status_line: Query<&StatusLine>,
    space_menu: Query<&SpaceMenu>,
) -> Result<()> {
    let mut change_buffer = change_buffer.get_single_mut()?;
    let command_line = command_line.get_single()?;
    let status_line = status_line.get_single()?;

    context.draw(|frame| {
        frame.render_widget(Clear, frame.area());

        let [buffer_area, status_line_area, command_line_area] = Layout::vertical([
            Constraint::Fill(1),
            Constraint::Length(1),
            Constraint::Length(1),
        ])
        .areas(frame.area());

        frame.render_stateful_widget(
            change_buffer.clone(),
            buffer_area,
            &mut change_buffer.viewport_y,
        );

        frame.render_widget(status_line, status_line_area);
        frame.render_widget(command_line, command_line_area);

        if let Ok(revset_prompt) = revset_prompt.get_single() {
            frame.render_widget(revset_prompt, buffer_area);
        }

        if let Ok(space_menu) = space_menu.get_single() {
            frame.render_widget(space_menu, buffer_area);
        }

        if let Ok(error_popup) = error_popup.get_single() {
            frame.render_widget(error_popup, buffer_area);
        }
    })?;

    Ok(())
}