mercuryworkshop-mirror/scramjet
1
0
Fork 0
mirror of https://github.com/MercuryWorkshop/scramjet.git synced 2025-05-16 23:50:02 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

add encoder function arg

Browse source
This commit is contained in:
Toshit Chawda 2024-07-26 22:06:13 -07:00
parent 544134f800
commit 11a0c09179
No known key found for this signature in database
GPG key ID: 91480ED99E2B3D9D
7 changed files with 104 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

10
rewriter/src/lib.rs
View file

@ -2,6 +2,7 @@ pub mod rewrite;
use std::{panic, str::FromStr};
use js_sys::encode_uri_component;
use rewrite::rewrite;
use url::Url;
use wasm_bindgen::prelude::*;
@ -12,6 +13,11 @@ extern "C" {
fn log(s: &str);
}
// import the SCRAM!!! jet encoder here later
fn encode(s: String) -> String {
encode_uri_component(&s).into()
}
#[wasm_bindgen]
pub fn init() {
panic::set_hook(Box::new(console_error_panic_hook::hook));
@ -19,7 +25,7 @@ pub fn init() {
#[wasm_bindgen]
pub fn rewrite_js(js: &str, url: &str) -> Vec<u8> {
rewrite(js, Url::from_str(url).unwrap())
rewrite(js, Url::from_str(url).unwrap(), Box::new(encode))
}
#[wasm_bindgen]
@ -28,5 +34,5 @@ pub fn rewrite_js_from_arraybuffer(js: &[u8], url: &str) -> Vec<u8> {
let js = unsafe { std::str::from_utf8_unchecked(js) };
rewrite(js, Url::from_str(url).unwrap())
rewrite(js, Url::from_str(url).unwrap(), Box::new(encode))
}