gbf_core/decompiler/ast/
function.rs

#![deny(missing_docs)]

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

use super::{ast_vec::AstVec, expr::ExprKind, AstKind, AstVisitable};

/// Represents a metadata node in the AST
#[derive(Debug, Clone, Serialize, Deserialize, Eq, AstNodeTransform)]
#[convert_to(AstKind::Function)]
pub struct FunctionNode {
    name: Option<String>,
    params: AstVec<ExprKind>,
    body: AstVec<AstKind>,
}

impl FunctionNode {
    /// Creates a new `FunctionNode` with the given `params` and `body`.
    ///
    /// # Arguments
    /// - `params` - The parameters of the function.
    /// - `body` - The body of the function.
    ///
    /// # Returns
    /// A new `FunctionNode`.
    pub fn new<N>(name: N, params: AstVec<ExprKind>, body: AstVec<AstKind>) -> Self
    where
        N: Into<Option<String>>,
    {
        Self {
            name: name.into(),
            params,
            body,
        }
    }

    /// Returns the parameters of the function.
    pub fn params(&self) -> &Vec<ExprKind> {
        &self.params
    }

    /// Returns the body of the function.
    pub fn body(&self) -> &Vec<AstKind> {
        &self.body
    }

    /// Returns the name of the function.
    pub fn name(&self) -> &Option<String> {
        &self.name
    }
}

// == Other implementations for literal ==
impl AstVisitable for FunctionNode {
    fn accept(&self, visitor: &mut dyn super::visitors::AstVisitor) {
        visitor.visit_function(self);
    }
}

impl PartialEq for FunctionNode {
    fn eq(&self, other: &Self) -> bool {
        self.params == other.params && self.body == other.body
    }
}
gbf_core/decompiler/ast/function.rs

gbf_core/decompiler/ast/
function.rs
