Implemented the final widget for confirmation prompts!

src/ui/widgets/confirmation_prompt.rs

@@ -0,0 +1,154 @@
+use crate::ui::styles::ManagarrStyle;
+use crate::ui::utils::{layout_paragraph_borderless, title_block_centered};
+use crate::ui::widgets::button::Button;
+use crate::ui::widgets::checkbox::Checkbox;
+use ratatui::buffer::Buffer;
+use ratatui::layout::{Constraint, Flex, Layout, Rect};
+use ratatui::widgets::{Paragraph, Widget};
+use std::iter;
+
+#[cfg(test)]
+#[path = "confirmation_prompt_tests.rs"]
+mod confirmation_prompt_tests;
+
+pub struct ConfirmationPrompt<'a> {
+  title: &'a str,
+  prompt: &'a str,
+  content: Option<Paragraph<'a>>,
+  checkboxes: Option<Vec<Checkbox<'a>>>,
+  yes_no_value: bool,
+  yes_no_highlighted: bool,
+}
+
+impl<'a> ConfirmationPrompt<'a> {
+  pub fn new() -> Self {
+    Self {
+      title: "",
+      prompt: "",
+      content: None,
+      checkboxes: None,
+      yes_no_value: false,
+      yes_no_highlighted: true,
+    }
+  }
+
+  pub fn title(mut self, title: &'a str) -> Self {
+    self.title = title;
+    self
+  }
+
+  pub fn prompt(mut self, prompt: &'a str) -> Self {
+    self.prompt = prompt;
+    self
+  }
+
+  pub fn content(mut self, content: Paragraph<'a>) -> Self {
+    self.content = Some(content);
+    self
+  }
+
+  pub fn checkboxes(mut self, checkboxes: Vec<Checkbox<'a>>) -> Self {
+    self.checkboxes = Some(checkboxes);
+    self
+  }
+
+  pub fn yes_no_value(mut self, yes_highlighted: bool) -> Self {
+    self.yes_no_value = yes_highlighted;
+    self
+  }
+
+  pub fn yes_no_highlighted(mut self, yes_highlighted: bool) -> Self {
+    self.yes_no_highlighted = yes_highlighted;
+    self
+  }
+
+  fn render_confirmation_prompt_with_checkboxes(self, area: Rect, buf: &mut Buffer) {
+    title_block_centered(self.title).render(area, buf);
+
+    if let Some(checkboxes) = self.checkboxes {
+      let mut constraints = vec![
+        Constraint::Length(4),
+        Constraint::Fill(0),
+        Constraint::Length(3),
+      ];
+      constraints.splice(
+        1..1,
+        iter::repeat(Constraint::Length(3)).take(checkboxes.len()),
+      );
+      let chunks = Layout::vertical(constraints).margin(1).split(area);
+      let [yes_area, no_area] =
+        Layout::horizontal([Constraint::Percentage(50), Constraint::Percentage(50)])
+          .areas(chunks[checkboxes.len() + 2]);
+
+      layout_paragraph_borderless(self.prompt).render(chunks[0], buf);
+
+      checkboxes
+        .into_iter()
+        .enumerate()
+        .for_each(|(i, checkbox)| {
+          checkbox.render(chunks[i + 1], buf);
+        });
+
+      Button::new()
+        .title("Yes")
+        .selected(self.yes_no_value && self.yes_no_highlighted)
+        .render(yes_area, buf);
+      Button::new()
+        .title("No")
+        .selected(!self.yes_no_value && self.yes_no_highlighted)
+        .render(no_area, buf);
+    }
+  }
+
+  fn render_confirmation_prompt(self, area: Rect, buf: &mut Buffer) {
+    title_block_centered(self.title).render(area, buf);
+
+    let [prompt_area, buttons_area] = if let Some(content_paragraph) = self.content {
+      let [prompt_area, content_area, _, buttons_area] = Layout::vertical([
+        Constraint::Length(4),
+        Constraint::Length(7),
+        Constraint::Fill(0),
+        Constraint::Length(3),
+      ])
+      .margin(1)
+      .areas(area);
+
+      content_paragraph.render(content_area, buf);
+
+      [prompt_area, buttons_area]
+    } else {
+      let [prompt_area, buttons_area] =
+        Layout::vertical([Constraint::Percentage(72), Constraint::Length(3)])
+          .margin(1)
+          .flex(Flex::SpaceBetween)
+          .areas(area);
+
+      [prompt_area, buttons_area]
+    };
+
+    layout_paragraph_borderless(self.prompt).render(prompt_area, buf);
+
+    let [yes_area, no_area] =
+      Layout::horizontal([Constraint::Percentage(50), Constraint::Percentage(50)])
+        .areas(buttons_area);
+
+    Button::new()
+      .title("Yes")
+      .selected(self.yes_no_value)
+      .render(yes_area, buf);
+    Button::new()
+      .title("No")
+      .selected(!self.yes_no_value)
+      .render(no_area, buf);
+  }
+}
+
+impl<'a> Widget for ConfirmationPrompt<'a> {
+  fn render(self, area: Rect, buf: &mut Buffer) {
+    if self.checkboxes.is_some() {
+      self.render_confirmation_prompt_with_checkboxes(area, buf);
+    } else {
+      self.render_confirmation_prompt(area, buf);
+    }
+  }
+}