solhint-community/lib/common/blank-line-counter.js

Solidity Code Linter

const { stopLine, lineOf } = require('./tokens')

class BlankLineCounter {
  constructor() {
    this.tokenLines = new Set()
  }

  countOfEmptyLinesBetween(start, end) {
    return this.countOfEmptyLinesBetweenTokens(stopLine(start), lineOf(end))
  }

  countOfEmptyLinesBetweenTokens(start, end) {
    let count = 0

    for (let i = start + 1; i < end; i += 1) {
      if (!this.tokenLines.has(i)) {
        count++
      }
    }

    return count
  }

  calcTokenLines(ctx) {
    if (this.tokenLines.size === 0) {
      ctx.parser._input.tokens.forEach((i) => this.addTokenLinesToMap(i))
    }
  }

  addTokenLinesToMap(token) {
    const HIDDEN = 1
    if (token.channel === HIDDEN) {
      const linesCount = token.text.split('\n').length
      for (let curLine = token.line; curLine < token.line + linesCount; curLine += 1) {
        this.tokenLines.add(curLine)
      }
    } else {
      this.tokenLines.add(token.line)
    }
  }
}

module.exports = BlankLineCounter