enum 允许Integer赋值

Author: xuhuanzy

crates/emmylua_code_analysis/src/diagnostic/test/assign_type_mismatch_test.rs

@@ -64,6 +64,52 @@ mod tests {
         ));
     }
 
+    #[test]
+    fn test_enum() {
+        let mut ws = VirtualWorkspace::new();
+        assert!(ws.check_code_for_namespace(
+            DiagnosticCode::AssignTypeMismatch,
+            r#" 
+                ---@enum SubscriberFlags
+                local SubscriberFlags = {
+                    None = 0,
+                    Tracking = 1 << 0,
+                    Recursed = 1 << 1,
+                    ToCheckDirty = 1 << 3,
+                    Dirty = 1 << 4,
+                }
+                ---@class Subscriber
+                ---@field flags SubscriberFlags
+
+                ---@type Subscriber
+                local subscriber
+
+                subscriber.flags = subscriber.flags & ~SubscriberFlags.Tracking -- 被推断为`integer`而不是实际整数值, 允许匹配
+            "#
+        ));
+
+        assert!(!ws.check_code_for_namespace(
+            DiagnosticCode::AssignTypeMismatch,
+            r#" 
+                ---@enum SubscriberFlags
+                local SubscriberFlags = {
+                    None = 0,
+                    Tracking = 1 << 0,
+                    Recursed = 1 << 1,
+                    ToCheckDirty = 1 << 3,
+                    Dirty = 1 << 4,
+                }
+                ---@class Subscriber
+                ---@field flags SubscriberFlags
+
+                ---@type Subscriber
+                local subscriber
+                
+                subscriber.flags = 9 -- 不允许匹配不上的实际值
+            "#
+        ));
+    }
+
     #[test]
     fn test_issue_193() {
         let mut ws = VirtualWorkspace::new();

crates/emmylua_code_analysis/src/semantic/type_check/ref_type.rs

@@ -83,6 +83,19 @@ pub fn check_ref_type_compact(
             }
         }
 
+        // 当 enum 的值全为整数常量时, 可能会用于位运算, 此时右值推断为整数
+        if union_types
+            .iter()
+            .all(|t| matches!(t, LuaType::DocIntegerConst(_)))
+        {
+            match compact_type {
+                LuaType::Integer => {
+                    return Ok(());
+                }
+                _ => {}
+            }
+        }
+
         let fake_union_type = LuaType::Union(LuaUnionType::new(union_types).into());
         return check_general_type_compact(
             db,