use ratatui::layout::{Constraint, Layout, Rect};
pub struct LayoutResult {
pub tool_history: Option<Rect>,
pub messages: Rect,
pub thinking: Option<Rect>,
pub delegate: Option<Rect>,
pub summary: Option<Rect>,
pub input: Rect,
pub footer: Rect,
}
pub struct MainLayout;
impl MainLayout {
pub const TOOL_HISTORY_WIDTH: u16 = 44;
pub const MIN_MAIN_WIDTH: u16 = 60;
pub fn split(
area: Rect,
input_lines: usize,
thinking_lines: usize,
delegate_lines: usize,
has_summary: bool,
) -> LayoutResult {
let (tool_history, main) = if area.width > Self::TOOL_HISTORY_WIDTH + Self::MIN_MAIN_WIDTH {
let chunks = Layout::horizontal([
Constraint::Length(Self::TOOL_HISTORY_WIDTH),
Constraint::Min(1),
])
.split(area);
(Some(chunks[0]), chunks[1])
} else {
(None, area)
};
let content = input_lines.clamp(1, 8) as u16;
let input_h = (content + 2).min(main.height.saturating_sub(4));
let thinking_h: u16 = if thinking_lines == 0 {
0
} else {
(1 + thinking_lines.min(4)) as u16
};
let delegate_h: u16 = delegate_lines.min(4) as u16;
let mut constraints: Vec<Constraint> = vec![Constraint::Min(1)];
if thinking_h > 0 { constraints.push(Constraint::Length(thinking_h)); }
if delegate_h > 0 { constraints.push(Constraint::Length(delegate_h)); }
if has_summary { constraints.push(Constraint::Length(1)); }
constraints.push(Constraint::Length(input_h));
constraints.push(Constraint::Length(1));
let rows = Layout::vertical(constraints).split(main);
let mut idx = 0;
let messages = rows[idx]; idx += 1;
let thinking = if thinking_h > 0 { let r = Some(rows[idx]); idx += 1; r } else { None };
let delegate = if delegate_h > 0 { let r = Some(rows[idx]); idx += 1; r } else { None };
let summary = if has_summary { let r = Some(rows[idx]); idx += 1; r } else { None };
let input = rows[idx]; idx += 1;
let footer = rows[idx];
LayoutResult { tool_history, messages, thinking, delegate, summary, input, footer }
}
}