...

src/compiler/Syntax/DefinitionTable.cs

@@ -1,158 +1,23 @@
-using Common;
-using Syntax.Binding;
 using Syntax.Node;
-using Syntax.Parsing;
 
 namespace Syntax;
 
 public class DefinitionTable
 {
-    private readonly IEnumerable<SyntaxTree> _syntaxTrees;
+    private readonly List<TopLevelNode> _topLevelNodes;
 
     public DefinitionTable(IEnumerable<SyntaxTree> syntaxTrees)
     {
-        _syntaxTrees = syntaxTrees;
-    }
-
-    public Optional<FuncDefinition> LookupFunction(string @namespace, string name)
-    {
-        var definition = _syntaxTrees
-            .Where(c => c.Namespace == @namespace)
-            .SelectMany(c => c.Definitions)
-            .OfType<FuncDefinition>()
-            .SingleOrDefault(f => f.Name == name);
-
-        return Optional.OfNullable(definition);
-    }
-
-    public Optional<StructDefinitionNode> LookupStruct(string @namespace, string name)
-    {
-        var definition = _syntaxTrees
-            .Where(c => c.Namespace == @namespace)
-            .SelectMany(c => c.Definitions)
-            .OfType<StructDefinitionNode>()
-            .SingleOrDefault(f => f.Name == name);
-
-        return Optional.OfNullable(definition);
-    }
-
-    public Optional<TraitDefinitionNode> LookupTrait(string @namespace, string name)
-    {
-        var definition = _syntaxTrees
-            .Where(c => c.Namespace == @namespace)
-            .SelectMany(c => c.Definitions)
-            .OfType<TraitDefinitionNode>()
-            .SingleOrDefault(f => f.Name == name);
-
-        return Optional.OfNullable(definition);
-    }
-
-    public Optional<Tuple<TraitImplementationDefinitionNode, ImplementationFuncNode>> LookupTraitImplementationForType(NubType type, string funcName)
-    {
-        var implementations = _syntaxTrees
-            .SelectMany(c => c.Definitions)
-            .OfType<TraitImplementationDefinitionNode>()
-            .Where(c => c.ForType == type);
-
-        var implementation = implementations.SingleOrDefault(c => c.Functions.Any(x => x.Name == funcName));
-
-        var value = implementation == null ? null : new Tuple<TraitImplementationDefinitionNode, ImplementationFuncNode>(implementation, implementation.Functions.First(x => x.Name == funcName));
-
-        return Optional.OfNullable(value);
-    }
-
-    public Optional<TraitFunc> LookupFunctionOnTrait(string @namespace, string name, string funcName)
-    {
-        var traitDef = LookupTrait(@namespace, name);
-        if (traitDef.HasValue)
-        {
-            var function = traitDef.Value.Functions.SingleOrDefault(x => x.Name == funcName);
-            return Optional.OfNullable(function);
-        }
-
-        return Optional<TraitFunc>.Empty();
+        _topLevelNodes = syntaxTrees.SelectMany(x => x.TopLevelNodes).ToList();
     }
 }
 
 public class BoundDefinitionTable
 {
-    private readonly IEnumerable<BoundSyntaxTree> _syntaxTrees;
+    private readonly List<BoundTopLevelNode> _topLevelNodes;
 
     public BoundDefinitionTable(IEnumerable<BoundSyntaxTree> syntaxTrees)
     {
-        _syntaxTrees = syntaxTrees;
-    }
-
-    public Optional<BoundFuncDefinition> LookupFunc(string @namespace, string name)
-    {
-        var definition = _syntaxTrees
-            .Where(c => c.Namespace == @namespace)
-            .SelectMany(c => c.Definitions)
-            .OfType<BoundFuncDefinition>()
-            .SingleOrDefault(f => f.Name == name);
-
-        return Optional.OfNullable(definition);
-    }
-
-    public Optional<BoundStructDefinitionNode> LookupStruct(string @namespace, string name)
-    {
-        var definition = _syntaxTrees
-            .Where(c => c.Namespace == @namespace)
-            .SelectMany(c => c.Definitions)
-            .OfType<BoundStructDefinitionNode>()
-            .SingleOrDefault(f => f.Name == name);
-
-        return Optional.OfNullable(definition);
-    }
-
-    public Optional<BoundTraitDefinitionNode> LookupTrait(string @namespace, string name)
-    {
-        var definition = _syntaxTrees
-            .Where(c => c.Namespace == @namespace)
-            .SelectMany(c => c.Definitions)
-            .OfType<BoundTraitDefinitionNode>()
-            .SingleOrDefault(f => f.Name == name);
-
-        return Optional.OfNullable(definition);
-    }
-
-    public Optional<Tuple<BoundTraitImplementationDefinitionNode, BoundImplementationFuncNode>> LookupTraitImplementationForType(NubType type, string funcName)
-    {
-        var implementations = _syntaxTrees
-            .SelectMany(c => c.Definitions)
-            .OfType<BoundTraitImplementationDefinitionNode>()
-            .Where(c => c.ForType == type);
-
-        var implementation = implementations.SingleOrDefault(c => c.Functions.Any(x => x.Name == funcName));
-
-        var value = implementation == null ? null : new Tuple<BoundTraitImplementationDefinitionNode, BoundImplementationFuncNode>(implementation, implementation.Functions.First(x => x.Name == funcName));
-
-        return Optional.OfNullable(value);
-    }
-
-    public Optional<BountTraitFunc> LookupFunctionOnTrait(string @namespace, string name, string funcName)
-    {
-        var traitDef = LookupTrait(@namespace, name);
-        if (traitDef.HasValue)
-        {
-            var function = traitDef.Value.Functions.SingleOrDefault(x => x.Name == funcName);
-            return Optional.OfNullable(function);
-        }
-
-        return Optional<BountTraitFunc>.Empty();
-    }
-
-    public IEnumerable<BoundStructDefinitionNode> GetStructs()
-    {
-        return _syntaxTrees
-            .SelectMany(c => c.Definitions)
-            .OfType<BoundStructDefinitionNode>();
-    }
-
-    public IEnumerable<BoundTraitDefinitionNode> GetTraits()
-    {
-        return _syntaxTrees
-            .SelectMany(c => c.Definitions)
-            .OfType<BoundTraitDefinitionNode>();
+        _topLevelNodes = syntaxTrees.SelectMany(x => x.TopLevelNodes).ToList();
     }
 }