Merge pull request #5 from NodeFactoryIo/hotfix/old-generated-single-line-comments

Fix problem when generating comments on edited contract

Author: mpetrunic

src/lib/contract/contract-comment.es6

@@ -1,4 +1,4 @@
-import { pad } from '../utils/string-utils';
+import {pad} from '../utils/string-utils';
 import CommentsGenerator from '../generators/comments-generator';
 
 // regexps
@@ -31,11 +31,11 @@ export class ContractComment {
       // extract old comments
       let oldCommentsParams = [];
       let oldCommentsMap = {};
-      let oldCommentPosition = line - 2;
+      let oldCommentPosition = line - 1;
       while (true) {
         let comment = this.contract.getLineAt(oldCommentPosition).trim();
         if (comment.startsWith(parameterCommentRegex)) {
-          oldCommentsParams.push({ line: oldCommentPosition, value: comment });
+          oldCommentsParams.push({line: oldCommentPosition, value: comment});
         } else if (comment.startsWith('//')) {
           oldCommentsMap[comment.match(generatedCommentRegex)[0]] = comment;
         } else if (!comment.startsWith('function')) {
@@ -58,16 +58,12 @@ export class ContractComment {
       }, {});
       // update params if changed
       if (newCommentsParams.length) {
-        for (let k in oldCommentsMap) {
-          const c = !k;
-          if (c in newCommentsMap) {
-            return true;
-          }
+        if (Object.keys(oldCommentsMap).length !== Object.keys(newCommentsMap).length) {
+          return true;
         }
         let firstCommentLine = oldCommentsParams
           .reduce((min, b) => Math.min(min, b.line), oldCommentsParams[0].line);
-        // remove old params comments and save additional information about
-        // params
+        // remove old params comments and save addi. information about params
         let savedComments = {};
         for (let oldComment of oldCommentsParams) {
           this.contract.removeLine(firstCommentLine);