change union index

Fix #796

Author: CppCXY

crates/emmylua_code_analysis/src/semantic/infer/infer_index.rs

@@ -664,10 +664,10 @@ fn infer_union_member(
             index_expr.clone(),
             &infer_guard.fork(),
         );
-        if let Ok(typ) = result
-            && !typ.is_nil()
-        {
+        if let Ok(typ) = result {
             member_types.push(typ);
+        } else {
+            member_types.push(LuaType::Nil);
         }
     }
 

crates/emmylua_ls/src/handlers/test/code_actions_test.rs

@@ -22,6 +22,16 @@ mod tests {
                 local _a = A:get(1):get(2):get(3)
             "#,
             vec![
+                VirtualCodeAction {
+                    title: "Disable current line diagnostic (need-check-nil)".to_string()
+                },
+                VirtualCodeAction {
+                    title: "Disable all diagnostics in current file (need-check-nil)".to_string()
+                },
+                VirtualCodeAction {
+                    title:
+                        "Disable all diagnostics in current project (need-check-nil)".to_string()
+                },
                 VirtualCodeAction {
                     title: "use cast to remove nil".to_string()
                 },
@@ -35,6 +45,16 @@ mod tests {
                     title:
                         "Disable all diagnostics in current project (need-check-nil)".to_string()
                 },
+                VirtualCodeAction {
+                    title: "Disable current line diagnostic (need-check-nil)".to_string()
+                },
+                VirtualCodeAction {
+                    title: "Disable all diagnostics in current file (need-check-nil)".to_string()
+                },
+                VirtualCodeAction {
+                    title:
+                        "Disable all diagnostics in current project (need-check-nil)".to_string()
+                },
                 VirtualCodeAction {
                     title: "use cast to remove nil".to_string()
                 },