ts-budgie
Version:
Converts TypeScript code to Budgie.
65 lines • 3.43 kB
JavaScript
;
var __extends = (this && this.__extends) || (function () {
var extendStatics = function (d, b) {
extendStatics = Object.setPrototypeOf ||
({ __proto__: [] } instanceof Array && function (d, b) { d.__proto__ = b; }) ||
function (d, b) { for (var p in b) if (b.hasOwnProperty(p)) d[p] = b[p]; };
return extendStatics(d, b);
};
return function (d, b) {
extendStatics(d, b);
function __() { this.constructor = d; }
d.prototype = b === null ? Object.create(b) : (__.prototype = b.prototype, new __());
};
})();
var __spreadArrays = (this && this.__spreadArrays) || function () {
for (var s = 0, i = 0, il = arguments.length; i < il; i++) s += arguments[i].length;
for (var r = Array(s), k = 0, i = 0; i < il; i++)
for (var a = arguments[i], j = 0, jl = a.length; j < jl; j++, k++)
r[k] = a[j];
return r;
};
Object.defineProperty(exports, "__esModule", { value: true });
var budgie_1 = require("budgie");
var budgieLine_1 = require("../../output/budgieLine");
var transformation_1 = require("../../output/transformation");
var visitor_1 = require("../visitor");
var IfStatementVisitor = /** @class */ (function (_super) {
__extends(IfStatementVisitor, _super);
function IfStatementVisitor() {
return _super !== null && _super.apply(this, arguments) || this;
}
IfStatementVisitor.prototype.visit = function (node) {
var expression = this.router.recurseIntoValue(node.expression);
var elseStatement = node.elseStatement, thenStatement = node.thenStatement;
var thenBody = this.router.recurseIntoNode(thenStatement);
var transformations = __spreadArrays(thenBody);
if (elseStatement !== undefined) {
transformations.push.apply(transformations, this.replaceWithElseCommands(elseStatement, this.router.recurseIntoNode(elseStatement)));
}
return [
transformation_1.Transformation.fromNode(node, this.sourceFile, __spreadArrays([
new budgieLine_1.BudgieLine(budgie_1.CommandNames.IfStart, expression)
], transformations, [
new budgieLine_1.BudgieLine(budgie_1.CommandNames.IfEnd),
])),
];
};
IfStatementVisitor.prototype.replaceWithElseCommands = function (elseStatement, transformations) {
// If there are no commands, just end with an else command
if (transformations.length === 0) {
return [transformation_1.Transformation.fromNode(elseStatement, this.sourceFile, [new budgieLine_1.BudgieLine(budgie_1.CommandNames.ElseStart)])];
}
// 'else if' commands are still stored as IfStatement nodes under the parent's elseStatment
var starter = transformations[0].output[0];
if (starter.command === budgie_1.CommandNames.IfStart) {
transformations[0].output[0] = new (budgieLine_1.BudgieLine.bind.apply(budgieLine_1.BudgieLine, __spreadArrays([void 0, budgie_1.CommandNames.ElseIfStart], starter.args)))();
}
// Removes any now-unnecessary 'if end' command
transformations[0].output.pop();
return transformations;
};
return IfStatementVisitor;
}(visitor_1.NodeVisitor));
exports.IfStatementVisitor = IfStatementVisitor;
//# sourceMappingURL=ifStatementVisitor.js.map