mirror of
https://github.com/notion-enhancer/notion-enhancer.git
synced 2025-04-03 12:19:02 +00:00
92 lines
3.9 KiB
JavaScript
Executable File
92 lines
3.9 KiB
JavaScript
Executable File
/**
|
|
* notion-enhancer
|
|
* (c) 2023 dragonwocky <thedragonring.bod@gmail.com> (https://dragonwocky.me/)
|
|
* (https://notion-enhancer.github.io/) under the MIT license
|
|
*/
|
|
|
|
// patch scripts within notion's sources to
|
|
// activate and respond to the notion-enhancer
|
|
const injectTriggerOnce = (file, content) =>
|
|
content +
|
|
(!/require\(['|"]notion-enhancer['|"]\)/.test(content)
|
|
? `\n\nrequire("notion-enhancer")("${file}",exports,(js)=>eval(js));`
|
|
: ""),
|
|
replaceIfNotFound = ({ string, mode = "replace" }, search, replacement) =>
|
|
string.includes(replacement)
|
|
? string
|
|
: string.replace(
|
|
search,
|
|
typeof replacement === "string" && mode === "append"
|
|
? `$&${replacement}`
|
|
: typeof replacement === "string" && mode === "prepend"
|
|
? `${replacement}$&`
|
|
: replacement
|
|
);
|
|
|
|
const patches = {
|
|
// prettier-ignore
|
|
".webpack/main/index.js": (file, content) => {
|
|
content = injectTriggerOnce(file, content);
|
|
const replace = (...args) =>
|
|
(content = replaceIfNotFound(
|
|
{ string: content, mode: "replace" },
|
|
...args
|
|
)),
|
|
prepend = (...args) =>
|
|
(content = replaceIfNotFound(
|
|
{ string: content, mode: "prepend" },
|
|
...args
|
|
)),
|
|
append = (...args) =>
|
|
(content = replaceIfNotFound(
|
|
{ string: content, mode: "append" },
|
|
...args
|
|
));
|
|
|
|
// https://github.com/notion-enhancer/notion-enhancer/issues/160:
|
|
// enable the notion:// protocol, windows-style tab layouts, and
|
|
// quitting the app when the last window is closed on linux
|
|
const isWindows =
|
|
/(?:"win32"===process\.platform(?:(?=,isFullscreen)|(?=&&\w\.BrowserWindow)|(?=&&\(\w\.app\.requestSingleInstanceLock)))/g,
|
|
isWindowsOrLinux = '["win32","linux"].includes(process.platform)';
|
|
replace(isWindows, isWindowsOrLinux);
|
|
|
|
// restore node integration in the renderer process
|
|
// so the notion-enhancer can be require()-d into it
|
|
replace(/sandbox:!0/g, `sandbox:!1,nodeIntegration:!0,session:require('electron').session.fromPartition("persist:notion")`);
|
|
|
|
// bypass webRequest filter to load enhancer menu
|
|
replace(/(\w)\.top!==\w\?(\w)\(\{cancel:!0\}\)/, "$1.top!==$1?$2({})");
|
|
|
|
// serve enhancer sources over the notion:// protocol
|
|
const protocolHandler = /try\{const \w=await \w\.assetCache\.handleRequest\(\w\);/,
|
|
protocolInterceptor = `{const n="notion://www.notion.so/__notion-enhancer/";if(e.url.startsWith(n))return require("electron").net.fetch(\`file://\${require("path").join(__dirname,"..","..","node_modules","notion-enhancer",e.url.slice(n.length))}\`)}`;
|
|
prepend(protocolHandler, protocolInterceptor);
|
|
|
|
// disable csp entirely. risky? yes. necessary? also yes.
|
|
// potentially could extend default policy instead someday.
|
|
const headerHandler = /\.onHeadersReceived\(\(\((\w),\w\)=>{/,
|
|
cspDisabler = `{const responseHeaders=$1.responseHeaders;if(responseHeaders){delete responseHeaders["content-security-policy"];return t({responseHeaders})}}`
|
|
append(headerHandler, cspDisabler);
|
|
|
|
// expose the app's config + cache + preferences to the global namespace
|
|
// e.g. to enable development mode or check if keep in background is enabled
|
|
prepend(/\w\.exports=JSON\.parse\('\{"env":"production"/, "globalThis.__notionConfig=");
|
|
prepend(/\w\.updatePreferences=\w\.updatePreferences/, "globalThis.__updatePreferences=");
|
|
prepend(/\w\.Store=\(0,\w\.configureStore\)/, "globalThis.__notionStore=");
|
|
|
|
return content;
|
|
},
|
|
".webpack/renderer/tabs/preload.js": injectTriggerOnce,
|
|
".webpack/renderer/tab_browser_view/preload.js": injectTriggerOnce,
|
|
};
|
|
|
|
const decoder = new TextDecoder(),
|
|
encoder = new TextEncoder();
|
|
export default (file, content) => {
|
|
if (!patches[file]) return content;
|
|
content = decoder.decode(content);
|
|
content = patches[file](file, content);
|
|
return encoder.encode(content);
|
|
};
|