Fix index member and global referencs bug

Author: CppCXY

EmmyLua/CodeAnalysis/Compilation/Search/IndexMembers.cs

@@ -354,6 +354,10 @@ public class IndexMembers(SearchContext context)
         {
             return new LuaSymbol(string.Empty, arrayType.BaseType, new VirtualInfo());
         }
+        else if (type is LuaGenericType genericType)
+        {
+            return FindGenericTypeMember(genericType, indexExpr);
+        }
         else if (type is LuaNamedType namedType && indexExpr.IndexKeyExpr is { } indexKeyExpr)
         {
             var keyType = context.Infer(indexKeyExpr);
@@ -426,4 +430,26 @@ public class IndexMembers(SearchContext context)
 
         return null;
     }
+
+    private LuaSymbol? FindGenericTypeMember(LuaGenericType type, LuaIndexExprSyntax indexExpr)
+    {
+        if (type.Name == "table")
+        {
+            if (type.GenericArgs.Count != 2)
+            {
+                return null;
+            }
+
+            if (indexExpr.IndexKeyExpr is { } indexKeyExpr)
+            {
+                var keyType = context.Infer(indexKeyExpr);
+                if (keyType.SubTypeOf(type.GenericArgs[0], context))
+                {
+                    return new LuaSymbol(string.Empty, type.GenericArgs[1], new VirtualInfo());
+                }
+            }
+        }
+
+        return null;
+    }
 }

EmmyLua/CodeAnalysis/Compilation/Search/References.cs

@@ -68,7 +68,7 @@ private IEnumerable<ReferenceResult> GlobalReferences(LuaSymbol symbol)
 
         foreach (var nameExpr in nameExprs)
         {
-            if (context.FindDeclaration(nameExpr) == symbol)
+            if (IsReferenceTo(nameExpr, symbol))
             {
                 references.Add(new ReferenceResult(nameExpr.Location, nameExpr));
             }

EmmyLua/CodeAnalysis/Syntax/Node/SyntaxNodes/Basic.cs

@@ -166,7 +166,7 @@ public LuaDocTypeSyntax? TypeField
                 {
                     if (i + 1 < finish)
                     {
-                        var element = Tree.GetElement(i);
+                        var element = Tree.GetElement(i + 1);
                         if (element is LuaDocTypeSyntax typeSyntax)
                         {
                             return typeSyntax;

EmmyLua/CodeAnalysis/Syntax/Node/SyntaxNodes/Expressions.cs

@@ -220,7 +220,7 @@ public LuaExprSyntax? IndexKeyExpr
                 {
                     if (i + 1 < finish)
                     {
-                        var element = Tree.GetElement(i);
+                        var element = Tree.GetElement(i + 1);
                         if (element is LuaExprSyntax exprSyntax)
                         {
                             return exprSyntax;