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
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
//! The **which-key overlay** (spec §8b) — the popup docked above the status
//! bar that documents the pending leader sequence. It renders the node the
//! `LeaderEngine` currently sits on, so it can never drift from the tree:
//! same data, two surfaces.
//!
//! Reveal policy: hidden while a sequence is typed fluently; shown once the
//! user hesitates past `leader_timeout_ms`. Hidden the instant the sequence
//! fires or cancels (the engine simply stops being pending).
use ratatui::Frame;
use ratatui::layout::Rect;
use ratatui::style::{Modifier, Style};
use ratatui::text::{Line, Span};
use ratatui::widgets::Paragraph;
use crate::components::panel::{ModalBg, ModalSpec, modal_chrome};
use crate::keys::leader::LeaderEngine;
use crate::settings::themes::Theme;
/// Minimum column width for a `key → target` cell.
const CELL_WIDTH: u16 = 24;
/// Rows the overlay needs for the current node (header + grid + borders).
/// The caller carves this out of the area directly above the status bar.
/// Counts `display_children()`, the same collapsed rows `render` lays out —
/// counting raw `children()` here would reserve extra blank rows wherever a
/// digit run (or any future collapse) shrinks the grid.
pub fn desired_height(engine: &LeaderEngine, width: u16) -> u16 {
let n = engine.current_node().display_children().len() as u16;
let cols = (width.saturating_sub(2) / CELL_WIDTH).max(1);
let grid_rows = n.div_ceil(cols);
grid_rows + 3 // top border + header + grid + bottom border
}
/// Render the overlay into `rect` (the caller positions it docked above the
/// status bar, full width).
pub fn render(
f: &mut Frame,
rect: Rect,
theme: &Theme,
engine: &LeaderEngine,
gateway_label: &str,
) {
let inner = modal_chrome(
f,
rect,
theme,
ModalSpec {
border: Some(Style::default().fg(theme.focus_border.to_ratatui())),
bg: ModalBg::Hard,
..Default::default()
},
);
if inner.height == 0 {
return;
}
let node = engine.current_node();
let keycap = Style::default()
.fg(theme.yellow.to_ratatui())
.add_modifier(Modifier::BOLD);
let muted = Style::default().fg(theme.gray.to_ratatui());
let caption_style = Style::default().fg(theme.fg_secondary.to_ratatui());
// ── Header: pressed keycaps · caption · right-aligned controls ─────────
let mut pressed = format!(" {gateway_label}");
for c in engine.path() {
pressed.push(' ');
pressed.push(*c);
}
let controls = "Esc cancel · BkSp up ";
let controls_w = controls.len() as u16;
let header_cols = ratatui::layout::Layout::default()
.direction(ratatui::layout::Direction::Horizontal)
.constraints([
ratatui::layout::Constraint::Min(0),
ratatui::layout::Constraint::Length(controls_w),
])
.split(Rect::new(inner.x, inner.y, inner.width, 1));
f.render_widget(
Paragraph::new(Line::from(vec![
Span::styled(pressed, keycap),
Span::styled(format!(" {}", node.label()), caption_style),
])),
header_cols[0],
);
f.render_widget(
Paragraph::new(Line::from(Span::styled(controls, muted)))
.alignment(ratatui::layout::Alignment::Right),
header_cols[1],
);
// ── Body: multi-column key → target grid ───────────────────────────────
let children = node.display_children();
if children.is_empty() {
return;
}
let cols = (inner.width / CELL_WIDTH).max(1) as usize;
let rows = children.len().div_ceil(cols);
let arrow = Span::styled(" → ", muted);
for (i, row) in children.iter().enumerate() {
// Column-major fill: read top-to-bottom within a column, like the
// spec mockup.
let col = i / rows;
let grid_row = i % rows;
let y = inner.y + 1 + grid_row as u16;
if y >= inner.bottom() {
continue;
}
let x = inner.x + (col as u16) * CELL_WIDTH;
if x >= inner.right() {
continue;
}
let target_style = if row.is_group {
Style::default().fg(theme.aqua.to_ratatui())
} else {
Style::default().fg(theme.fg.to_ratatui())
};
let cell = Rect::new(x, y, CELL_WIDTH.min(inner.right() - x), 1);
f.render_widget(
Paragraph::new(Line::from(vec![
Span::styled(format!(" {}", row.keys), keycap),
arrow.clone(),
Span::styled(row.label.clone(), target_style),
])),
cell,
);
}
}
#[cfg(test)]
mod tests {
use super::*;
/// `desired_height` must size the grid off the same rows `render` draws
/// — `display_children()` — not the raw `children()` count. At the root
/// the nine pinned-note digit leaves collapse to one row, so sizing off
/// the raw count would reserve several rows nothing draws into.
///
/// The `17` is a literal, hand-derived pin (not re-derived from
/// `display_children()`, or the assertion could pass even if the
/// collapse silently regressed): the root has 13 non-digit children
/// (`f n l o g v w m t a p q ?`) plus the nine `1`-`9` pinned-note
/// leaves, which collapse to one row — 14 display rows. `CELL_WIDTH` is
/// narrower than `2 × CELL_WIDTH`, so `desired_height` computes a single
/// column and `grid_rows` equals the row count directly: 14 rows + 3
/// (top border + header + bottom border) = 17.
#[test]
fn desired_height_counts_collapsed_rows_not_raw_children() {
let engine = LeaderEngine::new();
let root = engine.current_node();
let raw = root.children().len() as u16;
let width = CELL_WIDTH;
assert_eq!(desired_height(&engine, width), 17); // 14 collapsed root rows + 3 chrome
// Sizing off the raw, uncollapsed count would have asked for more.
assert_ne!(desired_height(&engine, width), raw + 3);
}
}