use crate::{
formatter::*,
utils::{
map::byte_span::{ByteSpan, LeafSpans},
Parenthesis,
},
};
use std::fmt::Write;
use sway_ast::{token::Delimiter, AbiCastArgs};
use sway_types::Spanned;
impl Format for AbiCastArgs {
fn format(
&self,
formatted_code: &mut FormattedCode,
formatter: &mut Formatter,
) -> Result<(), FormatterError> {
Self::open_parenthesis(formatted_code, formatter)?;
self.name.format(formatted_code, formatter)?;
write!(formatted_code, "{} ", self.comma_token.span().as_str())?;
self.address.format(formatted_code, formatter)?;
Self::close_parenthesis(formatted_code, formatter)?;
Ok(())
}
}
impl Parenthesis for AbiCastArgs {
fn open_parenthesis(
line: &mut FormattedCode,
_formatter: &mut Formatter,
) -> Result<(), FormatterError> {
write!(line, "{}", Delimiter::Parenthesis.as_open_char())?;
Ok(())
}
fn close_parenthesis(
line: &mut FormattedCode,
_formatter: &mut Formatter,
) -> Result<(), FormatterError> {
write!(line, "{}", Delimiter::Parenthesis.as_close_char())?;
Ok(())
}
}
impl LeafSpans for AbiCastArgs {
fn leaf_spans(&self) -> Vec<ByteSpan> {
let mut collected_spans = vec![ByteSpan::from(self.name.span())];
collected_spans.push(ByteSpan::from(self.comma_token.span()));
collected_spans.append(&mut self.address.leaf_spans());
collected_spans
}
}