php_codegen 0.4.0

Generate PHP code from Rust using a fluent API 🐘 🦀
Documentation
use std::fmt::Debug;

use crate::attribute::AttributeGroup;
use crate::body::Body;
use crate::comment::Document;
use crate::data_type::DataType;
use crate::modifiers::Modifier;
use crate::modifiers::VisibilityModifier;
use crate::parameter::Parameter;
use crate::Generator;
use crate::Indentation;

#[derive(Debug)]
pub struct Method {
    pub documentation: Option<Document>,
    pub attributes: Vec<AttributeGroup>,
    pub name: String,
    pub parameters: Vec<Parameter>,
    pub return_type: Option<DataType>,
    pub body: Body,
    pub modifiers: Vec<Modifier>,
    pub visibility: Option<VisibilityModifier>,
}

impl Method {
    pub fn new<T: ToString>(name: T) -> Self {
        Self {
            name: name.to_string(),
            parameters: vec![],
            return_type: None,
            body: Body::default(),
            modifiers: vec![],
            attributes: vec![],
            documentation: None,
            visibility: None,
        }
    }

    pub fn document(mut self, documentation: Document) -> Self {
        self.documentation = Some(documentation);

        self
    }

    pub fn attributes(mut self, attributes: AttributeGroup) -> Self {
        self.attributes.push(attributes);

        self
    }

    pub fn public(mut self) -> Self {
        self.visibility = Some(VisibilityModifier::Public);

        self
    }

    pub fn protected(mut self) -> Self {
        self.visibility = Some(VisibilityModifier::Protected);

        self
    }

    pub fn private(mut self) -> Self {
        self.visibility = Some(VisibilityModifier::Private);

        self
    }

    pub fn visibility(mut self, visibility: VisibilityModifier) -> Self {
        self.visibility = Some(visibility);

        self
    }

    pub fn modifier(mut self, modifier: Modifier) -> Self {
        self.modifiers.push(modifier);

        self
    }

    pub fn parameter(mut self, parameter: Parameter) -> Self {
        self.parameters.push(parameter);

        self
    }

    pub fn returns(mut self, return_type: DataType) -> Self {
        self.return_type = Some(return_type);

        self
    }

    pub fn body<T: Into<Body>>(mut self, body: T) -> Self {
        self.body = body.into().with_semicolon_for_empty(true);

        self
    }
}

impl Generator for Method {
    fn generate(&self, indentation: Indentation, level: usize) -> String {
        let mut code = String::new();

        if let Some(document) = &self.documentation {
            code.push_str(&document.generate(indentation, level));
        }

        for attribute in &self.attributes {
            code.push_str(&attribute.generate(indentation, level));
        }

        code.push_str(&indentation.value(level));
        if let Some(visibility) = &self.visibility {
            code.push_str(format!("{} ", visibility.generate(indentation, level)).as_str());
        }

        if !self.modifiers.is_empty() {
            code.push_str(
                &self
                    .modifiers
                    .iter()
                    .map(|modifier| modifier.generate(indentation, level))
                    .collect::<Vec<String>>()
                    .join(" "),
            );

            code.push(' ');
        }

        code.push_str(format!("function {}", self.name).as_str());
        code.push_str(&self.parameters.generate(indentation, level));

        if let Some(return_type) = &self.return_type {
            code.push_str(format!(": {}", return_type.generate(indentation, level)).as_str());
        }

        code.push_str(&self.body.generate(indentation, level));

        code
    }
}

impl Generator for Vec<Method> {
    fn generate(&self, indentation: Indentation, level: usize) -> String {
        let mut code = String::new();
        if self.is_empty() {
            return code;
        }

        for method in self {
            code.push_str(method.generate(indentation, level).as_str());
            code.push('\n');
        }

        code
    }
}