mirror of
https://github.com/notion-enhancer/notion-enhancer.git
synced 2025-04-05 13:19:03 +00:00
113 lines
3.2 KiB
JavaScript
113 lines
3.2 KiB
JavaScript
/*
|
|
* notion-enhancer
|
|
* (c) 2020 dragonwocky <thedragonring.bod@gmail.com>
|
|
* (https://dragonwocky.me/notion-enhancer) under the MIT license
|
|
*/
|
|
|
|
'use strict';
|
|
const os = require('os'),
|
|
path = require('path'),
|
|
fs = require('fs-extra'),
|
|
exec = require('util').promisify(require('child_process').exec);
|
|
|
|
// used to differentiate between "enhancer failed" and "code broken" errors.
|
|
class EnhancerError extends Error {
|
|
constructor(message) {
|
|
super(message);
|
|
this.name = 'EnhancerError';
|
|
}
|
|
}
|
|
|
|
// checks if being run on the windows subsystem for linux:
|
|
// used to modify windows notion app.
|
|
const is_wsl =
|
|
process.platform == 'linux' &&
|
|
os.release().toLowerCase().includes('microsoft'),
|
|
data_folder = path.join(os.homedir(), '.notion-enhancer');
|
|
|
|
// wait for console input, returns keys when enter pressed.
|
|
function readline() {
|
|
return new Promise((res, rej) => {
|
|
process.stdin.resume();
|
|
process.stdin.setEncoding('utf8');
|
|
process.stdin.on('data', (key) => {
|
|
if (key === '\u0003') process.exit(); // CTRL+C
|
|
process.stdin.pause();
|
|
res(key.trim());
|
|
});
|
|
});
|
|
}
|
|
|
|
// gets possible system notion app filepaths.
|
|
async function getNotion() {
|
|
let folder = '';
|
|
switch (process.platform) {
|
|
case 'darwin':
|
|
folder = '/Applications/Notion.app/Contents/Resources';
|
|
break;
|
|
case 'win32':
|
|
folder = process.env.LOCALAPPDATA + '\\Programs\\Notion\\resources';
|
|
break;
|
|
case 'linux':
|
|
if (is_wsl) {
|
|
const { stdout } = await exec('cmd.exe /c echo %localappdata%'),
|
|
drive = stdout[0];
|
|
folder = `/mnt/${drive.toLowerCase()}${stdout
|
|
.replace(/\\/g, '/')
|
|
.slice(2)
|
|
.trim()}/Programs/Notion/resources`;
|
|
} else {
|
|
for (let loc of [
|
|
'/usr/lib/notion-desktop/resources', // https://github.com/davidbailey00/notion-deb-builder/
|
|
'/opt/notion-app', // https://aur.archlinux.org/packages/notion-app/
|
|
'/opt/notion', // https://github.com/jaredallard/notion-app
|
|
]) {
|
|
if (await fs.pathExists(loc)) folder = loc;
|
|
}
|
|
}
|
|
}
|
|
if (!folder)
|
|
throw new EnhancerError(
|
|
'platform not supported: open a request in the github repo:\n' +
|
|
'https://github.com/dragonwocky/notion-enhancer/issues/new?assignees=&labels=enhancement&template=platform-support.md'
|
|
);
|
|
// check if actual app files are present.
|
|
// if app/app.asar are missing but app.asar.bak present it will be moved to app.asar
|
|
const app_asar = path.join(folder, 'app.asar');
|
|
if (
|
|
!(
|
|
(await fs.pathExists(folder)) &&
|
|
((await fs.pathExists(app_asar)) ||
|
|
(await fs.pathExists(path.join(folder, 'app'))))
|
|
)
|
|
) {
|
|
const asar_bak = path.join(folder, 'app.asar.bak');
|
|
if (await fs.pathExists(asar_bak)) {
|
|
await fs.move(asar_bak, app_asar);
|
|
} else
|
|
throw new EnhancerError(
|
|
'nothing found: notion installation is either corrupted or non-existent.'
|
|
);
|
|
}
|
|
return folder;
|
|
}
|
|
|
|
// attempts to read a JSON file, falls back to empty object.
|
|
function getJSON(from) {
|
|
try {
|
|
return fs.readJsonSync(from);
|
|
} catch (err) {
|
|
return {};
|
|
}
|
|
}
|
|
|
|
module.exports = {
|
|
EnhancerError,
|
|
is_wsl,
|
|
data_folder,
|
|
readline,
|
|
getNotion,
|
|
getJSON,
|
|
exec,
|
|
};
|