dragonwocky/notion-enhancer
1
0
Fork 0
mirror of https://github.com/notion-enhancer/notion-enhancer.git synced 2025-04-05 13:19:03 +00:00
Code Issues Packages Projects Releases Wiki Activity
6f6edb2cd8
notion-enhancer/customiser.py

215 lines
10 KiB
Python
Executable File
Raw Blame History

# 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.0~beta2'
# for toggling notion visibility
hotkey = 'CmdOrCtrl+Shift+A'
# '=== title ===' = headers
# '*' = information
# '...' = actions
# '##' = warnings
# '>' = exit
print('=== NOTION ENHANCER CUSTOMISATION LOG ===\n')
try:
filepath = ''
__folder__ = os.path.dirname(os.path.realpath(__file__)).replace('\\', '/')
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 = __folder__[5].capitalize() if __folder__.startswith(
'/mnt/') else 'C'
__folder__ = drive + ':/' + __folder__[6:]
elif sys.platform == 'win32':
filepath = subprocess.run(['echo', '%localappdata%'], shell=True, capture_output=True).stdout \
.rstrip().decode('utf-8').replace('\\', '/') + '/Programs/Notion/resources'
elif sys.platform == 'linux':
filepath = '/opt/notion-app'
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(filepath + '/app.asar'):
print(f' ## file {filepath}/app.asar not found!')
print(' * attempting to locate')
if os.path.exists(filepath + '/app'):
unpacking_asar = False
print(' * app.asar was already unpacked: checking version.')
cleaning_asar = True
if os.path.isfile(filepath + '/app/ENHANCER_VERSION.txt'):
with open(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(filepath + '/app'):
print(
f' ...removing folder {filepath}/app/')
rmtree(filepath + '/app')
else:
print(
f' * {filepath}/app/ was not found: step skipped.')
if os.path.isfile(filepath + '/app.asar.bak'):
print(' ...renaming asar.app.bak to asar.app')
os.rename(filepath + '/app.asar.bak',
filepath + '/app.asar')
else:
print(
f' * {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', filepath +
'/app.asar', filepath + '/app'], shell=(True if sys.platform == 'win32' else False))
print(' ...renaming asar.app to asar.app.bak')
os.rename(filepath + '/app.asar', filepath + '/app.asar.bak')
with open(filepath + '/app/ENHANCER_VERSION.txt', 'w', encoding='UTF-8') as write:
write.write(enhancer_version)
if os.path.isfile(filepath + '/app/renderer/preload.js'):
print(f' ...adding preload.js to {filepath}/app/renderer/preload.js')
with open(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(filepath + '/app/renderer/preload.js', 'w', encoding='UTF-8') as write:
write.writelines(original)
else:
with open(filepath + '/app/renderer/preload.js', 'a', encoding='UTF-8') as append:
append.write('\n\n')
with open(filepath + '/app/renderer/preload.js', 'a', encoding='UTF-8') as append:
print(' ...linking to ./resources/user.css')
with open('./resources/preload.js', 'r', encoding='UTF-8') as insert:
append.write(insert.read().replace(
'☃☃☃assets☃☃☃', __folder__
+ '/resources'))
else:
print(
f' * {filepath}/app/renderer/preload.js was not found: step skipped.')
if os.path.isfile(filepath + '/app/main/createWindow.js'):
with open(filepath + '/app/main/createWindow.js', 'r', encoding='UTF-8') as content:
content = content.read()
print(
f' ...making window frameless @ {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 {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(filepath + '/app/main/createWindow.js', 'w', encoding='UTF-8') as write:
write.write(content)
else:
print(
f' * {filepath}/app/main/createWindow.js was not found: step skipped.')
if os.path.isfile(filepath + '/app/renderer/index.js'):
with open(filepath + '/app/renderer/index.js', 'r', encoding='UTF-8') as content:
print(
f' ...adjusting drag area for frameless window in {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(filepath + '/app/renderer/index.js', 'w', encoding='UTF-8') as write:
write.write(content)
else:
print(
f' * {filepath}/app/renderer/index.js was not found: step skipped.')
if os.path.isfile(filepath + '/app/main/main.js'):
with open(filepath + '/app/main/main.js', 'r', encoding='UTF-8') as content:
print(
f' ...adding tray support (inc. context menu with settings) to {filepath}/app/main/main.js')
print(
f' ...adding window toggle hotkey to {filepath}/app/main/main.js')
content = content.read()
with open(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(filepath + '/app/main/main.js', 'a', encoding='UTF-8') as append:
with open('./resources/tray.js', 'r', encoding='UTF-8') as insert:
append.write('\n' + insert.read().replace(
'☃☃☃hotkey☃☃☃', hotkey))
print(
f' ...copying tray icon ./resources/notion.ico to {filepath}/app/main/')
copyfile('./resources/notion.ico',
filepath + '/app/main/notion.ico')
print(
f' ...copying datastore wrapper ./resources/store.js to {filepath}/app/')
copyfile('./resources/store.js', filepath + '/app/store.js')
else:
print(
f' * {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 {filepath}/notion-app')
subprocess.call(
['sed', '-i', 's/app\.asar/app/', '/usr/bin/notion-app'])
# patch this too just in case
subprocess.call(['sed', '-i', 's/app\.asar/app/',
f'{filepath}/notion-app'])
print('\n>>> SUCCESSFULLY CUSTOMISED <<<')
except Exception as e:
print(f'\n### ERROR ###\n{str(e)}')
print('\n=== END OF LOG ===')
Reference in New Issue View Git Blame Copy Permalink