atom-nuclide/pkg/nuclide-python/lib/outline.js

A unified developer experience for web and mobile development, built as a suite of features on top of Atom to provide hackability and the support of an active community.

Object.defineProperty(exports, '__esModule', {
  value: true
});

/*
 * Copyright (c) 2015-present, Facebook, Inc.
 * All rights reserved.
 *
 * This source code is licensed under the license found in the LICENSE file in
 * the root directory of this source tree.
 */

var _extends = Object.assign || function (target) { for (var i = 1; i < arguments.length; i++) { var source = arguments[i]; for (var key in source) { if (Object.prototype.hasOwnProperty.call(source, key)) { target[key] = source[key]; } } } return target; };

exports.itemsToOutline = itemsToOutline;

function _toConsumableArray(arr) { if (Array.isArray(arr)) { for (var i = 0, arr2 = Array(arr.length); i < arr.length; i++) arr2[i] = arr[i]; return arr2; } else { return Array.from(arr); } }

var _atom2;

function _atom() {
  return _atom2 = require('atom');
}

var _commonsNodeTokenizedText2;

function _commonsNodeTokenizedText() {
  return _commonsNodeTokenizedText2 = require('../../commons-node/tokenizedText');
}

function itemToOutlineTree(mode, item) {
  switch (item.kind) {
    case 'class':
      return classToOutlineTree('all', item);
    case 'function':
      return functionToOutlineTree(item);
    case 'statement':
      return statementToOutlineTree(mode, item);
  }
}

function classToOutlineTree(mode, item) {
  return _extends({
    tokenizedText: [(0, (_commonsNodeTokenizedText2 || _commonsNodeTokenizedText()).keyword)('class'), (0, (_commonsNodeTokenizedText2 || _commonsNodeTokenizedText()).whitespace)(' '), (0, (_commonsNodeTokenizedText2 || _commonsNodeTokenizedText()).method)(item.name)],
    representativeName: item.name,
    children: itemsToOutline(mode, item.children)
  }, itemToPositions(item));
}

function functionToOutlineTree(item) {
  return _extends({
    tokenizedText: [(0, (_commonsNodeTokenizedText2 || _commonsNodeTokenizedText()).keyword)('def'), (0, (_commonsNodeTokenizedText2 || _commonsNodeTokenizedText()).whitespace)(' '), (0, (_commonsNodeTokenizedText2 || _commonsNodeTokenizedText()).method)(item.name), (0, (_commonsNodeTokenizedText2 || _commonsNodeTokenizedText()).plain)('(')].concat(_toConsumableArray(argsToText(item.params || [])), [(0, (_commonsNodeTokenizedText2 || _commonsNodeTokenizedText()).plain)(')')]),
    representativeName: item.name,
    children: []
  }, itemToPositions(item));
}

function statementToOutlineTree(mode, item) {
  if (mode === 'none') {
    return null;
  }
  var name = item.name;
  // Only show initialization of constants, which according to python
  // style are all upper case.
  if (mode === 'constants' && name !== name.toUpperCase()) {
    return null;
  }

  return _extends({
    tokenizedText: [(0, (_commonsNodeTokenizedText2 || _commonsNodeTokenizedText()).plain)(name)],
    representativeName: name,
    children: []
  }, itemToPositions(item));
}

function argsToText(args) {
  var result = [];

  function startArg() {
    if (result.length > 0) {
      result.push((0, (_commonsNodeTokenizedText2 || _commonsNodeTokenizedText()).plain)(','));
      result.push((0, (_commonsNodeTokenizedText2 || _commonsNodeTokenizedText()).whitespace)(' '));
    }
  }
  args.forEach(function (arg) {
    startArg();
    if (arg.startsWith('**')) {
      result.push((0, (_commonsNodeTokenizedText2 || _commonsNodeTokenizedText()).plain)('**'));
      result.push((0, (_commonsNodeTokenizedText2 || _commonsNodeTokenizedText()).param)(arg.slice(2)));
    } else if (arg.startsWith('*')) {
      result.push((0, (_commonsNodeTokenizedText2 || _commonsNodeTokenizedText()).plain)('*'));
      result.push((0, (_commonsNodeTokenizedText2 || _commonsNodeTokenizedText()).param)(arg.slice(1)));
    } else {
      result.push((0, (_commonsNodeTokenizedText2 || _commonsNodeTokenizedText()).param)(arg));
    }
  });

  return result;
}

function itemToPositions(item) {
  var start = item.start;
  var end = item.end;

  return {
    startPosition: new (_atom2 || _atom()).Point(start.line - 1, start.column),
    // Outline's endPosition is inclusive, while Jedi's is exclusive.
    // By decrementing the end column, we avoid situations where
    // two items are highlighted at once. End column may end up as -1,
    // which still has the intended effect.
    endPosition: new (_atom2 || _atom()).Point(end.line - 1, end.column - 1)
  };
}

function itemsToOutline(mode, items) {
  if (!items || items.length === 0) {
    return [];
  }
  var result = [];
  items.map(function (i) {
    return itemToOutlineTree(mode, i);
  }).forEach(function (tree) {
    if (tree) {
      result.push(tree);
    }
  });
  return result;
}