use std::fmt::{self, Display, Formatter};

use serde::{Deserialize, Serialize};

use crate::{Identifier, Statement};

#[derive(Debug, Clone, Serialize, Deserialize, Eq, PartialEq, PartialOrd, Ord)]
pub struct Function {
    identifiers: Vec<Identifier>,
    statements: Vec<Statement>,
}

impl Function {
    pub fn new(identifiers: Vec<Identifier>, statements: Vec<Statement>) -> Self {
        Function {
            identifiers,
            statements,
        }
    }

    pub fn identifiers(&self) -> &Vec<Identifier> {
        &self.identifiers
    }

    pub fn statements(&self) -> &Vec<Statement> {
        &self.statements
    }
}

impl Display for Function {
    fn fmt(&self, f: &mut Formatter<'_>) -> fmt::Result {
        write!(
            f,
            "function < {:?} > {{ {:?} }}", // TODO: Correct this output
            self.identifiers, self.statements
        )
    }
}