977 lines
29 KiB
Rust
977 lines
29 KiB
Rust
|
//! Customizes the rendering of the elements.
|
|||
|
use std::{fmt, io};
|
|||
|
|
|||
|
use console::{measure_text_width, style, Style, StyledObject, Term};
|
|||
|
#[cfg(feature = "fuzzy-select")]
|
|||
|
use fuzzy_matcher::{skim::SkimMatcherV2, FuzzyMatcher};
|
|||
|
|
|||
|
/// Implements a theme for dialoguer.
|
|||
|
pub trait Theme {
|
|||
|
/// Formats a prompt.
|
|||
|
#[inline]
|
|||
|
fn format_prompt(&self, f: &mut dyn fmt::Write, prompt: &str) -> fmt::Result {
|
|||
|
write!(f, "{}:", prompt)
|
|||
|
}
|
|||
|
|
|||
|
/// Formats out an error.
|
|||
|
#[inline]
|
|||
|
fn format_error(&self, f: &mut dyn fmt::Write, err: &str) -> fmt::Result {
|
|||
|
write!(f, "error: {}", err)
|
|||
|
}
|
|||
|
|
|||
|
/// Formats a confirm prompt.
|
|||
|
fn format_confirm_prompt(
|
|||
|
&self,
|
|||
|
f: &mut dyn fmt::Write,
|
|||
|
prompt: &str,
|
|||
|
default: Option<bool>,
|
|||
|
) -> fmt::Result {
|
|||
|
if !prompt.is_empty() {
|
|||
|
write!(f, "{} ", &prompt)?;
|
|||
|
}
|
|||
|
match default {
|
|||
|
None => write!(f, "[y/n] ")?,
|
|||
|
Some(true) => write!(f, "[Y/n] ")?,
|
|||
|
Some(false) => write!(f, "[y/N] ")?,
|
|||
|
}
|
|||
|
Ok(())
|
|||
|
}
|
|||
|
|
|||
|
/// Formats a confirm prompt after selection.
|
|||
|
fn format_confirm_prompt_selection(
|
|||
|
&self,
|
|||
|
f: &mut dyn fmt::Write,
|
|||
|
prompt: &str,
|
|||
|
selection: Option<bool>,
|
|||
|
) -> fmt::Result {
|
|||
|
let selection = selection.map(|b| if b { "yes" } else { "no" });
|
|||
|
|
|||
|
match selection {
|
|||
|
Some(selection) if prompt.is_empty() => {
|
|||
|
write!(f, "{}", selection)
|
|||
|
}
|
|||
|
Some(selection) => {
|
|||
|
write!(f, "{} {}", &prompt, selection)
|
|||
|
}
|
|||
|
None if prompt.is_empty() => Ok(()),
|
|||
|
None => {
|
|||
|
write!(f, "{}", &prompt)
|
|||
|
}
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
/// Formats an input prompt.
|
|||
|
fn format_input_prompt(
|
|||
|
&self,
|
|||
|
f: &mut dyn fmt::Write,
|
|||
|
prompt: &str,
|
|||
|
default: Option<&str>,
|
|||
|
) -> fmt::Result {
|
|||
|
match default {
|
|||
|
Some(default) if prompt.is_empty() => write!(f, "[{}]: ", default),
|
|||
|
Some(default) => write!(f, "{} [{}]: ", prompt, default),
|
|||
|
None => write!(f, "{}: ", prompt),
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
/// Formats an input prompt after selection.
|
|||
|
#[inline]
|
|||
|
fn format_input_prompt_selection(
|
|||
|
&self,
|
|||
|
f: &mut dyn fmt::Write,
|
|||
|
prompt: &str,
|
|||
|
sel: &str,
|
|||
|
) -> fmt::Result {
|
|||
|
write!(f, "{}: {}", prompt, sel)
|
|||
|
}
|
|||
|
|
|||
|
/// Formats a password prompt.
|
|||
|
#[inline]
|
|||
|
#[cfg(feature = "password")]
|
|||
|
fn format_password_prompt(&self, f: &mut dyn fmt::Write, prompt: &str) -> fmt::Result {
|
|||
|
self.format_input_prompt(f, prompt, None)
|
|||
|
}
|
|||
|
|
|||
|
/// Formats a password prompt after selection.
|
|||
|
#[inline]
|
|||
|
#[cfg(feature = "password")]
|
|||
|
fn format_password_prompt_selection(
|
|||
|
&self,
|
|||
|
f: &mut dyn fmt::Write,
|
|||
|
prompt: &str,
|
|||
|
) -> fmt::Result {
|
|||
|
self.format_input_prompt_selection(f, prompt, "[hidden]")
|
|||
|
}
|
|||
|
|
|||
|
/// Formats a select prompt.
|
|||
|
#[inline]
|
|||
|
fn format_select_prompt(&self, f: &mut dyn fmt::Write, prompt: &str) -> fmt::Result {
|
|||
|
self.format_prompt(f, prompt)
|
|||
|
}
|
|||
|
|
|||
|
/// Formats a select prompt after selection.
|
|||
|
#[inline]
|
|||
|
fn format_select_prompt_selection(
|
|||
|
&self,
|
|||
|
f: &mut dyn fmt::Write,
|
|||
|
prompt: &str,
|
|||
|
sel: &str,
|
|||
|
) -> fmt::Result {
|
|||
|
self.format_input_prompt_selection(f, prompt, sel)
|
|||
|
}
|
|||
|
|
|||
|
/// Formats a multi select prompt.
|
|||
|
#[inline]
|
|||
|
fn format_multi_select_prompt(&self, f: &mut dyn fmt::Write, prompt: &str) -> fmt::Result {
|
|||
|
self.format_prompt(f, prompt)
|
|||
|
}
|
|||
|
|
|||
|
/// Formats a sort prompt.
|
|||
|
#[inline]
|
|||
|
fn format_sort_prompt(&self, f: &mut dyn fmt::Write, prompt: &str) -> fmt::Result {
|
|||
|
self.format_prompt(f, prompt)
|
|||
|
}
|
|||
|
|
|||
|
/// Formats a multi_select prompt after selection.
|
|||
|
fn format_multi_select_prompt_selection(
|
|||
|
&self,
|
|||
|
f: &mut dyn fmt::Write,
|
|||
|
prompt: &str,
|
|||
|
selections: &[&str],
|
|||
|
) -> fmt::Result {
|
|||
|
write!(f, "{}: ", prompt)?;
|
|||
|
for (idx, sel) in selections.iter().enumerate() {
|
|||
|
write!(f, "{}{}", if idx == 0 { "" } else { ", " }, sel)?;
|
|||
|
}
|
|||
|
Ok(())
|
|||
|
}
|
|||
|
|
|||
|
/// Formats a sort prompt after selection.
|
|||
|
#[inline]
|
|||
|
fn format_sort_prompt_selection(
|
|||
|
&self,
|
|||
|
f: &mut dyn fmt::Write,
|
|||
|
prompt: &str,
|
|||
|
selections: &[&str],
|
|||
|
) -> fmt::Result {
|
|||
|
self.format_multi_select_prompt_selection(f, prompt, selections)
|
|||
|
}
|
|||
|
|
|||
|
/// Formats a select prompt item.
|
|||
|
fn format_select_prompt_item(
|
|||
|
&self,
|
|||
|
f: &mut dyn fmt::Write,
|
|||
|
text: &str,
|
|||
|
active: bool,
|
|||
|
) -> fmt::Result {
|
|||
|
write!(f, "{} {}", if active { ">" } else { " " }, text)
|
|||
|
}
|
|||
|
|
|||
|
/// Formats a multi select prompt item.
|
|||
|
fn format_multi_select_prompt_item(
|
|||
|
&self,
|
|||
|
f: &mut dyn fmt::Write,
|
|||
|
text: &str,
|
|||
|
checked: bool,
|
|||
|
active: bool,
|
|||
|
) -> fmt::Result {
|
|||
|
write!(
|
|||
|
f,
|
|||
|
"{} {}",
|
|||
|
match (checked, active) {
|
|||
|
(true, true) => "> [x]",
|
|||
|
(true, false) => " [x]",
|
|||
|
(false, true) => "> [ ]",
|
|||
|
(false, false) => " [ ]",
|
|||
|
},
|
|||
|
text
|
|||
|
)
|
|||
|
}
|
|||
|
|
|||
|
/// Formats a sort prompt item.
|
|||
|
fn format_sort_prompt_item(
|
|||
|
&self,
|
|||
|
f: &mut dyn fmt::Write,
|
|||
|
text: &str,
|
|||
|
picked: bool,
|
|||
|
active: bool,
|
|||
|
) -> fmt::Result {
|
|||
|
write!(
|
|||
|
f,
|
|||
|
"{} {}",
|
|||
|
match (picked, active) {
|
|||
|
(true, true) => "> [x]",
|
|||
|
(false, true) => "> [ ]",
|
|||
|
(_, false) => " [ ]",
|
|||
|
},
|
|||
|
text
|
|||
|
)
|
|||
|
}
|
|||
|
|
|||
|
/// Formats a fuzzy select prompt item.
|
|||
|
#[cfg(feature = "fuzzy-select")]
|
|||
|
fn format_fuzzy_select_prompt_item(
|
|||
|
&self,
|
|||
|
f: &mut dyn fmt::Write,
|
|||
|
text: &str,
|
|||
|
active: bool,
|
|||
|
highlight_matches: bool,
|
|||
|
matcher: &SkimMatcherV2,
|
|||
|
search_term: &str,
|
|||
|
) -> fmt::Result {
|
|||
|
write!(f, "{} ", if active { ">" } else { " " })?;
|
|||
|
|
|||
|
if highlight_matches {
|
|||
|
if let Some((_score, indices)) = matcher.fuzzy_indices(text, &search_term) {
|
|||
|
for (idx, c) in text.chars().into_iter().enumerate() {
|
|||
|
if indices.contains(&idx) {
|
|||
|
write!(f, "{}", style(c).for_stderr().bold())?;
|
|||
|
} else {
|
|||
|
write!(f, "{}", c)?;
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
return Ok(());
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
write!(f, "{}", text)
|
|||
|
}
|
|||
|
|
|||
|
/// Formats a fuzzy select prompt.
|
|||
|
#[cfg(feature = "fuzzy-select")]
|
|||
|
fn format_fuzzy_select_prompt(
|
|||
|
&self,
|
|||
|
f: &mut dyn fmt::Write,
|
|||
|
prompt: &str,
|
|||
|
search_term: &str,
|
|||
|
cursor_pos: usize,
|
|||
|
) -> fmt::Result {
|
|||
|
if !prompt.is_empty() {
|
|||
|
write!(f, "{} ", prompt,)?;
|
|||
|
}
|
|||
|
|
|||
|
if cursor_pos < search_term.len() {
|
|||
|
let st_head = search_term[0..cursor_pos].to_string();
|
|||
|
let st_tail = search_term[cursor_pos..search_term.len()].to_string();
|
|||
|
let st_cursor = "|".to_string();
|
|||
|
write!(f, "{}{}{}", st_head, st_cursor, st_tail)
|
|||
|
} else {
|
|||
|
let cursor = "|".to_string();
|
|||
|
write!(f, "{}{}", search_term.to_string(), cursor)
|
|||
|
}
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
/// The default theme.
|
|||
|
pub struct SimpleTheme;
|
|||
|
|
|||
|
impl Theme for SimpleTheme {}
|
|||
|
|
|||
|
/// A colorful theme
|
|||
|
pub struct ColorfulTheme {
|
|||
|
/// The style for default values
|
|||
|
pub defaults_style: Style,
|
|||
|
/// The style for prompt
|
|||
|
pub prompt_style: Style,
|
|||
|
/// Prompt prefix value and style
|
|||
|
pub prompt_prefix: StyledObject<String>,
|
|||
|
/// Prompt suffix value and style
|
|||
|
pub prompt_suffix: StyledObject<String>,
|
|||
|
/// Prompt on success prefix value and style
|
|||
|
pub success_prefix: StyledObject<String>,
|
|||
|
/// Prompt on success suffix value and style
|
|||
|
pub success_suffix: StyledObject<String>,
|
|||
|
/// Error prefix value and style
|
|||
|
pub error_prefix: StyledObject<String>,
|
|||
|
/// The style for error message
|
|||
|
pub error_style: Style,
|
|||
|
/// The style for hints
|
|||
|
pub hint_style: Style,
|
|||
|
/// The style for values on prompt success
|
|||
|
pub values_style: Style,
|
|||
|
/// The style for active items
|
|||
|
pub active_item_style: Style,
|
|||
|
/// The style for inactive items
|
|||
|
pub inactive_item_style: Style,
|
|||
|
/// Active item in select prefix value and style
|
|||
|
pub active_item_prefix: StyledObject<String>,
|
|||
|
/// Inctive item in select prefix value and style
|
|||
|
pub inactive_item_prefix: StyledObject<String>,
|
|||
|
/// Checked item in multi select prefix value and style
|
|||
|
pub checked_item_prefix: StyledObject<String>,
|
|||
|
/// Unchecked item in multi select prefix value and style
|
|||
|
pub unchecked_item_prefix: StyledObject<String>,
|
|||
|
/// Picked item in sort prefix value and style
|
|||
|
pub picked_item_prefix: StyledObject<String>,
|
|||
|
/// Unpicked item in sort prefix value and style
|
|||
|
pub unpicked_item_prefix: StyledObject<String>,
|
|||
|
/// Formats the cursor for a fuzzy select prompt
|
|||
|
#[cfg(feature = "fuzzy-select")]
|
|||
|
pub fuzzy_cursor_style: Style,
|
|||
|
// Formats the highlighting if matched characters
|
|||
|
#[cfg(feature = "fuzzy-select")]
|
|||
|
pub fuzzy_match_highlight_style: Style,
|
|||
|
/// Show the selections from certain prompts inline
|
|||
|
pub inline_selections: bool,
|
|||
|
}
|
|||
|
|
|||
|
impl Default for ColorfulTheme {
|
|||
|
fn default() -> ColorfulTheme {
|
|||
|
ColorfulTheme {
|
|||
|
defaults_style: Style::new().for_stderr().cyan(),
|
|||
|
prompt_style: Style::new().for_stderr().bold(),
|
|||
|
prompt_prefix: style("?".to_string()).for_stderr().yellow(),
|
|||
|
prompt_suffix: style("›".to_string()).for_stderr().black().bright(),
|
|||
|
success_prefix: style("✔".to_string()).for_stderr().green(),
|
|||
|
success_suffix: style("·".to_string()).for_stderr().black().bright(),
|
|||
|
error_prefix: style("✘".to_string()).for_stderr().red(),
|
|||
|
error_style: Style::new().for_stderr().red(),
|
|||
|
hint_style: Style::new().for_stderr().black().bright(),
|
|||
|
values_style: Style::new().for_stderr().green(),
|
|||
|
active_item_style: Style::new().for_stderr().cyan(),
|
|||
|
inactive_item_style: Style::new().for_stderr(),
|
|||
|
active_item_prefix: style("❯".to_string()).for_stderr().green(),
|
|||
|
inactive_item_prefix: style(" ".to_string()).for_stderr(),
|
|||
|
checked_item_prefix: style("✔".to_string()).for_stderr().green(),
|
|||
|
unchecked_item_prefix: style("✔".to_string()).for_stderr().black(),
|
|||
|
picked_item_prefix: style("❯".to_string()).for_stderr().green(),
|
|||
|
unpicked_item_prefix: style(" ".to_string()).for_stderr(),
|
|||
|
#[cfg(feature = "fuzzy-select")]
|
|||
|
fuzzy_cursor_style: Style::new().for_stderr().black().on_white(),
|
|||
|
#[cfg(feature = "fuzzy-select")]
|
|||
|
fuzzy_match_highlight_style: Style::new().for_stderr().bold(),
|
|||
|
inline_selections: true,
|
|||
|
}
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
impl Theme for ColorfulTheme {
|
|||
|
/// Formats a prompt.
|
|||
|
fn format_prompt(&self, f: &mut dyn fmt::Write, prompt: &str) -> fmt::Result {
|
|||
|
if !prompt.is_empty() {
|
|||
|
write!(
|
|||
|
f,
|
|||
|
"{} {} ",
|
|||
|
&self.prompt_prefix,
|
|||
|
self.prompt_style.apply_to(prompt)
|
|||
|
)?;
|
|||
|
}
|
|||
|
|
|||
|
write!(f, "{}", &self.prompt_suffix)
|
|||
|
}
|
|||
|
|
|||
|
/// Formats an error
|
|||
|
fn format_error(&self, f: &mut dyn fmt::Write, err: &str) -> fmt::Result {
|
|||
|
write!(
|
|||
|
f,
|
|||
|
"{} {}",
|
|||
|
&self.error_prefix,
|
|||
|
self.error_style.apply_to(err)
|
|||
|
)
|
|||
|
}
|
|||
|
|
|||
|
/// Formats an input prompt.
|
|||
|
fn format_input_prompt(
|
|||
|
&self,
|
|||
|
f: &mut dyn fmt::Write,
|
|||
|
prompt: &str,
|
|||
|
default: Option<&str>,
|
|||
|
) -> fmt::Result {
|
|||
|
if !prompt.is_empty() {
|
|||
|
write!(
|
|||
|
f,
|
|||
|
"{} {} ",
|
|||
|
&self.prompt_prefix,
|
|||
|
self.prompt_style.apply_to(prompt)
|
|||
|
)?;
|
|||
|
}
|
|||
|
|
|||
|
match default {
|
|||
|
Some(default) => write!(
|
|||
|
f,
|
|||
|
"{} {} ",
|
|||
|
self.hint_style.apply_to(&format!("({})", default)),
|
|||
|
&self.prompt_suffix
|
|||
|
),
|
|||
|
None => write!(f, "{} ", &self.prompt_suffix),
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
/// Formats a confirm prompt.
|
|||
|
fn format_confirm_prompt(
|
|||
|
&self,
|
|||
|
f: &mut dyn fmt::Write,
|
|||
|
prompt: &str,
|
|||
|
default: Option<bool>,
|
|||
|
) -> fmt::Result {
|
|||
|
if !prompt.is_empty() {
|
|||
|
write!(
|
|||
|
f,
|
|||
|
"{} {} ",
|
|||
|
&self.prompt_prefix,
|
|||
|
self.prompt_style.apply_to(prompt)
|
|||
|
)?;
|
|||
|
}
|
|||
|
|
|||
|
match default {
|
|||
|
None => write!(
|
|||
|
f,
|
|||
|
"{} {}",
|
|||
|
self.hint_style.apply_to("(y/n)"),
|
|||
|
&self.prompt_suffix
|
|||
|
),
|
|||
|
Some(true) => write!(
|
|||
|
f,
|
|||
|
"{} {} {}",
|
|||
|
self.hint_style.apply_to("(y/n)"),
|
|||
|
&self.prompt_suffix,
|
|||
|
self.defaults_style.apply_to("yes")
|
|||
|
),
|
|||
|
Some(false) => write!(
|
|||
|
f,
|
|||
|
"{} {} {}",
|
|||
|
self.hint_style.apply_to("(y/n)"),
|
|||
|
&self.prompt_suffix,
|
|||
|
self.defaults_style.apply_to("no")
|
|||
|
),
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
/// Formats a confirm prompt after selection.
|
|||
|
fn format_confirm_prompt_selection(
|
|||
|
&self,
|
|||
|
f: &mut dyn fmt::Write,
|
|||
|
prompt: &str,
|
|||
|
selection: Option<bool>,
|
|||
|
) -> fmt::Result {
|
|||
|
if !prompt.is_empty() {
|
|||
|
write!(
|
|||
|
f,
|
|||
|
"{} {} ",
|
|||
|
&self.success_prefix,
|
|||
|
self.prompt_style.apply_to(prompt)
|
|||
|
)?;
|
|||
|
}
|
|||
|
let selection = selection.map(|b| if b { "yes" } else { "no" });
|
|||
|
|
|||
|
match selection {
|
|||
|
Some(selection) => {
|
|||
|
write!(
|
|||
|
f,
|
|||
|
"{} {}",
|
|||
|
&self.success_suffix,
|
|||
|
self.values_style.apply_to(selection)
|
|||
|
)
|
|||
|
}
|
|||
|
None => {
|
|||
|
write!(f, "{}", &self.success_suffix)
|
|||
|
}
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
/// Formats an input prompt after selection.
|
|||
|
fn format_input_prompt_selection(
|
|||
|
&self,
|
|||
|
f: &mut dyn fmt::Write,
|
|||
|
prompt: &str,
|
|||
|
sel: &str,
|
|||
|
) -> fmt::Result {
|
|||
|
if !prompt.is_empty() {
|
|||
|
write!(
|
|||
|
f,
|
|||
|
"{} {} ",
|
|||
|
&self.success_prefix,
|
|||
|
self.prompt_style.apply_to(prompt)
|
|||
|
)?;
|
|||
|
}
|
|||
|
|
|||
|
write!(
|
|||
|
f,
|
|||
|
"{} {}",
|
|||
|
&self.success_suffix,
|
|||
|
self.values_style.apply_to(sel)
|
|||
|
)
|
|||
|
}
|
|||
|
|
|||
|
/// Formats a password prompt after selection.
|
|||
|
#[cfg(feature = "password")]
|
|||
|
fn format_password_prompt_selection(
|
|||
|
&self,
|
|||
|
f: &mut dyn fmt::Write,
|
|||
|
prompt: &str,
|
|||
|
) -> fmt::Result {
|
|||
|
self.format_input_prompt_selection(f, prompt, "********")
|
|||
|
}
|
|||
|
|
|||
|
/// Formats a multi select prompt after selection.
|
|||
|
fn format_multi_select_prompt_selection(
|
|||
|
&self,
|
|||
|
f: &mut dyn fmt::Write,
|
|||
|
prompt: &str,
|
|||
|
selections: &[&str],
|
|||
|
) -> fmt::Result {
|
|||
|
if !prompt.is_empty() {
|
|||
|
write!(
|
|||
|
f,
|
|||
|
"{} {} ",
|
|||
|
&self.success_prefix,
|
|||
|
self.prompt_style.apply_to(prompt)
|
|||
|
)?;
|
|||
|
}
|
|||
|
|
|||
|
write!(f, "{} ", &self.success_suffix)?;
|
|||
|
|
|||
|
if self.inline_selections {
|
|||
|
for (idx, sel) in selections.iter().enumerate() {
|
|||
|
write!(
|
|||
|
f,
|
|||
|
"{}{}",
|
|||
|
if idx == 0 { "" } else { ", " },
|
|||
|
self.values_style.apply_to(sel)
|
|||
|
)?;
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
Ok(())
|
|||
|
}
|
|||
|
|
|||
|
/// Formats a select prompt item.
|
|||
|
fn format_select_prompt_item(
|
|||
|
&self,
|
|||
|
f: &mut dyn fmt::Write,
|
|||
|
text: &str,
|
|||
|
active: bool,
|
|||
|
) -> fmt::Result {
|
|||
|
let details = if active {
|
|||
|
(
|
|||
|
&self.active_item_prefix,
|
|||
|
self.active_item_style.apply_to(text),
|
|||
|
)
|
|||
|
} else {
|
|||
|
(
|
|||
|
&self.inactive_item_prefix,
|
|||
|
self.inactive_item_style.apply_to(text),
|
|||
|
)
|
|||
|
};
|
|||
|
|
|||
|
write!(f, "{} {}", details.0, details.1)
|
|||
|
}
|
|||
|
|
|||
|
/// Formats a multi select prompt item.
|
|||
|
fn format_multi_select_prompt_item(
|
|||
|
&self,
|
|||
|
f: &mut dyn fmt::Write,
|
|||
|
text: &str,
|
|||
|
checked: bool,
|
|||
|
active: bool,
|
|||
|
) -> fmt::Result {
|
|||
|
let details = match (checked, active) {
|
|||
|
(true, true) => (
|
|||
|
&self.checked_item_prefix,
|
|||
|
self.active_item_style.apply_to(text),
|
|||
|
),
|
|||
|
(true, false) => (
|
|||
|
&self.checked_item_prefix,
|
|||
|
self.inactive_item_style.apply_to(text),
|
|||
|
),
|
|||
|
(false, true) => (
|
|||
|
&self.unchecked_item_prefix,
|
|||
|
self.active_item_style.apply_to(text),
|
|||
|
),
|
|||
|
(false, false) => (
|
|||
|
&self.unchecked_item_prefix,
|
|||
|
self.inactive_item_style.apply_to(text),
|
|||
|
),
|
|||
|
};
|
|||
|
|
|||
|
write!(f, "{} {}", details.0, details.1)
|
|||
|
}
|
|||
|
|
|||
|
/// Formats a sort prompt item.
|
|||
|
fn format_sort_prompt_item(
|
|||
|
&self,
|
|||
|
f: &mut dyn fmt::Write,
|
|||
|
text: &str,
|
|||
|
picked: bool,
|
|||
|
active: bool,
|
|||
|
) -> fmt::Result {
|
|||
|
let details = match (picked, active) {
|
|||
|
(true, true) => (
|
|||
|
&self.picked_item_prefix,
|
|||
|
self.active_item_style.apply_to(text),
|
|||
|
),
|
|||
|
(false, true) => (
|
|||
|
&self.unpicked_item_prefix,
|
|||
|
self.active_item_style.apply_to(text),
|
|||
|
),
|
|||
|
(_, false) => (
|
|||
|
&self.unpicked_item_prefix,
|
|||
|
self.inactive_item_style.apply_to(text),
|
|||
|
),
|
|||
|
};
|
|||
|
|
|||
|
write!(f, "{} {}", details.0, details.1)
|
|||
|
}
|
|||
|
|
|||
|
/// Formats a fuzzy select prompt item.
|
|||
|
#[cfg(feature = "fuzzy-select")]
|
|||
|
fn format_fuzzy_select_prompt_item(
|
|||
|
&self,
|
|||
|
f: &mut dyn fmt::Write,
|
|||
|
text: &str,
|
|||
|
active: bool,
|
|||
|
highlight_matches: bool,
|
|||
|
matcher: &SkimMatcherV2,
|
|||
|
search_term: &str,
|
|||
|
) -> fmt::Result {
|
|||
|
write!(
|
|||
|
f,
|
|||
|
"{} ",
|
|||
|
if active {
|
|||
|
&self.active_item_prefix
|
|||
|
} else {
|
|||
|
&self.inactive_item_prefix
|
|||
|
}
|
|||
|
)?;
|
|||
|
|
|||
|
if highlight_matches {
|
|||
|
if let Some((_score, indices)) = matcher.fuzzy_indices(text, &search_term) {
|
|||
|
for (idx, c) in text.chars().into_iter().enumerate() {
|
|||
|
if indices.contains(&idx) {
|
|||
|
if active {
|
|||
|
write!(
|
|||
|
f,
|
|||
|
"{}",
|
|||
|
self.active_item_style
|
|||
|
.apply_to(self.fuzzy_match_highlight_style.apply_to(c))
|
|||
|
)?;
|
|||
|
} else {
|
|||
|
write!(f, "{}", self.fuzzy_match_highlight_style.apply_to(c))?;
|
|||
|
}
|
|||
|
} else {
|
|||
|
if active {
|
|||
|
write!(f, "{}", self.active_item_style.apply_to(c))?;
|
|||
|
} else {
|
|||
|
write!(f, "{}", c)?;
|
|||
|
}
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
return Ok(());
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
write!(f, "{}", text)
|
|||
|
}
|
|||
|
|
|||
|
/// Formats a fuzzy-selectprompt after selection.
|
|||
|
#[cfg(feature = "fuzzy-select")]
|
|||
|
fn format_fuzzy_select_prompt(
|
|||
|
&self,
|
|||
|
f: &mut dyn fmt::Write,
|
|||
|
prompt: &str,
|
|||
|
search_term: &str,
|
|||
|
cursor_pos: usize,
|
|||
|
) -> fmt::Result {
|
|||
|
if !prompt.is_empty() {
|
|||
|
write!(
|
|||
|
f,
|
|||
|
"{} {} ",
|
|||
|
&self.prompt_prefix,
|
|||
|
self.prompt_style.apply_to(prompt)
|
|||
|
)?;
|
|||
|
}
|
|||
|
|
|||
|
if cursor_pos < search_term.len() {
|
|||
|
let st_head = search_term[0..cursor_pos].to_string();
|
|||
|
let st_tail = search_term[cursor_pos + 1..search_term.len()].to_string();
|
|||
|
let st_cursor = self
|
|||
|
.fuzzy_cursor_style
|
|||
|
.apply_to(search_term.to_string().chars().nth(cursor_pos).unwrap());
|
|||
|
write!(
|
|||
|
f,
|
|||
|
"{} {}{}{}",
|
|||
|
&self.prompt_suffix, st_head, st_cursor, st_tail
|
|||
|
)
|
|||
|
} else {
|
|||
|
let cursor = self.fuzzy_cursor_style.apply_to(" ");
|
|||
|
write!(
|
|||
|
f,
|
|||
|
"{} {}{}",
|
|||
|
&self.prompt_suffix,
|
|||
|
search_term.to_string(),
|
|||
|
cursor
|
|||
|
)
|
|||
|
}
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
/// Helper struct to conveniently render a theme of a term.
|
|||
|
pub(crate) struct TermThemeRenderer<'a> {
|
|||
|
term: &'a Term,
|
|||
|
theme: &'a dyn Theme,
|
|||
|
height: usize,
|
|||
|
prompt_height: usize,
|
|||
|
prompts_reset_height: bool,
|
|||
|
}
|
|||
|
|
|||
|
impl<'a> TermThemeRenderer<'a> {
|
|||
|
pub fn new(term: &'a Term, theme: &'a dyn Theme) -> TermThemeRenderer<'a> {
|
|||
|
TermThemeRenderer {
|
|||
|
term,
|
|||
|
theme,
|
|||
|
height: 0,
|
|||
|
prompt_height: 0,
|
|||
|
prompts_reset_height: true,
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
#[cfg(feature = "password")]
|
|||
|
pub fn set_prompts_reset_height(&mut self, val: bool) {
|
|||
|
self.prompts_reset_height = val;
|
|||
|
}
|
|||
|
|
|||
|
#[cfg(feature = "password")]
|
|||
|
pub fn term(&self) -> &Term {
|
|||
|
self.term
|
|||
|
}
|
|||
|
|
|||
|
pub fn add_line(&mut self) {
|
|||
|
self.height += 1;
|
|||
|
}
|
|||
|
|
|||
|
fn write_formatted_str<
|
|||
|
F: FnOnce(&mut TermThemeRenderer, &mut dyn fmt::Write) -> fmt::Result,
|
|||
|
>(
|
|||
|
&mut self,
|
|||
|
f: F,
|
|||
|
) -> io::Result<usize> {
|
|||
|
let mut buf = String::new();
|
|||
|
f(self, &mut buf).map_err(|err| io::Error::new(io::ErrorKind::Other, err))?;
|
|||
|
self.height += buf.chars().filter(|&x| x == '\n').count();
|
|||
|
self.term.write_str(&buf)?;
|
|||
|
Ok(measure_text_width(&buf))
|
|||
|
}
|
|||
|
|
|||
|
fn write_formatted_line<
|
|||
|
F: FnOnce(&mut TermThemeRenderer, &mut dyn fmt::Write) -> fmt::Result,
|
|||
|
>(
|
|||
|
&mut self,
|
|||
|
f: F,
|
|||
|
) -> io::Result<()> {
|
|||
|
let mut buf = String::new();
|
|||
|
f(self, &mut buf).map_err(|err| io::Error::new(io::ErrorKind::Other, err))?;
|
|||
|
self.height += buf.chars().filter(|&x| x == '\n').count() + 1;
|
|||
|
self.term.write_line(&buf)
|
|||
|
}
|
|||
|
|
|||
|
fn write_formatted_prompt<
|
|||
|
F: FnOnce(&mut TermThemeRenderer, &mut dyn fmt::Write) -> fmt::Result,
|
|||
|
>(
|
|||
|
&mut self,
|
|||
|
f: F,
|
|||
|
) -> io::Result<()> {
|
|||
|
self.write_formatted_line(f)?;
|
|||
|
if self.prompts_reset_height {
|
|||
|
self.prompt_height = self.height;
|
|||
|
self.height = 0;
|
|||
|
}
|
|||
|
Ok(())
|
|||
|
}
|
|||
|
|
|||
|
fn write_paging_info(buf: &mut dyn fmt::Write, paging_info: (usize, usize)) -> fmt::Result {
|
|||
|
write!(buf, " [Page {}/{}] ", paging_info.0, paging_info.1)
|
|||
|
}
|
|||
|
|
|||
|
pub fn error(&mut self, err: &str) -> io::Result<()> {
|
|||
|
self.write_formatted_line(|this, buf| this.theme.format_error(buf, err))
|
|||
|
}
|
|||
|
|
|||
|
pub fn confirm_prompt(&mut self, prompt: &str, default: Option<bool>) -> io::Result<usize> {
|
|||
|
self.write_formatted_str(|this, buf| this.theme.format_confirm_prompt(buf, prompt, default))
|
|||
|
}
|
|||
|
|
|||
|
pub fn confirm_prompt_selection(&mut self, prompt: &str, sel: Option<bool>) -> io::Result<()> {
|
|||
|
self.write_formatted_prompt(|this, buf| {
|
|||
|
this.theme.format_confirm_prompt_selection(buf, prompt, sel)
|
|||
|
})
|
|||
|
}
|
|||
|
|
|||
|
#[cfg(feature = "fuzzy-select")]
|
|||
|
pub fn fuzzy_select_prompt(
|
|||
|
&mut self,
|
|||
|
prompt: &str,
|
|||
|
search_term: &str,
|
|||
|
cursor_pos: usize,
|
|||
|
) -> io::Result<()> {
|
|||
|
self.write_formatted_prompt(|this, buf| {
|
|||
|
this.theme
|
|||
|
.format_fuzzy_select_prompt(buf, prompt, search_term, cursor_pos)
|
|||
|
})
|
|||
|
}
|
|||
|
|
|||
|
pub fn input_prompt(&mut self, prompt: &str, default: Option<&str>) -> io::Result<usize> {
|
|||
|
self.write_formatted_str(|this, buf| this.theme.format_input_prompt(buf, prompt, default))
|
|||
|
}
|
|||
|
|
|||
|
pub fn input_prompt_selection(&mut self, prompt: &str, sel: &str) -> io::Result<()> {
|
|||
|
self.write_formatted_prompt(|this, buf| {
|
|||
|
this.theme.format_input_prompt_selection(buf, prompt, sel)
|
|||
|
})
|
|||
|
}
|
|||
|
|
|||
|
#[cfg(feature = "password")]
|
|||
|
pub fn password_prompt(&mut self, prompt: &str) -> io::Result<usize> {
|
|||
|
self.write_formatted_str(|this, buf| {
|
|||
|
write!(buf, "\r")?;
|
|||
|
this.theme.format_password_prompt(buf, prompt)
|
|||
|
})
|
|||
|
}
|
|||
|
|
|||
|
#[cfg(feature = "password")]
|
|||
|
pub fn password_prompt_selection(&mut self, prompt: &str) -> io::Result<()> {
|
|||
|
self.write_formatted_prompt(|this, buf| {
|
|||
|
this.theme.format_password_prompt_selection(buf, prompt)
|
|||
|
})
|
|||
|
}
|
|||
|
|
|||
|
pub fn select_prompt(
|
|||
|
&mut self,
|
|||
|
prompt: &str,
|
|||
|
paging_info: Option<(usize, usize)>,
|
|||
|
) -> io::Result<()> {
|
|||
|
self.write_formatted_prompt(|this, buf| {
|
|||
|
this.theme.format_select_prompt(buf, prompt)?;
|
|||
|
|
|||
|
if let Some(paging_info) = paging_info {
|
|||
|
TermThemeRenderer::write_paging_info(buf, paging_info)?;
|
|||
|
}
|
|||
|
|
|||
|
Ok(())
|
|||
|
})
|
|||
|
}
|
|||
|
|
|||
|
pub fn select_prompt_selection(&mut self, prompt: &str, sel: &str) -> io::Result<()> {
|
|||
|
self.write_formatted_prompt(|this, buf| {
|
|||
|
this.theme.format_select_prompt_selection(buf, prompt, sel)
|
|||
|
})
|
|||
|
}
|
|||
|
|
|||
|
pub fn select_prompt_item(&mut self, text: &str, active: bool) -> io::Result<()> {
|
|||
|
self.write_formatted_line(|this, buf| {
|
|||
|
this.theme.format_select_prompt_item(buf, text, active)
|
|||
|
})
|
|||
|
}
|
|||
|
|
|||
|
#[cfg(feature = "fuzzy-select")]
|
|||
|
pub fn fuzzy_select_prompt_item(
|
|||
|
&mut self,
|
|||
|
text: &str,
|
|||
|
active: bool,
|
|||
|
highlight: bool,
|
|||
|
matcher: &SkimMatcherV2,
|
|||
|
search_term: &str,
|
|||
|
) -> io::Result<()> {
|
|||
|
self.write_formatted_line(|this, buf| {
|
|||
|
this.theme.format_fuzzy_select_prompt_item(
|
|||
|
buf,
|
|||
|
text,
|
|||
|
active,
|
|||
|
highlight,
|
|||
|
matcher,
|
|||
|
search_term,
|
|||
|
)
|
|||
|
})
|
|||
|
}
|
|||
|
|
|||
|
pub fn multi_select_prompt(
|
|||
|
&mut self,
|
|||
|
prompt: &str,
|
|||
|
paging_info: Option<(usize, usize)>,
|
|||
|
) -> io::Result<()> {
|
|||
|
self.write_formatted_prompt(|this, buf| {
|
|||
|
this.theme.format_multi_select_prompt(buf, prompt)?;
|
|||
|
|
|||
|
if let Some(paging_info) = paging_info {
|
|||
|
TermThemeRenderer::write_paging_info(buf, paging_info)?;
|
|||
|
}
|
|||
|
|
|||
|
Ok(())
|
|||
|
})
|
|||
|
}
|
|||
|
|
|||
|
pub fn multi_select_prompt_selection(&mut self, prompt: &str, sel: &[&str]) -> io::Result<()> {
|
|||
|
self.write_formatted_prompt(|this, buf| {
|
|||
|
this.theme
|
|||
|
.format_multi_select_prompt_selection(buf, prompt, sel)
|
|||
|
})
|
|||
|
}
|
|||
|
|
|||
|
pub fn multi_select_prompt_item(
|
|||
|
&mut self,
|
|||
|
text: &str,
|
|||
|
checked: bool,
|
|||
|
active: bool,
|
|||
|
) -> io::Result<()> {
|
|||
|
self.write_formatted_line(|this, buf| {
|
|||
|
this.theme
|
|||
|
.format_multi_select_prompt_item(buf, text, checked, active)
|
|||
|
})
|
|||
|
}
|
|||
|
|
|||
|
pub fn sort_prompt(
|
|||
|
&mut self,
|
|||
|
prompt: &str,
|
|||
|
paging_info: Option<(usize, usize)>,
|
|||
|
) -> io::Result<()> {
|
|||
|
self.write_formatted_prompt(|this, buf| {
|
|||
|
this.theme.format_sort_prompt(buf, prompt)?;
|
|||
|
|
|||
|
if let Some(paging_info) = paging_info {
|
|||
|
TermThemeRenderer::write_paging_info(buf, paging_info)?;
|
|||
|
}
|
|||
|
|
|||
|
Ok(())
|
|||
|
})
|
|||
|
}
|
|||
|
|
|||
|
pub fn sort_prompt_selection(&mut self, prompt: &str, sel: &[&str]) -> io::Result<()> {
|
|||
|
self.write_formatted_prompt(|this, buf| {
|
|||
|
this.theme.format_sort_prompt_selection(buf, prompt, sel)
|
|||
|
})
|
|||
|
}
|
|||
|
|
|||
|
pub fn sort_prompt_item(&mut self, text: &str, picked: bool, active: bool) -> io::Result<()> {
|
|||
|
self.write_formatted_line(|this, buf| {
|
|||
|
this.theme
|
|||
|
.format_sort_prompt_item(buf, text, picked, active)
|
|||
|
})
|
|||
|
}
|
|||
|
|
|||
|
pub fn clear(&mut self) -> io::Result<()> {
|
|||
|
self.term
|
|||
|
.clear_last_lines(self.height + self.prompt_height)?;
|
|||
|
self.height = 0;
|
|||
|
self.prompt_height = 0;
|
|||
|
Ok(())
|
|||
|
}
|
|||
|
|
|||
|
pub fn clear_preserve_prompt(&mut self, size_vec: &[usize]) -> io::Result<()> {
|
|||
|
let mut new_height = self.height;
|
|||
|
let prefix_width = 2;
|
|||
|
//Check each item size, increment on finding an overflow
|
|||
|
for size in size_vec {
|
|||
|
if *size > self.term.size().1 as usize {
|
|||
|
new_height += (((*size as f64 + prefix_width as f64) / self.term.size().1 as f64)
|
|||
|
.ceil()) as usize
|
|||
|
- 1;
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
self.term.clear_last_lines(new_height)?;
|
|||
|
self.height = 0;
|
|||
|
Ok(())
|
|||
|
}
|
|||
|
}
|