Customize the number of lines to get

Fix #52

2 files changed, 3 insertions, 1 deletions

diff --git a/src/Config.ts b/src/Config.ts
index cce8f72..3a98a71 100644
--- a/src/Config.ts
+++ b/src/Config.ts
@@ -19,6 +19,7 @@ export class Config {
         values.paramTemplate = cfg.get<string>("paramTemplate", values.paramTemplate);
         values.tparamTemplate = cfg.get<string>("tparamTemplate", values.tparamTemplate);
         values.returnTemplate = cfg.get<string>("returnTemplate", values.returnTemplate);
+        values.linesToGet = cfg.get<number>("linesToGet", values.linesToGet);
 
         return values;
     }
@@ -39,4 +40,5 @@ export class Config {
     public paramTemplate: string = "@param {param} ";
     public tparamTemplate: string = "@tparam {param} ";
     public returnTemplate: string = "@return {type} ";
+    public linesToGet: number = 20;
 }
diff --git a/src/Lang/Cpp/CppParser.ts b/src/Lang/Cpp/CppParser.ts
index 9f8916a..c1a0add 100644
--- a/src/Lang/Cpp/CppParser.ts
+++ b/src/Lang/Cpp/CppParser.ts
@@ -274,7 +274,7 @@ export default class CppParser implements ICodeParser {
         logicalLine = nextLineTxt;
 
         // Get method end line
-        let linesToGet: number = 20;
+        let linesToGet: number = this.cfg.linesToGet;
         while (linesToGet-- > 0) { // Check for end of expression.
             nextLine = new Position(nextLine.line + 1, nextLine.character);
             nextLineTxt = this.activeEditor.document.lineAt(nextLine.line).text.trim();