| 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); |
| } |