Add config option to detect symbols

Author: krvajal

CHANGELOG.md

@@ -66,3 +66,14 @@ and this project adheres to [Semantic Versioning](http://semver.org/).
 ### Fixed
 
 * Fixed some highlighting issues by @pedro-ricardo
+
+## [0.6.3] - 2018-01-27
+
+### Added
+
+* Configuration option to set types of symbols shown
+
+  ### Fixed
+
+* Implementation of the symbol provider now returns a promise (Fixes #21)
+* Symbol provider now ignores case when searching for subroutines

src/features/document-symbol-provider.ts

@@ -4,6 +4,8 @@ import * as vscode from "vscode";
 import { getDeclaredFunctions, getDeclaredSubroutines } from "../lib/functions";
 import { getDeclaredVars } from "../lib/variables";
 
+type SymbolType = "subroutine" | "function" | "variable";
+
 export class FortranDocumentSymbolProvider
   implements vscode.DocumentSymbolProvider {
   vars: Array<vscode.SymbolInformation>;
@@ -18,12 +20,33 @@ export class FortranDocumentSymbolProvider
       token.onCancellationRequested(e => {
         reject();
       });
-      this.updateFunctionDefinitions(document);
-      this.updateSubroutineDefinitions(document);
-      this.updateVariablesDefiniton(document);
-      resolve([...this.functions, ...this.subroutines, ...this.vars]);
+      const symbolTypes = this.getSymbolTypes();
+      const documentSymbols = symbolTypes.reduce<vscode.SymbolInformation[]>(
+        (symbols, type: SymbolType) => {
+          return [...symbols, ...this.getSymbolsOfType(type, document)];
+        },
+        []
+      );
+
+      resolve(documentSymbols);
     });
   }
+  getSymbolsOfType(
+    type: "subroutine" | "function" | "variable",
+    document: TextDocument
+  ) {
+    switch (type) {
+      case "subroutine":
+        this.updateSubroutineDefinitions(document);
+        return this.subroutines;
+      case "function":
+        this.updateFunctionDefinitions(document);
+        return this.functions;
+      case "variable":
+        this.updateVariablesDefiniton(document);
+        return this.vars;
+    }
+  }
 
   private updateFunctionDefinitions(document: TextDocument) {
     this.functions = getDeclaredFunctions(document).map(fun => {
@@ -61,4 +84,12 @@ export class FortranDocumentSymbolProvider
       );
     });
   }
+  getSymbolTypes() {
+    let config = vscode.workspace.getConfiguration("fortran");
+    const symbolTypes = config.get<SymbolType[]>("symbols", [
+      "subroutine",
+      "function"
+    ]);
+    return symbolTypes;
+  }
 }