Provide document symbols for psd1 file

Author: Kapil Borle

src/PowerShellEditorServices/Language/AstOperations.cs

@@ -209,14 +209,96 @@ static public IEnumerable<SymbolReference> FindSymbolsInDocument(Ast scriptAst,
 //            }
 //            else
             {
-                FindSymbolsVisitor findSymbolsVisitor = new FindSymbolsVisitor();
-                scriptAst.Visit(findSymbolsVisitor);
-                symbolReferences = findSymbolsVisitor.SymbolReferences;
+                // we don't have reliable access to the filename here
+                // so we employ the following heuristic to check if the
+                // contents are part of a psd1 file.
+                if (IsPowerShellDataFile(scriptAst))
+                {
+                    var findHashtableSymbolsVisitor = new FindHashtabeSymbolsVisitor();
+                    scriptAst.Visit(findHashtableSymbolsVisitor);
+                    symbolReferences = findHashtableSymbolsVisitor.SymbolReferences;
+                }
+                else
+                {
+                    FindSymbolsVisitor findSymbolsVisitor = new FindSymbolsVisitor();
+                    scriptAst.Visit(findSymbolsVisitor);
+                    symbolReferences = findSymbolsVisitor.SymbolReferences;
+                }
             }
 
             return symbolReferences;
         }
 
+        static private bool IsPowerShellDataFile(Ast ast)
+        {
+            //var node = new { Item = ast, Children = new List<dynamic>() };
+            //GenerateTree(node);
+            //return node.Children.Count == 1 && node.Children[0].Item is NamedBlockAst
+            //    && node.Children[0].Children.Count == 1 && node.Children[0].Children[0].Item is PipelineAst
+            //    && node.Children[0].Children[0].Children.Count == 1 && ;
+
+            return IsPowerShellDataFile(
+                new { Item = ast, Children = new List<dynamic>() },
+                new Type[] {
+                    typeof(ScriptBlockAst),
+                    typeof(NamedBlockAst),
+                    typeof(PipelineAst),
+                    typeof(CommandExpressionAst),
+                    typeof(HashtableAst) },
+                0);
+        }
+
+        static private bool IsPowerShellDataFile(dynamic node, Type[] levelAstMap, int level)
+        {
+            var levelAstTypeMatch = node.Item.GetType().Equals(levelAstMap[level]);
+            if (!levelAstTypeMatch)
+            {
+                return false;
+            }
+
+            if (level == levelAstMap.Length - 1)
+            {
+                return levelAstTypeMatch;
+            }
+
+            var astsFound = (node.Item as Ast).FindAll(a => a is Ast, false);
+            if (astsFound != null)
+            {
+                foreach (var astFound in astsFound)
+                {
+                    if (!astFound.Equals(node.Item) && node.Item.Equals(astFound.Parent))
+                    {
+                        if (IsPowerShellDataFile(new { Item = astFound, Children = new List<dynamic>() }, levelAstMap, level + 1))
+                        {
+                            return true;
+                        }
+                    }
+                }
+            }
+
+            return false;
+        }
+
+        static private void GenerateTree(dynamic node)
+        {
+            var astsFound = (node.Item as Ast).FindAll(a => a is Ast, false);
+            if (astsFound != null)
+            {
+                foreach (var astFound in astsFound)
+                {
+                    if (!astFound.Equals(node.Item) && node.Item.Equals(astFound.Parent))
+                    {
+                        node.Children.Add(new { Item = astFound, Children = new List<dynamic>() });
+                    }
+                }
+            }
+
+            foreach (var child in node.Children)
+            {
+                GenerateTree(child);
+            }
+        }
+
         /// <summary>
         /// Finds all files dot sourced in a script
         /// </summary>

src/PowerShellEditorServices/Language/FindSymbolsVisitor.cs

@@ -89,4 +89,48 @@ private bool IsAssignedAtScriptScope(VariableExpressionAst variableExpressionAst
             return false;
         }
     }
+
+    internal class FindHashtabeSymbolsVisitor : AstVisitor
+    {
+        public List<SymbolReference> SymbolReferences { get; private set; }
+
+        public FindHashtabeSymbolsVisitor()
+        {
+            SymbolReferences = new List<SymbolReference>();
+        }
+
+        public override AstVisitAction VisitHashtable(HashtableAst hashtableAst)
+        {
+            if (hashtableAst.KeyValuePairs == null)
+            {
+                return AstVisitAction.Continue;
+            }
+
+            foreach (var kvp in hashtableAst.KeyValuePairs)
+            {
+                var keyStrConstExprAst = kvp.Item1 as StringConstantExpressionAst;
+                if (keyStrConstExprAst != null)
+                {
+                    IScriptExtent nameExtent = new ScriptExtent()
+                    {
+                        Text = keyStrConstExprAst.Value,
+                        StartLineNumber = kvp.Item1.Extent.StartLineNumber,
+                        EndLineNumber = kvp.Item2.Extent.EndLineNumber,
+                        StartColumnNumber = kvp.Item1.Extent.StartColumnNumber,
+                        EndColumnNumber = kvp.Item2.Extent.EndColumnNumber
+                    };
+
+                    SymbolType symbolType = SymbolType.HashtableKey;
+
+                    this.SymbolReferences.Add(
+                        new SymbolReference(
+                            symbolType,
+                            nameExtent));
+
+                }
+            }
+
+            return AstVisitAction.Continue;
+        }
+    }
 }

src/PowerShellEditorServices/Language/SymbolType.cs

@@ -14,17 +14,17 @@ public enum SymbolType
         /// The symbol type is unknown
         /// </summary>
         Unknown = 0,
-        
+
         /// <summary>
         /// The symbol is a vairable
         /// </summary>
         Variable,
-        
+
         /// <summary>
         /// The symbol is a function
         /// </summary>
         Function,
-        
+
         /// <summary>
         /// The symbol is a parameter
         /// </summary>
@@ -39,6 +39,10 @@ public enum SymbolType
         /// The symbol is a workflow
         /// </summary>
         Workflow,
+
+        /// <summary>
+        /// The symbol is a hashtable key
+        /// </summary>
+        HashtableKey
     }
 }
-