mirror of
https://github.com/notion-enhancer/notion-enhancer.git
synced 2025-10-31 14:18:08 +11:00
222 lines
11 KiB
Python
Executable File
222 lines
11 KiB
Python
Executable File
|
|
# notion-enhancer
|
|
# (c) 2020 dragonwocky <thedragonring.bod@gmail.com>
|
|
# (c) 2020 TarasokUA
|
|
# (https://dragonwocky.me/) under the MIT license
|
|
|
|
import re
|
|
import os
|
|
import sys
|
|
import platform
|
|
import subprocess
|
|
from shutil import copyfile, rmtree
|
|
from time import sleep
|
|
|
|
# to smooth the update process
|
|
enhancer_version = '0.6.1~beta3'
|
|
|
|
# for toggling notion visibility
|
|
hotkey = 'CmdOrCtrl+Shift+A'
|
|
|
|
# '=== title ===' = headers
|
|
# '*' = information
|
|
# '...' = actions
|
|
# '##' = warnings
|
|
# '>' = exit
|
|
|
|
print('=== NOTION ENHANCER CUSTOMISATION LOG ===\n')
|
|
|
|
try:
|
|
filepath = ''
|
|
__dirname__ = os.path.dirname(__file__)
|
|
enhancer_folder = os.path.realpath(__dirname__)
|
|
if 'microsoft' in platform.uname()[3].lower() and sys.platform == 'linux':
|
|
filepath = '/mnt/c/' + \
|
|
subprocess.run(
|
|
['cmd.exe', '/c', 'echo', '%localappdata%'], stdout=subprocess.PIPE).stdout \
|
|
.rstrip().decode('utf-8')[3:].replace('\\', '/') + '/Programs/Notion/resources'
|
|
drive = enhancer_folder[5].capitalize() if enhancer_folder.startswith(
|
|
'/mnt/') else 'C'
|
|
enhancer_folder = drive + ':\\' + enhancer_folder[6:]
|
|
elif sys.platform == 'win32':
|
|
filepath = subprocess.run(['echo', '%localappdata%'], shell=True, capture_output=True).stdout \
|
|
.rstrip().decode('utf-8') + '\\Programs\\Notion\\resources'
|
|
elif sys.platform == 'linux':
|
|
filepath = '/opt/notion-app' if os.path.exists(
|
|
'/opt/notion-app') else '/opt/notion'
|
|
elif sys.platform == 'darwin':
|
|
filepath = '/Applications/Notion.app/Contents/Resources'
|
|
else:
|
|
print(' > script not compatible with your os!\n (report this to dragonwocky#8449 on discord)')
|
|
exit()
|
|
|
|
unpacking_asar = True
|
|
if not os.path.isfile(os.path.join(filepath, 'app.asar')):
|
|
print(f' ## file {os.path.join(filepath, "app.asar")} not found!')
|
|
print(' * attempting to locate')
|
|
if os.path.exists(os.path.join(filepath, 'app')):
|
|
unpacking_asar = False
|
|
print(' * app.asar was already unpacked: checking version.')
|
|
cleaning_asar = True
|
|
if os.path.isfile(os.path.join(filepath, 'app', 'ENHANCER_VERSION.txt')):
|
|
with open(os.path.join(filepath, 'app', 'ENHANCER_VERSION.txt'), 'r', encoding='UTF-8') as content:
|
|
if content.read() == enhancer_version:
|
|
cleaning_asar = False
|
|
if cleaning_asar:
|
|
unpacking_asar = True
|
|
print(' * version does not match: cleaning.')
|
|
if os.path.exists(os.path.join(filepath, 'app')):
|
|
print(
|
|
f' ...removing folder {os.path.join(filepath, "app")}')
|
|
rmtree(os.path.join(filepath, 'app'))
|
|
else:
|
|
print(
|
|
f' * {os.path.join(filepath, "app")} was not found: step skipped.')
|
|
if os.path.isfile(os.path.join(filepath, 'app.asar.bak')):
|
|
print(' ...renaming asar.app.bak to asar.app')
|
|
os.rename(os.path.join(filepath, 'app.asar.bak'),
|
|
os.path.join(filepath, 'app.asar'))
|
|
else:
|
|
print(
|
|
f' * {os.path.join(filepath, "app.asar.bak")} was not found: exiting. notion install is corrupted.')
|
|
exit()
|
|
else:
|
|
print(' * version matches: continuing.')
|
|
else:
|
|
print(
|
|
' > nothing found: exiting. notion install is either corrupted or non-existent.')
|
|
exit()
|
|
if unpacking_asar:
|
|
print(' ...unpacking app.asar')
|
|
subprocess.run(['asar', 'extract', os.path.join(filepath, 'app.asar'), os.path.join(
|
|
filepath, 'app')], shell=(True if sys.platform == 'win32' else False))
|
|
print(' ...renaming asar.app to asar.app.bak')
|
|
os.rename(os.path.join(filepath, 'app.asar'),
|
|
os.path.join(filepath, 'app.asar.bak'))
|
|
with open(os.path.join(filepath, 'app', 'ENHANCER_VERSION.txt'), 'w', encoding='UTF-8') as write:
|
|
write.write(enhancer_version)
|
|
|
|
if os.path.isfile(os.path.join(filepath, "app", "renderer", "preload.js")):
|
|
print(
|
|
f' ...adding preload.js to {os.path.join(filepath, "app","renderer","preload.js")}')
|
|
with open(os.path.join(filepath, "app", "renderer", "preload.js"), 'r', encoding='UTF-8') as content:
|
|
if '/* === INJECTION MARKER === */' in content.read():
|
|
print(' * preload.js already added. replacing it.')
|
|
content.seek(0)
|
|
original = []
|
|
for num, line in enumerate(content):
|
|
if '/* === INJECTION MARKER === */' in line:
|
|
break
|
|
original += line
|
|
with open(os.path.join(filepath, "app", "renderer", "preload.js"), 'w', encoding='UTF-8') as write:
|
|
write.writelines(original)
|
|
else:
|
|
with open(os.path.join(filepath, "app", "renderer", "preload.js"), 'a', encoding='UTF-8') as append:
|
|
append.write('\n\n')
|
|
with open(os.path.join(filepath, "app", "renderer", "preload.js"), 'a', encoding='UTF-8') as append:
|
|
print(
|
|
f' ...linking to {os.path.join(".", "resources", "user.css")}')
|
|
with open(os.path.join(__dirname__, 'resources', 'preload.js'), 'r', encoding='UTF-8') as insert:
|
|
append.write(insert.read().replace(
|
|
'☃☃☃assets☃☃☃', os.path.join(enhancer_folder, 'resources').replace('\\', '/')))
|
|
else:
|
|
print(
|
|
f' * {os.path.join(filepath, "app","renderer","preload.js")} was not found: step skipped.')
|
|
|
|
if os.path.isfile(os.path.join(filepath, "app", "main", "createWindow.js")):
|
|
with open(os.path.join(filepath, "app", "main", "createWindow.js"), 'r', encoding='UTF-8') as content:
|
|
content = content.read()
|
|
print(
|
|
f' ...making window frameless @ {os.path.join(filepath, "app", "main", "createWindow.js")}')
|
|
if '{ frame: false, show: false' not in content:
|
|
content = content.replace(
|
|
'{ show: false', '{ frame: false, show: false')
|
|
print(
|
|
f' ...adding "open hidden" capabilities to {os.path.join(filepath, "app", "main", "createWindow.js")}')
|
|
content = re.sub('\\s*\\/\\* === INJECTION START === \\*\\/.*?\\/\\* === INJECTION END === \\*\\/\\s*',
|
|
'window.show()', content, flags=re.DOTALL).replace('window.show()', """
|
|
/* === INJECTION START === */
|
|
const path = require('path'),
|
|
store = require(path.join(__dirname, '..', 'store.js'))({
|
|
config: 'user-preferences',
|
|
defaults: {
|
|
openhidden: false,
|
|
maximized: false
|
|
}
|
|
});
|
|
if (!store.openhidden || electron_1.BrowserWindow.getAllWindows().some(win => win.isVisible()))
|
|
{ window.show(); if (store.maximized) window.maximize(); }
|
|
/* === INJECTION END === */
|
|
""")
|
|
with open(os.path.join(filepath, "app", "main", "createWindow.js"), 'w', encoding='UTF-8') as write:
|
|
write.write(content)
|
|
else:
|
|
print(
|
|
f' * {os.path.join(filepath, "app", "main", "createWindow.js")} was not found: step skipped.')
|
|
|
|
if os.path.isfile(os.path.join(filepath, "app", "renderer", "index.js")):
|
|
with open(os.path.join(filepath, "app", "renderer", "index.js"), 'r', encoding='UTF-8') as content:
|
|
print(
|
|
f' ...adjusting drag area for frameless window in {os.path.join(filepath, "app", "renderer", "index.js")}')
|
|
content = content.read()
|
|
loc = content.rfind('dragRegionStyle')
|
|
content = content[:loc] + content[loc:] \
|
|
.replace('top: 0', 'top: 1', 1) \
|
|
.replace('height: 34', 'height: 10', 1)
|
|
with open(os.path.join(filepath, "app", "renderer", "index.js"), 'w', encoding='UTF-8') as write:
|
|
write.write(content)
|
|
else:
|
|
print(
|
|
f' * {os.path.join(filepath, "app", "renderer", "index.js")} was not found: step skipped.')
|
|
|
|
if os.path.isfile(os.path.join(filepath, "app", "main", "main.js")):
|
|
with open(os.path.join(filepath, "app", "main", "main.js"), 'r', encoding='UTF-8') as content:
|
|
print(
|
|
f' ...adding tray support (inc. context menu with settings) to {os.path.join(filepath, "app", "main", "main.js")}')
|
|
print(
|
|
f' ...adding window toggle hotkey to {os.path.join(filepath, "app", "main", "main.js")}')
|
|
content = content.read()
|
|
with open(os.path.join(filepath, "app", "main", "main.js"), 'w', encoding='UTF-8') as write:
|
|
if '/* === INJECTION MARKER === */' in content:
|
|
print(' * tray.js already added. replacing it.')
|
|
original = []
|
|
for line in content.splitlines():
|
|
if '/* === INJECTION MARKER === */' in line:
|
|
break
|
|
original.append(line)
|
|
write.write('\n'.join(original))
|
|
else:
|
|
write.write(content.replace(
|
|
'electron_1.app.on("ready", handleReady);',
|
|
'electron_1.app.on("ready", () => handleReady() && enhancements());') + '\n')
|
|
with open(os.path.join(filepath, "app", "main", "main.js"), 'a', encoding='UTF-8') as append:
|
|
with open(os.path.join(__dirname__, 'resources', 'tray.js'), 'r', encoding='UTF-8') as insert:
|
|
append.write('\n' + insert.read().replace(
|
|
'☃☃☃hotkey☃☃☃', hotkey))
|
|
print(
|
|
f' ...copying tray icon {os.path.join(".", "logo.png")} to {os.path.join(filepath, "app", "main")}')
|
|
copyfile(os.path.join(__dirname__, 'logo.png'),
|
|
os.path.join(filepath, "app", "main", "logo.png"))
|
|
print(
|
|
f' ...copying datastore wrapper {os.path.join(".", "resources", "store.js")} to {os.path.join(filepath, "app")}')
|
|
copyfile(os.path.join(__dirname__, 'resources', 'store.js'),
|
|
os.path.join(filepath, "app", "store.js"))
|
|
else:
|
|
print(
|
|
f' * {os.path.join(filepath, "app", "main", "main.js")} was not found: step skipped.')
|
|
|
|
if sys.platform == 'linux' and 'microsoft' not in platform.uname()[3].lower():
|
|
print(
|
|
f' ...patching app launcher')
|
|
subprocess.call(
|
|
['sed', '-i', r's/electron\ app\.asar/electron\ app/',
|
|
'/usr/bin/notion-app' if os.path.exists('/usr/bin/notion-app') else '/usr/bin/notion'])
|
|
|
|
print('\n>>> SUCCESSFULLY CUSTOMISED <<<')
|
|
|
|
except Exception as e:
|
|
print(
|
|
f'\n### ERROR (report this to dragonwocky#8449 on discord) ###\n{str(e)}')
|
|
|
|
print('\n=== END OF LOG ===')
|