71 lines
2.4 KiB
JavaScript
71 lines
2.4 KiB
JavaScript
"use strict";
|
|
|
|
Object.defineProperty(exports, "__esModule", {
|
|
value: true
|
|
});
|
|
exports.program = exports.expression = exports.statement = exports.statements = exports.smart = void 0;
|
|
|
|
var t = _interopRequireWildcard(require("@babel/types"));
|
|
|
|
function _getRequireWildcardCache() { if (typeof WeakMap !== "function") return null; var cache = new WeakMap(); _getRequireWildcardCache = function () { return cache; }; return cache; }
|
|
|
|
function _interopRequireWildcard(obj) { if (obj && obj.__esModule) { return obj; } if (obj === null || typeof obj !== "object" && typeof obj !== "function") { return { default: obj }; } var cache = _getRequireWildcardCache(); if (cache && cache.has(obj)) { return cache.get(obj); } var newObj = {}; var hasPropertyDescriptor = Object.defineProperty && Object.getOwnPropertyDescriptor; for (var key in obj) { if (Object.prototype.hasOwnProperty.call(obj, key)) { var desc = hasPropertyDescriptor ? Object.getOwnPropertyDescriptor(obj, key) : null; if (desc && (desc.get || desc.set)) { Object.defineProperty(newObj, key, desc); } else { newObj[key] = obj[key]; } } } newObj.default = obj; if (cache) { cache.set(obj, newObj); } return newObj; }
|
|
|
|
function makeStatementFormatter(fn) {
|
|
return {
|
|
code: str => `/* @babel/template */;\n${str}`,
|
|
validate: () => {},
|
|
unwrap: ast => {
|
|
return fn(ast.program.body.slice(1));
|
|
}
|
|
};
|
|
}
|
|
|
|
const smart = makeStatementFormatter(body => {
|
|
if (body.length > 1) {
|
|
return body;
|
|
} else {
|
|
return body[0];
|
|
}
|
|
});
|
|
exports.smart = smart;
|
|
const statements = makeStatementFormatter(body => body);
|
|
exports.statements = statements;
|
|
const statement = makeStatementFormatter(body => {
|
|
if (body.length === 0) {
|
|
throw new Error("Found nothing to return.");
|
|
}
|
|
|
|
if (body.length > 1) {
|
|
throw new Error("Found multiple statements but wanted one");
|
|
}
|
|
|
|
return body[0];
|
|
});
|
|
exports.statement = statement;
|
|
const expression = {
|
|
code: str => `(\n${str}\n)`,
|
|
validate: ast => {
|
|
if (ast.program.body.length > 1) {
|
|
throw new Error("Found multiple statements but wanted one");
|
|
}
|
|
|
|
if (expression.unwrap(ast).start === 0) {
|
|
throw new Error("Parse result included parens.");
|
|
}
|
|
},
|
|
unwrap: ({
|
|
program
|
|
}) => {
|
|
const [stmt] = program.body;
|
|
t.assertExpressionStatement(stmt);
|
|
return stmt.expression;
|
|
}
|
|
};
|
|
exports.expression = expression;
|
|
const program = {
|
|
code: str => str,
|
|
validate: () => {},
|
|
unwrap: ast => ast.program
|
|
};
|
|
exports.program = program; |