Handle comments in a multiline definition

CHANGELOG.md

@@ -4,6 +4,7 @@
 
 -   Added doxygen format support ([#120](https://github.com/NilsJPWerner/autoDocstring/issues/120)) (@daluar @PhilipNelson5)
 -   Disable HTML-escaping behavior globally ([#235](https://github.com/NilsJPWerner/autoDocstring/issues/235)) (@PhilipNelson5)
+-   Handle comments in multiline definitions ([#252](https://github.com/NilsJPWerner/autoDocstring/issues/252)) (@GabrielSchoenweiler @PhilipNelson5)
 
 [All Changes](https://github.com/NilsJPWerner/autoDocstring/compare/v0.6.1...master)
 

src/parse/tokenize_definition.ts

@@ -7,7 +7,7 @@ export function tokenizeDefinition(functionDefinition: string): string[] {
         return [];
     }
 
-    const tokens = tokenizeParameterString(match[1]);
+    const tokens = tokenizeParameterString(stripComments(match[1]));
 
     if (match[2] != undefined) {
         tokens.push(match[2]);
@@ -16,6 +16,22 @@ export function tokenizeDefinition(functionDefinition: string): string[] {
     return tokens;
 }
 
+function stripComments(parameterString: string): string {
+    let cleanString = "";
+    let position = 0;
+
+    while (position < parameterString.length) {
+        // When a comment is encountered, skip ahead to the end of the line
+        if (parameterString[position] === "#") {
+            position = parameterString.indexOf("\n", position);
+        }
+
+        cleanString += parameterString[position++];
+    }
+
+    return cleanString;
+}
+
 function tokenizeParameterString(parameterString: string): string[] {
     const stack: string[] = [];
     const parameters: string[] = [];

src/test/parse/tokenize_definition.spec.ts

@@ -113,4 +113,24 @@ describe("tokenizeDefinition()", () => {
 
         expect(result).to.have.ordered.members(["arg", "arg_2"]);
     });
+
+    it("should ignore comments in the middle of the function definition", () => {
+        const functionDefinition = `def abc_c(
+    arg, # comment
+    arg_2, # comment, comment
+    arg_3 # comment with special characters "'"({[]})
+    ):`;
+        const result = tokenizeDefinition(functionDefinition);
+
+        expect(result).to.have.ordered.members(["arg", "arg_2", "arg_3"]);
+    });
+    it("should ignore comments in the middle of the class definition", () => {
+        const functionDefinition = `class abc_c(
+    arg, # comment,
+    arg_2 # comment with special characters "'"({[]})
+    ):`;
+        const result = tokenizeDefinition(functionDefinition);
+
+        expect(result).to.have.ordered.members(["arg", "arg_2"]);
+    });
 });