From bba53b05a85e4ccad2eca31124ea9d0b80b4e392 Mon Sep 17 00:00:00 2001
From: Christoph Schlosser <christoph@linux.com>
Date: Sat, 24 Feb 2018 10:45:06 +0100
Subject: Customize the number of lines to get

Fix #52
---
 src/Config.ts             | 2 ++
 src/Lang/Cpp/CppParser.ts | 2 +-
 2 files changed, 3 insertions(+), 1 deletion(-)

(limited to 'src')

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();
-- 
cgit v1.2.3