just/src/parameter.rs

use crate::common::*;

/// A single function parameter
#[derive(PartialEq, Debug)]
pub(crate) struct Parameter<'src> {
  /// The parameter name
  pub(crate) name:    Name<'src>,
  /// The kind of parameter
  pub(crate) kind:    ParameterKind,
  /// An optional default expression
  pub(crate) default: Option<Expression<'src>>,
}

impl<'src> Display for Parameter<'src> {
  fn fmt(&self, f: &mut Formatter) -> Result<(), fmt::Error> {
    let color = Color::fmt(f);
    if let Some(prefix) = self.kind.prefix() {
      write!(f, "{}", color.annotation().paint(prefix))?;
    }
    write!(f, "{}", color.parameter().paint(self.name.lexeme()))?;
    if let Some(ref default) = self.default {
      write!(f, "={}", color.string().paint(&default.to_string()))?;
    }
    Ok(())
  }
}