Handle global statements

Author: ATrefzer

CodeParser/Parser/Parser.Phase1.cs

@@ -9,6 +9,9 @@ public partial class Parser
 {
     private readonly List<INamedTypeSymbol> _allNamedTypesInSolution = new();
 
+    private readonly Dictionary<IAssemblySymbol, List<GlobalStatementSyntax>> _globalStatementsByAssembly = new(SymbolEqualityComparer.Default);
+
+
     private async Task BuildHierarchy(Solution solution)
     {
         foreach (var project in solution.Projects)
@@ -40,6 +43,7 @@ private void BuildHierarchy(Compilation compilation)
         // Assembly has no source location.
         var assemblySymbol = compilation.Assembly;
         var assemblyElement = GetOrCreateCodeElement(assemblySymbol, CodeElementType.Assembly, null!, null!);
+        _globalStatementsByAssembly[assemblySymbol] = new List<GlobalStatementSyntax>();
 
         foreach (var syntaxTree in compilation.SyntaxTrees)
         {
@@ -146,6 +150,12 @@ private void ProcessNodeForHierarchy(SyntaxNode node, SemanticModel semanticMode
                 symbol = semanticModel.GetDeclaredSymbol(node) as IEventSymbol;
                 elementType = CodeElementType.Event;
                 break;
+
+            case GlobalStatementSyntax globalStatementSyntax:
+                var assemblySymbol = semanticModel.Compilation.Assembly;
+                _globalStatementsByAssembly[assemblySymbol].Add(globalStatementSyntax);
+                return; // We'll handle these collectively later
+                        
             // Add more cases as needed (e.g., for events, delegates, etc.)
         }
 

CodeParser/Parser/Parser.Phase2.cs

@@ -57,8 +57,57 @@ private void AnalyzeDependencies(Solution solution)
 
             ++loop;
         }
+
+        // Analyze global statements for each assembly
+        AnalyzeGlobalStatementsForAssembly(solution);
     }
 
+
+    private void AnalyzeGlobalStatementsForAssembly(Solution solution)
+    {
+        foreach (var statement in _globalStatementsByAssembly)
+        {
+            var assemblySymbol = statement.Key;
+            var globalStatements = statement.Value;
+            if (globalStatements.Count == 0)
+            {
+                continue;
+            }
+
+            // Find the existing assembly element
+            var symbolKey = GetSymbolKey(assemblySymbol);
+            var assemblyElement = _symbolKeyToElementMap[symbolKey];
+
+            // Create a dummy class for this assembly's global statements
+            var dummyClassId = Guid.NewGuid().ToString();
+            var dummyClassName = "GlobalStatements";
+            var dummyClassFullName = BuildSymbolName(assemblySymbol) + "." + dummyClassName;
+            var dummyClass = new CodeElement(dummyClassId, CodeElementType.Class, dummyClassName, dummyClassFullName,
+                assemblyElement);
+            _codeGraph.Nodes[dummyClassId] = dummyClass;
+            assemblyElement.Children.Add(dummyClass);
+
+            // Create a dummy method to contain global statements
+            var dummyMethodId = Guid.NewGuid().ToString();
+            var dummyMethodName = "Execute";
+            var dummyMethodFullName = $"{dummyClassName}.{dummyMethodName}";
+            var dummyMethod = new CodeElement(dummyMethodId, CodeElementType.Method, dummyMethodName,
+                dummyMethodFullName, dummyClass);
+            _codeGraph.Nodes[dummyMethodId] = dummyMethod;
+            dummyClass.Children.Add(dummyMethod);
+
+            // Analyze global statements within the context of the dummy method
+            foreach (var globalStatement in globalStatements)
+            {
+                var document = solution.GetDocument(globalStatement.SyntaxTree);
+                var semanticModel = document?.GetSemanticModelAsync().Result;
+                if (semanticModel != null)
+                {
+                    AnalyzeMethodBody(dummyMethod, globalStatement, semanticModel);
+                }
+            }
+        }
+    }
     private void AnalyzeAttributeDependencies(CodeElement element, ISymbol symbol)
     {
         foreach (var attributeData in symbol.GetAttributes())

CodeParser/Parser/Parser.cs

@@ -42,7 +42,7 @@ public async Task<CodeGraph> ParseSolution(string solutionPath)
 
         // Makes the cycle detection easier because I never get to the assembly as shared ancestor
         // for a nested dependency.
-        InsertGlobalNamesapceIfUsed();
+        InsertGlobalNamespaceIfUsed();
         return _codeGraph;
     }
 
@@ -57,10 +57,10 @@ private void Clear()
     ///     If any assembly uses the global namespace we add the global namespace to all assemblies.
     ///     For example a unit test assembly may have the autogenerated Main.
     /// </summary>
-    private void InsertGlobalNamesapceIfUsed()
+    private void InsertGlobalNamespaceIfUsed()
     {
         var global = "global";
-        var assemblies = _codeGraph.Nodes.Values.Where(n => n.Parent is null);
+        var assemblies = _codeGraph.Nodes.Values.Where(n => n.Parent is null).ToList();
         Debug.Assert(assemblies.All(a => a.ElementType == CodeElementType.Assembly));
         var isGlobalNsUsed = assemblies.Any(a => a.Children.Any(c => c.ElementType != CodeElementType.Namespace));