emmylua_code_analysis 0.22.0

use emmylua_parser::{LuaAstNode, LuaCallExpr, LuaExpr, LuaIndexKey, LuaTableField};

use crate::{
    InFiled, LuaOperator, LuaOperatorMetaMethod, LuaOperatorOwner, LuaSignatureId, OperatorFunction,
};

use super::LuaAnalyzer;

pub fn analyze_setmetatable(analyzer: &mut LuaAnalyzer, call_expr: LuaCallExpr) -> Option<()> {
    let arg_list = call_expr.get_args_list()?;
    let args = arg_list.get_args().collect::<Vec<_>>();

    if args.len() != 2 {
        return Some(());
    }

    let table = args[0].clone();
    let metatable = args[1].clone();
    let LuaExpr::TableExpr(metatable) = metatable else {
        return Some(());
    };

    let file_id = analyzer.file_id;
    analyzer.db.get_metatable_index_mut().add(
        InFiled::new(file_id, table.get_range()),
        InFiled::new(file_id, metatable.get_range()),
    );

    let operator_owner = LuaOperatorOwner::Table(InFiled::new(file_id, metatable.get_range()));
    for field in metatable.get_fields() {
        analyze_metable_field(analyzer, &field, &operator_owner);
    }

    Some(())
}

fn analyze_metable_field(
    analyzer: &mut LuaAnalyzer,
    field: &LuaTableField,
    operator_owner: &LuaOperatorOwner,
) -> Option<()> {
    let field_name = match field.get_field_key()? {
        LuaIndexKey::Name(n) => n.get_name_text().to_string(),
        LuaIndexKey::String(s) => s.get_value(),
        _ => return None,
    };

    let meta_method = LuaOperatorMetaMethod::from_metatable_name(&field_name)?;
    let field_value = field.get_value_expr()?;
    let file_id = analyzer.file_id;

    let signature_id = match field_value {
        LuaExpr::ClosureExpr(closure) => LuaSignatureId::from_closure(file_id, &closure),
        _ => return None,
    };

    let operator = LuaOperator::new(
        operator_owner.clone(),
        meta_method,
        file_id,
        field.get_range(),
        OperatorFunction::Signature(signature_id),
    );
    analyzer.db.get_operator_index_mut().add_operator(operator);

    Some(())
}