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

small changes

Browse source
This commit is contained in:
Avad3 2024-07-11 18:59:51 -04:00
parent 8c7255ccf8
commit 3a85734b7f
22 changed files with 244 additions and 200 deletions
Download patch file Download diff file Expand all files Collapse all files

58
src/client/element.ts
View file

@ -1,3 +1,6 @@
import { encodeUrl, rewriteCss, rewriteJs, rewriteSrcset } from "../bundle";
import { rewriteHtml } from "../html";
// object
// iframe
// embed
@ -37,25 +40,30 @@ Object.keys(attribs).forEach((attrib: string) => {
const descriptor = Object.getOwnPropertyDescriptor(element.prototype, attrib);
Object.defineProperty(element.prototype, attrib, {
get() {
return descriptor.get.call(this, [this.dataset[`_${attrib}`]]);
return this.dataset[`${attrib}`];
},
set(value) {
this.dataset[`_${attrib}`] = value;
this.dataset[`${attrib}`] = value;
if (/nonce|integrity|csp/.test(attrib)) {
this.removeAttribute(attrib);
} else if (/src|href|data|action|formaction/.test(attrib)) {
// @ts-expect-error
// TrustedScriptURL does not exist as a type yet, but it is a real thing
if (value instanceof TrustedScriptURL) {
return;
}
value = self.__scramjet$bundle.rewriters.url.encodeUrl(value);
value = encodeUrl(value);
} else if (attrib === "srcdoc") {
value = self.__scramjet$bundle.rewriters.rewriteHtml(value);
// @ts-ignore
// This needs to be ignored because I'm bad at TypeScript
value = rewriteHtml(value).documentElement.innerHTML;
} else if (/(image)?srcset/.test(attrib)) {
value = self.__scramjet$bundle.rewriters.rewriteSrcset(value);
value = rewriteSrcset(value);
} else if (attrib === "style") {
value = self.__scramjet$bundle.rewriters.rewriteCss(value);
value = rewriteCss(value);
}
descriptor.set.call(this, value);
@ -64,34 +72,31 @@ Object.keys(attribs).forEach((attrib: string) => {
})
});
HTMLElement.prototype.getAttribute = new Proxy(Element.prototype.getAttribute, {
Element.prototype.getAttribute = new Proxy(Element.prototype.getAttribute, {
apply(target, thisArg, argArray) {
console.log(thisArg);
if (Object.keys(attribs).includes(argArray[0])) {
argArray[0] = `_${argArray[0]}`;
if (Object.keys(attribs).includes(argArray[0]) && thisArg.dataset[`${argArray[0]}`]) {
return thisArg.dataset[`${argArray[0]}`];
}
return Reflect.apply(target, thisArg, argArray);
},
});
// setAttribute proxy is currently broken
HTMLElement.prototype.setAttribute = new Proxy(Element.prototype.setAttribute, {
Element.prototype.setAttribute = new Proxy(Element.prototype.setAttribute, {
apply(target, thisArg, argArray) {
if (Object.keys(attribs).includes(argArray[0])) {
thisArg.dataset[`_${argArray[0]}`] = argArray[1];
thisArg.dataset[`${argArray[0]}`] = argArray[1];
if (/nonce|integrity|csp/.test(argArray[0])) {
return;
} else if (/src|href|data|action|formaction/.test(argArray[0])) {
console.log(thisArg);
argArray[1] = self.__scramjet$bundle.rewriters.url.encodeUrl(argArray[1]);
argArray[1] = encodeUrl(argArray[1]);
} else if (argArray[0] === "srcdoc") {
argArray[1] = self.__scramjet$bundle.rewriters.rewriteHtml(argArray[1]);
// @ts-ignore
argArray[1] = rewriteHtml(argArray[1]).documentElement.innerHTML;
} else if (/(image)?srcset/.test(argArray[0])) {
argArray[1] = self.__scramjet$bundle.rewriters.rewriteSrcset(argArray[1]);
argArray[1] = rewriteSrcset(argArray[1]);
} else if (argArray[1] === "style") {
argArray[1] = self.__scramjet$bundle.rewriters.rewriteCss(argArray[1]);
argArray[1] = rewriteCss(argArray[1]);
}
}
@ -104,17 +109,22 @@ const innerHTML = Object.getOwnPropertyDescriptor(Element.prototype, "innerHTML"
Object.defineProperty(HTMLElement.prototype, "innerHTML", {
set(value) {
if (this instanceof HTMLScriptElement) {
// @ts-expect-error
// TrustedScript does not exist as a type yet, but it is a real thing
if (!(value instanceof TrustedScript)) {
value = self.__scramjet$bundle.rewriters.rewriteJs(value);
value = rewriteJs(value);
}
} else if (this instanceof HTMLStyleElement) {
value = self.__scramjet$bundle.rewriters.rewriteCss(value);
value = rewriteCss(value);
} else {
// @ts-expect-error
// TrustedHTML does not exist as a type, but it is a real thing
if (!(value instanceof TrustedHTML)) {
value = self.__scramjet$bundle.rewriters.rewriteHtml(value);
// @ts-ignore
value = rewriteHtml(value).documentElement.innerHTML;
}
}
return innerHTML.set.call(this, value);
innerHTML.set.call(this, value);
},
})
});