File size: 2,561 Bytes
bee6636 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 |
import { rewriteCss, unrewriteCss } from "../../shared/rewriters/css";
import { ScramjetClient } from "../client";
export default function (client: ScramjetClient) {
client.Proxy("CSSStyleDeclaration.prototype.setProperty", {
apply(ctx) {
if (!ctx.args[1]) return;
ctx.args[1] = rewriteCss(ctx.args[1], client.meta);
},
});
client.Proxy("CSSStyleDeclaration.prototype.getPropertyValue", {
apply(ctx) {
const v = ctx.call();
if (!v) return v;
ctx.return(unrewriteCss(v));
},
});
client.Trap("CSSStyleDeclaration.prototype.cssText", {
set(ctx, value: string) {
ctx.set(rewriteCss(value, client.meta));
},
get(ctx) {
return unrewriteCss(ctx.get());
},
});
client.Proxy("CSSStyleSheet.prototype.insertRule", {
apply(ctx) {
ctx.args[0] = rewriteCss(ctx.args[0], client.meta);
},
});
client.Proxy("CSSStyleSheet.prototype.replace", {
apply(ctx) {
ctx.args[0] = rewriteCss(ctx.args[0], client.meta);
},
});
client.Proxy("CSSStyleSheet.prototype.replaceSync", {
apply(ctx) {
ctx.args[0] = rewriteCss(ctx.args[0], client.meta);
},
});
client.Trap("CSSRule.prototype.cssText", {
set(ctx, value: string) {
ctx.set(rewriteCss(value, client.meta));
},
get(ctx) {
return unrewriteCss(ctx.get());
},
});
client.Proxy("CSSStyleValue.parse", {
apply(ctx) {
if (!ctx.args[1]) return;
ctx.args[1] = rewriteCss(ctx.args[1], client.meta);
},
});
client.Trap("HTMLElement.prototype.style", {
get(ctx) {
// unfortunate and dumb hack. we have to trap every property of this
// since the prototype chain is fucked
const style = ctx.get() as CSSStyleDeclaration;
return new Proxy(style, {
get(target, prop) {
const value = Reflect.get(target, prop);
if (typeof value === "function") {
return new Proxy(value, {
apply(target, that, args) {
return Reflect.apply(target, style, args);
},
});
}
if (prop in CSSStyleDeclaration.prototype) return value;
if (!value) return value;
return unrewriteCss(value);
},
set(target, prop, value) {
if (prop == "cssText" || value == "" || typeof value !== "string") {
return Reflect.set(target, prop, value);
}
return Reflect.set(target, prop, rewriteCss(value, client.meta));
},
});
},
set(ctx, value: string) {
// this will actually run the trap for cssText. don't rewrite it here
ctx.set(value);
},
});
}
|