gbf_core/decompiler/ast/
ret.rs

#![deny(missing_docs)]

use gbf_macros::AstNodeTransform;
use serde::{Deserialize, Serialize};

use super::{expr::ExprKind, visitors::AstVisitor, AstKind, AstVisitable};

/// Represents a return node in the AST, such as `return 5`.
#[derive(Debug, Clone, Serialize, Deserialize, Eq, AstNodeTransform)]
#[convert_to(AstKind::Return)]
pub struct ReturnNode {
    /// The value to return.
    pub ret: Box<ExprKind>,
}

impl ReturnNode {
    /// Creates a new return node.
    ///
    /// # Arguments
    /// - `ret`: The value to return.
    ///
    /// # Returns
    /// The return node.
    pub fn new(ret: Box<ExprKind>) -> Self {
        Self { ret }
    }
}

impl AstVisitable for ReturnNode {
    fn accept(&self, visitor: &mut dyn AstVisitor) {
        visitor.visit_return(self);
    }
}

// == Other implementations for return ==
impl PartialEq for ReturnNode {
    fn eq(&self, other: &Self) -> bool {
        self.ret == other.ret
    }
}

#[cfg(test)]
mod tests {
    use crate::decompiler::ast::{emit, literal::LiteralNode};

    use super::*;

    #[test]
    fn test_return_node() {
        let ret = ReturnNode::new(Box::new(ExprKind::Literal(LiteralNode::new_number(5))));
        assert_eq!(
            ret.ret,
            Box::new(ExprKind::Literal(LiteralNode::new_number(5)))
        );
    }

    #[test]
    fn test_emit() {
        let ret = ReturnNode::new(Box::new(ExprKind::Literal(LiteralNode::new_number(5))));
        assert_eq!(emit(ret), "return 5;");
    }

    #[test]
    fn test_equality() {
        let ret = ReturnNode::new(Box::new(ExprKind::Literal(LiteralNode::new_number(5))));
        let ret2 = ReturnNode::new(Box::new(ExprKind::Literal(LiteralNode::new_number(5))));
        let ret3 = ReturnNode::new(Box::new(ExprKind::Literal(LiteralNode::new_number(6))));
        assert_eq!(ret, ret2);
        assert_ne!(ret, ret3);
    }
}
gbf_core/decompiler/ast/ret.rs

gbf_core/decompiler/ast/
ret.rs
