cst/lib/elements/types/ClassBody.js

JavaScript CST Implementation

'use strict';

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

var _getPrototypeOf = require('babel-runtime/core-js/object/get-prototype-of');

var _getPrototypeOf2 = _interopRequireDefault(_getPrototypeOf);

var _classCallCheck2 = require('babel-runtime/helpers/classCallCheck');

var _classCallCheck3 = _interopRequireDefault(_classCallCheck2);

var _createClass2 = require('babel-runtime/helpers/createClass');

var _createClass3 = _interopRequireDefault(_createClass2);

var _possibleConstructorReturn2 = require('babel-runtime/helpers/possibleConstructorReturn');

var _possibleConstructorReturn3 = _interopRequireDefault(_possibleConstructorReturn2);

var _inherits2 = require('babel-runtime/helpers/inherits');

var _inherits3 = _interopRequireDefault(_inherits2);

var _Node2 = require('../Node');

var _Node3 = _interopRequireDefault(_Node2);

function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }

function isCode(_ref) {
    var currentElement = _ref.currentElement;

    var _ref2 = currentElement || {},
        isCode = _ref2.isCode;

    return isCode !== false;
}

var ClassBody = function (_Node) {
    (0, _inherits3.default)(ClassBody, _Node);

    function ClassBody(childNodes) {
        (0, _classCallCheck3.default)(this, ClassBody);
        return (0, _possibleConstructorReturn3.default)(this, (ClassBody.__proto__ || (0, _getPrototypeOf2.default)(ClassBody)).call(this, 'ClassBody', childNodes));
    }

    (0, _createClass3.default)(ClassBody, [{
        key: '_acceptChildren',
        value: function _acceptChildren(children) {
            var body = [];

            children.assertToken('Punctuator', '{');
            children.moveNext();
            children.skipNonCode();

            while (!children.isToken('Punctuator', '}')) {
                if (children.isToken('Punctuator', ';')) {
                    // For the class Test { x() {}; } case
                    children.passToken('Punctuator', ';');
                } else if (!isCode(children)) {
                    children.skipNonCode();
                } else {
                    children.assertOneOfNode(['ClassProperty', 'ClassMethod']);
                    if (children.isNode('ClassProperty')) {
                        body.push(children.passNode('ClassProperty'));
                    } else if (children.isNode('ClassMethod')) {
                        body.push(children.passNode('ClassMethod'));
                    }
                }
            }

            children.passToken('Punctuator', '}');
            children.assertEnd();

            this.body = body;
        }
    }]);
    return ClassBody;
}(_Node3.default);

exports.default = ClassBody;
//# sourceMappingURL=ClassBody.js.map