diff --git a/worker.js b/worker.js
index 813126a..9486837 100644
--- a/worker.js
+++ b/worker.js
@@ -207,6 +207,8 @@ class HeadRewriter {
class BodyRewriter {
constructor(SLUG_TO_PAGE) {
this.SLUG_TO_PAGE = SLUG_TO_PAGE;
+ this.onDarkHTML = '
';
+ this.onLightHTML = '';
}
element(element) {
element.append(
@@ -237,12 +239,12 @@ class BodyRewriter {
}
}
function onDark() {
- el.innerHTML = '';
+ el.innerHTML = atob('${btoa(this.onDarkHTML)}');
document.body.classList.add('dark');
__console.environment.ThemeStore.setState({ mode: 'dark' });
};
function onLight() {
- el.innerHTML = '';
+ el.innerHTML = atob('${btoa(this.onLightHTML)}');
document.body.classList.remove('dark');
__console.environment.ThemeStore.setState({ mode: 'light' });
}
@@ -316,7 +318,7 @@ class BodyRewriter {
arguments[1] = arguments[1].replace('${MY_DOMAIN}', 'www.notion.so');
return open.apply(this, [].slice.call(arguments));
};
- ${CUSTOM_SCRIPT}`,
+ ${CUSTOM_SCRIPT}`.replaceAll("\n", ""),
{
html: true
}