@marko/compiler/dist/babel-utils/loc.js

Marko template to JS compiler.

"use strict";exports.__esModule = true;exports.getEnd = getEnd;exports.getLoc = getLoc;exports.getLocRange = getLocRange;exports.getStart = getStart;exports.withLoc = withLoc;const LINE_INDEX_KEY = Symbol();

function getLoc(file, index) {
  return findLoc(getLineIndexes(file), 0, index);
}

function getLocRange(file, start, end) {
  const lineIndexes = getLineIndexes(file);
  const startLoc = findLoc(lineIndexes, 0, start);

  if (startLoc) {
    const endLoc =
    start === end ? startLoc : findLoc(lineIndexes, startLoc.line - 1, end);

    return {
      start: startLoc,
      end: endLoc
    };
  }
}

function withLoc(file, node, start, end) {
  node.loc = getLocRange(file, start, end);
  node.start = start;
  node.end = end;
  return node;
}

function getStart(file, node) {
  // Restore if merged: https://github.com/babel/babel/pull/16849
  // if (node.start != null) {
  //   return node.start;
  // }

  if (node.loc) {
    return locToIndex(file, node.loc.start);
  }

  return null;
}

function getEnd(file, node) {
  // Restore if merged: https://github.com/babel/babel/pull/16849
  // if (node.end != null) {
  //   return node.end;
  // }

  if (node.loc) {
    return locToIndex(file, node.loc.end);
  }

  return null;
}

function locToIndex(file, loc) {
  const { line, column } = loc;
  return line === 1 ? column : getLineIndexes(file)[line - 1] + column + 1;
}

function getLineIndexes(file) {
  let lineIndexes = file.metadata.marko[LINE_INDEX_KEY];

  if (!lineIndexes) {
    lineIndexes = [-1];
    for (let i = 0; i < file.code.length; i++) {
      if (file.code[i] === "\n") {
        lineIndexes.push(i);
      }
    }

    file.metadata.marko[LINE_INDEX_KEY] = lineIndexes;
  }

  return lineIndexes;
}

function findLoc(lineIndexes, startLine, index) {
  if (index <= 0) {
    return {
      line: 1,
      column: 0
    };
  }

  const endLine = lineIndexes.length - 1;
  let max = endLine;
  let line = startLine;

  while (line < max) {
    const mid = line + max >>> 1;
    if (lineIndexes[mid] < index) {
      line = mid + 1;
    } else {
      max = mid;
    }
  }

  let lineIndex = lineIndexes[line];
  if (lineIndex >= index) {
    lineIndex = lineIndexes[--line];
  }

  if (line === 0) {
    return {
      line: 1,
      column: index
    };
  }

  return {
    line: line + 1,
    column: index - lineIndex - 1
  };
}