awv3/plugins/sketcher/command/ast/module.js

⚡ AWV3 embedded CAD

'use strict';

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

var _getIterator2 = require('babel-runtime/core-js/get-iterator');

var _getIterator3 = _interopRequireDefault(_getIterator2);

var _toConsumableArray2 = require('babel-runtime/helpers/toConsumableArray');

var _toConsumableArray3 = _interopRequireDefault(_toConsumableArray2);

var _set = require('babel-runtime/core-js/set');

var _set2 = _interopRequireDefault(_set);

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 _ast = require('./ast');

var _ast2 = _interopRequireDefault(_ast);

var _namevariable = require('./namevariable');

var _namevariable2 = _interopRequireDefault(_namevariable);

var _variabledef = require('./variabledef');

var _variabledef2 = _interopRequireDefault(_variabledef);

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

var Module = function (_Ast) {
    (0, _inherits3.default)(Module, _Ast);

    function Module() {
        (0, _classCallCheck3.default)(this, Module);
        return (0, _possibleConstructorReturn3.default)(this, (Module.__proto__ || (0, _getPrototypeOf2.default)(Module)).apply(this, arguments));
    }

    (0, _createClass3.default)(Module, [{
        key: 'unparse',
        value: function unparse() {
            var variables = new _set2.default();
            var _iteratorNormalCompletion = true;
            var _didIteratorError = false;
            var _iteratorError = undefined;

            try {
                for (var _iterator = (0, _getIterator3.default)(this.walk()), _step; !(_iteratorNormalCompletion = (_step = _iterator.next()).done); _iteratorNormalCompletion = true) {
                    var node = _step.value;

                    if (node instanceof _namevariable2.default) variables.add(node);
                }
            } catch (err) {
                _didIteratorError = true;
                _iteratorError = err;
            } finally {
                try {
                    if (!_iteratorNormalCompletion && _iterator.return) {
                        _iterator.return();
                    }
                } finally {
                    if (_didIteratorError) {
                        throw _iteratorError;
                    }
                }
            }

            var stmts = variables.size !== 0 ? [new _variabledef2.default([].concat((0, _toConsumableArray3.default)(variables)))] : [];
            return stmts.concat(this.body).map(function (stmt) {
                return stmt.unparse();
            }).join(' ');
        }
    }]);
    return Module;
}(_ast2.default);

Module._fields = ['body'];
exports.default = Module;