forked from cloudflare/html-rewriter-wasm
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
20 changed files
with
494 additions
and
5,961 deletions.
There are no files selected for viewing
File renamed without changes.
This file was deleted.
Oops, something went wrong.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,23 @@ | ||
| wasm-pack-build: | ||
| @echo "---> Building custom wasm-pack with asyncify enabled..." | ||
| cd ./wasm-pack; cargo build --release | ||
|
|
||
| wasm-pack: wasm-pack-build | ||
| @echo "---> Building WebAssembly with wasm-pack..." | ||
| ./wasm-pack/target/release/wasm-pack build --target web | ||
|
|
||
| # Wraps write/end with asyncify magic and adds this returns for chaining | ||
| # diff -uN pkg/html_rewriter.js pkg2/html_rewriter.js > html_rewriter.js.patch | ||
| patch: wasm-pack | ||
| @echo "---> Patching JavaScript glue code..." | ||
| patch -uN pkg/html_rewriter.js < html_rewriter.js.patch | ||
|
|
||
| dist: patch | ||
| @echo "---> Copying required files to root..." | ||
| cp pkg/html_rewriter.js . | ||
| cp pkg/html_rewriter_bg.wasm . | ||
| cp src/asyncify.js . | ||
| cp src/html_rewriter.d.ts . | ||
|
|
||
| test: | ||
| deno test ./test |
File renamed without changes.
File renamed without changes.
File renamed without changes.
File renamed without changes.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,106 @@ | ||
| import { TextEncoder, TextDecoder } from "util"; | ||
| import { Macro } from "ava"; | ||
| import { | ||
| Comment, | ||
| DocumentHandlers, | ||
| Element, | ||
| ElementHandlers, | ||
| HTMLRewriter as RawHTMLRewriter, | ||
| HTMLRewriterOptions as RawHTMLRewriterOptions, | ||
| TextChunk, | ||
| } from ".."; | ||
|
|
||
| const encoder = new TextEncoder(); | ||
| const decoder = new TextDecoder(); | ||
|
|
||
| export class HTMLRewriter { | ||
| private elementHandlers: [selector: string, handlers: ElementHandlers][] = []; | ||
| private documentHandlers: DocumentHandlers[] = []; | ||
|
|
||
| constructor(private readonly options?: RawHTMLRewriterOptions) {} | ||
|
|
||
| on(selector: string, handlers: ElementHandlers): this { | ||
| this.elementHandlers.push([selector, handlers]); | ||
| return this; | ||
| } | ||
|
|
||
| onDocument(handlers: DocumentHandlers): this { | ||
| this.documentHandlers.push(handlers); | ||
| return this; | ||
| } | ||
|
|
||
| async transform(input: string): Promise<string> { | ||
| let output = ""; | ||
| const rewriter = new RawHTMLRewriter((chunk) => { | ||
| output += decoder.decode(chunk); | ||
| }, this.options); | ||
| for (const [selector, handlers] of this.elementHandlers) { | ||
| rewriter.on(selector, handlers); | ||
| } | ||
| for (const handlers of this.documentHandlers) { | ||
| rewriter.onDocument(handlers); | ||
| } | ||
| try { | ||
| await rewriter.write(encoder.encode(input)); | ||
| await rewriter.end(); | ||
| return output; | ||
| } finally { | ||
| rewriter.free(); | ||
| } | ||
| } | ||
| } | ||
|
|
||
| export function wait(t: number): Promise<void> { | ||
| return new Promise((resolve) => setTimeout(resolve, t)); | ||
| } | ||
|
|
||
| export const mutationsMacro: Macro< | ||
| [ | ||
| ( | ||
| rw: HTMLRewriter, | ||
| handler: (token: Element | TextChunk | Comment) => void | ||
| ) => HTMLRewriter, | ||
| string, | ||
| { | ||
| beforeAfter: string; | ||
| replace: string; | ||
| replaceHtml: string; | ||
| remove: string; | ||
| } | ||
| ] | ||
| > = async (t, func, input, expected) => { | ||
| // In all these tests, only process text chunks containing text. All test | ||
| // inputs for text handlers will be single characters, so we'll only process | ||
| // text nodes once. | ||
|
|
||
| // before/after | ||
| let res = await func(new HTMLRewriter(), (token) => { | ||
| if ("text" in token && !token.text) return; | ||
| token.before("<span>before</span>"); | ||
| token.before("<span>before html</span>", { html: true }); | ||
| token.after("<span>after</span>"); | ||
| token.after("<span>after html</span>", { html: true }); | ||
| }).transform(input); | ||
| t.is(res, expected.beforeAfter); | ||
|
|
||
| // replace | ||
| res = await func(new HTMLRewriter(), (token) => { | ||
| if ("text" in token && !token.text) return; | ||
| token.replace("<span>replace</span>"); | ||
| }).transform(input); | ||
| t.is(res, expected.replace); | ||
| res = await func(new HTMLRewriter(), (token) => { | ||
| if ("text" in token && !token.text) return; | ||
| token.replace("<span>replace</span>", { html: true }); | ||
| }).transform(input); | ||
| t.is(res, expected.replaceHtml); | ||
|
|
||
| // remove | ||
| res = await func(new HTMLRewriter(), (token) => { | ||
| if ("text" in token && !token.text) return; | ||
| t.false(token.removed); | ||
| token.remove(); | ||
| t.true(token.removed); | ||
| }).transform(input); | ||
| t.is(res, expected.remove); | ||
| }; |
File renamed without changes.
File renamed without changes.
File renamed without changes.
This file was deleted.
Oops, something went wrong.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.