mod.rs

mod item_fn;
pub mod utils;

use proc_macro2::Ident;
use syn::spanned::Spanned;

use crate::parse::Ast;

use self::item_fn::ImplItemFnModel;

#[derive(Debug)]
pub struct Model {
    pub original_ast: Ast,
    pub struct_name: Ident,
    pub methods: Vec<ImplItemFnModel>,
}

impl Model {
    pub fn new(original_ast: Ast, struct_name: Ident) -> Self {
        Self {
            original_ast,
            struct_name,
            methods: vec![],
        }
    }
}

pub fn analyze(ast: Ast) -> syn::Result<Model> {
    let struct_name = match *ast.self_ty {
        syn::Type::Path(ref path) => path.path.get_ident(),
        _ => None,
    };
    let struct_name = struct_name.ok_or(syn::Error::new(
        ast.self_ty.span(),
        "Impl block must have a name".to_string(),
    ))?;

    let mut model = Model::new(ast.clone(), struct_name.clone());

    for item in ast.items {
        if let syn::ImplItem::Fn(impl_item_fn) = item {
            let fn_model = ImplItemFnModel::try_from(&impl_item_fn)?;
            model.methods.push(fn_model);
        }
    }

    Ok(model)
}