blob: 398e86fc66331ab120df0732e3ec89e3c4ca1dec [file] [log] [blame]
function _typeof(obj) { if (typeof Symbol === "function" && typeof Symbol.iterator === "symbol") { _typeof = function _typeof(obj) { return typeof obj; }; } else { _typeof = function _typeof(obj) { return obj && typeof Symbol === "function" && obj.constructor === Symbol && obj !== Symbol.prototype ? "symbol" : typeof obj; }; } return _typeof(obj); }
function _sliceIterator(arr, i) { var _arr = []; var _n = true; var _d = false; var _e = undefined; try { for (var _i = arr[Symbol.iterator](), _s; !(_n = (_s = _i.next()).done); _n = true) { _arr.push(_s.value); if (i && _arr.length === i) break; } } catch (err) { _d = true; _e = err; } finally { try { if (!_n && _i["return"] != null) _i["return"](); } finally { if (_d) throw _e; } } return _arr; }
function _slicedToArray(arr, i) { if (Array.isArray(arr)) { return arr; } else if (Symbol.iterator in Object(arr)) { return _sliceIterator(arr, i); } else { throw new TypeError("Invalid attempt to destructure non-iterable instance"); } }
import { isAnonymous, isInstruction } from "@webassemblyjs/ast";
import Long from "@xtuc/long";
var compact = false;
var space = " ";
var quote = function quote(str) {
return "\"".concat(str, "\"");
};
function indent(nb) {
return Array(nb).fill(space + space).join("");
} // TODO(sven): allow arbitrary ast nodes
export function print(n) {
if (n.type === "Program") {
return printProgram(n, 0);
} else {
throw new Error("Unsupported node in print of type: " + String(n.type));
}
}
function printProgram(n, depth) {
return n.body.reduce(function (acc, child) {
if (child.type === "Module") {
acc += printModule(child, depth + 1);
}
if (child.type === "Func") {
acc += printFunc(child, depth + 1);
}
if (child.type === "BlockComment") {
acc += printBlockComment(child);
}
if (child.type === "LeadingComment") {
acc += printLeadingComment(child);
}
if (compact === false) {
acc += "\n";
}
return acc;
}, "");
}
function printTypeInstruction(n) {
var out = "";
out += "(";
out += "type";
out += space;
if (n.id != null) {
out += printIndex(n.id);
out += space;
}
out += "(";
out += "func";
n.functype.params.forEach(function (param) {
out += space;
out += "(";
out += "param";
out += space;
out += printFuncParam(param);
out += ")";
});
n.functype.results.forEach(function (result) {
out += space;
out += "(";
out += "result";
out += space;
out += result;
out += ")";
});
out += ")"; // func
out += ")";
return out;
}
function printModule(n, depth) {
var out = "(";
out += "module";
if (typeof n.id === "string") {
out += space;
out += n.id;
}
if (compact === false) {
out += "\n";
} else {
out += space;
}
n.fields.forEach(function (field) {
if (compact === false) {
out += indent(depth);
}
switch (field.type) {
case "Func":
{
out += printFunc(field, depth + 1);
break;
}
case "TypeInstruction":
{
out += printTypeInstruction(field);
break;
}
case "Table":
{
out += printTable(field);
break;
}
case "Global":
{
out += printGlobal(field, depth + 1);
break;
}
case "ModuleExport":
{
out += printModuleExport(field);
break;
}
case "ModuleImport":
{
out += printModuleImport(field);
break;
}
case "Memory":
{
out += printMemory(field);
break;
}
case "BlockComment":
{
out += printBlockComment(field);
break;
}
case "LeadingComment":
{
out += printLeadingComment(field);
break;
}
case "Start":
{
out += printStart(field);
break;
}
case "Elem":
{
out += printElem(field, depth);
break;
}
case "Data":
{
out += printData(field, depth);
break;
}
default:
throw new Error("Unsupported node in printModule: " + String(field.type));
}
if (compact === false) {
out += "\n";
}
});
out += ")";
return out;
}
function printData(n, depth) {
var out = "";
out += "(";
out += "data";
out += space;
out += printIndex(n.memoryIndex);
out += space;
out += printInstruction(n.offset, depth);
out += space;
out += '"';
n.init.values.forEach(function (byte) {
// Avoid non-displayable characters
if (byte <= 31 || byte == 34 || byte == 92 || byte >= 127) {
out += "\\";
out += ("00" + byte.toString(16)).substr(-2);
} else if (byte > 255) {
throw new Error("Unsupported byte in data segment: " + byte);
} else {
out += String.fromCharCode(byte);
}
});
out += '"';
out += ")";
return out;
}
function printElem(n, depth) {
var out = "";
out += "(";
out += "elem";
out += space;
out += printIndex(n.table);
var _n$offset = _slicedToArray(n.offset, 1),
firstOffset = _n$offset[0];
out += space;
out += "(";
out += "offset";
out += space;
out += printInstruction(firstOffset, depth);
out += ")";
n.funcs.forEach(function (func) {
out += space;
out += printIndex(func);
});
out += ")";
return out;
}
function printStart(n) {
var out = "";
out += "(";
out += "start";
out += space;
out += printIndex(n.index);
out += ")";
return out;
}
function printLeadingComment(n) {
// Don't print leading comments in compact mode
if (compact === true) {
return "";
}
var out = "";
out += ";;";
out += n.value;
out += "\n";
return out;
}
function printBlockComment(n) {
// Don't print block comments in compact mode
if (compact === true) {
return "";
}
var out = "";
out += "(;";
out += n.value;
out += ";)";
out += "\n";
return out;
}
function printSignature(n) {
var out = "";
n.params.forEach(function (param) {
out += space;
out += "(";
out += "param";
out += space;
out += printFuncParam(param);
out += ")";
});
n.results.forEach(function (result) {
out += space;
out += "(";
out += "result";
out += space;
out += result;
out += ")";
});
return out;
}
function printModuleImportDescr(n) {
var out = "";
if (n.type === "FuncImportDescr") {
out += "(";
out += "func";
if (isAnonymous(n.id) === false) {
out += space;
out += printIdentifier(n.id);
}
out += printSignature(n.signature);
out += ")";
}
if (n.type === "GlobalType") {
out += "(";
out += "global";
out += space;
out += printGlobalType(n);
out += ")";
}
if (n.type === "Table") {
out += printTable(n);
}
return out;
}
function printModuleImport(n) {
var out = "";
out += "(";
out += "import";
out += space;
out += quote(n.module);
out += space;
out += quote(n.name);
out += space;
out += printModuleImportDescr(n.descr);
out += ")";
return out;
}
function printGlobalType(n) {
var out = "";
if (n.mutability === "var") {
out += "(";
out += "mut";
out += space;
out += n.valtype;
out += ")";
} else {
out += n.valtype;
}
return out;
}
function printGlobal(n, depth) {
var out = "";
out += "(";
out += "global";
out += space;
if (n.name != null && isAnonymous(n.name) === false) {
out += printIdentifier(n.name);
out += space;
}
out += printGlobalType(n.globalType);
out += space;
n.init.forEach(function (i) {
out += printInstruction(i, depth + 1);
});
out += ")";
return out;
}
function printTable(n) {
var out = "";
out += "(";
out += "table";
out += space;
if (n.name != null && isAnonymous(n.name) === false) {
out += printIdentifier(n.name);
out += space;
}
out += printLimit(n.limits);
out += space;
out += n.elementType;
out += ")";
return out;
}
function printFuncParam(n) {
var out = "";
if (typeof n.id === "string") {
out += "$" + n.id;
out += space;
}
out += n.valtype;
return out;
}
function printFunc(n, depth) {
var out = "";
out += "(";
out += "func";
if (n.name != null) {
if (n.name.type === "Identifier" && isAnonymous(n.name) === false) {
out += space;
out += printIdentifier(n.name);
}
}
if (n.signature.type === "Signature") {
out += printSignature(n.signature);
} else {
var index = n.signature;
out += space;
out += "(";
out += "type";
out += space;
out += printIndex(index);
out += ")";
}
if (n.body.length > 0) {
// func is empty since we ignore the default end instruction
if (n.body.length === 1 && n.body[0].id === "end") {
out += ")";
return out;
}
if (compact === false) {
out += "\n";
}
n.body.forEach(function (i) {
if (i.id !== "end") {
out += indent(depth);
out += printInstruction(i, depth);
if (compact === false) {
out += "\n";
}
}
});
out += indent(depth - 1) + ")";
} else {
out += ")";
}
return out;
}
function printInstruction(n, depth) {
switch (n.type) {
case "Instr":
// $FlowIgnore
return printGenericInstruction(n, depth + 1);
case "BlockInstruction":
// $FlowIgnore
return printBlockInstruction(n, depth + 1);
case "IfInstruction":
// $FlowIgnore
return printIfInstruction(n, depth + 1);
case "CallInstruction":
// $FlowIgnore
return printCallInstruction(n, depth + 1);
case "CallIndirectInstruction":
// $FlowIgnore
return printCallIndirectIntruction(n, depth + 1);
case "LoopInstruction":
// $FlowIgnore
return printLoopInstruction(n, depth + 1);
default:
throw new Error("Unsupported instruction: " + JSON.stringify(n.type));
}
}
function printCallIndirectIntruction(n, depth) {
var out = "";
out += "(";
out += "call_indirect";
if (n.signature.type === "Signature") {
out += printSignature(n.signature);
} else if (n.signature.type === "Identifier") {
out += space;
out += "(";
out += "type";
out += space;
out += printIdentifier(n.signature);
out += ")";
} else {
throw new Error("CallIndirectInstruction: unsupported signature " + JSON.stringify(n.signature.type));
}
out += space;
if (n.intrs != null) {
// $FlowIgnore
n.intrs.forEach(function (i, index) {
// $FlowIgnore
out += printInstruction(i, depth + 1); // $FlowIgnore
if (index !== n.intrs.length - 1) {
out += space;
}
});
}
out += ")";
return out;
}
function printLoopInstruction(n, depth) {
var out = "";
out += "(";
out += "loop";
if (n.label != null && isAnonymous(n.label) === false) {
out += space;
out += printIdentifier(n.label);
}
if (typeof n.resulttype === "string") {
out += space;
out += "(";
out += "result";
out += space;
out += n.resulttype;
out += ")";
}
if (n.instr.length > 0) {
n.instr.forEach(function (e) {
if (compact === false) {
out += "\n";
}
out += indent(depth);
out += printInstruction(e, depth + 1);
});
if (compact === false) {
out += "\n";
out += indent(depth - 1);
}
}
out += ")";
return out;
}
function printCallInstruction(n, depth) {
var out = "";
out += "(";
out += "call";
out += space;
out += printIndex(n.index);
if (_typeof(n.instrArgs) === "object") {
// $FlowIgnore
n.instrArgs.forEach(function (arg) {
out += space;
out += printFuncInstructionArg(arg, depth + 1);
});
}
out += ")";
return out;
}
function printIfInstruction(n, depth) {
var out = "";
out += "(";
out += "if";
if (n.testLabel != null && isAnonymous(n.testLabel) === false) {
out += space;
out += printIdentifier(n.testLabel);
}
if (typeof n.result === "string") {
out += space;
out += "(";
out += "result";
out += space;
out += n.result;
out += ")";
}
if (n.test.length > 0) {
out += space;
n.test.forEach(function (i) {
out += printInstruction(i, depth + 1);
});
}
if (n.consequent.length > 0) {
if (compact === false) {
out += "\n";
}
out += indent(depth);
out += "(";
out += "then";
depth++;
n.consequent.forEach(function (i) {
if (compact === false) {
out += "\n";
}
out += indent(depth);
out += printInstruction(i, depth + 1);
});
depth--;
if (compact === false) {
out += "\n";
out += indent(depth);
}
out += ")";
} else {
if (compact === false) {
out += "\n";
out += indent(depth);
}
out += "(";
out += "then";
out += ")";
}
if (n.alternate.length > 0) {
if (compact === false) {
out += "\n";
}
out += indent(depth);
out += "(";
out += "else";
depth++;
n.alternate.forEach(function (i) {
if (compact === false) {
out += "\n";
}
out += indent(depth);
out += printInstruction(i, depth + 1);
});
depth--;
if (compact === false) {
out += "\n";
out += indent(depth);
}
out += ")";
} else {
if (compact === false) {
out += "\n";
out += indent(depth);
}
out += "(";
out += "else";
out += ")";
}
if (compact === false) {
out += "\n";
out += indent(depth - 1);
}
out += ")";
return out;
}
function printBlockInstruction(n, depth) {
var out = "";
out += "(";
out += "block";
if (n.label != null && isAnonymous(n.label) === false) {
out += space;
out += printIdentifier(n.label);
}
if (typeof n.result === "string") {
out += space;
out += "(";
out += "result";
out += space;
out += n.result;
out += ")";
}
if (n.instr.length > 0) {
n.instr.forEach(function (i) {
if (compact === false) {
out += "\n";
}
out += indent(depth);
out += printInstruction(i, depth + 1);
});
if (compact === false) {
out += "\n";
}
out += indent(depth - 1);
out += ")";
} else {
out += ")";
}
return out;
}
function printGenericInstruction(n, depth) {
var out = "";
out += "(";
if (typeof n.object === "string") {
out += n.object;
out += ".";
}
out += n.id;
n.args.forEach(function (arg) {
out += space;
out += printFuncInstructionArg(arg, depth + 1);
});
out += ")";
return out;
}
function printLongNumberLiteral(n) {
if (typeof n.raw === "string") {
return n.raw;
}
var _n$value = n.value,
low = _n$value.low,
high = _n$value.high;
var v = new Long(low, high);
return v.toString();
}
function printFloatLiteral(n) {
if (typeof n.raw === "string") {
return n.raw;
}
return String(n.value);
}
function printFuncInstructionArg(n, depth) {
var out = "";
if (n.type === "NumberLiteral") {
out += printNumberLiteral(n);
}
if (n.type === "LongNumberLiteral") {
out += printLongNumberLiteral(n);
}
if (n.type === "Identifier" && isAnonymous(n) === false) {
out += printIdentifier(n);
}
if (n.type === "ValtypeLiteral") {
out += n.name;
}
if (n.type === "FloatLiteral") {
out += printFloatLiteral(n);
}
if (isInstruction(n)) {
out += printInstruction(n, depth + 1);
}
return out;
}
function printNumberLiteral(n) {
if (typeof n.raw === "string") {
return n.raw;
}
return String(n.value);
}
function printModuleExport(n) {
var out = "";
out += "(";
out += "export";
out += space;
out += quote(n.name);
if (n.descr.exportType === "Func") {
out += space;
out += "(";
out += "func";
out += space;
out += printIndex(n.descr.id);
out += ")";
} else if (n.descr.exportType === "Global") {
out += space;
out += "(";
out += "global";
out += space;
out += printIndex(n.descr.id);
out += ")";
} else if (n.descr.exportType === "Memory" || n.descr.exportType === "Mem") {
out += space;
out += "(";
out += "memory";
out += space;
out += printIndex(n.descr.id);
out += ")";
} else if (n.descr.exportType === "Table") {
out += space;
out += "(";
out += "table";
out += space;
out += printIndex(n.descr.id);
out += ")";
} else {
throw new Error("printModuleExport: unknown type: " + n.descr.exportType);
}
out += ")";
return out;
}
function printIdentifier(n) {
return "$" + n.value;
}
function printIndex(n) {
if (n.type === "Identifier") {
return printIdentifier(n);
} else if (n.type === "NumberLiteral") {
return printNumberLiteral(n);
} else {
throw new Error("Unsupported index: " + n.type);
}
}
function printMemory(n) {
var out = "";
out += "(";
out += "memory";
if (n.id != null) {
out += space;
out += printIndex(n.id);
out += space;
}
out += printLimit(n.limits);
out += ")";
return out;
}
function printLimit(n) {
var out = "";
out += n.min + "";
if (n.max != null) {
out += space;
out += String(n.max);
}
return out;
}