Fix issue resolving comparison binary expr

2 files changed, 33 insertions, 0 deletions

diff --git a/src/server/analysis.odin b/src/server/analysis.odin
index e1ebfd6..c4de6dd 100644
--- a/src/server/analysis.odin
+++ b/src/server/analysis.odin
@@ -1467,6 +1467,8 @@ internal_resolve_type_identifier :: proc(ast_context: ^AstContext, node: ast.Ide
 			return_symbol, ok = resolve_basic_lit(ast_context, v^)
 			return_symbol.name = node.name
 			return_symbol.type = local.variable ? .Variable : .Constant
+		case ^ast.Binary_Expr:
+			return_symbol, ok = resolve_binary_expression(ast_context, v)
 		case:
 			return_symbol, ok = internal_resolve_type_expression(ast_context, local.rhs)
 		}
@@ -2415,6 +2417,14 @@ resolve_binary_expression :: proc(ast_context: ^AstContext, binary: ^ast.Binary_
 	symbol_a, symbol_b: Symbol
 	ok_a, ok_b: bool
 
+	#partial switch binary.op.kind {
+	case .Cmp_Eq, .Gt, .Gt_Eq, .Lt, .Lt_Eq:
+		symbol_a.value = SymbolUntypedValue {
+			type = .Bool,
+		}
+		return symbol_a, true
+	}
+
 	if expr, ok := binary.left.derived.(^ast.Binary_Expr); ok {
 		symbol_a, ok_a = resolve_binary_expression(ast_context, expr)
 	} else {
diff --git a/tests/hover_test.odin b/tests/hover_test.odin
index af9990e..a69b494 100644
--- a/tests/hover_test.odin
+++ b/tests/hover_test.odin
@@ -2559,6 +2559,29 @@ ast_hover_struct_field_should_show_docs_and_comments_matrix :: proc(t: ^testing.
 	}
 	test.expect_hover( t, &source, "Foo.bar: matrix[4,5]int // bar comment\n bar docs")
 }
+
+@(test)
+ast_hover_variable_from_comparison :: proc(t: ^testing.T) {
+	source := test.Source {
+		main = `package test
+
+		Foo :: enum {
+			A,
+			B,
+		}
+
+		Bar :: struct {
+			foo: Foo,
+		}
+
+		main :: proc() {
+			bar: Bar
+			b{*}azz := bar.bar == .A
+		}
+		`,
+	}
+	test.expect_hover( t, &source, "test.bazz: bool")
+}
 /*
 
 Waiting for odin fix