use ratatui::layout::{Constraint, Direction, Layout, Rect};
use ratatui::style::Style;
use ratatui::text::{Line, Span};
use ratatui::widgets::{Paragraph, Row, Table};
use ratatui::Frame;
use crate::tui::action::LensData;
use crate::tui::lenses::{compare, dnssec};
use crate::tui::panes::Panes;
use crate::tui::theme::Theme;
use crate::tui::widgets::{panel, row_style, scroll_to};
const NS_LABELS: &[&str] = &["system", "8.8.8.8", "1.1.1.1"];
#[allow(clippy::too_many_arguments)]
pub fn render(
f: &mut Frame,
area: Rect,
theme: &Theme,
tab: usize,
data: &LensData,
focused: bool,
sel: usize,
panes: &Panes,
) {
if tab == 1 {
dnssec::render(f, area, theme, data);
return;
}
if tab == 2 {
compare::render(f, area, theme, data);
return;
}
let LensData::Dns(records) = data else { return };
let title = format!("dig · {} records", panes.dns.record_type);
let inner = panel::render(f, area, theme, &title, theme.sky, focused);
let chunks = Layout::default()
.direction(Direction::Vertical)
.constraints([
Constraint::Length(1),
Constraint::Length(1),
Constraint::Min(0),
])
.split(inner);
let selected_ns = panes.dns.ns_idx;
let mut ns_spans: Vec<Span> = Vec::new();
for (i, label) in NS_LABELS.iter().enumerate() {
if i > 0 {
ns_spans.push(Span::raw(" "));
}
let style = if i == selected_ns {
Style::default().fg(theme.base).bg(theme.sky)
} else {
Style::default().fg(theme.subtext).bg(theme.surface0)
};
ns_spans.push(Span::styled(format!(" {label} "), style));
}
f.render_widget(Paragraph::new(Line::from(ns_spans)), chunks[0]);
f.render_widget(
Paragraph::new(Line::from(Span::styled(
"s nameserver",
Style::default().fg(theme.overlay0),
))),
chunks[1],
);
let header =
Row::new(["TYPE", "NAME", "DATA", "TTL"]).style(Style::default().fg(theme.overlay0));
let rows = records.iter().enumerate().map(|(i, r)| {
Row::new(vec![
r.record_type.to_string(),
r.name.clone(),
r.format_short(),
r.ttl.to_string(),
])
.style(row_style(theme, focused && i == sel))
});
let table = Table::new(
rows,
[
Constraint::Length(7),
Constraint::Percentage(35),
Constraint::Percentage(45),
Constraint::Length(8),
],
)
.header(header)
.column_spacing(1);
let mut state = scroll_to(focused.then_some(sel));
f.render_stateful_widget(table, chunks[2], &mut state);
}
#[cfg(test)]
mod tests {
use super::*;
use crate::tui::test_util::render_text;
use seer_core::dns::{RecordData, RecordType};
use seer_core::DnsRecord;
fn a_record() -> DnsRecord {
DnsRecord {
name: "example.com".into(),
record_type: RecordType::A,
ttl: 300,
data: RecordData::A {
address: "93.184.215.14".into(),
},
}
}
#[test]
fn renders_an_a_record() {
let theme = Theme::frappe();
let data = LensData::Dns(vec![a_record()]);
let panes = Panes::default();
let text = render_text(70, 10, |f| {
render(f, f.area(), &theme, 0, &data, false, 0, &panes);
});
assert!(text.contains("93.184.215.14"));
assert!(text.contains("example.com"));
assert!(text.contains("dig · A records"), "title names the type");
}
#[test]
fn title_names_the_selected_record_type() {
let theme = Theme::frappe();
let data = LensData::Dns(vec![]);
let mut panes = Panes::default();
panes.dns.record_type = RecordType::MX;
let text = render_text(70, 10, |f| {
render(f, f.area(), &theme, 0, &data, false, 0, &panes);
});
assert!(text.contains("dig · MX records"));
}
#[test]
fn renders_nameserver_chip_row() {
let theme = Theme::frappe();
let data = LensData::Dns(vec![a_record()]);
let panes = Panes::default();
let text = render_text(70, 10, |f| {
render(f, f.area(), &theme, 0, &data, false, 0, &panes);
});
assert!(text.contains("system"), "chip row should show 'system'");
assert!(text.contains("8.8.8.8"), "chip row should show '8.8.8.8'");
assert!(text.contains("1.1.1.1"), "chip row should show '1.1.1.1'");
}
}