Add detection of variables for symbol search

loganch · Apr 27, 2017 · b62f2d0 · b62f2d0
1 parent 5824e8f
commit b62f2d0
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/out/src/ai_symbols.js b/out/src/ai_symbols.js
@@ -4,24 +4,34 @@ var { languages, SymbolInformation, SymbolKind,
     Location, Position } = require('vscode')
 
 const _funcPattern = /Func\s(.+)\(/
+const _varPattern = /(\$\w+)/
 
 module.exports = languages.registerDocumentSymbolProvider(
     { language: 'autoit', scheme: 'file' },
     {
         provideDocumentSymbols(doc, token) {
 
             let result = []
+            let found = []
             let funcName
+            let varName
 
             // Get the number of lines in the document to loop through
             const lineCount = Math.min(doc.lineCount, 10000)
             for (let line = 0; line < lineCount; line++) {
                 const {text}  = doc.lineAt(line)
 
                 funcName = _funcPattern.exec(text)
-                if(funcName) {
+                if(funcName && found.indexOf(funcName[1]) === -1) {
                     result.push(new SymbolInformation(funcName[1], SymbolKind.Function, 
                     '', new Location(doc.uri, new Position(line, 0))))
+                    found.push(funcName[1])
+                }
+                varName = _varPattern.exec(text)
+                if (varName && found.indexOf(varName[1]) === -1) {
+                    result.push(new SymbolInformation(varName[1], SymbolKind.Variable,
+                    '', new Location(doc.uri, new Position(line, 0))))
+                    found.push(varName[1])
                 }
             }