|  | 
| 1 |  | -import {pad} from "../utils/string-utils"; | 
| 2 |  | -import CommentsGenerator from "../generators/comments-generator"; | 
|  | 1 | +import { pad } from '../utils/string-utils'; | 
|  | 2 | +import CommentsGenerator from '../generators/comments-generator'; | 
| 3 | 3 | 
 | 
| 4 | 4 | // regexps | 
| 5 | 5 | const generatedCommentRegex = new RegExp('/// @([a-zA-Z]*)\\b'); | 
| 6 | 6 | const parameterCommentRegex = '/// @param'; | 
| 7 | 7 | 
 | 
| 8 | 8 | export class ContractComment { | 
| 9 | 9 | 
 | 
| 10 |  | -    constructor(contract) { | 
| 11 |  | -        this.contract = contract; | 
| 12 |  | -        this.generator = new CommentsGenerator(); | 
| 13 |  | -    } | 
|  | 10 | +  constructor(contract) { | 
|  | 11 | +    this.contract = contract; | 
|  | 12 | +    this.generator = new CommentsGenerator(); | 
|  | 13 | +  } | 
| 14 | 14 | 
 | 
| 15 |  | -    insertComment(node) { | 
| 16 |  | -        const comment = this.generator.generate(node).trim(); | 
| 17 |  | -        if (!comment) return; | 
| 18 |  | -        let commentLines = comment.split('\n'); | 
| 19 |  | -        if (this.updateComment(commentLines, node.loc)) return; | 
| 20 |  | -        commentLines = pad( | 
| 21 |  | -            node.loc.start.column, | 
| 22 |  | -            commentLines, | 
| 23 |  | -            this.isTab(this.contract.getOriginalLineAt(node.loc.start.line - 1)) | 
| 24 |  | -        ); | 
| 25 |  | -        this.contract.insertLinesBefore(commentLines, node.loc.start.line - 1); | 
| 26 |  | -    } | 
|  | 15 | +  insertComment(node) { | 
|  | 16 | +    const comment = this.generator.generate(node).trim(); | 
|  | 17 | +    if (!comment) return; | 
|  | 18 | +    let commentLines = comment.split('\n'); | 
|  | 19 | +    if (this.updateComment(commentLines, node.loc)) return; | 
|  | 20 | +    commentLines = pad( | 
|  | 21 | +      node.loc.start.column, | 
|  | 22 | +      commentLines, | 
|  | 23 | +      this.isTab(this.contract.getOriginalLineAt(node.loc.start.line - 1)) | 
|  | 24 | +    ); | 
|  | 25 | +    this.contract.insertLinesBefore(commentLines, node.loc.start.line - 1); | 
|  | 26 | +  } | 
| 27 | 27 | 
 | 
| 28 |  | -    updateComment(commentLines, location) { | 
| 29 |  | -        let line = location.start.line; | 
| 30 |  | -        if (this.hasComment(line)) { | 
| 31 |  | -            // extract old comments | 
| 32 |  | -            let oldCommentsParams = []; | 
| 33 |  | -            let oldCommentsMap = {}; | 
| 34 |  | -            let oldCommentPosition = line - 2; | 
| 35 |  | -            while (true) { | 
| 36 |  | -                let comment = this.contract.getLineAt(oldCommentPosition).trim(); | 
| 37 |  | -                if (comment.startsWith(parameterCommentRegex)) { | 
| 38 |  | -                    oldCommentsParams.push({line: oldCommentPosition, value: comment}) | 
| 39 |  | -                } else if (comment.startsWith('//')) { | 
| 40 |  | -                    oldCommentsMap[comment.match(generatedCommentRegex)[0]] = comment | 
| 41 |  | -                } else if (!comment.startsWith('function')) { | 
| 42 |  | -                    break; | 
| 43 |  | -                } | 
| 44 |  | -                oldCommentPosition--; | 
| 45 |  | -            } | 
| 46 |  | -            // check if old comment is generated comment | 
| 47 |  | -            if (this.isEmptyObject(oldCommentsMap)) { | 
| 48 |  | -                return true; | 
| 49 |  | -            } | 
| 50 |  | -            // extract new comments | 
| 51 |  | -            let newCommentsParams = []; | 
| 52 |  | -            let newCommentsMap = commentLines.reduce(function (map, obj) { | 
| 53 |  | -                let key = obj.match(generatedCommentRegex)[0]; | 
| 54 |  | -                if (key === parameterCommentRegex) { | 
| 55 |  | -                    newCommentsParams.push(obj); | 
| 56 |  | -                } else map[key] = obj; | 
| 57 |  | -                return map; | 
| 58 |  | -            }, {}); | 
| 59 |  | -            // update params if changed | 
| 60 |  | -            if (newCommentsParams.length) { | 
| 61 |  | -                for (let k in oldCommentsMap) { | 
| 62 |  | -                    if (!k in newCommentsMap) { | 
| 63 |  | -                        return true; | 
| 64 |  | -                    } | 
| 65 |  | -                } | 
| 66 |  | -                let firstCommentLine = oldCommentsParams | 
| 67 |  | -                    .reduce((min, b) => Math.min(min, b.line), oldCommentsParams[0].line); | 
| 68 |  | -                // remove old params comments and save additional information about params | 
| 69 |  | -                let savedComments = {}; | 
| 70 |  | -                for (let oldComment of oldCommentsParams) { | 
| 71 |  | -                    this.contract.removeLine(firstCommentLine); | 
| 72 |  | -                    // save old right part of comment | 
| 73 |  | -                    let c = oldComment.value.toString().trim().split(' '); | 
| 74 |  | -                    if (c.length > 3) { | 
| 75 |  | -                        savedComments[c[2]] = c.slice(3).join(' '); | 
| 76 |  | -                    } | 
| 77 |  | -                } | 
| 78 |  | -                // insert new params comments | 
| 79 |  | -                newCommentsParams = pad( | 
| 80 |  | -                    location.start.column, | 
| 81 |  | -                    newCommentsParams, | 
| 82 |  | -                    this.isTab(this.contract.getOriginalLineAt(location.start.line - 1)) | 
| 83 |  | -                ); | 
| 84 |  | -                for (let newComment of newCommentsParams.reverse()) { | 
| 85 |  | -                    let oldCommentParamName = newComment.trim().split(' ')[2]; | 
| 86 |  | -                    let savedComment = savedComments[oldCommentParamName]; | 
| 87 |  | -                    if (typeof savedComment !== "undefined") { | 
| 88 |  | -                        newComment = newComment + " " + savedComment; | 
| 89 |  | -                    } | 
| 90 |  | -                    this.contract.insertLinesBeforeWithoutCalculatingAndAddingOffset(newComment.split(), firstCommentLine); | 
| 91 |  | -                } | 
| 92 |  | -                this.contract.addOffset(firstCommentLine, newCommentsParams.length - oldCommentsParams.length); | 
| 93 |  | -                return true; | 
| 94 |  | -            } | 
|  | 28 | +  updateComment(commentLines, location) { | 
|  | 29 | +    let line = location.start.line; | 
|  | 30 | +    if (this.hasComment(line)) { | 
|  | 31 | +      // extract old comments | 
|  | 32 | +      let oldCommentsParams = []; | 
|  | 33 | +      let oldCommentsMap = {}; | 
|  | 34 | +      let oldCommentPosition = line - 2; | 
|  | 35 | +      while (true) { | 
|  | 36 | +        let comment = this.contract.getLineAt(oldCommentPosition).trim(); | 
|  | 37 | +        if (comment.startsWith(parameterCommentRegex)) { | 
|  | 38 | +          oldCommentsParams.push({ line: oldCommentPosition, value: comment }); | 
|  | 39 | +        } else if (comment.startsWith('//')) { | 
|  | 40 | +          oldCommentsMap[comment.match(generatedCommentRegex)[0]] = comment; | 
|  | 41 | +        } else if (!comment.startsWith('function')) { | 
|  | 42 | +          break; | 
|  | 43 | +        } | 
|  | 44 | +        oldCommentPosition--; | 
|  | 45 | +      } | 
|  | 46 | +      // check if old comment is generated comment | 
|  | 47 | +      if (this.isEmptyObject(oldCommentsMap)) { | 
|  | 48 | +        return true; | 
|  | 49 | +      } | 
|  | 50 | +      // extract new comments | 
|  | 51 | +      let newCommentsParams = []; | 
|  | 52 | +      let newCommentsMap = commentLines.reduce(function(map, obj) { | 
|  | 53 | +        let key = obj.match(generatedCommentRegex)[0]; | 
|  | 54 | +        if (key === parameterCommentRegex) { | 
|  | 55 | +          newCommentsParams.push(obj); | 
|  | 56 | +        } else map[key] = obj; | 
|  | 57 | +        return map; | 
|  | 58 | +      }, {}); | 
|  | 59 | +      // update params if changed | 
|  | 60 | +      if (newCommentsParams.length) { | 
|  | 61 | +        for (let k in oldCommentsMap) { | 
|  | 62 | +          const c = !k; | 
|  | 63 | +          if (c in newCommentsMap) { | 
| 95 | 64 |             return true; | 
|  | 65 | +          } | 
| 96 | 66 |         } | 
| 97 |  | -        return false; | 
| 98 |  | -    } | 
| 99 |  | - | 
| 100 |  | -    hasComment(line) { | 
| 101 |  | -        let counter = 1; | 
| 102 |  | -        while (true) { | 
| 103 |  | -            counter++; | 
| 104 |  | -            let lineText = this.contract.getOriginalLineAt(line - counter); | 
| 105 |  | -            if (lineText.trim().startsWith('function')) { | 
| 106 |  | -                lineText = this.contract.getOriginalLineAt(line - counter - 1); | 
| 107 |  | -            } | 
| 108 |  | -            if (lineText === undefined) return false; | 
| 109 |  | -            lineText = lineText.trim(); | 
| 110 |  | -            if (lineText.startsWith('*') || lineText.startsWith('//')) return true; | 
| 111 |  | -            if (!lineText.replace(/\s/g, '').length) continue; | 
| 112 |  | -            return false; | 
|  | 67 | +        let firstCommentLine = oldCommentsParams | 
|  | 68 | +          .reduce((min, b) => Math.min(min, b.line), oldCommentsParams[0].line); | 
|  | 69 | +        // remove old params comments and save additional information about | 
|  | 70 | +        // params | 
|  | 71 | +        let savedComments = {}; | 
|  | 72 | +        for (let oldComment of oldCommentsParams) { | 
|  | 73 | +          this.contract.removeLine(firstCommentLine); | 
|  | 74 | +          // save old right part of comment | 
|  | 75 | +          let c = oldComment.value.toString().trim().split(' '); | 
|  | 76 | +          if (c.length > 3) { | 
|  | 77 | +            savedComments[c[2]] = c.slice(3).join(' '); | 
|  | 78 | +          } | 
| 113 | 79 |         } | 
| 114 |  | -    } | 
| 115 |  | - | 
| 116 |  | -    isEmptyObject(obj) { | 
| 117 |  | -        for (let name in obj) { | 
| 118 |  | -            return false; | 
|  | 80 | +        // insert new params comments | 
|  | 81 | +        newCommentsParams = pad( | 
|  | 82 | +          location.start.column, | 
|  | 83 | +          newCommentsParams, | 
|  | 84 | +          this.isTab(this.contract.getOriginalLineAt(location.start.line - 1)) | 
|  | 85 | +        ); | 
|  | 86 | +        for (let newComment of newCommentsParams.reverse()) { | 
|  | 87 | +          let oldCommentParamName = newComment.trim().split(' ')[2]; | 
|  | 88 | +          let savedComment = savedComments[oldCommentParamName]; | 
|  | 89 | +          if (typeof savedComment !== 'undefined') { | 
|  | 90 | +            newComment = newComment + ' ' + savedComment; | 
|  | 91 | +          } | 
|  | 92 | +          this.contract.insertLinesBeforeWithoutCalculatingAndAddingOffset(newComment.split(), firstCommentLine); | 
| 119 | 93 |         } | 
|  | 94 | +        this.contract.addOffset(firstCommentLine, newCommentsParams.length - oldCommentsParams.length); | 
| 120 | 95 |         return true; | 
|  | 96 | +      } | 
|  | 97 | +      return true; | 
| 121 | 98 |     } | 
|  | 99 | +    return false; | 
|  | 100 | +  } | 
| 122 | 101 | 
 | 
| 123 |  | -    isTab(originalLineAt) { | 
| 124 |  | -        return originalLineAt.startsWith('\t'); | 
|  | 102 | +  hasComment(line) { | 
|  | 103 | +    let counter = 1; | 
|  | 104 | +    while (true) { | 
|  | 105 | +      counter++; | 
|  | 106 | +      let lineText = this.contract.getOriginalLineAt(line - counter); | 
|  | 107 | +      if (lineText.trim().startsWith('function')) { | 
|  | 108 | +        lineText = this.contract.getOriginalLineAt(line - counter - 1); | 
|  | 109 | +      } | 
|  | 110 | +      if (lineText === undefined) return false; | 
|  | 111 | +      lineText = lineText.trim(); | 
|  | 112 | +      if (lineText.startsWith('*') || lineText.startsWith('//')) return true; | 
|  | 113 | +      if (!lineText.replace(/\s/g, '').length) continue; | 
|  | 114 | +      return false; | 
| 125 | 115 |     } | 
| 126 |  | -} | 
|  | 116 | +  } | 
| 127 | 117 | 
 | 
|  | 118 | +  isEmptyObject(obj) { | 
|  | 119 | +    for (let name in obj) { | 
|  | 120 | +      return false; | 
|  | 121 | +    } | 
|  | 122 | +    return true; | 
|  | 123 | +  } | 
|  | 124 | + | 
|  | 125 | +  isTab(originalLineAt) { | 
|  | 126 | +    return originalLineAt.startsWith('\t'); | 
|  | 127 | +  } | 
|  | 128 | +} | 
| 128 | 129 | 
 | 
| 129 | 130 | 
 | 
0 commit comments