lay-out/helpers/lines.js

project structure visualizer

var _this = this;

var settings = require('./settings.js');

var lineLevelIndentation = function (level) {
  return level.replace(/\t/g, '  ').length / settings.get('indentation_level');
};

/**
 * determine the indentation level of a given line (object).
 *
 * @param  {Object} line
 * @return {Number}
 */
exports.getLineLevel = function (line) {
  switch (_this.getLineType(line)) {
    case 'folder':
      line.level = line.initial.match(/^(\s*?)[^-\s\?]/)[1];
      return lineLevelIndentation(line.level);
      break;

    case 'file':
      line.level = line.initial.match(/^(\s+)-/)[1];
      return lineLevelIndentation(line.level);
      break;

    default:
      return 0;
      break;
  }
};

exports.getLineIndentation = function (line) {
  var num_indentation = line.level * settings.get('indentation_level');
  var indentation = 0;

  for (var i = num_indentation - 1; i >= 0; i--) {
    indentation++;
  }

  return indentation;
};

exports.getLineComment = function (line) {
  var regex;

  switch (_this.getLineType(line)) {
    case 'folder':
      regex = /^\s*?[^-\s]+\s?\?\s(.+)/;
      break;

    case 'file':
      regex = /^\s+-\s[^?\s]+\s*?\?\s?(?=[^:])(.+)/;
      break;
  }

  if (line.initial.match(/\s?(\?)\:$/)) {
    line.multiline_comment = true;
  }

  if (line.initial.match(regex) && line.initial.match(regex)[1]) {
    return line.initial.match(regex)[1];
  } else {
    return false;
  }
};

exports.getLineString = function (line) {
  switch (_this.getLineType(line)) {
    case 'folder':
      line = line.initial.match(/^\s*?(\/?[^?\s-][^?\s]*)/)[1];
      return line.replace(/\//g, '') + '/';
      break;

    case 'file':
      return line.initial.match(/\-\s([^?\s]+)/)[1];
      break;
  }
};

exports.fillLineObj = function (line) {
  line.string       = _this.getLineString(line);
  line.type         = _this.getLineType(line);
  line.level        = _this.getLineLevel(line);
  line.comment      = _this.getLineComment(line);
  line.indentation  = _this.getLineIndentation(line);

  return line;
};

exports.setLineObj = function (el, i) {
  return {
    initial: el,
    line: i + 1,
    indentation: 0
  };
};

exports.getLineType = function (line) {
  // element is a folder
  if (line.initial.match(/^\s*?([^?\s|-]+)/)) {
    return 'folder';

  // element is a file
  } else if (line.initial.match(/^\s+-/)) {
    return 'file';
  }
};

exports.getLineObj = function (el, i) {
  return _this.fillLineObj(_this.setLineObj(el, i));
};