✨ fix(inlay_hints): extend inlay hint support for variable, call, and reference symbols

Author: harehare

crates/mq-check/src/builtin.rs

@@ -79,12 +79,12 @@ fn register_arithmetic(ctx: &mut InferenceContext) {
     register_binary(ctx, "+", Type::String, Type::String, Type::String);
     register_binary(ctx, "add", Type::String, Type::String, Type::String);
 
-    // Addition: string + number -> string (coercion)
-    register_binary(ctx, "+", Type::String, Type::Number, Type::String);
-    register_binary(ctx, "add", Type::String, Type::Number, Type::String);
-
-    // Addition: [a] + [a] -> [a] (array concatenation)
     for name in ["+", "add"] {
+        // Addition: string + number -> string (coercion)
+        register_binary(ctx, name, Type::String, Type::Number, Type::String);
+        register_binary(ctx, name, Type::Number, Type::String, Type::String);
+
+        // Addition: [a] + [a] -> [a] (array concatenation)
         let a = ctx.fresh_var();
         register_binary(
             ctx,

crates/mq-lsp/src/inlay_hints.rs

@@ -46,7 +46,7 @@ pub fn response(
             let type_label = format!(": {}", type_scheme.ty);
 
             match &symbol.kind {
-                mq_hir::SymbolKind::Variable => {
+                mq_hir::SymbolKind::Variable | mq_hir::SymbolKind::Call | mq_hir::SymbolKind::Ref => {
                     // Place hint after the variable name (end of the symbol)
                     let hint_col = (text_range.end.column as u32).saturating_sub(1);
                     let hint_line = text_range.end.line.saturating_sub(1);
@@ -61,21 +61,6 @@ pub fn response(
                         data: None,
                     })
                 }
-                mq_hir::SymbolKind::Function(_) | mq_hir::SymbolKind::Macro(_) => {
-                    // Place hint after the function name showing the return type
-                    let hint_col = (text_range.end.column as u32).saturating_sub(1);
-                    let hint_line = text_range.end.line.saturating_sub(1);
-                    Some(InlayHint {
-                        position: Position::new(hint_line, hint_col),
-                        label: InlayHintLabel::String(type_label),
-                        kind: Some(InlayHintKind::TYPE),
-                        text_edits: None,
-                        tooltip: None,
-                        padding_left: Some(false),
-                        padding_right: Some(true),
-                        data: None,
-                    })
-                }
                 _ => None,
             }
         })