Added symbolProvider to give outline on rule names

yara/src/extension.ts

@@ -8,6 +8,7 @@ import { YaraDefinitionProvider } from './lib/definitionProvider';
 import { YaraReferenceProvider } from './lib/referenceProvider';
 import { YaraSnippetCompletionItemProvider } from './lib/snippetProvider';
 import { YaraHexStringHoverProvider } from './lib/hoverProvider';
+import { YaraDocumentSymbolProvider } from "./lib/symbolProvider";
 
 
 export function activate(context: vscode.ExtensionContext): void {
@@ -39,6 +40,9 @@ export function activate(context: vscode.ExtensionContext): void {
         }
     });
     context.subscriptions.push(configWatcher);
+    const symbolDisposable: vscode.Disposable = vscode.languages.registerDocumentSymbolProvider(YARA, new YaraDocumentSymbolProvider());
+    context.subscriptions.push(symbolDisposable);
+
     if (debug) { log('Registered configuration watcher'); }
 }
 

yara/src/lib/symbolProvider.ts

@@ -0,0 +1,25 @@
+'use strict';
+
+import vscode = require('vscode');
+
+export class YaraDocumentSymbolProvider implements vscode.DocumentSymbolProvider {
+    public provideDocumentSymbols(
+        document: vscode.TextDocument,
+        token: vscode.CancellationToken): Promise<vscode.DocumentSymbol[]> {
+        return new Promise((resolve, reject) => {
+            const symbols: vscode.DocumentSymbol[] = [];
+            for (let i = 0; i < document.lineCount; i++) {
+                const line = document.lineAt(i);
+                if (line.text.startsWith("rule")) {
+                    const name = line.text.split(" ")[1]
+                    const symbol = new vscode.DocumentSymbol(
+                        name, 'Rule',
+                        vscode.SymbolKind.Class,
+                        line.range, line.range)
+                    symbols.push(symbol)
+                }
+            }
+            resolve(symbols);
+        });
+    }
+}