use crate::{
formatter::*,
utils::map::byte_span::{ByteSpan, LeafSpans},
};
use std::fmt::Write;
use sway_ast::{
keywords::{ColonToken, Keyword, Token, WhereToken},
CommaToken, WhereBound, WhereClause,
};
use sway_types::Spanned;
impl Format for WhereClause {
fn format(
&self,
formatted_code: &mut FormattedCode,
formatter: &mut Formatter,
) -> Result<(), FormatterError> {
writeln!(
formatted_code,
"{}{}",
formatter.indent_to_str()?,
WhereToken::AS_STR,
)?;
formatter.indent();
let value_pairs = self.bounds.value_separator_pairs.clone();
for (bound, _comma_token) in value_pairs.iter() {
bound.format(formatted_code, formatter)?;
writeln!(formatted_code, "{}", CommaToken::AS_STR)?;
}
if let Some(final_value) = &self.bounds.final_value_opt {
final_value.format(formatted_code, formatter)?;
writeln!(formatted_code, "{}", CommaToken::AS_STR)?;
}
formatter.unindent();
Ok(())
}
}
impl Format for WhereBound {
fn format(
&self,
formatted_code: &mut FormattedCode,
formatter: &mut Formatter,
) -> Result<(), FormatterError> {
write!(
formatted_code,
"{}{}{} ",
formatter.indent_to_str()?,
self.ty_name.as_str(),
ColonToken::AS_STR,
)?;
self.bounds.format(formatted_code, formatter)?;
Ok(())
}
}
impl LeafSpans for WhereBound {
fn leaf_spans(&self) -> Vec<ByteSpan> {
let mut collected_spans = self.ty_name.leaf_spans();
collected_spans.append(&mut self.colon_token.leaf_spans());
collected_spans.append(&mut self.bounds.leaf_spans());
collected_spans
}
}
impl LeafSpans for WhereClause {
fn leaf_spans(&self) -> Vec<ByteSpan> {
let mut collected_spans = vec![ByteSpan::from(self.where_token.span())];
collected_spans.append(&mut self.bounds.leaf_spans());
collected_spans
}
}