1
  2
  3
  4
  5
  6
  7
  8
  9
 10
 11
 12
 13
 14
 15
 16
 17
 18
 19
 20
 21
 22
 23
 24
 25
 26
 27
 28
 29
 30
 31
 32
 33
 34
 35
 36
 37
 38
 39
 40
 41
 42
 43
 44
 45
 46
 47
 48
 49
 50
 51
 52
 53
 54
 55
 56
 57
 58
 59
 60
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103

use crate::ast;
use crate::ast::expr::EagerBrace;
use crate::{ParseError, Parser, Spanned, ToTokens};
use runestick::Span;
use std::fmt;

/// A unary expression.
///
/// # Examples
///
/// ```rust
/// use rune::{testing, ast};
///
/// testing::roundtrip::<ast::ExprUnary>("!0");
/// testing::roundtrip::<ast::ExprUnary>("*foo");
/// testing::roundtrip::<ast::ExprUnary>("&foo");
/// testing::roundtrip::<ast::ExprUnary>("&Foo {
///     a: 42,
/// }");
/// ```
#[derive(Debug, Clone, PartialEq, Eq, ToTokens, Spanned)]
pub struct ExprUnary {
    /// Attributes associated with expression.
    #[rune(iter)]
    pub attributes: Vec<ast::Attribute>,
    /// Token associated with operator.
    pub op_token: ast::Token,
    /// The expression of the operation.
    pub expr: ast::Expr,
    /// The operation to apply.
    #[rune(skip)]
    pub op: UnOp,
}

impl ExprUnary {
    /// Get the span of the op.
    pub fn op_span(&self) -> Span {
        self.op_token.span()
    }

    /// Parse the uniary expression with the given meta and configuration.
    pub(crate) fn parse_with_meta(
        parser: &mut Parser,
        attributes: Vec<ast::Attribute>,
        eager_brace: EagerBrace,
    ) -> Result<Self, ParseError> {
        let op_token = parser.next()?;
        let op = UnOp::from_token(op_token)?;

        Ok(Self {
            attributes,
            op_token,
            expr: ast::Expr::parse_with(
                parser,
                eager_brace,
                ast::expr::EagerBinary(false),
                ast::expr::Callable(true),
            )?,
            op,
        })
    }
}

expr_parse!(Unary, ExprUnary, "try expression");

/// A unary operation.
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum UnOp {
    /// Not `!<thing>`.
    Not,
    /// Negation `-<thing>`.
    Neg,
    /// Reference `&<thing>`.
    BorrowRef,
    /// Dereference `*<thing>`.
    Deref,
}

impl UnOp {
    /// Convert a unary operator from a token.
    pub fn from_token(t: ast::Token) -> Result<Self, ParseError> {
        match t.kind {
            K![!] => Ok(Self::Not),
            K![-] => Ok(Self::Neg),
            K![&] => Ok(Self::BorrowRef),
            K![*] => Ok(Self::Deref),
            _ => Err(ParseError::expected(&t, "unary operator, like `!` or `-`")),
        }
    }
}

impl fmt::Display for UnOp {
    fn fmt(&self, fmt: &mut fmt::Formatter) -> fmt::Result {
        match self {
            Self::Not => write!(fmt, "!")?,
            Self::Neg => write!(fmt, "-")?,
            Self::BorrowRef => write!(fmt, "&")?,
            Self::Deref => write!(fmt, "*")?,
        }

        Ok(())
    }
}