mirror of
https://github.com/notion-enhancer/notion-enhancer.git
synced 2025-04-06 05:29:02 +00:00
44 lines
1.6 KiB
JavaScript
44 lines
1.6 KiB
JavaScript
/*
|
|
* notion-enhancer
|
|
* (c) 2021 dragonwocky <thedragonring.bod@gmail.com> (https://dragonwocky.me/)
|
|
* (https://notion-enhancer.github.io/) under the MIT license
|
|
*/
|
|
|
|
'use strict';
|
|
|
|
import fsp from 'fs/promises';
|
|
|
|
export default async function (filepath) {
|
|
// https://github.com/notion-enhancer/desktop/issues/291
|
|
// bypass csp issues by intercepting notion:// protocol
|
|
const contents = await fsp.readFile(filepath, 'utf8');
|
|
await fsp.writeFile(
|
|
filepath,
|
|
contents.replace(
|
|
/const success = protocol\.registerStreamProtocol\(config_1.default.protocol, async \(req, callback\) => \{/,
|
|
`const success = protocol.registerStreamProtocol(config_1.default.protocol, async (req, callback) => {
|
|
{
|
|
// notion-enhancer
|
|
const schemePrefix = 'notion://www.notion.so/__notion-enhancer/';
|
|
if (req.url.startsWith(schemePrefix)) {
|
|
const resolvePath = (path) => require('path').resolve(\`\${__dirname}/\${path}\`),
|
|
fileExt = req.url.split('.').reverse()[0],
|
|
filePath = resolvePath(
|
|
\`../node_modules/notion-enhancer/\${req.url.slice(schemePrefix.length)}\`
|
|
),
|
|
mimeDB = Object.entries(require('notion-enhancer/dep/mime-db.json')),
|
|
mimeType = mimeDB
|
|
.filter(([mime, data]) => data.extensions)
|
|
.find(([mime, data]) => data.extensions.includes(fileExt));
|
|
callback({
|
|
data: require('fs').createReadStream(filePath),
|
|
headers: { 'content-type': mimeType },
|
|
});
|
|
}
|
|
}`
|
|
)
|
|
);
|
|
|
|
return true;
|
|
}
|