36 lines
1.3 KiB
JavaScript
36 lines
1.3 KiB
JavaScript
|
import { traverse, getSectionMetadatas, shiftSection } from "@webassemblyjs/ast";
|
||
|
|
import { overrideBytesInBuffer } from "@webassemblyjs/helper-buffer";
|
||
|
|
export function removeSections(ast, uint8Buffer, section) {
|
||
|
|
var sectionMetadatas = getSectionMetadatas(ast, section);
|
||
|
|
|
||
|
|
if (sectionMetadatas.length === 0) {
|
||
|
|
throw new Error("Section metadata not found");
|
||
|
|
}
|
||
|
|
|
||
|
|
return sectionMetadatas.reverse().reduce(function (uint8Buffer, sectionMetadata) {
|
||
|
|
var startsIncludingId = sectionMetadata.startOffset - 1;
|
||
|
|
var ends = section === "start" ? sectionMetadata.size.loc.end.column + 1 : sectionMetadata.startOffset + sectionMetadata.size.value + 1;
|
||
|
|
var delta = -(ends - startsIncludingId);
|
||
|
|
/**
|
||
|
|
* update AST
|
||
|
|
*/
|
||
|
|
// Once we hit our section every that is after needs to be shifted by the delta
|
||
|
|
|
||
|
|
var encounteredSection = false;
|
||
|
|
traverse(ast, {
|
||
|
|
SectionMetadata: function SectionMetadata(path) {
|
||
|
|
if (path.node.section === section) {
|
||
|
|
encounteredSection = true;
|
||
|
|
return path.remove();
|
||
|
|
}
|
||
|
|
|
||
|
|
if (encounteredSection === true) {
|
||
|
|
shiftSection(ast, path.node, delta);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}); // replacement is nothing
|
||
|
|
|
||
|
|
var replacement = [];
|
||
|
|
return overrideBytesInBuffer(uint8Buffer, startsIncludingId, ends, replacement);
|
||
|
|
}, uint8Buffer);
|
||
|
|
}
|