use crate::*;
use std::{ops::{Range, RangeBounds, RangeFrom, RangeInclusive, RangeTo, RangeToInclusive}, str::FromStr};
pub fn read_range<T, R>(range: R, mut prompt: String, default: Option<T>) -> BoxResult<T>
where
T: Display + FromStr + PartialOrd<T>,
R: RangeBounds<T>,
<T as FromStr>::Err: Display,
{
if let Some(default) = default.as_ref() {
prompt += &format!("(default: {default}) ");
}
loop {
print!("{prompt}");
let output_string = read_stdin()?;
if output_string.is_empty() && let Some(default) = default {
return Ok(default);
}
let output = match output_string.parse::<T>() {
Ok(v) => v,
Err(err) => {
println!();
println!("Could not parse input ({err})");
continue;
}
};
if range.contains(&output) {
return Ok(output);
}
println!();
println!("Invalid input, not within bounds");
}
}
impl<T> TryRead for Range<T>
where
T: Display + FromStr + PartialOrd<T>,
<T as FromStr>::Err: Display,
{
type Output = T;
type Default = T;
fn try_read_line(self, prompt: Option<String>, default: Option<Self::Output>) -> BoxResult<Self::Output> {
let prompt = prompt.unwrap_or_else(|| format!("Enter a number within the range [{:.1}, {:.1}): ", self.start, self.end));
read_range(self, prompt, default)
}
}
impl<T> TryRead for RangeInclusive<T>
where
T: Display + FromStr + PartialOrd<T>,
<T as FromStr>::Err: Display,
{
type Output = T;
type Default = T;
fn try_read_line(self, prompt: Option<String>, default: Option<Self::Output>) -> BoxResult<Self::Output> {
let prompt = prompt.unwrap_or_else(|| format!("Enter a number within the range [{:.1}, {:.1}]: ", self.start(), self.end()));
read_range(self, prompt, default)
}
}
impl<T> TryRead for RangeTo<T>
where
T: Display + FromStr + PartialOrd<T>,
<T as FromStr>::Err: Display,
{
type Output = T;
type Default = T;
fn try_read_line(self, prompt: Option<String>, default: Option<Self::Output>) -> BoxResult<Self::Output> {
let prompt = prompt.unwrap_or_else(|| format!("Enter a number which is less than {:.1}: ", self.end));
read_range(self, prompt, default)
}
}
impl<T> TryRead for RangeFrom<T>
where
T: Display + FromStr + PartialOrd<T>,
<T as FromStr>::Err: Display,
{
type Output = T;
type Default = T;
fn try_read_line(self, prompt: Option<String>, default: Option<Self::Output>) -> BoxResult<Self::Output> {
let prompt = prompt.unwrap_or_else(|| format!("Enter a number which is at least {:.1}", self.start));
read_range(self, prompt, default)
}
}
impl<T> TryRead for RangeToInclusive<T>
where
T: Display + FromStr + PartialOrd<T>,
<T as FromStr>::Err: Display,
{
type Output = T;
type Default = T;
fn try_read_line(self, prompt: Option<String>, default: Option<Self::Output>) -> BoxResult<Self::Output> {
let prompt = prompt.unwrap_or_else(|| format!("Enter a number which is at most {:.1}: ", self.end));
read_range(self, prompt, default)
}
}